forked from daliu87/spykshrk_realtime
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
26 lines (17 loc) · 1.15 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
from setuptools import setup, find_packages
from setuptools.extension import Extension
from Cython.Build import cythonize
import numpy
rst_sourcefiles = ['RSTPython.pyx',
'RStarTree.c','RSTInterUtil.c','RSTInOut.c','RSTInstDel.c','RSTUtil.c','RSTQuery.c','RSTJoin.c','RSTFunctions.c']
rst_sourcefiles = ['./spykshrk/realtime/rst/' + src for src in rst_sourcefiles]
binary_rec_sourcefiles = ['./spykshrk/realtime/binary_record_cy.pyx']
pp_decode_sourcefiles = ['./spykshrk/franklab/pp_decoder/pp_clusterless_cy.pyx']
extensions = [Extension('spykshrk.realtime.rst.RSTPython', rst_sourcefiles, include_dirs=[numpy.get_include()]),
Extension('spykshrk.realtime.binary_record_cy', binary_rec_sourcefiles, include_dirs=[numpy.get_include()]),
Extension('spykshrk.franklab.pp_decoder.pp_clusterless_cy', pp_decode_sourcefiles, include_dirs=[numpy.get_include()], define_macros=[('CYTHON_TRACE', '1')])
]
setup (name = 'spykshrk_realtime',
ext_modules = cythonize(extensions, compiler_directives={'linetrace': True}),
packages=find_packages('./'),
package_dir={'spykshrk':'./spykshrk'})