-
Notifications
You must be signed in to change notification settings - Fork 3
/
setup.py
27 lines (22 loc) · 885 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
import os
import sys
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
import numpy as np
"""
The module uses Eigen C++ library and therefore
setup.py needs to be able to locate it.
Change the last directory in include_dirs
for your local directory containing Eigen
header files.
"""
cur_dir = os.path.dirname(__file__)
lib_dir = os.path.join(cur_dir, r'lib')
os.chdir(cur_dir)
sourcefiles = [r"pyhawkes.pyx", r"lib\exp_hawkes.cpp",
r"lib\power_hawkes.cpp", r"lib\general_hawkes.cpp", r"lib\additional_functions.cpp"]
setup(cmdclass={'build_ext': build_ext},
ext_modules=[Extension("pyhawkes", sourcefiles, language="c++",
include_dirs=[".", np.get_include(), lib_dir,
r"D:\Materials\Programming\C++\Libraries\include"])])