add_cosmic_rays.py

from ObservationSim.Instrument.Chip import ChipUtils as chip_utils

def add_cosmic_rays(self, chip, filt, tel, pointing, catalog, obs_param):
    self.chip_output.Log_info("  Adding Cosmic-Ray")
    
    # Get exposure time
    if (obs_param) and ("exptime" in obs_param) and (obs_param["exptime"] is not None):
        exptime = obs_param["exptime"]
    else:
        exptime = pointing.exp_time

    chip.img, crmap_gsimg, cr_event_num = chip_utils.add_cosmic_rays(
        img=chip.img, 
        chip=chip, 
        exptime=exptime, 
        seed=self.overall_config["random_seeds"]["seed_CR"]+pointing.id*30+chip.chipID)
    # Save cosmic ray image
    if (obs_param) and ("save_cosmic_img" in obs_param) and (obs_param["save_cosmic_img"] is not None):
        if obs_param["save_cosmic_img"]:
            chip_utils.output_fits_image(
                chip=chip,
                pointing=pointing,
                img=crmap_gsimg,
                output_dir=self.chip_output.subdir,
                img_type='CRS',
                img_type_code=pointing.pointing_type_code,
                project_cycle=self.overall_config["project_cycle"],
                run_counter=self.overall_config["run_counter"]
            )
    return chip, filt, tel, pointing