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
61
62
63
64
65
66
67
68
69
70
71
72
|
#!/usr/bin/env python
#If a
import reflex
import sys
import os
import_ok = 1
try:
import pyfits
#import_ok = 1
except ImportError:
try:
from astropy.io import fits as pyfits
# import_ok = 1
except ImportError:
import_ok = 0
try:
import math
except ImportError:
import_ok = 0
if __name__ == '__main__':
parser = reflex.ReflexIOParser()
#Define inputs
parser.add_option("-i", "--in_sof", dest="in_sof")
#Define outputs
parser.add_output("-o", "--out_sof", dest="out_sof")
inputs = parser.get_inputs()
outputs = parser.get_outputs()
#Retrieve input
in_sof = inputs.in_sof
#Get the input files
files = in_sof.files
#Do the stuff
if import_ok == 0:
outputs.out_sof = inputs.in_sof
parser.write_outputs()
sys.exit()
#Read the content of TRACE_TABLE_06 (static, if present)
for file in files:
if file.category == 'TRACE_TABLE_06':
static = pyfits.open(file.name,mode='readonly')
newdata = static[1].data
#Read the content of the TRATE_TABLE-06.fits, produced by the pipeline
for file in files:
if os.path.basename(file.name) == 'TRACE_TABLE-06.fits':
hdulist = pyfits.open(file.name,mode='update')
DataTable = hdulist[1].data
Width = DataTable.field('Width')
total = sum(Width)
#If there is a NaN, it replaces the content of TRATE_TABLE-06.fits
#with that of the static table.
if math.isnan(total):
hdulist[1].data = newdata
hdulist.flush()
DataTable = hdulist[1].data
Width = DataTable.field('Width')
total = sum(Width)
outputs.out_sof = inputs.in_sof
parser.write_outputs()
sys.exit()
|
