overall and obs configs for SCIE observation (8f6e5f8f) · Commits · csst-sims / csst_msc_sim

ObservationSim/Config/ChipOutput.py

+1 −3

Original line number	Original line	Diff line number	Diff line
	@@ -60,7 +60,7 @@ class ChipOutput(object):
	self.hdr += additional_column_names		self.hdr += additional_column_names

	def create_output_file(self):		def create_output_file(self):
	if self.pointing_type == 'SCI':		if self.pointing_type == 'SCIE':
	self.cat = open(os.path.join(self.subdir, self.cat_name), "w")		self.cat = open(os.path.join(self.subdir, self.cat_name), "w")
	self.logger.info("Creating catalog file %s ...\n"%(os.path.join(self.subdir, self.cat_name)))		self.logger.info("Creating catalog file %s ...\n"%(os.path.join(self.subdir, self.cat_name)))
	if not self.hdr.endswith("\n"):		if not self.hdr.endswith("\n"):
	@@ -68,8 +68,6 @@ class ChipOutput(object):
	self.cat.write(self.hdr)		self.cat.write(self.hdr)

	def cat_add_obj(self, obj, pos_img, pos_shear):		def cat_add_obj(self, obj, pos_img, pos_shear):
	# ximg = pos_img.x - self.chip.bound.xmin + 1.0
	# yimg = pos_img.y - self.chip.bound.ymin + 1.0
	# self.logger.info('-------------debug-----------------')		# self.logger.info('-------------debug-----------------')
	# self.logger.info('from global',ximg, yimg)		# self.logger.info('from global',ximg, yimg)
	ximg = obj.real_pos.x + 1.0		ximg = obj.real_pos.x + 1.0

ObservationSim/Config/Config.py

+1 −1

Original line number	Original line	Diff line number	Diff line
	@@ -19,7 +19,7 @@ def config_dir(config, work_dir=None, data_dir=None):
	path_dict["data_dir"] = data_dir		path_dict["data_dir"] = data_dir
	# PSF data directory		# PSF data directory
	if config["psf_setting"]["psf_model"] == "Interp":		if config["psf_setting"]["psf_model"] == "Interp":
	path_dict["psf_dir"] = os.path.join(path_dict["data_dir"], config["psf_setting"]["psf_dir"])		path_dict["psf_pho_dir"] = os.path.join(path_dict["data_dir"], config["psf_setting"]["psf_pho_dir"])
	path_dict["psf_sls_dir"] = os.path.join(path_dict["data_dir"], config["psf_setting"]["psf_sls_dir"])		path_dict["psf_sls_dir"] = os.path.join(path_dict["data_dir"], config["psf_setting"]["psf_sls_dir"])

	return path_dict		return path_dict

ObservationSim/Config/Pointing.py

+2 −1

Original line number	Original line	Diff line number	Diff line
	@@ -71,12 +71,13 @@ class Pointing(object):
	self.survey_field_type = "DEEP"		self.survey_field_type = "DEEP"

	# Load the configuration file for this particular pointing		# Load the configuration file for this particular pointing
			# [TODO]
	self.obs_config_file = "/share/home/fangyuedong/20231211/csst-simulation/config/obs_config_SCI_WIDE_phot.yaml"		self.obs_config_file = "/share/home/fangyuedong/20231211/csst-simulation/config/obs_config_SCI_WIDE_phot.yaml"
	with open(self.obs_config_file, "r") as stream:		with open(self.obs_config_file, "r") as stream:
	try:		try:
	self.obs_param = yaml.safe_load(stream)		self.obs_param = yaml.safe_load(stream)
	except yaml.YAMLError as exc:		except yaml.YAMLError as exc:
	print(exc)		print(exc)
	self.pointing_type = self.obs_config_file["obs_type"]		self.pointing_type = self.obs_param["obs_type"]
	else:		else:
	self.timestamp = t		self.timestamp = t

ObservationSim/Instrument/Chip/Chip.py

+23 −31

Original line number	Original line	Diff line number	Diff line
	@@ -57,10 +57,8 @@ class Chip(FocalPlane):
	for key in chip_dict:		for key in chip_dict:
	setattr(self, key, chip_dict[key])		setattr(self, key, chip_dict[key])

	if self.filter_type == "FGS":
	if ("field_dist" in config) and (config["ins_effects"]["field_dist"]) == False:
	self.fdModel = None		self.fdModel = None
	else:		if self.filter_type == "FGS":
	fgs_name = self.chip_name[0:4]		fgs_name = self.chip_name[0:4]
	try:		try:
	with pkg_resources.files('ObservationSim.Instrument.data.field_distortion').joinpath("FieldDistModelGlobal_pr4_%s.pickle"%(fgs_name.lower())) as field_distortion:		with pkg_resources.files('ObservationSim.Instrument.data.field_distortion').joinpath("FieldDistModelGlobal_pr4_%s.pickle"%(fgs_name.lower())) as field_distortion:
	@@ -72,9 +70,6 @@ class Chip(FocalPlane):
	self.fdModel = pickle.load(f)		self.fdModel = pickle.load(f)
	else:		else:
	# Get the corresponding field distortion model		# Get the corresponding field distortion model
	if ("field_dist" in config) and (config["ins_effects"]["field_dist"] == False):
	self.fdModel = None
	else:
	try:		try:
	with pkg_resources.files('ObservationSim.Instrument.data.field_distortion').joinpath("FieldDistModel_v2.0.pickle") as field_distortion:		with pkg_resources.files('ObservationSim.Instrument.data.field_distortion').joinpath("FieldDistModel_v2.0.pickle") as field_distortion:
	with open(field_distortion, "rb") as f:		with open(field_distortion, "rb") as f:
	@@ -117,10 +112,7 @@ class Chip(FocalPlane):
	self.effCurve = self._getChipEffCurve(self.filter_type)		self.effCurve = self._getChipEffCurve(self.filter_type)
	self._getCRdata()		self._getCRdata()

	# Define the sensor model		# # Define the sensor model
	if "bright_fatter" in config["ins_effects"] and config["ins_effects"]["bright_fatter"] == True and self.survey_type == "photometric":
	self.sensor = galsim.SiliconSensor(strength=self.df_strength, treering_func=treering_func)
	else:
	self.sensor = galsim.Sensor()		self.sensor = galsim.Sensor()

	self.flat_cube = None # for spectroscopic flat field cube simulation		self.flat_cube = None # for spectroscopic flat field cube simulation
	@@ -132,9 +124,9 @@ class Chip(FocalPlane):
	self.rotate_angle = 0.		self.rotate_angle = 0.
	self.overscan = 1000		self.overscan = 1000
	# Override default values		# Override default values
	for key in ["gain", "bias_level, dark_exptime", "flat_exptime", "readout_time", "full_well", "read_noise", "dark_noise", "overscan"]:		# for key in ["gain", "bias_level, dark_exptime", "flat_exptime", "readout_time", "full_well", "read_noise", "dark_noise", "overscan"]:
	if key in config["ins_effects"]:		# if key in config["ins_effects"]:
	setattr(self, key, config["ins_effects"][key])		# setattr(self, key, config["ins_effects"][key])

	def _getChipRowCol(self):		def _getChipRowCol(self):
	self.rowID, self.colID = self.getChipRowCol(self.chipID)		self.rowID, self.colID = self.getChipRowCol(self.chipID)

ObservationSim/MockObject/Galaxy.py

+13 −15

Original line number	Original line	Diff line number	Diff line
	import numpy as np		import numpy as np
	import galsim		import galsim
	import os, sys
	import astropy.constants as cons
	from astropy.table import Table		from astropy.table import Table
	from scipy import interpolate

	from ObservationSim.MockObject._util import eObs, integrate_sed_bandpass, getNormFactorForSpecWithABMAG, getObservedSED, getABMAG,convolveGaussXorders		from ObservationSim.MockObject._util import eObs, integrate_sed_bandpass, getNormFactorForSpecWithABMAG
	from ObservationSim.MockObject.SpecDisperser import SpecDisperser		from ObservationSim.MockObject.SpecDisperser import SpecDisperser
	from ObservationSim.MockObject.MockObject import MockObject		from ObservationSim.MockObject.MockObject import MockObject

	@@ -64,6 +61,11 @@ class Galaxy(MockObject):
	bulge_shape = galsim.Shear(g1=self.e1_bulge, g2=self.e2_bulge)		bulge_shape = galsim.Shear(g1=self.e1_bulge, g2=self.e2_bulge)
	bulge = bulge.shear(bulge_shape)		bulge = bulge.shear(bulge_shape)

			if self.bfrac == 0:
			gal = disk
			elif self.bfrac == 1:
			gal = bulge
			else:
	gal = self.bfrac * bulge + (1.0 - self.bfrac) * disk		gal = self.bfrac * bulge + (1.0 - self.bfrac) * disk
	gal = gal.withFlux(nphotons)		gal = gal.withFlux(nphotons)
	if fd_shear is not None:		if fd_shear is not None:
	@@ -155,9 +157,13 @@ class Galaxy(MockObject):
	# Get PSF model		# Get PSF model
	psf, pos_shear = psf_model.get_PSF(chip=chip, pos_img=pos_img, bandpass=bandpass, folding_threshold=folding_threshold)		psf, pos_shear = psf_model.get_PSF(chip=chip, pos_img=pos_img, bandpass=bandpass, folding_threshold=folding_threshold)

			if self.bfrac == 0:
			gal_temp = disk
			elif self.bfrac == 1:
			gal_temp = bulge
			else:
	gal_temp = self.bfrac * bulge + (1.0 - self.bfrac) * disk		gal_temp = self.bfrac * bulge + (1.0 - self.bfrac) * disk
	gal_temp = gal_temp.shear(gal_shear)		gal_temp = gal_temp.shear(gal_shear)

	gal_temp = gal_temp.withFlux(nphotons)		gal_temp = gal_temp.withFlux(nphotons)
	if not big_galaxy: # Not apply PSF for very big galaxy		if not big_galaxy: # Not apply PSF for very big galaxy
	gal_temp = galsim.Convolve(psf, gal_temp)		gal_temp = galsim.Convolve(psf, gal_temp)
	@@ -172,14 +178,6 @@ class Galaxy(MockObject):
	# kfrac = np.random.random()*(1.0 - self.bfrac)		# kfrac = np.random.random()*(1.0 - self.bfrac)
	# gal = self.bfrac * bulge + (1.0 - self.bfrac - kfrac) * disk + kfrac * knots		# gal = self.bfrac * bulge + (1.0 - self.bfrac - kfrac) * disk + kfrac * knots

	# # [C6 TEST]
	# print('xmax = %d, ymax = %d '%(xmax, ymax))
	# # Output memory usage
	# snapshot = tracemalloc.take_snapshot()
	# top_stats = snapshot.statistics('lineno')
	# for stat in top_stats[:10]:
	# print(stat)

	# stamp = gal.drawImage(wcs=chip_wcs_local, method='phot', offset=offset, save_photons=True)		# stamp = gal.drawImage(wcs=chip_wcs_local, method='phot', offset=offset, save_photons=True)
	stamp = gal.drawImage(wcs=chip_wcs_local, offset=offset)		stamp = gal.drawImage(wcs=chip_wcs_local, offset=offset)
	if np.sum(np.isnan(stamp.array)) > 0:		if np.sum(np.isnan(stamp.array)) > 0: