setup.py

'''
Author: Zhang Xin zhangx@bao.ac.cn
Date: 2023-08-07 11:23:28
LastEditors: Zhang Xin zhangx@bao.ac.cn
LastEditTime: 2023-10-08 14:44:19
FilePath: /undefined/Users/zhangxin/Work/SlitlessSim/CSST_SIM/CSST_develop/csst-simulation/setup.py
Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
'''
from setuptools import setup, find_packages

from setuptools.extension import Extension
from setuptools.config import read_configuration
from distutils.command.build_ext import build_ext

from Cython.Build import cythonize

import numpy

class CTypes(Extension): pass

class build_ext(build_ext):

    def build_extension(self, ext):
        self._ctypes = isinstance(ext, CTypes)
        return super().build_extension(ext)

    def get_export_symbols(self, ext):
        if self._ctypes:
            return ext.export_symbols
        return super().get_export_symbols(ext)

    def get_ext_filename(self, ext_name):
        if self._ctypes:
            return ext_name + '.so'
        return super().get_ext_filename(ext_name)

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"]),
]

df_module = [CTypes('ObservationSim.Instrument.Chip.libBF.libmoduleBF',
                    ['ObservationSim/Instrument/Chip/libBF/diffusion_X1.c', 'ObservationSim/Instrument/Chip/libBF/nrutil.c'],
                    include_dirs=['ObservationSim/Instrument/Chip/libBF/', '/usr/include']
)]
cti_module = [CTypes('ObservationSim.Instrument.Chip.libCTI.libmoduleCTI',
                    ['ObservationSim/Instrument/Chip/libCTI/src/add_CTI.c', 'ObservationSim/Instrument/Chip/libCTI/src/nrutil.c', 'ObservationSim/Instrument/Chip/libCTI/src/ran1.c', 'ObservationSim/Instrument/Chip/libCTI/src/ran2.c', 'ObservationSim/Instrument/Chip/libCTI/src/poidev.c', 'ObservationSim/Instrument/Chip/libCTI/src/gammln.c', 'ObservationSim/Instrument/Chip/libCTI/src/gasdev.c', 'ObservationSim/Instrument/Chip/libCTI/src/sort.c', 'ObservationSim/Instrument/Chip/libCTI/src/creattraps.c'],
                    include_dirs=['ObservationSim/Instrument/Chip/libCTI/src/', '/usr/include']
)]


# setup(
#     name = "slssim_disperse",
#     ext_modules = cythonize(extensions),
# )
# 读取依赖列表requirements.txt
# 忽略#开头或者版本号不明确指定的条目
with open("requirements.txt", "r") as f:
    requirements = [
        req.strip()
        for req in f.readlines()
        if not req.startswith("#") and req.__contains__("==")
    ]

setup(name='CSSTSim', 
    version='2.1.0', 
    packages=find_packages(),
    # install_requires=[
    #     # 'numpy>=1.18.5',
    #     # 'galsim>=2.2.4',
    #     # 'pyyaml>=5.3.1',
    #     # 'astropy>=4.0.1',
    #     # 'scipy>=1.5.0',
    #     # 'mpi4py>=3.0.3',
    #     # 'sep>=1.0.3',
    #     # 'healpy>=1.14.0',
    #     # 'h5py>=2.10.0',
    #     # 'Cython>=0.29.21',
    #     # 'numba>=0.50.1'
    # ],
    
    package_data = {
        'ObservationSim.Astrometry.lib': ['libshao.so'], 
        'ObservationSim.Instrument.Chip.libBF': ['libmoduleBF.so'],
        'ObservationSim.Instrument.Chip.libCTI': ['libmoduleCTI.so'],
        'ObservationSim.MockObject.data': ['*.dat'], 
        'ObservationSim.MockObject.data.led': ['*.fits'],
        'ObservationSim.Instrument.data': ['*.txt', '*.dat', '*.json'],
        'ObservationSim.Instrument.data.field_distortion': ['*.pickle'],
        'ObservationSim.Instrument.data.ccd': ['*.txt','*.json'], 
        'ObservationSim.Instrument.data.filters': ['*.txt', '*.list', '*.dat'], 
        'ObservationSim.Instrument.data.throughputs': ['*.txt', '*.dat'], 
        'ObservationSim.Instrument.data.sls_conf': ['*.conf', '*.fits'],
        'ObservationSim.Instrument.data.flatCube': ['*.fits'],
        'Catalog.data': ['*.fits','*.so'],
        'ObservationSim.Config.Header':['*.header','*.lst'],
        'ObservationSim.Straylight.data': ['*.dat'],
        'ObservationSim.Straylight.data.sky': ['*.dat'],
        'ObservationSim.Straylight.lib': ['*'],
    },
    python_requires=">=3.11",  # Python版本要求
    install_requires=requirements,

    ext_modules = cythonize(extensions) + df_module + cti_module,
    cmdclass={'build_ext': build_ext}
)