fABBA/setup.py at master · nla-group/fABBA · GitHub

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
from setuptools import setup, Extension
import numpy
import os
from Cython.Build import cythonize

ext_modules = [
    Extension("fABBA.extmod.chainApproximation_c",
              ["fABBA/extmod/chainApproximation_c.pyx"]),
    Extension("fABBA.extmod.chainApproximation_cm",
              ["fABBA/extmod/chainApproximation_cm.pyx"]),
    Extension("fABBA.extmod.fabba_agg_c",
              ["fABBA/extmod/fabba_agg_c.pyx"]),
    Extension("fABBA.extmod.inverse_tc",
              ["fABBA/extmod/inverse_tc.pyx"]),
    Extension("fABBA.extmod.fabba_agg_cm",
              ["fABBA/extmod/fabba_agg_cm.pyx"]),
    Extension("fABBA.extmod.fabba_agg_cm_win",
              ["fABBA/extmod/fabba_agg_cm_win.pyx"]),

    Extension("fABBA.separate.aggregation_c",
              ["fABBA/separate/aggregation_c.pyx"]),
    Extension("fABBA.separate.aggregation_cm",
              ["fABBA/separate/aggregation_cm.pyx"]),

    Extension("fABBA.jabba.compmem",
              ["fABBA/jabba/compmem.pyx"]),
    Extension("fABBA.jabba.aggwin",
              ["fABBA/jabba/aggwin.pyx"]),
    Extension("fABBA.jabba.aggmem",
              ["fABBA/jabba/aggmem.pyx"]),
    Extension("fABBA.jabba.inversetc",
              ["fABBA/jabba/inversetc.pyx"])
]

setup(
    ext_modules=([] if os.environ.get("FABBA_NO_EXTENSIONS") == "1" else
                 cythonize(ext_modules, build_dir="build/cython",
                           compiler_directives={"language_level": 3})),
    include_dirs=[numpy.get_include()],
    options={"build": {"build_base": "build/python" if os.environ.get("FABBA_NO_EXTENSIONS") == "1" else "build/compiled"}},
)