Re-fix constant type to int64_t
[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_ENABLE_DEVICE_FISSION", True,
76             "Enable device fission extension, if present"),
77         Option("CL_PRETEND_VERSION", None,
78             "Dotted CL version (e.g. 1.2) which you'd like to use."),
79
80         IncludeDir("CL", []),
81         LibraryDir("CL", []),
82         Libraries("CL", default_libs),
83
84         StringListOption("CXXFLAGS", default_cxxflags,
85             help="Any extra C++ compiler options to include"),
86         StringListOption("LDFLAGS", default_ldflags,
87             help="Any extra linker options to include"),
88         ])
89
90
91 def main():
92     from setuptools import find_packages
93     from aksetup_helper import (hack_distutils, get_config, setup,
94             check_git_submodules)
95     check_git_submodules()
96
97     hack_distutils()
98     conf = get_config(get_config_schema(),
99             warn_about_no_config=False)
100
101     extra_defines = {}
102
103     extra_defines["PYGPU_PACKAGE"] = "pyopencl"
104     extra_defines["PYGPU_PYOPENCL"] = "1"
105
106     if conf["CL_TRACE"]:
107         extra_defines["PYOPENCL_TRACE"] = 1
108
109     if conf["CL_ENABLE_GL"]:
110         extra_defines["HAVE_GL"] = 1
111
112     if conf["CL_ENABLE_DEVICE_FISSION"]:
113         extra_defines["PYOPENCL_USE_DEVICE_FISSION"] = 1
114     if conf["CL_PRETEND_VERSION"]:
115         try:
116             major, minor = [int(x) for x in conf["CL_PRETEND_VERSION"].split(".")]
117             extra_defines["PYOPENCL_PRETEND_CL_VERSION"] = \
118                     0x1000*major + 0x10 * minor
119         except:
120             print("CL_PRETEND_VERSION must be of the form M.N, "
121                     "with two integers M and N")
122             raise
123
124     conf["EXTRA_DEFINES"] = extra_defines
125
126     ver_dic = {}
127     version_file = open("pyopencl/version.py")
128     try:
129         version_file_contents = version_file.read()
130     finally:
131         version_file.close()
132
133     exec(compile(version_file_contents, "pyopencl/version.py", 'exec'), ver_dic)
134
135     separator = "-"*75
136
137     try:
138         import mako  # noqa
139     except ImportError:
140         print(separator)
141         print("Mako is not installed.")
142         print(separator)
143         print("That is not a problem, as most of PyOpenCL will be just fine ")
144         print("without it.Some higher-level parts of pyopencl (such as ")
145         print("pyopencl.reduction) will not function without the templating engine ")
146         print("Mako [1] being installed. If you would like this functionality to ")
147         print("work, you might want to install Mako after you finish ")
148         print("installing PyOpenCL.")
149         print("")
150         print("[1] http://www.makotemplates.org/")
151         print(separator)
152         print("Hit Ctrl-C now if you'd like to think about the situation.")
153         print(separator)
154
155         from aksetup_helper import count_down_delay
156         count_down_delay(delay=5)
157
158     # {{{ write cffi build script
159
160     with open("cffi_build.py.in", "rt") as f:
161         build_script_template = f.read()
162
163     format_args = {}
164     for k, v in conf.items():
165         format_args[k] = repr(v)
166
167     build_script = build_script_template.format(**format_args)
168
169     with open("cffi_build.py", "wt") as f:
170         f.write(build_script)
171
172     # }}}
173
174     setup(name="pyopencl",
175             # metadata
176             version=ver_dic["VERSION_TEXT"],
177             description="Python wrapper for OpenCL",
178             long_description=open("README.rst", "rt").read(),
179             author="Andreas Kloeckner",
180             author_email="inform@tiker.net",
181             license="MIT",
182             url="http://mathema.tician.de/software/pyopencl",
183             classifiers=[
184                 'Environment :: Console',
185                 'Development Status :: 5 - Production/Stable',
186                 'Intended Audience :: Developers',
187                 'Intended Audience :: Other Audience',
188                 'Intended Audience :: Science/Research',
189                 'License :: OSI Approved :: MIT License',
190                 'Natural Language :: English',
191                 'Programming Language :: C++',
192                 'Programming Language :: Python',
193                 'Programming Language :: Python :: 2',
194                 'Programming Language :: Python :: 2.6',
195                 'Programming Language :: Python :: 2.7',
196                 'Programming Language :: Python :: 3',
197                 'Programming Language :: Python :: 3.2',
198                 'Programming Language :: Python :: 3.3',
199                 'Topic :: Scientific/Engineering',
200                 'Topic :: Scientific/Engineering :: Mathematics',
201                 'Topic :: Scientific/Engineering :: Physics',
202                 ],
203
204             # build info
205             packages=find_packages(),
206
207             setup_requires=[
208                 "numpy",
209                 "cffi>=1.1.0",
210                 ],
211
212             install_requires=[
213                 "numpy",
214                 "pytools>=2015.1.2",
215                 "pytest>=2",
216                 "decorator>=3.2.0",
217                 "cffi>=1.1.0",
218                 "appdirs>=1.4.0",
219                 "six>=1.9.0",
220                 # "Mako>=0.3.6",
221                 ],
222
223             cffi_modules=["cffi_build.py:ffi"],
224
225             include_package_data=True,
226             package_data={
227                     "pyopencl": ["cl/*.cl", "cl/*.h"]
228                     },
229
230             zip_safe=False)
231
232
233 if __name__ == '__main__':
234     main()