__init__.py 1.48 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import os

class SimSteps:
    def __init__(self, overall_config, chip_output, all_filters):
        self.overall_config = overall_config
        self.chip_output = chip_output
        self.all_filters = all_filters

    from .prepare_headers import prepare_headers
    from .add_sky_background import add_sky_background_sci, add_sky_flat_calibration, add_sky_background
    from .add_objects import add_objects
    from .add_cosmic_rays import add_cosmic_rays
    from .add_pattern_noise import apply_PRNU, add_poisson_and_dark, add_detector_defects, add_nonlinearity, add_blooming, add_bias
    from .add_brighter_fatter_CTE import add_brighter_fatter, apply_CTE
    from .readout_output import add_prescan_overscan, add_readout_noise, apply_gain, quantization_and_output
Zhang Xin's avatar
Zhang Xin committed
16
    from .add_LED_flat import add_LED_Flat
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

SIM_STEP_TYPES = {
    "scie_obs": "add_objects",
    "sky_background": "add_sky_background",
    "cosmic_rays": "add_cosmic_rays",
    "PRNU_effect": "apply_PRNU",
    "poisson_and_dark": "add_poisson_and_dark",
    "bright_fatter": "add_brighter_fatter",
    "detector_defects": "add_detector_defects",
    "nonlinearity": "add_nonlinearity",
    "blooming": "add_blooming",
    "CTE_effect": "apply_CTE",
    "prescan_overscan": "add_prescan_overscan",
    "bias": "add_bias",
    "readout_noise": "add_readout_noise",
    "gain": "apply_gain",
    "quantization_and_output": "quantization_and_output",
    "led_calib_model":"add_LED_Flat",
    "sky_flatField":"add_sky_flat_calibration",
}