Upload New File

CpicImgSim/config.py

+import os, yaml
+import warnings
+
+cpism_refdata = 'D:/CPIC_image_process/cpic-img-sim-master/cpism_refdata'
+
+if not os.path.exists(cpism_refdata):  # pragma: no cover
+    raise Exception(
+        "Can not find CPISM reference data.")
+
+
+if not os.path.exists(os.environ.get('PYSYN_CDBS', './trd')):  # pragma: no cover
+    raise Exception(
+        "Can not find PYSYN Stellar reference data.")
+
+# we need to ignore the warning from pysynphot, because we only use the star models.
+with warnings.catch_warnings():  # pragma: no cover
+    warnings.filterwarnings("ignore")
+    import pysynphot as S
+
+solar_spectrum = S.FileSpectrum(
+    f"{os.environ['PYSYN_CDBS']}/grid/solsys/solar_spec.fits")
+solar_spectrum.convert('photlam')
+
+config_file = cpism_refdata + '/optics/optics_config.yaml'
+if not os.path.exists(config_file):  # pragma: no cover
+    raise FileNotFoundError(f"光学配置文件不存在({config_file})")
+with open(config_file, 'r') as f:
+    optics_config = yaml.load(f, Loader=yaml.FullLoader)
+
+MAG_SYSTEM = 'abmag'
+
+__version__ = '1.0.0'
+
+def which_focalplane(band):
+    """
+    Return the name of the focalplane which the band belongs to.
+
+    Parameters
+    -----------
+    band: str
+        The name of the band.
+        from ['f565', 'f661', 'f743', 'f883', 'f940', 'f1265', 'f1425', 'f1542', 'wfs']
+
+    Returns
+    --------
+    str
+        The name of the focalplane.
+        'vis' or 'nir' or 'wfs'
+
+    Raises
+    -------
+    ValueError
+        If the band is not in ['f565', 'f661', 'f743', 'f883', 'f940', 'f1265', 'f1425', 'f1542', 'wfs']
+    """
+    band = band.lower()
+    if band in ['f520', 'f662', 'f720', 'f850']:
+        return 'vis'
+    if band in ['f940', 'f1265', 'f1425', 'f1542']:
+        return 'nir'
+    if band in ['wfs']:
+        return 'wfs'
+
+    raise ValueError(f"未知的波段{band}")
\ No newline at end of file