Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import sys
from datetime import datetime
from csst_dfs_api.facility.level0 import Level0DataApi
from csst_dfs_api.facility.level0prc import Level0PrcApi
from csst_dfs_api.facility.calmerge import CalMergeApi
from csst_msc_instrument_effect.msc_iec import InstrumentEffectCorrection
l0Api = Level0DataApi()
prcApi = Level0PrcApi()
calApi = CalMergeApi()
# -----------------------------------------------------
# get data
l0_id = sys.argv[1]
rec = l0Api.get(level0_id=l0_id)
print(rec)
print("Find Level0 Data :" + rec.data.file_path)
bias_rec = calApi.get_latest_by_l0(level0_id=l0_id, ref_type="bias")
dark_rec = calApi.get_latest_by_l0(level0_id=l0_id, ref_type="dark")
flat_rec = calApi.get_latest_by_l0(level0_id=l0_id, ref_type="flat")
cray_rec = calApi.get_latest_by_l0(level0_id=l0_id, ref_type="cray") # temp
print("Find Level0 Bias :" + bias_rec.data.file_path)
print("Find Level0 Dark :" + dark_rec.data.file_path)
print("Find Level0 Flat :" + flat_rec.data.file_path)
# ------------------------------------------------------
# to-do data processing here
iec = InstrumentEffectCorrection(
data_input=rec.data.file_path,
bias_input=bias_rec.data.file_path,
dark_input=dark_rec.data.file_path,
flat_input=flat_rec.data.file_path,
output_path="/home/csstpipeline/data/L05_test/",
config_path="/home/csstpipeline/csst-msc-instrument-effect-master/MSC_crmask.ini",
cray_input=cray_rec.data.file_path, # temp
)
iec.run()
params_file_path = iec.config_path
result_file_path = ', '.join([
iec.data_output,
iec.flag_output,
iec.weight_output,
iec.header_output,
])
dt = datetime.now()
# ------------------------------------------------------
# write result
rec = prcApi.write(
level0_id=l0_id,
pipeline_id="P1",
prc_module="MSC-IE", # csst-msc-instrument-effect
params_file_path=params_file_path,
prc_status=0,
prc_time=dt.strftime('%Y-%m-%d %H:%M:%S'),
result_file_path=result_file_path)
print('Write Result:', rec)