-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
57 lines (48 loc) · 1.28 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
#! /usr/bin/env python
import os
import sys
import numpy as np
import versioneer
from setuptools import find_packages, setup
from distutils.extension import Extension
from model_metadata.utils import get_cmdclass, get_entry_points
common_flags = {
"include_dirs": [np.get_include(), os.path.join(sys.prefix, "include")],
"library_dirs": [],
"define_macros": [],
"undef_macros": [],
"extra_compile_args": [],
"language": "c",
}
libraries = []
ext_modules = [
Extension(
"pymt_cem.lib.cem",
["pymt_cem/lib/cem.pyx"],
libraries=libraries + ["bmi_cem"],
**common_flags,
),
Extension(
"pymt_cem.lib.waves",
["pymt_cem/lib/waves.pyx"],
libraries=libraries + ["bmi_waves"],
**common_flags,
),
]
packages = find_packages()
pymt_components = [
("Cem=pymt_cem.bmi:Cem", "meta/Cem"),
("Waves=pymt_cem.bmi:Waves", "meta/Waves"),
]
cmdclass = get_cmdclass(pymt_components, cmdclass=versioneer.get_cmdclass())
setup(
name="pymt_cem",
author="Eric Hutton",
description="PyMT plugin for cem",
version=versioneer.get_version(),
setup_requires=["cython"],
ext_modules=ext_modules,
packages=packages,
cmdclass=cmdclass,
entry_points=get_entry_points(pymt_components),
)