test

straylight.py

@@ -472,15 +472,78 @@ def testZodiacal(lon = 285.04312526255366, lat = 30.):
 # sl = StrayLight(jtime = ju, sat = np.array([5873.752, -1642.066, 2896.744]), pointing = np.array([-0.445256,-0.76061,0.47246]))
 
 
-fn = '/Users/zhangxin/Work/SurveyPlan/point/csst_survey_sim_20211028/E17.5_b17.5_beta_11.6_opt_transtime_1_CMG_1_dp_2_0.25_da_10_Texp_1.5_DEC60_500_0.1_800_1000_+5deg.dat'
+# fn = '/Users/zhangxin/Work/SurveyPlan/point/csst_survey_sim_20211028/E17.5_b17.5_beta_11.6_opt_transtime_1_CMG_1_dp_2_0.25_da_10_Texp_1.5_DEC60_500_0.1_800_1000_+5deg.dat'
+
+# surveylist = np.loadtxt(fn)
+# sky_pix = np.zeros([surveylist.shape[0],7])
+
+
+# i = 693438 
+# c_eclip = coord.SkyCoord(surveylist[:,2]*u.degree, surveylist[:,1]*u.degree,frame='barycentrictrueecliptic')
+# c_equtor = c_eclip.transform_to('icrs')
+
+
+# # pointing = transRaDec2D((c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value)
+# #     # print(ju, pointing, surveylist[i,3:9])
+# # ju = surveylist[i,0]
+# # sl = StrayLight(jtime = ju, sat = surveylist[i,3:6], pointing = pointing)
+# # sl.caculateStrayLightGrating(grating = 'GI', pixel_size_phy = 10, normFilter = 'g')
+
+# for i in np.arange(surveylist.shape[0]):
+#     print(i)
+#     if i > 300:
+#         break
+#     # if i != 300:
+#     #     continue
+#     # if i != 693438:
+#     #      continue
+#     ju = surveylist[i,0]
+#     pointing = transRaDec2D((c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value)
+#     # print(ju, pointing, surveylist[i,3:9])
+#     sl = StrayLight(jtime = ju, sat = surveylist[i,3:6], radec = np.array([(c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value]))
+#     # strayl_i,s_zoldical ,s_earth, s_earth1 = sl.caculateStrayLightFilter(filter = 'i')
+#     # print(i,strayl_i,s_zoldical,s_earth, s_earth1)
+#     p_cart= transRaDec2D((c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value)
+#     sky_pix[i,6] = getAngle132(x1 = surveylist[i,6], y1 = surveylist[i,7], z1 = surveylist[i,8], x2 = p_cart[0], y2 = p_cart[1], z2 = p_cart[2], x3 = 0, y3 = 0, z3 = 0)
+
+#     earthZenithAngle,isInSunSide = calculateAnglePwithEarth(sat = surveylist[i,3:6], pointing = pointing, sun = surveylist[i,6:9])
+#     sky_pix[i,4] = earthZenithAngle
+#     sky_pix[i,5] = isInSunSide
+
+#     e1_,py = sl.caculateEarthShineFilter(filter = 'i')
+#     # e2, e2_ = sl.calculateZodiacalFilter1(filter = 'i')
+#     e3, v_mag = sl.calculateZaodiacalFilter2(filter = 'i', sun_pos=surveylist[i,6:9])
+#     # e4 = sl.caculateStarLightFilter(filter = 'i',pointYaxis = py)
+#     # e4 = 0 
+
+#     e1,py = sl.caculateEarthshineGrating(grating = 'GI', pixel_size_phy = 10, normFilter = 'g')
+
+#     # e2 = sl.caculateStarLightGrating(grating = 'GV', pointYaxis = py)
+#     e2 = sl.caculateStarLightGrating(grating = 'GI', pointYaxis = py)
+#     e4 = sl.caculateStarLightFilter(filter = 'i',pointYaxis = py)
+
+#     e5=sl.caculateStrayLightFilter(filter = 'i', pixel_size_phy = 10, pixel_scale = 0.074, sun_pos = surveylist[i,6:9])
+#     e6,_=sl.caculateStrayLightGrating(grating = 'GI', normFilter_es = 'g', sun_pos = surveylist[i,6:9])
+
+#     sky_pix[i,0] = e1
+#     sky_pix[i,1] = e2
+#     sky_pix[i,2] = e3
+#     sky_pix[i,3] = e4
+#     print(e1+e2,e1_+e3+e4,e5,e6)
+
+    # print(e1,e2,e3,e4)
+
+
+import sys
+fn = sys.argv[1]
 
 surveylist = np.loadtxt(fn)
-sky_pix = np.zeros([surveylist.shape[0],7])
+sky_pix = np.zeros([surveylist.shape[0],4])
 
 
-i = 693438 
-c_eclip = coord.SkyCoord(surveylist[:,2]*u.degree, surveylist[:,1]*u.degree,frame='barycentrictrueecliptic')
-c_equtor = c_eclip.transform_to('icrs')
+# i = 693438 
+# c_eclip = coord.SkyCoord(surveylist[:,2]*u.degree, surveylist[:,1]*u.degree,frame='barycentrictrueecliptic')
+# c_equtor = c_eclip.transform_to('icrs')
 
 
 # pointing = transRaDec2D((c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value)
@@ -489,47 +552,32 @@ c_equtor = c_eclip.transform_to('icrs')
 # sl = StrayLight(jtime = ju, sat = surveylist[i,3:6], pointing = pointing)
 # sl.caculateStrayLightGrating(grating = 'GI', pixel_size_phy = 10, normFilter = 'g')
 
+strRes = ""
 for i in np.arange(surveylist.shape[0]):
-    print(i)
-    if i > 300:
-        break
-    # if i != 300:
-    #     continue
-    # if i != 693438:
-    #      continue
-    ju = surveylist[i,0]
-    pointing = transRaDec2D((c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value)
+    
+  
+    ju = surveylist[i,5]
+    pointing = transRaDec2D(surveylist[i,0],surveylist[i,1])
     # print(ju, pointing, surveylist[i,3:9])
-    sl = StrayLight(jtime = ju, sat = surveylist[i,3:6], radec = np.array([(c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value]))
-    # strayl_i,s_zoldical ,s_earth, s_earth1 = sl.caculateStrayLightFilter(filter = 'i')
-    # print(i,strayl_i,s_zoldical,s_earth, s_earth1)
-    p_cart= transRaDec2D((c_equtor[i].ra*u.degree).value, (c_equtor[i].dec*u.degree).value)
-    sky_pix[i,6] = getAngle132(x1 = surveylist[i,6], y1 = surveylist[i,7], z1 = surveylist[i,8], x2 = p_cart[0], y2 = p_cart[1], z2 = p_cart[2], x3 = 0, y3 = 0, z3 = 0)
+    sl = StrayLight(jtime = ju, sat = surveylist[i,6:9], radec = np.array([surveylist[i,0],surveylist[i,1]]))
 
-    earthZenithAngle,isInSunSide = calculateAnglePwithEarth(sat = surveylist[i,3:6], pointing = pointing, sun = surveylist[i,6:9])
-    sky_pix[i,4] = earthZenithAngle
-    sky_pix[i,5] = isInSunSide
 
-    e1_,py = sl.caculateEarthShineFilter(filter = 'i')
-    # e2, e2_ = sl.calculateZodiacalFilter1(filter = 'i')
-    e3, v_mag = sl.calculateZaodiacalFilter2(filter = 'i', sun_pos=surveylist[i,6:9])
-    # e4 = sl.caculateStarLightFilter(filter = 'i',pointYaxis = py)
-    # e4 = 0 
+    e1,py = sl.caculateEarthShineFilter(filter = 'i')
 
-    e1,py = sl.caculateEarthshineGrating(grating = 'GI', pixel_size_phy = 10, normFilter = 'g')
+    e2, v_mag = sl.calculateZaodiacalFilter2(filter = 'i', sun_pos=surveylist[i,9:12])
 
-    # e2 = sl.caculateStarLightGrating(grating = 'GV', pointYaxis = py)
-    e2 = sl.caculateStarLightGrating(grating = 'GI', pointYaxis = py)
-    e4 = sl.caculateStarLightFilter(filter = 'i',pointYaxis = py)
+    e3 = sl.caculateStarLightFilter(filter = 'i',pointYaxis = py)
 
-    e5=sl.caculateStrayLightFilter(filter = 'i', pixel_size_phy = 10, pixel_scale = 0.074, sun_pos = surveylist[i,6:9])
-    e6,_=sl.caculateStrayLightGrating(grating = 'GI', normFilter_es = 'g', sun_pos = surveylist[i,6:9])
+    print(i, e1, e2, e3)
 
     sky_pix[i,0] = e1
     sky_pix[i,1] = e2
     sky_pix[i,2] = e3
-    sky_pix[i,3] = e4
-    print(e1+e2,e1_+e3+e4,e5,e6)
+    sky_pix[i,3] = e1+e2+e3
 
-    # print(e1,e2,e3,e4)
+    strRes = str(e1) + '    ' + str(e2) + '    ' + str(e3) + '    ' + str(e1+e2+e3) + '\n'
 
+outF = open(fn[0:-4]+'stralight.dat',"w")
+outF.write(strRes)
+outF.flush()
+outF.close()