Reduction fixes
[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         Option("CL_PRETEND_VERSION", None,
76             "Dotted CL version (e.g. 1.2) which you'd like to use."),
77
78         IncludeDir("CL", []),
79         LibraryDir("CL", []),
80         Libraries("CL", default_libs),
81
82         StringListOption("CXXFLAGS", default_cxxflags,
83             help="Any extra C++ compiler options to include"),
84         StringListOption("LDFLAGS", default_ldflags,
85             help="Any extra linker options to include"),
86         ])
87
88
89 def main():
90     from setuptools import find_packages
91     from aksetup_helper import (hack_distutils, get_config, setup,
92             check_git_submodules)
93     check_git_submodules()
94
95     hack_distutils()
96     conf = get_config(get_config_schema(),
97             warn_about_no_config=False)
98
99     extra_defines = {}
100
101     extra_defines["PYGPU_PACKAGE"] = "pyopencl"
102     extra_defines["PYGPU_PYOPENCL"] = "1"
103
104     if conf["CL_TRACE"]:
105         extra_defines["PYOPENCL_TRACE"] = 1
106
107     if conf["CL_ENABLE_GL"]:
108         extra_defines["HAVE_GL"] = 1
109
110     if conf["CL_PRETEND_VERSION"]:
111         try:
112             major, minor = [int(x) for x in conf["CL_PRETEND_VERSION"].split(".")]
113             extra_defines["PYOPENCL_PRETEND_CL_VERSION"] = \
114                     0x1000*major + 0x10 * minor
115         except:
116             print("CL_PRETEND_VERSION must be of the form M.N, "
117                     "with two integers M and N")
118             raise
119
120     conf["EXTRA_DEFINES"] = extra_defines
121
122     ver_dic = {}
123     version_file = open("pyopencl/version.py")
124     try:
125         version_file_contents = version_file.read()
126     finally:
127         version_file.close()
128
129     exec(compile(version_file_contents, "pyopencl/version.py", 'exec'), ver_dic)
130
131     separator = "-"*75
132
133     try:
134         import mako  # noqa
135     except ImportError:
136         print(separator)
137         print("Mako is not installed.")
138         print(separator)
139         print("That is not a problem, as most of PyOpenCL will be just fine ")
140         print("without it. Some higher-level parts of pyopencl (such as ")
141         print("pyopencl.reduction) will not function without the templating engine ")
142         print("Mako [1] being installed. If you would like this functionality to ")
143         print("work, you might want to install Mako after you finish ")
144         print("installing PyOpenCL.")
145         print("")
146         print("Simply type")
147         print("python -m pip install mako")
148         print("either now or after the installation completes to fix this.")
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": [
228                         "cl/*.cl",
229                         "cl/*.h",
230                         "cl/pyopencl-random123/*.cl",
231                         "cl/pyopencl-random123/*.h",
232                         ]
233                     },
234
235             zip_safe=False)
236
237
238 if __name__ == '__main__':
239     main()