Pointing.py 3.35 KB
Newer Older
1
2
import numpy as np
import galsim
Fang Yuedong's avatar
Fang Yuedong committed
3
import yaml
4
5
from astropy.time import Time

6
7
import ObservationSim.Instrument._util as _util

8
class Pointing(object):
9
    def __init__(self, id=0, ra=0., dec=0., img_pa=0., timestamp=1621915200, sat_x=0., sat_y=0., sat_z=0., sun_x=0., sun_y=0., sun_z=0., sat_vx=0., sat_vy=0., sat_vz=0., exp_time=150., pointing_type='SCI'):
10
11
12
13
14
15
        self.id = id
        self.ra = ra
        self.dec = dec
        self.img_pa = img_pa * galsim.degrees
        self.timestamp = timestamp
        self.sat_x, self.sat_y, self.sat_z = sat_x, sat_y, sat_z
16
        self.sun_x, self.sun_y, self.sun_z = sun_x, sun_y, sun_z
17
18
19
        self.sat_vx, self.sat_vy, self.sat_vz = sat_vx, sat_vy, sat_vz
        self.exp_time = exp_time
        self.pointing_type = pointing_type
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
        self.survey_field_type = 'WIDE'
        self.jdt = 0. 
    
    def get_full_depth_exptime(self, filter_type):
        if self.survey_field_type == 'WIDE':
            if filter_type in _util.SPEC_FILTERS:
                return 150. * 4
            else:
                if filter_type.lower() in ['nuv', 'y']:
                    return 150. * 4
                elif filter_type.lower() in ['u', 'g', 'r', 'i', 'z']:
                    return 150. * 2
                else:
                    return max(150., self.exp_time) # [TODO] for FGS
        elif self.survey_field_type == 'DEEP':
            if filter_type in _util.SPEC_FILTERS:
                return 250. * 4 * 4
            else:
                if filter_type.lower() in ['nuv', 'y']:
                    return 250. * 4 * 4
                elif filter_type.lower() in ['u', 'g', 'r', 'i', 'z']:
                    return 250. * 2 * 4
                else:
                    return max(150., self.exp_time) # [TODO] for FGS

45

Fang Yuedong's avatar
Fang Yuedong committed
46
    def read_pointing_columns(self, columns, id=0, t=1621915200, pointing_type='SCIE'):
47
48
49
50
51
        self.id = id
        col_len = len(columns)
        self.ra = float(columns[0])
        self.dec = float(columns[1])
        self.img_pa = float(columns[4]) * galsim.degrees
Fang Yuedong's avatar
Fang Yuedong committed
52
        # self.pointing_type = pointing_type
53
54
55
        if col_len > 5:
            jdt = np.double(columns[5])
            t_temp = Time(jdt, format='jd')
56
            self.jdt = jdt
57
58
59
60
            self.timestamp = t_temp.unix
            self.sat_x = float(columns[6])
            self.sat_y = float(columns[7])
            self.sat_z = float(columns[8])
61
62
            self.sun_x = float(columns[9])
            self.sun_y = float(columns[10])
63
            self.sun_z = float(columns[11])
64
65
66
67
            self.sat_vx = float(columns[15])
            self.sat_vy = float(columns[16])
            self.sat_vz = float(columns[17])
            self.exp_time = float(columns[18])
68
69
70
71
            is_deep = float(columns[19])
            # [TODO] Can also define other survey types
            if is_deep != -1.0:
                self.survey_field_type = "DEEP"
Fang Yuedong's avatar
Fang Yuedong committed
72
73
74
75
76
77
78
79
80
            
            # Load the configuration file for this particular pointing
            self.obs_config_file = "/share/home/fangyuedong/20231211/csst-simulation/config/obs_config_SCI_WIDE_phot.yaml"
            with open(self.obs_config_file, "r") as stream:
                try:
                    self.obs_param = yaml.safe_load(stream)
                except yaml.YAMLError as exc:
                    print(exc)
            self.pointing_type = self.obs_config_file["obs_type"]
81
82
        else:
            self.timestamp = t