update ifs_sim

csst_ifs_sim/csst_ifs_sim.py

@@ -13,10 +13,21 @@ import os
 from scipy.interpolate import interp1d
 import astropy.coordinates as coord
 import ctypes
-from csst_ifs_sim.support import IFSinstrumentModel
-from csst_ifs_sim.support import cosmicrays
-from csst_ifs_sim.support import logger as lg
-from csst_ifs_sim.CTI import CTI
+
+# from csst_ifs_sim.support import IFSinstrumentModel
+# from csst_ifs_sim.support import cosmicrays
+# from csst_ifs_sim.support import logger as lg
+# from csst_ifs_sim.CTI import CTI
+
+
+sys.path.append('../')
+
+from CTI import CTI
+from support import logger as lg
+from support import cosmicrays
+from support import IFSinstrumentModel
+
+
 # from optparse import OptionParser
 import configparser as ConfigParser
 import cmath
@@ -355,6 +366,7 @@ class StrayLight(object):
             DESCRIPTION.
 
         """
+        filterIndex = {'nuv': 0, 'u': 1, 'g': 2, 'r': 3, 'i': 4, 'z': 5, 'y': 6}
 
         sat = (ctypes.c_double*3)()
         sat[:] = self.sat
@@ -392,6 +404,8 @@ class StrayLight(object):
             DESCRIPTION.
 
         """
+        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)()
@@ -402,7 +416,7 @@ class StrayLight(object):
         self.slcdll.ComposeY(ob, py1, py2)
 
         earth_e1 = (ctypes.c_double*7)()
-        self.slcdll.EarthShine(self.jtime, sat, ob, py1,  earth_e1) 
+        self.slcdll.EarthShine(self.jtime, sat, ob, py1,  earth_e1)
          # e[7]代表7个波段的照度
 
         earth_e2 = (ctypes.c_double*7)()