Commit 5d54f005 authored by yuedong0607's avatar yuedong0607
Browse files

update to meet the L1 input format. Codes reformatted

parent 36b23b01

config/config_injection.yaml

+5 −4
Original line number Diff line number Diff line
@@ -19,10 +19,11 @@ pos_sampling: 
  # type: "uniform"

  # object_density: 37 # arcmin^-2



input_dir: "/home/yuedong/Work/CSST_sim/50sqDeg_Photo_W2/"

output_dir: "/home/yuedong/Work/demo_csst_injection/"

pointing_label_list: ["MSC_0000000"]

chip_label_list: ["09"]

input_dir: "/home/yuedong/Work/CSST_sim/img_data/"

output_dir: "/home/yuedong/Work/demo_csst_injection/new_L1/"

# pointing_label_list: ["MSC_0000000"]

# chip_label_list: ["09"]

input_image_list: ["/home/yuedong/Work/CSST_sim/img_data/CSST_MSC_MS_WIDE_20260820131350_20260820131620_10100328199_09_L1_V01.fits"]



# input_dir: "/public/home/fangyuedong/project/50sqDeg_L1_outputs/50sqDeg_Photo_W3/"

# output_dir: "/public/home/fangyuedong/project/injected_50sqDeg_L1_outputs/g1_002_g_2002_50sqDeg_Photo_W3/"

injection_pipeline/Catalog/C6_SimCat.py

+8 −14
Original line number Diff line number Diff line 
import os

import galsim

import random

import numpy as np

import h5py as h5
@@ -7,25 +6,20 @@ import healpy as hp 
import astropy.constants as cons

import traceback

from itertools import cycle

from astropy.coordinates import spherical_to_cartesian

from astropy.table import Table

from scipy import interpolate

from datetime import datetime



from observation_sim.mock_objects import CatalogBase, Star, Galaxy, Quasar

from observation_sim.mock_objects._util import tag_sed, getObservedSED, getABMAG, integrate_sed_bandpass, comoving_dist

from observation_sim.astrometry.Astrometry_util import on_orbit_obs_position

from observation_sim.mock_objects import CatalogBase, Galaxy

from observation_sim.mock_objects._util import getObservedSED



# (TEST)

from astropy.cosmology import FlatLambdaCDM

from astropy import constants

from astropy import units as U



try:

    import importlib.resources as pkg_resources

except ImportError:

    # Try backported to PY<37 'importlib_resources'

    import importlib_resources as pkg_resources

    pass



# CONSTANTS

NSIDE = 128

injection_pipeline/Grid.py

+16 −18
Original line number Diff line number Diff line 
import math

import numpy as np

import matplotlib.pyplot as plt





class BaseGrid(object):

injection_pipeline/InjectionCatalog.py

+14 −15
Original line number Diff line number Diff line 
import numpy as np

import galsim



import injection_pipeline.Grid as Grid

import injection_pipeline.mathutil as util





class InjectionCatalog(object):



    def __init__(self, image):
@@ -61,7 +61,6 @@ class InjectionCatalog(object): 
        else:

            self.rotate = None





    def _build_grid_kwargs(self, pstype, ps):

        if pstype == 'MixedGrid':

            gtype = ps['grid_type']

injection_pipeline/Injector.py

+4 −7
Original line number Diff line number Diff line 
import galsim

import galsim.config.stamp as stamp

import logging

import os

import numpy as np



import injection_pipeline.Grid as grid



class AddOnImageBuilder(galsim.config.image_scattered.ScatteredImageBuilder):
@@ -22,6 +17,7 @@ class AddOnImageBuilder(galsim.config.image_scattered.ScatteredImageBuilder): 


galsim.config.RegisterImageType('AddOnImage', AddOnImageBuilder())





class InjectImageBuilder(AddOnImageBuilder):



    def setup(self, config, base, image_num, obj_num, ignore, logger):
@@ -87,4 +83,5 @@ class InjectImageBuilder(AddOnImageBuilder): 
                                                            current_var,

                                                            logger)





galsim.config.RegisterImageType('InjectImage', InjectImageBuilder())