-
Notifications
You must be signed in to change notification settings - Fork 0
/
asol6_loc.py
110 lines (72 loc) · 3 KB
/
asol6_loc.py
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
# PURPOSE: Return the location of various data files that come with the
# PURPOSE: observarion dataset.
# v3: * based on v1
# * make sure that if both asolfile.fits and asolfile.fits.gz, are present
# only the first is used.
# * write absolute paths in asol_list.txt
# v6_loc: add MTL file types
import pdb
import glob
import os
import numpy as n
def asol(ob, ftype) :
# Identifying the ASOL file(s).. (or more generally asol, flt1, pbk, msk, bpix or evt1 files)
########################################## ASOL
if ftype == 'asol' :
asolfits = sorted(glob.glob('orig_*/'+ob+'/primary/*asol*.fits*'))
if len(asolfits) != 0 : obpath = asolfits[0].partition('/primary/')[0]
if len(asolfits) == 0 :
print 'NO ASOL FILE FOUND FOR OBSID ' + ob
acao=[]
elif len(asolfits) == 1 : acao = asolfits[0]
else : # --------- Multiple ASOL files OR .fits and .fits.gz both present
#if asolfits[0]+'.gz' == asolfits[1] : acao = asolfits[0]
newasolfits=[asolfits[0]]
for i in range(1,len(asolfits)) :
if asolfits[i-1]+'.gz' != asolfits[i] : newasolfits.append(asolfits[i])
if len(newasolfits) == 1 : acao = newasolfits[0]
else :
vernum = n.zeros(len(newasolfits), dtype=n.int)
for i in range(len(newasolfits)) :
vernum[i] = int( newasolfits[i].split('_asol')[0][-1] )
allequal=True
for i in range(1,len(vernum)) :
if vernum[i] != vernum[0] : allequal=False
if allequal :
if not os.path.exists('asol_list'+ob+'.txt') :
asoltxtfile = open('asol_list'+ob+'.txt', 'w')
for i in range(len(newasolfits)) :
asoltxtfile.write( newasolfits[i]+'\n' )
asoltxtfile.close()
acao = '@'+'asol_list'+ob+'.txt'
else :
maxwh = n.where(vernum == max(vernum))[0]
if len(maxwh) == 1 : acao = newasolfits[maxwh[0]]
else :
if not os.path.exists('asol_list'+ob+'.txt') :
asoltxtfile = open('asol_list'+ob+'.txt', 'w')
for wh in range(len(maxwh)) :
asoltxtfile.write( newasolfits[wh]+'\n' )
asoltxtfile.close()
acao = '@'+'asol_list'+ob+'.txt'
return acao
########################################## anything other than ASOL:
elif ftype == 'bpix' or ftype == 'evt1' or ftype == 'pbk' or ftype == 'flt1' or ftype == 'msk1' or ftype == 'mtl' :
if ftype == 'bpix' : matches = sorted( glob.glob('orig_*/'+ob+'/primary/*'+ftype+'*') )
else : matches = sorted( glob.glob('orig_*/'+ob+'/secondary/*'+ftype+'*') )
if len(matches) == 0 :
print 'NO '+ftype+' FILE FOUND FOR OBSID ' + ob
acao=[]
elif len(matches) == 1 : acao = matches[0]
else : # --------- Multiple ASOL files OR .fits and .fits.gz both present
newmatches=[matches[0]]
for i in range(1,len(matches)) :
if matches[i-1]+'.gz' != matches[i] : newmatches.append(matches[i])
if len(newmatches) == 1 : acao = newmatches[0]
else : acao = newmatches[-1]
return acao
else :
print
print '============ WRONG USAGE OF ASOL3.PY ========================='
print
return ''