Use Py3 to build dcos for constants
[pyopencl.git] / setup.py
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3
4 from __future__ import absolute_import, print_function
5
6 __copyright__ = """
7 Copyright (C) 2009-15 Andreas Kloeckner
8 Copyright (C) 2013 Marko Bencun
9 Copyright (C) 2013 Marko Bencun
10 """
11
12 __license__ = """
13 Permission is hereby granted, free of charge, to any person obtaining a copy
14 of this software and associated documentation files (the "Software"), to deal
15 in the Software without restriction, including without limitation the rights
16 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
17 copies of the Software, and to permit persons to whom the Software is
18 furnished to do so, subject to the following conditions:
19
20 The above copyright notice and this permission notice shall be included in
21 all copies or substantial portions of the Software.
22
23 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
24 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
25 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
26 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
27 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
28 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
29 THE SOFTWARE.
30 """
31
32
33 import sys
34
35
36 def get_config_schema():
37     from aksetup_helper import ConfigSchema, Option, \
38             IncludeDir, LibraryDir, Libraries, \
39             Switch, StringListOption
40
41     default_cxxflags = ['-std=c++0x']
42
43     if 'darwin' in sys.platform:
44         import platform
45         osx_ver, _, _ = platform.mac_ver()
46         osx_ver = '.'.join(osx_ver.split('.')[:2])
47
48         sysroot_paths = [
49                 "/Applications/Xcode.app/Contents/Developer/Platforms/"
50                 "MacOSX.platform/Developer/SDKs/MacOSX%s.sdk" % osx_ver,
51                 "/Developer/SDKs/MacOSX%s.sdk" % osx_ver
52                 ]
53
54         default_libs = []
55         default_cxxflags = default_cxxflags + [
56                 '-stdlib=libc++', '-mmacosx-version-min=10.7',
57                 '-arch', 'i386', '-arch', 'x86_64'
58                 ]
59
60         from os.path import isdir
61         for srp in sysroot_paths:
62             if isdir(srp):
63                 default_cxxflags.extend(['-isysroot', srp])
64                 break
65
66         default_ldflags = default_cxxflags[:] + ["-Wl,-framework,OpenCL"]
67
68     else:
69         default_libs = ["OpenCL"]
70         default_ldflags = []
71
72     return ConfigSchema([
73         Switch("CL_TRACE", False, "Enable OpenCL API tracing"),
74         Switch("CL_ENABLE_GL", False, "Enable OpenCL<->OpenGL interoperability"),
75         Switch("CL_USE_SHIPPED_EXT", True,
76             "Use the pyopencl version of CL/cl_ext.h which includes" +
77             " a broader range of vendor-specific OpenCL extension attributes" +
78             " than the standard Khronos (or vendor specific) CL/cl_ext.h."
79         ),
80         Option("CL_PRETEND_VERSION", None,
81             "Dotted CL version (e.g. 1.2) which you'd like to use."),
82
83         IncludeDir("CL", []),
84         LibraryDir("CL", []),
85         Libraries("CL", default_libs),
86
87         StringListOption("CXXFLAGS", default_cxxflags,
88             help="Any extra C++ compiler options to include"),
89         StringListOption("LDFLAGS", default_ldflags,
90             help="Any extra linker options to include"),
91         ])
92
93
94 def main():
95     from setuptools import find_packages
96     from aksetup_helper import (hack_distutils, get_config, setup,
97             check_git_submodules)
98     check_git_submodules()
99
100     hack_distutils()
101     conf = get_config(get_config_schema(),
102             warn_about_no_config=False)
103
104     extra_defines = {}
105
106     extra_defines["PYGPU_PACKAGE"] = "pyopencl"
107     extra_defines["PYGPU_PYOPENCL"] = "1"
108
109     if conf["CL_TRACE"]:
110         extra_defines["PYOPENCL_TRACE"] = 1
111
112     if conf["CL_ENABLE_GL"]:
113         extra_defines["HAVE_GL"] = 1
114
115     if conf["CL_USE_SHIPPED_EXT"]:
116         extra_defines["PYOPENCL_USE_SHIPPED_EXT"] = 1
117
118     if conf["CL_PRETEND_VERSION"]:
119         try:
120             major, minor = [int(x) for x in conf["CL_PRETEND_VERSION"].split(".")]
121             extra_defines["PYOPENCL_PRETEND_CL_VERSION"] = \
122                     0x1000*major + 0x10 * minor
123         except:
124             print("CL_PRETEND_VERSION must be of the form M.N, "
125                     "with two integers M and N")
126             raise
127
128     conf["EXTRA_DEFINES"] = extra_defines
129
130     ver_dic = {}
131     version_file = open("pyopencl/version.py")
132     try:
133         version_file_contents = version_file.read()
134     finally:
135         version_file.close()
136
137     exec(compile(version_file_contents, "pyopencl/version.py", 'exec'), ver_dic)
138
139     separator = "-"*75
140
141     try:
142         import mako  # noqa
143     except ImportError:
144         print(separator)
145         print("Mako is not installed.")
146         print(separator)
147         print("That is not a problem, as most of PyOpenCL will be just fine ")
148         print("without it. Some higher-level parts of pyopencl (such as ")
149         print("pyopencl.reduction) will not function without the templating engine ")
150         print("Mako [1] being installed. If you would like this functionality to ")
151         print("work, you might want to install Mako after you finish ")
152         print("installing PyOpenCL.")
153         print("")
154         print("Simply type")
155         print("python -m pip install mako")
156         print("either now or after the installation completes to fix this.")
157         print("")
158         print("[1] http://www.makotemplates.org/")
159         print(separator)
160         print("Hit Ctrl-C now if you'd like to think about the situation.")
161         print(separator)
162
163         from aksetup_helper import count_down_delay
164         count_down_delay(delay=5)
165
166     # {{{ write cffi build script
167
168     with open("cffi_build.py.in", "rt") as f:
169         build_script_template = f.read()
170
171     format_args = {}
172     for k, v in conf.items():
173         format_args[k] = repr(v)
174
175     build_script = build_script_template.format(**format_args)
176
177     with open("cffi_build.py", "wt") as f:
178         f.write(build_script)
179
180     # }}}
181
182     setup(name="pyopencl",
183             # metadata
184             version=ver_dic["VERSION_TEXT"],
185             description="Python wrapper for OpenCL",
186             long_description=open("README.rst", "rt").read(),
187             author="Andreas Kloeckner",
188             author_email="inform@tiker.net",
189             license="MIT",
190             url="http://mathema.tician.de/software/pyopencl",
191             classifiers=[
192                 'Environment :: Console',
193                 'Development Status :: 5 - Production/Stable',
194                 'Intended Audience :: Developers',
195                 'Intended Audience :: Other Audience',
196                 'Intended Audience :: Science/Research',
197                 'License :: OSI Approved :: MIT License',
198                 'Natural Language :: English',
199                 'Programming Language :: C++',
200                 'Programming Language :: Python',
201                 'Programming Language :: Python :: 2',
202                 'Programming Language :: Python :: 2.6',
203                 'Programming Language :: Python :: 2.7',
204                 'Programming Language :: Python :: 3',
205                 'Programming Language :: Python :: 3.2',
206                 'Programming Language :: Python :: 3.3',
207                 'Topic :: Scientific/Engineering',
208                 'Topic :: Scientific/Engineering :: Mathematics',
209                 'Topic :: Scientific/Engineering :: Physics',
210                 ],
211
212             # build info
213             packages=find_packages(),
214
215             setup_requires=[
216                 "numpy",
217                 "cffi>=1.1.0",
218                 ],
219
220             install_requires=[
221                 "numpy",
222                 "pytools>=2015.1.2",
223                 "pytest>=2",
224                 "decorator>=3.2.0",
225                 "cffi>=1.1.0",
226                 "appdirs>=1.4.0",
227                 "six>=1.9.0",
228                 # "Mako>=0.3.6",
229                 ],
230
231             cffi_modules=["cffi_build.py:ffi"],
232
233             include_package_data=True,
234             package_data={
235                     "pyopencl": [
236                         "cl/*.cl",
237                         "cl/*.h",
238                         "cl/pyopencl-random123/*.cl",
239                         "cl/pyopencl-random123/*.h",
240                         ]
241                     },
242
243             zip_safe=False)
244
245
246 if __name__ == '__main__':
247     main()