debug (3c8dbef0) · Commits · csst-sims / csst_mci_sim

csst_mci_sim/CTI/pycache/CTI.cpython-311.pyc

0 → 100644

+8.5 KiB

File added.

No diff preview for this file type.

View file

csst_mci_sim/CTI/pycache/init.cpython-311.pyc

0 → 100644

+169 B

File added.

No diff preview for this file type.

View file

csst_mci_sim/csst_mci_sim.py

+9 −84

Original line number	Diff line number	Diff line
		@@ -61,8 +61,8 @@ from datetime import datetime, timedelta
		from astropy.time import Time
		from astropy.coordinates import get_sun

		import scipy.io as sio
		import cmath
		# import scipy.io as sio
		# import cmath

		import numpy as np

		@@ -70,7 +70,7 @@ from tqdm import tqdm

		from astropy.table import Table

		from astropy.wcs import WCS as WCS
		# from astropy.wcs import WCS as WCS

		from astropy.io import fits

		@@ -80,9 +80,9 @@ import os, sys, math

		import configparser as ConfigParser

		from optparse import OptionParser
		# from optparse import OptionParser

		from matplotlib import pyplot as plt
		# from matplotlib import pyplot as plt

		from scipy import ndimage

		@@ -115,20 +115,6 @@ import pandas
		# sys.path.append('../MCI_inputData/SED_Code')






		# import jax
		#import jax.numpy as jnp

		# from jax import config
		# config.update("jax_enable_x64", True)

		# os.environ['CUDA_VISIBLE_DEVICES'] = '0'
		# devices = jax.local_devices()


		# set the folder
		FOLDER ='../'

		@@ -141,19 +127,6 @@ import astropy.coordinates as coord

		from scipy.interpolate import interp1d

		#from sky_bkg import sky_bkg


		filterPivotWave = {'nuv':2875.5,'u':3629.6,'g':4808.4,'r':6178.2, 'i':7609.0, 'z':9012.9,'y':9627.9}
		filterIndex = {'nuv':0,'u':1,'g':2,'r':3, 'i':4, 'z':5,'y':6}
		# filterCCD = {'nuv':'UV0','u':'UV0','g':'Astro_MB','r':'Astro_MB', 'i':'Basic_NIR', 'z':'Basic_NIR','y':'Basic_NIR'}
		# bandRange = {'nuv':[2504.0,3230.0],'u':[3190.0,4039.0],'g':[3989.0,5498.0],'r':[5438.0,6956.0], 'i':[6886.0,8469.0],
		# 'z':[8379.0,10855.0],'y':[9217.0, 10900.0], 'GU':[2550, 4000],'GV':[4000, 6200],'GI':[6200,10000]}
		# Instrument_dir = '/Users/linlin/Data/csst/straylightsim-master/Instrument/'
		# SpecOrder = ['-2','-1','0','1','2']
		#
		# filterMirrorEff = {'nuv':0.54,'u':0.68,'g':0.8,'r':0.8, 'i':0.8, 'z':0.8,'y':0.8}


		def transRaDec2D(ra, dec):
		# radec转为竞天程序里的ob, 赤道坐标系下的笛卡尔三维坐标xyz.
		@@ -163,53 +136,6 @@ def transRaDec2D(ra, dec):
		return np.array([x1, y1, z1])
		###############################################################################

		# def flux2ill(wave, flux):

		# # erg/s/cm^2/A/arcsec^2 to W/m^2

		# # 1 W/m^2/sr/μm = 0.10 erg/cm^2/s/sr/A
		# # 1 sr = 1 rad^2 = 4.25452e10 arcsec^2
		# # 1 J/s = 1 W
		# # 1 J = 10^7 erg

		# # convert erg/s/cm^2/A/arcsec^2 to erg/s/cm^2/A/sr
		# flux1 = flux / (1/4.25452e10)
		# # convert erg/s/cm^2/A/sr to W/m^2/sr/um
		# flux2 = flux1 * 10

		# # 对接收面积积分，输出单位 W/m^2/nm
		# D = 2 # meter
		# f = 28 # meter
		# flux3 = flux2 * np.pi * D2 / 4 / f2 / 10**3

		# # # # u_lambda: W/m^2/nm
		# # # obj0:
		# # V = 73 * 1e-8 # W/m^2/nm
		# # Es = V * np.pi * D2 / 4 / f2 / 10**3
		# # c1 = 3.7418e-16
		# # c2 = 1.44e-2
		# # t = 5700
		# # # wave需要代入meter.
		# # wave0 = np.arange(380, 780) # nm
		# # delta_lamba0 = 1 # nm
		# # u_lambda = c1 / (wave01e-9)5 / (np.exp(c2 / (wave01e-9) / t) - 1)
		# # f_lambda = (u_lambda / u_lambda[wave0 == 500]) * Es
		# # E0 = np.sum(f_lambda * 1)
		# # # plt.plot(wave, u_lambda)
		# # # plt.show()

		# # 对波长积分
		# f = interp1d(wave, flux3)
		# wave_interp = np.arange(3800, 7800)
		# flux3_interp = f(wave_interp)
		# # 输出单位 W/m^2
		# delta_lamba = 0.1 # nm
		# E = np.sum(flux3_interp * delta_lamba)

		# # pdb.set_trace()

		# return E

		################################################################
		def ill2flux(E,path):

		@@ -356,6 +282,8 @@ class StrayLight(object):

		def caculateStarLightFilter(self, filter='i'):

		filterIndex = {'nuv':0,'u':1,'g':2,'r':3, 'i':4, 'z':5,'y':6}

		sat = (ctypes.c_double*3)()
		sat[:] = self.sat
		ob = (ctypes.c_double*3)()
		@@ -376,6 +304,8 @@ class StrayLight(object):
		return max(band_star_e1, band_star_e2)

		def caculateEarthShineFilter(self, filter='i'):

		filterIndex = {'nuv':0,'u':1,'g':2,'r':3, 'i':4, 'z':5,'y':6}
		sat = (ctypes.c_double*3)()
		sat[:] = self.sat
		ob = (ctypes.c_double*3)()
		@@ -1893,9 +1823,6 @@ class MCIsimulator():

		# EarthShine from straylight
		sl = StrayLight(self.information['dir_path'],jtime=time_jd, sat=np.array([x_sat, y_sat, z_sat]), radec=np.array([(rau.degree).value, (decu.degree).value]))



		earth_e = sl.caculateEarthShineFilter(filter='r')
		star_e = sl.caculateStarLightFilter(filter='r')
		angle_earth = earth_angle(time_jd, x_sat, y_sat, z_sat, ra, dec)
		@@ -1905,8 +1832,6 @@ class MCIsimulator():

		earthshine_wave0, earthshine_flux0 = ill2flux(earth_e+star_e, self.information['dir_path'])



		# sample as mci wavelength
		wave_mci = np.linspace(2500, 11000, 8501) #np.arange(2500, 11000, 1)
		f1 = interp1d(wave0, zodi0)

csst_mci_sim/earthshine.py

0 → 100644

+216 −0

Original line number	Diff line number	Diff line

		import numpy as np
		import julian
		from datetime import datetime
		from astropy.time import Time
		from astropy.coordinates import get_sun
		from astropy.coordinates import SkyCoord
		import astropy.coordinates as coord
		import pandas as pd
		from astropy import units as u

		from scipy.interpolate import interp1d

		from scipy import interpolate

		import ctypes


		def transRaDec2D(ra, dec):
		# radec转为竞天程序里的ob, 赤道坐标系下的笛卡尔三维坐标xyz.
		x1 = np.cos(dec / 57.2957795) * np.cos(ra / 57.2957795)
		y1 = np.cos(dec / 57.2957795) * np.sin(ra / 57.2957795)
		z1 = np.sin(dec / 57.2957795)
		return np.array([x1, y1, z1])


		def earth_angle(time_jd, x_sat, y_sat, z_sat, ra_obj, dec_obj):

		ra_sat = np.arctan2(y_sat, x_sat) / np.pi * 180
		dec_sat = np.arctan2(z_sat, np.sqrt(x_sat2+y_sat2)) / np.pi * 180
		radec_sat = SkyCoord(ra=ra_sat*u.degree,
		dec=dec_sat*u.degree, frame='gcrs')
		lb_sat = radec_sat.transform_to('geocentrictrueecliptic')

		# get the obj location
		radec_obj = SkyCoord(ra=ra_obj*u.degree,
		dec=dec_obj*u.degree, frame='gcrs')
		lb_obj = radec_obj.transform_to('geocentrictrueecliptic')

		# calculate the angle between sub-satellite point and the earth side
		earth_radius = 6371 # km
		sat_height = np.sqrt(x_sat2 + y_sat2 + z_sat**2)
		angle_a = np.arcsin(earth_radius/sat_height) / np.pi * 180

		# calculate the angle between satellite position and the target position
		angle_b = lb_sat.separation(lb_obj)

		# calculat the earth angle
		angle = 180 - angle_a - angle_b.degree

		return angle

		###############################################################################


		def ill2flux(E, path):

		# use template from sky_bkg (background_spec_hst.dat)
		filename = path+'MCI_inputData/refs/background_spec_hst.dat'
		cat_spec = pd.read_csv(filename, sep='\s+', header=None, comment='#')
		wave0 = cat_spec[0].values # A
		spec0 = cat_spec[2].values # erg/s/cm^2/A/arcsec^2

		# convert erg/s/cm^2/A/arcsec^2 to erg/s/cm^2/A/sr
		flux1 = spec0 / (1/4.25452e10)
		# convert erg/s/cm^2/A/sr to W/m^2/sr/um
		flux2 = flux1 * 10

		# 对接收面积积分，输出单位 W/m^2/nm
		D = 2 # meter
		f = 28 # meter, 焦距，转换关系来源于王维notes.
		flux3 = flux2 * np.pi * D2 / 4 / f2 / 10**3

		f = interp1d(wave0, flux3)
		wave_range = np.arange(3800, 7800)
		flux3_mean = f(wave_range)
		delta_lamba = 0.1 # nm
		E0 = np.sum(flux3_mean * delta_lamba)

		factor = E / E0
		spec_scaled = factor * spec0

		return wave0, spec_scaled

		##############################################################


		def earthshine(path, time_jd, x_sat, y_sat, z_sat, ra, dec):
		# EarthShine from straylight
		sl = StrayLight(path, jtime=time_jd, sat=np.array(
		[x_sat, y_sat, z_sat]), radec=np.array([(rau.degree).value, (decu.degree).value]))

		earth_e = sl.caculateEarthShineFilter(filter='r')

		angle_earth = earth_angle(time_jd, x_sat, y_sat, z_sat, ra, dec)

		if angle_earth < 0:
		earth_e = 0

		earthshine_wave0, earthshine_flux0 = ill2flux(earth_e, path)

		# sample as mci wavelength
		wave_mci = np.linspace(2500, 11000, 8501) # np.arange(2500, 11000, 1)

		f2 = interp1d(earthshine_wave0, earthshine_flux0)
		earthshine = f2(wave_mci)

		return earthshine
		#################################################################


		class StrayLight(object):

		def __init__(self, path, jtime=2460843., sat=np.array([0, 0, 0]), radec=np.array([0, 0])):

		self.path = path
		self.jtime = jtime
		self.sat = sat
		self.equator = coord.SkyCoord(
		radec[0]u.degree, radec[1]u.degree, frame='icrs')
		self.ecliptic = self.equator.transform_to('barycentrictrueecliptic')
		self.pointing = transRaDec2D(radec[0], radec[1])
		self.slcdll = ctypes.CDLL(
		self.path+'MCI_inputData/refs/libstraylight.so') # dylib

		self.slcdll.Calculate.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.POINTER(
		ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.c_char_p]

		self.slcdll.PointSource.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.POINTER(
		ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.c_char_p]

		self.slcdll.EarthShine.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.POINTER(
		ctypes.c_double),
		ctypes.POINTER(ctypes.c_double)]

		self.slcdll.Zodiacal.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double)]
		self.slcdll.ComposeY.argtypes = [ctypes.POINTER(ctypes.c_double), ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double)]
		self.slcdll.Init.argtypes = [
		ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p]
		self.deFn = self.path+"MCI_inputData/refs/DE405"
		self.PSTFn = self.path+"MCI_inputData/refs/PST"
		self.RFn = self.path+"MCI_inputData/refs/R"
		self.ZolFn = self.path+"MCI_inputData/refs/Zodiacal"
		self.brightStarTabFn = self.path+"MCI_inputData/refs/BrightGaia_with_csst_mag"

		self.slcdll.Init(str.encode(self.deFn), str.encode(
		self.PSTFn), str.encode(self.RFn), str.encode(self.ZolFn))

		def caculateStarLightFilter(self, filter='i'):
		filterIndex = {'nuv': 0, 'u': 1, 'g': 2,
		'r': 3, 'i': 4, 'z': 5, 'y': 6}

		sat = (ctypes.c_double*3)()
		sat[:] = self.sat
		ob = (ctypes.c_double*3)()
		ob[:] = self.pointing

		py1 = (ctypes.c_double*3)()
		py2 = (ctypes.c_double*3)()
		self.slcdll.ComposeY(ob, py1, py2)

		star_e1 = (ctypes.c_double*7)()
		self.slcdll.PointSource(self.jtime, sat, ob, py1,
		star_e1, str.encode(self.brightStarTabFn))
		star_e2 = (ctypes.c_double*7)()
		self.slcdll.PointSource(self.jtime, sat, ob, py2,
		star_e2, str.encode(self.brightStarTabFn))

		band_star_e1 = star_e1[:][filterIndex[filter]]
		band_star_e2 = star_e2[:][filterIndex[filter]]

		return max(band_star_e1, band_star_e2)

		def caculateEarthShineFilter(self, filter='i'):

		filterIndex = {'nuv': 0, 'u': 1, 'g': 2,
		'r': 3, 'i': 4, 'z': 5, 'y': 6}
		sat = (ctypes.c_double*3)()
		sat[:] = self.sat
		ob = (ctypes.c_double*3)()
		ob[:] = self.pointing

		py1 = (ctypes.c_double*3)()
		py2 = (ctypes.c_double*3)()
		self.slcdll.ComposeY(ob, py1, py2)

		earth_e1 = (ctypes.c_double*7)()
		self.slcdll.EarthShine(self.jtime, sat, ob, py1,
		earth_e1) # e[7]代表7个波段的照度

		earth_e2 = (ctypes.c_double*7)()
		self.slcdll.EarthShine(self.jtime, sat, ob, py2, earth_e2)

		band_earth_e1 = earth_e1[:][filterIndex[filter]]
		band_earth_e2 = earth_e2[:][filterIndex[filter]]

		return max(band_earth_e1, band_earth_e2)


		###############################################################################
		### test
		# path='/home/yan/MCI/'
		# time_jd = 2460417.59979167
		# x_sat = -4722.543136
		# y_sat = -1478.219213
		# z_sat = 4595.402769
		# ra = 116.18081536720157
		# dec= 39.42316681016602
		# earthshine0=earthshine(path,time_jd, x_sat, y_sat, z_sat, ra, dec)

csst_mci_sim/shraylight.py

0 → 100644

+216 −0

Original line number	Diff line number	Diff line

		import numpy as np
		import julian
		from datetime import datetime
		from astropy.time import Time
		from astropy.coordinates import get_sun
		from astropy.coordinates import SkyCoord
		import astropy.coordinates as coord
		import pandas as pd
		from astropy import units as u

		from scipy.interpolate import interp1d

		from scipy import interpolate

		import ctypes


		def transRaDec2D(ra, dec):
		# radec转为竞天程序里的ob, 赤道坐标系下的笛卡尔三维坐标xyz.
		x1 = np.cos(dec / 57.2957795) * np.cos(ra / 57.2957795)
		y1 = np.cos(dec / 57.2957795) * np.sin(ra / 57.2957795)
		z1 = np.sin(dec / 57.2957795)
		return np.array([x1, y1, z1])


		def earth_angle(time_jd, x_sat, y_sat, z_sat, ra_obj, dec_obj):

		ra_sat = np.arctan2(y_sat, x_sat) / np.pi * 180
		dec_sat = np.arctan2(z_sat, np.sqrt(x_sat2+y_sat2)) / np.pi * 180
		radec_sat = SkyCoord(ra=ra_sat*u.degree,
		dec=dec_sat*u.degree, frame='gcrs')
		lb_sat = radec_sat.transform_to('geocentrictrueecliptic')

		# get the obj location
		radec_obj = SkyCoord(ra=ra_obj*u.degree,
		dec=dec_obj*u.degree, frame='gcrs')
		lb_obj = radec_obj.transform_to('geocentrictrueecliptic')

		# calculate the angle between sub-satellite point and the earth side
		earth_radius = 6371 # km
		sat_height = np.sqrt(x_sat2 + y_sat2 + z_sat**2)
		angle_a = np.arcsin(earth_radius/sat_height) / np.pi * 180

		# calculate the angle between satellite position and the target position
		angle_b = lb_sat.separation(lb_obj)

		# calculat the earth angle
		angle = 180 - angle_a - angle_b.degree

		return angle

		###############################################################################


		def ill2flux(E, path):

		# use template from sky_bkg (background_spec_hst.dat)
		filename = path+'MCI_inputData/refs/background_spec_hst.dat'
		cat_spec = pd.read_csv(filename, sep='\s+', header=None, comment='#')
		wave0 = cat_spec[0].values # A
		spec0 = cat_spec[2].values # erg/s/cm^2/A/arcsec^2

		# convert erg/s/cm^2/A/arcsec^2 to erg/s/cm^2/A/sr
		flux1 = spec0 / (1/4.25452e10)
		# convert erg/s/cm^2/A/sr to W/m^2/sr/um
		flux2 = flux1 * 10

		# 对接收面积积分，输出单位 W/m^2/nm
		D = 2 # meter
		f = 28 # meter, 焦距，转换关系来源于王维notes.
		flux3 = flux2 * np.pi * D2 / 4 / f2 / 10**3

		f = interp1d(wave0, flux3)
		wave_range = np.arange(3800, 7800)
		flux3_mean = f(wave_range)
		delta_lamba = 0.1 # nm
		E0 = np.sum(flux3_mean * delta_lamba)

		factor = E / E0
		spec_scaled = factor * spec0

		return wave0, spec_scaled

		##############################################################


		def stray_light(path, time_jd, x_sat, y_sat, z_sat, ra, dec):
		# EarthShine from straylight
		sl = StrayLight(path, jtime=time_jd, sat=np.array(
		[x_sat, y_sat, z_sat]), radec=np.array([(rau.degree).value, (decu.degree).value]))

		star_e = sl.caculateStarLightFilter(filter='r')

		# angle_earth = earth_angle(time_jd, x_sat, y_sat, z_sat, ra, dec)

		# if angle_earth < 0:
		# earth_e = 0

		earthshine_wave0, earthshine_flux0 = ill2flux(star_e, path)

		# sample as mci wavelength
		wave_mci = np.linspace(2500, 11000, 8501) # np.arange(2500, 11000, 1)

		f2 = interp1d(earthshine_wave0, earthshine_flux0)
		star_stray = f2(wave_mci)

		return star_stray
		#################################################################


		class StrayLight(object):

		def __init__(self, path, jtime=2460843., sat=np.array([0, 0, 0]), radec=np.array([0, 0])):

		self.path = path
		self.jtime = jtime
		self.sat = sat
		self.equator = coord.SkyCoord(
		radec[0]u.degree, radec[1]u.degree, frame='icrs')
		self.ecliptic = self.equator.transform_to('barycentrictrueecliptic')
		self.pointing = transRaDec2D(radec[0], radec[1])
		self.slcdll = ctypes.CDLL(
		self.path+'MCI_inputData/refs/libstraylight.so') # dylib

		self.slcdll.Calculate.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.POINTER(
		ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.c_char_p]

		self.slcdll.PointSource.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.POINTER(
		ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.c_char_p]

		self.slcdll.EarthShine.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double), ctypes.POINTER(
		ctypes.c_double),
		ctypes.POINTER(ctypes.c_double)]

		self.slcdll.Zodiacal.argtypes = [ctypes.c_double, ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double)]
		self.slcdll.ComposeY.argtypes = [ctypes.POINTER(ctypes.c_double), ctypes.POINTER(ctypes.c_double),
		ctypes.POINTER(ctypes.c_double)]
		self.slcdll.Init.argtypes = [
		ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p]
		self.deFn = self.path+"MCI_inputData/refs/DE405"
		self.PSTFn = self.path+"MCI_inputData/refs/PST"
		self.RFn = self.path+"MCI_inputData/refs/R"
		self.ZolFn = self.path+"MCI_inputData/refs/Zodiacal"
		self.brightStarTabFn = self.path+"MCI_inputData/refs/BrightGaia_with_csst_mag"

		self.slcdll.Init(str.encode(self.deFn), str.encode(
		self.PSTFn), str.encode(self.RFn), str.encode(self.ZolFn))

		def caculateStarLightFilter(self, filter='i'):
		filterIndex = {'nuv': 0, 'u': 1, 'g': 2,
		'r': 3, 'i': 4, 'z': 5, 'y': 6}

		sat = (ctypes.c_double*3)()
		sat[:] = self.sat
		ob = (ctypes.c_double*3)()
		ob[:] = self.pointing

		py1 = (ctypes.c_double*3)()
		py2 = (ctypes.c_double*3)()
		self.slcdll.ComposeY(ob, py1, py2)

		star_e1 = (ctypes.c_double*7)()
		self.slcdll.PointSource(self.jtime, sat, ob, py1,
		star_e1, str.encode(self.brightStarTabFn))
		star_e2 = (ctypes.c_double*7)()
		self.slcdll.PointSource(self.jtime, sat, ob, py2,
		star_e2, str.encode(self.brightStarTabFn))

		band_star_e1 = star_e1[:][filterIndex[filter]]
		band_star_e2 = star_e2[:][filterIndex[filter]]

		return max(band_star_e1, band_star_e2)

		def caculateEarthShineFilter(self, filter='i'):

		filterIndex = {'nuv': 0, 'u': 1, 'g': 2,
		'r': 3, 'i': 4, 'z': 5, 'y': 6}
		sat = (ctypes.c_double*3)()
		sat[:] = self.sat
		ob = (ctypes.c_double*3)()
		ob[:] = self.pointing

		py1 = (ctypes.c_double*3)()
		py2 = (ctypes.c_double*3)()
		self.slcdll.ComposeY(ob, py1, py2)

		earth_e1 = (ctypes.c_double*7)()
		self.slcdll.EarthShine(self.jtime, sat, ob, py1,
		earth_e1) # e[7]代表7个波段的照度

		earth_e2 = (ctypes.c_double*7)()
		self.slcdll.EarthShine(self.jtime, sat, ob, py2, earth_e2)

		band_earth_e1 = earth_e1[:][filterIndex[filter]]
		band_earth_e2 = earth_e2[:][filterIndex[filter]]

		return max(band_earth_e1, band_earth_e2)


		###############################################################################
		## test
		# path='/home/yan/MCI/'
		# time_jd = 2460417.59979167
		# x_sat = -4722.543136
		# y_sat = -1478.219213
		# z_sat = 4595.402769
		# ra = 116.18081536720157
		# dec= 39.42316681016602
		# earthshine0=stray_light(path,time_jd, x_sat, y_sat, z_sat, ra, dec)