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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
from __future__ import division # confidence high
import os
from stsci.tools import fileutil
def diff_angles(a,b):
"""
Perform angle subtraction a-b taking into account
small-angle differences across 360degree line.
"""
diff = a - b
if diff > 180.0:
diff -= 360.0
if diff < -180.0:
diff += 360.0
return diff
def getBinning(fobj, extver=1):
# Return the binning factor
binned = 1
if fobj[0].header['INSTRUME'] == 'WFPC2':
mode = fobj[0].header.get('MODE', "")
if mode == 'AREA': binned = 2
else:
binned = fobj['SCI', extver].header.get('BINAXIS',1)
return binned
def extract_rootname(kwvalue):
""" Returns the rootname from a full reference filename
If a non-valid value (any of ['','N/A','NONE','INDEF',None]) is input,
simply return a string value of 'NONE'
"""
# check to see whether a valid kwvalue has been provided as input
if kwvalue.strip() in ['','N/A','NONE','INDEF',None]:
return 'NONE' # no valid value, so return 'NONE'
# for a valid kwvalue, parse out the rootname
# strip off any environment variable from input filename, if any are given
if '$' in kwvalue:
fullval = kwvalue[kwvalue.find('$')+1:]
else:
fullval = kwvalue
# Extract filename without path from kwvalue
fname = os.path.basename(fullval).strip()
# Now, rip out just the rootname from the full filename
if '_' in fname:
rootname = fname[:fname.rfind('_')]
elif '.fit' in fname: # on some systems, only .fit is used instead of .fits
rootname = fname[:fname.rfind('.fit')]
elif '.' in fname: # account for non-standard file extensions
rootname = fname[:fname.rfind('.')]
else:
rootname = fname
return rootname.strip()
def construct_distname(fobj,wcsobj):
"""
This function constructs the value for the keyword 'DISTNAME'.
It relies on the reference files specified by the keywords 'IDCTAB',
'NPOLFILE', and 'D2IMFILE'.
The final constructed value will be of the form:
<idctab rootname>-<npolfile rootname>-<d2imfile rootname>
and have a value of 'NONE' if no reference files are specified.
"""
idcname = extract_rootname(fobj[0].header.get('IDCTAB', "NONE"))
if idcname is None and wcsobj.sip is not None:
idcname = 'UNKNOWN'
npolname = build_npolname(fobj)
if npolname is None and wcsobj.cpdis1 is not None:
npolname = 'UNKNOWN'
d2imname = build_d2imname(fobj)
if d2imname is None and wcsobj.det2im is not None:
d2imname = 'UNKNOWN'
sipname = build_sipname(fobj)
distname = build_distname(sipname,npolname,d2imname)
return {'DISTNAME':distname,'SIPNAME':sipname}
def build_distname(sipname,npolname,d2imname):
"""
Core function to build DISTNAME keyword value without the HSTWCS input.
"""
distname = sipname.strip()
if npolname != 'NONE' or d2imname != 'NONE':
if d2imname != 'NONE':
distname+= '-'+npolname.strip() + '-'+d2imname.strip()
else:
distname+='-'+npolname.strip()
return distname
def build_sipname(fobj):
try:
sipname = fobj[0].header["SIPNAME"]
except KeyError:
try:
sipname = fobj[0].header['rootname']+'_'+ \
extract_rootname(fobj[0].header["IDCTAB"])
except KeyError:
if 'A_ORDER' in fobj[1].header or 'B_ORDER' in fobj[1].header:
sipname = 'UNKNOWN'
else:
sipname = 'NOMODEL'
return sipname
def build_npolname(fobj):
try:
npolfile = extract_rootname(fobj[0].header["NPOLFILE"])
except KeyError:
if fileutil.countExtn(f, 'WCSDVARR'):
npolfile = 'UNKNOWN'
else:
npolfile = 'NOMODEL'
return npolfile
def build_d2imname(fobj):
try:
d2imfile = extract_rootname(fobj[0].header["D2IMFILE"])
except KeyError:
if fileutil.countExtn(f, 'D2IMARR'):
d2imfile = 'UNKNOWN'
else:
d2imfile = 'NOMODEL'
return d2imfile
|