-
Notifications
You must be signed in to change notification settings - Fork 84
/
setup.py
106 lines (90 loc) · 3.04 KB
/
setup.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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import os.path
import platform
import subprocess
from warnings import warn
import numpy
from setuptools import Extension
from setuptools import setup
CONDA_PREFIX = os.getenv("MSP_CONDA_PREFIX", None)
IS_WINDOWS = platform.system() == "Windows"
class PathConfigurator:
"""
A class to attempt configuration of the compile search paths
on various platforms.
"""
def __init__(self):
self.include_dirs = []
self.library_dirs = []
try:
self._configure_gsl()
except OSError as e:
warn(f"Error occured getting GSL path config: {e}")
# If the conda prefix is defined, then we are compiling in a conda
# context. All include and lib paths should come from within this prefix.
if CONDA_PREFIX is not None:
prefix = CONDA_PREFIX
if IS_WINDOWS:
prefix = os.path.join(prefix, "Library")
self.library_dirs.append(os.path.join(prefix, "lib"))
self.include_dirs.append(os.path.join(prefix, "include"))
def _run_command(self, args):
return subprocess.check_output(args, text=True)
def _configure_gsl(self):
output = self._run_command(["gsl-config", "--cflags"]).split()
if len(output) > 0:
token = output[0]
self.include_dirs.append(token[2:])
output = self._run_command(["gsl-config", "--libs"]).split()
for token in output:
if token.startswith("-L"):
self.library_dirs.append(token[2:])
libdir = "lib"
tskroot = os.path.join("git-submodules", "tskit", "c")
tskdir = os.path.join(tskroot, "tskit")
kasdir = os.path.join(tskroot, "subprojects", "kastore")
includes = ["lwt_interface", libdir, tskroot, tskdir, kasdir]
configurator = PathConfigurator()
msp_source_files = [
"msprime.c",
"fenwick.c",
"avl.c",
"util.c",
"object_heap.c",
"rate_map.c",
"mutgen.c",
"likelihood.c",
]
tsk_source_files = ["core.c", "tables.c", "trees.c"]
kas_source_files = ["kastore.c"]
sources = (
["msprime/_msprimemodule.c"]
+ [os.path.join(libdir, f) for f in msp_source_files]
+ [os.path.join(tskdir, f) for f in tsk_source_files]
+ [os.path.join(kasdir, f) for f in kas_source_files]
)
libraries = ["gsl", "gslcblas"]
defines = []
if IS_WINDOWS:
# Needed for generating UUIDs
libraries.append("Advapi32")
defines += [
# These two are required for GSL to compile and link against the
# conda-forge version.
("GSL_DLL", None),
("WIN32", None),
]
_msprime_module = Extension(
"msprime._msprime",
sources=sources,
extra_compile_args=["-std=c99"],
libraries=libraries,
define_macros=defines,
include_dirs=includes + configurator.include_dirs + [numpy.get_include()],
library_dirs=configurator.library_dirs,
)
setup(
# The package name along with all the other metadata is specified in setup.cfg
# However, GitHub's dependency graph can't see the package unless we put this here.
name="msprime",
ext_modules=[_msprime_module],
)