forked from abelcarreras/DynaPhoPy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup_openmp.py
58 lines (47 loc) · 1.85 KB
/
setup_openmp.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
try:
from setuptools import setup, Extension
use_setuptools = True
print("setuptools is used.")
except ImportError:
from distutils.core import setup, Extension
use_setuptools = False
print("distutils is used.")
import numpy
include_dirs_numpy = [numpy.get_include()]
def get_version_number():
__version__ = None
for l in open('dynaphopy/__init__.py', 'r').readlines():
if not(l.find('__version__')):
exec(l)
return __version__
correlation = Extension('dynaphopy.power_spectrum.correlation',
extra_compile_args=['-std=c99', '-fopenmp'],
extra_link_args=['-lgomp'],
include_dirs = include_dirs_numpy,
sources=['c/correlation.c'])
mem = Extension('dynaphopy.power_spectrum.mem',
extra_compile_args=['-std=c99', '-fopenmp'],
extra_link_args=['-lgomp'],
include_dirs = include_dirs_numpy,
sources=['c/mem.c'])
displacements = Extension('dynaphopy.displacements',
extra_compile_args=['-std=c99', '-fopenmp'],
extra_link_args=['-lgomp'],
include_dirs = include_dirs_numpy,
sources=['c/displacements.c'])
setup(name='dynaphopy',
version=get_version_number(),
description='dynaphopy module',
author='Abel Carreras',
url='https://github.com/abelcarreras/DynaPhoPy',
author_email='abelcarreras83@gmail.com',
packages=['dynaphopy',
'dynaphopy.orm',
'dynaphopy.power_spectrum',
'dynaphopy.analysis',
'dynaphopy.analysis.fitting',
'dynaphopy.interface',
'dynaphopy.interface.iofile'],
scripts=['scripts/dynaphopy'],
ext_modules=[correlation, mem, displacements])
exit()