-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
30 lines (24 loc) · 848 Bytes
/
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
from __future__ import division, absolute_import, print_function
from glob import glob
import os
import numpy as np
from setuptools import setup
from setuptools import Extension
from Cython.Build import cythonize
from Cython.Distutils import build_ext
from glob import glob
import platform
extra_args = ['/openmp'] if platform.system() == "Windows" else ['-fopenmp']
here = os.path.abspath(os.path.dirname(__file__))
srvf_ext = Extension("srvf_register.dynamic_programming_q2",
sources = ["src/dynamic_programming_q2.pyx", "src/dp_grid.c"],
extra_compile_args=extra_args,
extra_link_args=extra_args,
language="c")
setup(
name="srvf_register",
version="0.0",
packages=["srvf_register"],
ext_modules = cythonize(srvf_ext,language="c"),
include_dirs = [np.get_include(),"."]
)