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='csst_msc_sim',
      version='3.0.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': ['*.fits', '*.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}
      )