update codes for fitsStamp (6f2f8a75) · Commits · Liu Dezi / csst_msc_sim

Catalog/C6_fits_Catalog.py

0 → 100644

+674 −0

File added.

Preview size limit exceeded, changes collapsed.

ObservationSim/MockObject/MockObject.py

+4 −0

Original line number	Diff line number	Diff line
		@@ -23,6 +23,10 @@ class MockObject(object):
		self.type = "star"
		elif self.param["star"] == 2:
		self.type = "quasar"
		###mock_stamp_START
		elif self.param["star"] == 3:
		self.type = "stamp"
		###mock_stamp_END

		# self.id = self.param["id"]
		# self.ra = self.param["ra"]

ObservationSim/MockObject/Stamp.py

0 → 100644

+123 −0

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

		import astropy.io.fits as fitsio

		import galsim
		import gc
		from ObservationSim.MockObject.MockObject import MockObject

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

		class Stamp(MockObject):
		def __init__(self, param):
		super().__init__(param)

		def unload_SED(self):
		"""(Test) free up SED memory
		"""
		del self.sed

		def drawObj_multiband(self, tel, pos_img, psf_model, bandpass_list, filt, chip, nphotons_tot=None, g1=0, g2=0, exptime=150.):
		if nphotons_tot == None:
		nphotons_tot = self.getElectronFluxFilt(filt, tel, exptime)

		try:
		full = integrate_sed_bandpass(sed=self.sed, bandpass=filt.bandpass_full)
		except Exception as e:
		print(e)
		self.logger.error(e)
		return False

		nphotons_sum = 0
		photons_list = []
		xmax, ymax = 0, 0

		if self.getMagFilter(filt) <= 15:
		folding_threshold = 5.e-4
		else:
		folding_threshold = 5.e-3
		gsp = galsim.GSParams(folding_threshold=folding_threshold)

		self.real_pos = self.getRealPos(chip.img, global_x=self.posImg.x, global_y=self.posImg.y,
		img_real_wcs=self.real_wcs)

		x, y = self.real_pos.x + 0.5, self.real_pos.y + 0.5
		x_nominal = int(np.floor(x + 0.5))
		y_nominal = int(np.floor(y + 0.5))
		dx = x - x_nominal
		dy = y - y_nominal
		offset = galsim.PositionD(dx, dy)

		real_wcs_local = self.real_wcs.local(self.real_pos)

		for i in range(len(bandpass_list)):
		bandpass = bandpass_list[i]

		try:
		sub = integrate_sed_bandpass(sed=self.sed, bandpass=bandpass)
		except Exception as e:
		print(e)
		self.logger.error(e)
		# return False
		continue

		ratio = sub/full
		if not (ratio == -1 or (ratio != ratio)):
		nphotons = ratio * nphotons_tot
		else:
		# return False
		continue
		nphotons_sum += nphotons

		psf, pos_shear = psf_model.get_PSF(chip=chip, pos_img=pos_img, bandpass=bandpass, folding_threshold=folding_threshold)

		_gal = self.param['image']
		galIm = galsim.ImageF(_gal, scale=self.param['pixScale'])
		gal = galsim.InterpolatedImage(galIm)
		gal = gal.withFlux(nphotons)
		#gal_shear = galsim.Shear(g1=g1, g2=g2)
		#gal = gal.shear(gal_shear)

		gal = galsim.Convolve(psf, gal)

		stamp = gal.drawImage(wcs=real_wcs_local, method='phot', offset=self.offset, save_photons=True)

		xmax = max(xmax, stamp.xmax - stamp.xmin)
		ymax = max(ymax, stamp.ymax - stamp.ymin)

		photons = stamp.photons
		photons.x += x_nominal
		photons.y += y_nominal
		photons_list.append(photons)
		del gal

		# print('xmax = %d, ymax = %d '%(xmax, ymax))

		stamp = galsim.ImageF(int(xmax1.1), int(ymax1.1))
		stamp.wcs = real_wcs_local
		stamp.setCenter(x_nominal, y_nominal)
		bounds = stamp.bounds & galsim.BoundsI(0, chip.npix_x - 1, 0, chip.npix_y - 1)

		if bounds.area() > 0:
		chip.img.setOrigin(0, 0)
		stamp[bounds] = chip.img[bounds]
		for i in range(len(photons_list)):
		if i == 0:
		chip.sensor.accumulate(photons_list[i], stamp)
		else:
		chip.sensor.accumulate(photons_list[i], stamp, resume=True)

		chip.img[bounds] = stamp[bounds]

		chip.img.setOrigin(chip.bound.xmin, chip.bound.ymin)

		del photons_list
		del stamp
		gc.collect()
		return True, pos_shear

ObservationSim/ObservationSim.py

+20 −3

Original line number	Diff line number	Diff line
		@@ -192,12 +192,29 @@ class Observation(object):

		obj = self.cat.objs[j]

		if obj.type == 'star' and self.config["run_option"]["galaxy_only"]:
		#if obj.type == 'star' and self.config["run_option"]["galaxy_only"]:
		# continue
		#elif obj.type == 'galaxy' and self.config["run_option"]["star_only"]:
		# continue
		#elif obj.type == 'quasar' and self.config["run_option"]["star_only"]:
		# continue

		###mock_stamp_START
		if obj.type == 'star' and not self.config["run_option"]["star_yes"]:
		continue
		elif obj.type == 'galaxy' and not self.config["run_option"]["galaxy_yes"]:
		continue
		elif obj.type == 'galaxy' and self.config["run_option"]["star_only"]:
		elif obj.type == 'quasar' and not self.config["run_option"]["galaxy_yes"]:
		continue
		elif obj.type == 'quasar' and self.config["run_option"]["star_only"]:
		elif obj.type == 'stamp' and not self.config["run_option"]["stamp_yes"]:
		continue
		#if obj.type == 'star' or obj.type == 'galaxy':
		# continue #for test

		#mu_temp = 1.
		#if obj.type == 'stamp':
		# mu_temp = obj.param["mu"]
		###mock_stamp_END

		# load and convert SED; also caculate object's magnitude in all CSST bands
		try:

run_sim.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -104,5 +104,5 @@ if __name__=='__main__':
		# run_sim(Catalog=NGPCatalog)
		# from Catalog.NJU_Catalog import NJU_Catalog
		# run_sim(Catalog=NJU_Catalog)
		from Catalog.C6_Catalog import C6_Catalog
		run_sim(Catalog=C6_Catalog)
		from Catalog.C6_fits_Catalog import C6_fits_Catalog
		run_sim(Catalog=C6_fits_Catalog)