diff --git a/ObservationSim/MockObject/SpecDisperser/disperse_c/__init__.py b/ObservationSim/MockObject/SpecDisperser/disperse_c/__init__.py index 8cd2688031ac6c1d1e4614aa67ff0ef0b9b19cdd..89127a04aa30fadbf54cf3817aa7158f1a6a8d55 100644 --- a/ObservationSim/MockObject/SpecDisperser/disperse_c/__init__.py +++ b/ObservationSim/MockObject/SpecDisperser/disperse_c/__init__.py @@ -4,7 +4,7 @@ Cython """ -from . import disperse -from . import interp -#from .disperse import * -#from .interp import * +# from . import disperse +# from . import interp +from .disperse import * +from .interp import * diff --git a/setup.py b/setup.py index e750979c8d4a970aa310e42b99d5b0102991c8fb..5162fa91a410aa42c8ea6c665e0547dceb9371a4 100644 --- a/setup.py +++ b/setup.py @@ -1,5 +1,30 @@ from setuptools import setup, find_packages +from setuptools.extension import Extension +from setuptools.config import read_configuration + +from Cython.Build import cythonize + +import numpy + +extensions = [ + Extension("ObservationSim.MockObject.SpecDisperser.disperse_c.interp", ["ObservationSim/MockObject/SpecDisperser/disperse_c/interp.pyx"], + include_dirs = [numpy.get_include()], + libraries=["m"]), + + Extension("ObservationSim.MockObject.SpecDisperser.disperse_c.disperse", ["ObservationSim/MockObject/SpecDisperser/disperse_c/disperse.pyx"], + include_dirs = [numpy.get_include()], + libraries=["m"]), +] + + +# setup( +# name = "slssim_disperse", +# ext_modules = cythonize(extensions), +# ) + + + setup(name='CSSTSim', version='0.3', packages=find_packages(), @@ -16,4 +41,5 @@ setup(name='CSSTSim', # 'Cython>=0.29.21' # 'numba>=0.50.1' # ] + ext_modules = cythonize(extensions), ) \ No newline at end of file