Skip to content
GitLab
Commit 22e2bedd authored by Yan Zhaojun's avatar Yan Zhaojun
Browse files

test

parent 04b66f9b
Pipeline #4050 failed with stage
in 0 seconds
Hide whitespace changes
Inline Side-by-side
csst_mci_sim/csst_mci_sim.py
View file @ 22e2bedd
...@@ -1738,8 +1738,6 @@ class MCIsimulator(): ...@@ -1738,8 +1738,6 @@ class MCIsimulator():
# np.save('umag.npy', df3['umag'] ) # np.save('umag.npy', df3['umag'] )
self.star=df3 self.star=df3
del da,df2,df3 del da,df2,df3
self.information['ra_obj'] = self.star['ra_gaia'].mean() self.information['ra_obj'] = self.star['ra_gaia'].mean()
...@@ -1760,109 +1758,109 @@ class MCIsimulator(): ...@@ -1760,109 +1758,109 @@ class MCIsimulator():
#################### cal_PSF_array ################################# #################### cal_PSF_array #################################
if self.save_starpsf==100: # if self.save_starpsf==100:
self.log.info('calculate and save star PSF data......') # self.log.info('calculate and save star PSF data......')
primary_g=fits.PrimaryHDU() # primary_g=fits.PrimaryHDU()
PSF_g=fits.HDUList([primary_g]) # PSF_g=fits.HDUList([primary_g])
primary_r=fits.PrimaryHDU() # primary_r=fits.PrimaryHDU()
PSF_r=fits.HDUList([primary_r]) # PSF_r=fits.HDUList([primary_r])
primary_i=fits.PrimaryHDU() # primary_i=fits.PrimaryHDU()
PSF_i=fits.HDUList([primary_i]) # PSF_i=fits.HDUList([primary_i])
fov=0.05*min(self.information['xsize'],self.information['ysize']) # fov=0.05*min(self.information['xsize'],self.information['ysize'])
dec_arr,ra_arr=make_c_coor(fov,25) # dec_arr,ra_arr=make_c_coor(fov,25)
k=0; # k=0;
####################################################################### # #######################################################################
for ii in range(len(ra_arr)): # for ii in range(len(ra_arr)):
for jj in range(len(ra_arr)): # for jj in range(len(ra_arr)):
################################################################## # ##################################################################
galRa = ra_arr[ii,jj] +center_ra*3600 # ra of PFS, arcsecond # galRa = ra_arr[ii,jj] +center_ra*3600 # ra of PFS, arcsecond
galDec = dec_arr[ii,jj]+center_dec*3600 # dec of PSF, arcsecond # galDec = dec_arr[ii,jj]+center_dec*3600 # dec of PSF, arcsecond
fsx,fsy=cal_pos(center_ra,center_dec, rotTelPos, rotSkyPos, galRa, galDec) # fsx,fsy=cal_pos(center_ra,center_dec, rotTelPos, rotSkyPos, galRa, galDec)
############# do field distottion ########## # ############# do field distottion ##########
if self.distortion: # if self.distortion:
for i in range(3): # for i in range(3):
ch=channel[i] # ch=channel[i]
fpx,fpy=distortField(fsx, fsy, ch) # fpx,fpy=distortField(fsx, fsy, ch)
else: # else:
fpx=fsx # fpx=fsx
fpy=fsy # fpy=fsy
################################################################### # ###################################################################
psf=dict() # psf=dict()
for i in range(3): # for i in range(3):
ch=channel[i] # ch=channel[i]
psfmat=self.get_PSF(fpx, fpy, ch) # psfmat=self.get_PSF(fpx, fpy, ch)
temp=0 # temp=0
for iwave in range(7): # for iwave in range(7):
temp=temp+1/7.0*psfmat[:,:,iwave] # temp=temp+1/7.0*psfmat[:,:,iwave]
temp=temp/temp.sum() # temp=temp/temp.sum()
####rotate the PSF data # ####rotate the PSF data
if abs(theta.deg)>0: # if abs(theta.deg)>0:
psf[ch]=ndimage.rotate(temp, theta.deg, order=1, reshape=False) # here we choose reshape=False, the rotated image will # psf[ch]=ndimage.rotate(temp, theta.deg, order=1, reshape=False) # here we choose reshape=False, the rotated image will
else: # else:
psf[ch]=temp # psf[ch]=temp
################################### # ###################################
hdu_g=fits.ImageHDU(psf['g']) # hdu_g=fits.ImageHDU(psf['g'])
hdu_g.header['ra'] = ra_arr[ii,jj] # hdu_g.header['ra'] = ra_arr[ii,jj]
hdu_g.header['dec'] = dec_arr[ii,jj] # hdu_g.header['dec'] = dec_arr[ii,jj]
hdu_g.header['rot_deg']=-theta.deg # hdu_g.header['rot_deg']=-theta.deg
PSF_g.append(hdu_g) # PSF_g.append(hdu_g)
################################### # ###################################
hdu_r=fits.ImageHDU(psf['r']) # hdu_r=fits.ImageHDU(psf['r'])
hdu_r.header['ra'] = ra_arr[ii,jj] # hdu_r.header['ra'] = ra_arr[ii,jj]
hdu_r.header['dec'] = dec_arr[ii,jj] # hdu_r.header['dec'] = dec_arr[ii,jj]
hdu_r.header['rot_deg']=-theta.deg # hdu_r.header['rot_deg']=-theta.deg
PSF_r.append(hdu_r) # PSF_r.append(hdu_r)
################################### # ###################################
hdu_i=fits.ImageHDU(psf['i']) # hdu_i=fits.ImageHDU(psf['i'])
hdu_i.header['ra'] = ra_arr[ii,jj] # hdu_i.header['ra'] = ra_arr[ii,jj]
hdu_i.header['dec']= dec_arr[ii,jj] # hdu_i.header['dec']= dec_arr[ii,jj]
hdu_i.header['rot_deg']=-theta.deg # hdu_i.header['rot_deg']=-theta.deg
PSF_i.append(hdu_i) # PSF_i.append(hdu_i)
################################### # ###################################
del hdu_g # del hdu_g
del hdu_r # del hdu_r
del hdu_i # del hdu_i
k=k+1 # k=k+1
############################################ # ############################################
file_g=self.result_path+'/PSF_Data/'+'PSF_sim_No.'+str(self.information['simnumber'])+'_C1.fits' # file_g=self.result_path+'/PSF_Data/'+'PSF_sim_No.'+str(self.information['simnumber'])+'_C1.fits'
PSF_g.writeto(file_g,overwrite=True) # PSF_g.writeto(file_g,overwrite=True)
file_r=self.result_path+'/PSF_Data/'+'PSF_sim_No.'+str(self.information['simnumber'])+'_C2.fits' # file_r=self.result_path+'/PSF_Data/'+'PSF_sim_No.'+str(self.information['simnumber'])+'_C2.fits'
PSF_r.writeto(file_r,overwrite=True) # PSF_r.writeto(file_r,overwrite=True)
file_i=self.result_path+'/PSF_Data/'+'PSF_sim_No.'+str(self.information['simnumber'])+'_C3.fits' # file_i=self.result_path+'/PSF_Data/'+'PSF_sim_No.'+str(self.information['simnumber'])+'_C3.fits'
PSF_i.writeto(file_i,overwrite=True) # PSF_i.writeto(file_i,overwrite=True)
del PSF_g # del PSF_g
del PSF_r # del PSF_r
del PSF_i # del PSF_i
########## finish save PSF fits ########################################## ########## finish save PSF fits ##########################################
############################################################################ ############################################################################
...@@ -1878,6 +1876,7 @@ class MCIsimulator(): ...@@ -1878,6 +1876,7 @@ class MCIsimulator():
nsrcs=len(self.star['ra_gaia']) + 8 nsrcs=len(self.star['ra_gaia']) + 8
self.log.info('load star catlog successfully') self.log.info('load star catlog successfully')
################################################################## ##################################################################
...@@ -1910,6 +1909,8 @@ class MCIsimulator(): ...@@ -1910,6 +1909,8 @@ class MCIsimulator():
####################################################################### #######################################################################
nlayccd = 0 nlayccd = 0
self.log.info('load star catlog successfully 0000000')
############################################ ############################################
ra = self.information['ra_pnt0'] ra = self.information['ra_pnt0']
...@@ -1922,11 +1923,11 @@ class MCIsimulator(): ...@@ -1922,11 +1923,11 @@ class MCIsimulator():
z_sat=float(self.orbit_pars[self.orbit_exp_num,3]) z_sat=float(self.orbit_pars[self.orbit_exp_num,3])
self.log.info('load star catlog 111') self.log.info('load star catlog 111111111111111')
wave0, zodi0 = self.zodiacal(ra, dec, self.TianCe_day) # erg/s/cm^2/A/arcsec^2 wave0, zodi0 = self.zodiacal(ra, dec, self.TianCe_day) # erg/s/cm^2/A/arcsec^2
self.log.info('load star catlog 2222') self.log.info('load star catlog 2222222222222222222')
# EarthShine from straylight # EarthShine from straylight
sl = StrayLight(self.information['dir_path'],jtime=time_jd, sat=np.array([x_sat, y_sat, z_sat]), sl = StrayLight(self.information['dir_path'],jtime=time_jd, sat=np.array([x_sat, y_sat, z_sat]),
radec=np.array([(ra*u.degree).value, (dec*u.degree).value])) radec=np.array([(ra*u.degree).value, (dec*u.degree).value]))
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment