-
Notifications
You must be signed in to change notification settings - Fork 3
/
build_cffi.py
44 lines (37 loc) · 1.26 KB
/
build_cffi.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
from cffi import FFI
import os
ffi = FFI()
LOCATION = os.path.dirname(os.path.abspath(__file__))
CLOC = os.path.join(LOCATION, 'src', 'py21cmmc', '_21cmfast', 'src')
include_dirs = [CLOC]
# =================================================================
# Set compilation arguments dependent on environment... a bit buggy
# =================================================================
if "DEBUG" in os.environ:
extra_compile_args = ['-fopenmp', '-w', '-g', '-O0']
else:
extra_compile_args = ['-fopenmp', '-Ofast', '-w']
library_dirs = []
for k,v in os.environ.items():
if "inc" in k.lower():
include_dirs += [v]
elif "lib" in k.lower():
library_dirs += [v]
# =================================================================
# This is the overall C code.
ffi.set_source(
"py21cmmc._21cmfast._21cmfast", # Name/Location of shared library module
'''
#include "GenerateICs.c"
''',
include_dirs = include_dirs,
library_dirs=library_dirs,
libraries=['m','gsl','gslcblas','fftw3f_omp', 'fftw3f'],
extra_compile_args = extra_compile_args,
extra_link_args=['-fopenmp']
)
# This is the Header file
with open(os.path.join(CLOC, "21CMMC.h")) as f:
ffi.cdef(f.read())
if __name__ == "__main__":
ffi.compile()