source: trunk/loadparameters.py@ 17

Last change on this file since 17 was 17, checked in by JohnLightfoot, 10 years ago

initial import
File size: 8.8 KB
Line 
1from __future__ import absolute_import
2
3import collections
4import os
5import pprint
6import xlrd
7
8from mako.lookup import TemplateLookup
9
10class LoadParameters:
11 """Class to compute interferograms.
12 """
13
14 def __init__(self, sky_spreadsheet='Sky.xlsx', sky_sheet='1point',
15 instrument_spreadsheet='FIInS_Instrument_cor3.xlsx'):
16 self.result = collections.OrderedDict()
17 self.result['substages'] = collections.OrderedDict()
18
19 # access the relevant sheet of the spreadsheet
20 book = xlrd.open_workbook(sky_spreadsheet)
21 print 'names', book.sheet_names()
22 sheet = book.sheet_by_name(sky_sheet)
23
24 sheet_dict = collections.OrderedDict()
25
26 # Sheet has names in col 0, values in columns 1 to n
27 # Ignore row 0.
28
29 # read column names
30 for row in range(1,sheet.nrows):
31 colname = sheet.cell(row, 0)
32 sheet_dict[colname.value] = {}
33 for col in range(1,sheet.ncols):
34 sheet_dict[colname.value][col] = sheet.cell(row,col).value
35
36 self.result['substages']['Sky'] = sheet_dict
37
38 # the instrument spreadsheet
39 instrument_book = xlrd.open_workbook(instrument_spreadsheet)
40 for sheet in instrument_book.sheets():
41
42 if sheet.name == 'FTSpectrograph':
43 sheet_dict = collections.OrderedDict()
44
45 # ignore first row of sheet
46 # second row of sheet has column names
47 name_row = 1
48 # subsequent rows have entries, only one of which is
49 # 'selected'
50 select_col = None
51
52 # read column names
53 for col in range(sheet.ncols):
54 sheet_dict[sheet.cell(name_row,col).value] = \
55 collections.OrderedDict()
56 if 'Selection' in sheet.cell(name_row,col).value:
57 select_col = col
58
59 # sanity check
60 if select_col is None:
61 raise Exception, \
62 'No Selection column in FTSpectrograph spreadsheet'
63
64 # read entries from first 'selected' row
65 for row in range(name_row+1,sheet.nrows):
66 if int(sheet.cell(row,select_col).value) > 0:
67 for col in range(sheet.ncols):
68 sheet_dict[sheet.cell(name_row,col).value][row] = \
69 sheet.cell(row,col).value
70 break
71
72 # resulting dict should contain the entries from the
73 # selected spreadsheet row
74
75 self.result['substages'][sheet.name] = sheet_dict
76
77 elif sheet.name == 'FTSMechanical':
78 sheet_dict = collections.OrderedDict()
79
80 # Sheet has names in col 0, values in col 1.
81 # Ignore row 0.
82 name_col = 0
83 entry_col = 1
84
85 # read column names
86 for row in range(1,sheet.nrows):
87 sheet_dict[sheet.cell(row,name_col).value] = \
88 collections.OrderedDict()
89 sheet_dict[sheet.cell(row,name_col).value][entry_col] = \
90 sheet.cell(row,entry_col).value
91
92 self.result['substages'][sheet.name] = sheet_dict
93
94 elif sheet.name == 'Telescope':
95 sheet_dict = collections.OrderedDict()
96
97 # Sheet has names in col 0, values in col 1, units in col 3.
98 # Ignore rows 0 and 1.
99 name_col = 0
100 entry_col = 1
101
102 # read column names
103 for row in range(2,sheet.nrows):
104 sheet_dict[sheet.cell(row,name_col).value] = \
105 sheet.cell(row,entry_col).value
106
107 self.result['substages'][sheet.name] = sheet_dict
108
109 elif sheet.name == 'Interferometer':
110 sheet_dict = collections.OrderedDict()
111
112 # ignore first 4 rows of sheet
113 # 5th row of sheet has column names
114 name_row = 4
115 # subsequent rows have entries, only one of which is
116 # 'selected'
117 select_col = None
118
119 # read column names
120 for col in range(sheet.ncols):
121 sheet_dict[sheet.cell(name_row,col).value] = \
122 collections.OrderedDict()
123 if 'Select' in sheet.cell(name_row,col).value:
124 select_col = col
125
126 # sanity check
127 if select_col is None:
128 raise Exception, \
129 'No Selection column in Interferometer spreadsheet'
130
131 # read entries from first 'selected' row
132 for row in range(name_row+1,sheet.nrows):
133 select_cell = str(sheet.cell(row,select_col).value)
134 select_val = 0
135 try:
136 select_val = int(float(select_cell))
137 except:
138 pass
139
140 if select_val == 1:
141 for col in range(sheet.ncols):
142 sheet_dict[sheet.cell(name_row,col).value][row] = \
143 sheet.cell(row,col).value
144 break
145
146 # resulting dict should contain the entries from the
147 # selected spreadsheet row
148
149 self.result['substages'][sheet.name] = sheet_dict
150
151 elif sheet.name == 'ColdOptics':
152 sheet_dict = collections.OrderedDict()
153
154 # Sheet has names in col 0, values in col 1.
155 # Ignore rows 0 and 1 and final row.
156 # Ignore cols 2 and higher - duplicate information.
157 name_col = 0
158 entry_col = 1
159
160 # read column names
161 for row in range(2,sheet.nrows-1):
162 sheet_dict[sheet.cell(row,name_col).value] = \
163 collections.OrderedDict()
164 sheet_dict[sheet.cell(row,name_col).value][entry_col] = \
165 sheet.cell(row,entry_col).value
166
167 self.result['substages'][sheet.name] = sheet_dict
168
169 elif sheet.name == 'Background':
170 sheet_dict = collections.OrderedDict()
171
172 # Sheet has names in col 0, values in col 1.
173 # Ignore rows 0 and 1.
174 name_col = 0
175 entry_col = 1
176
177 for row in range(2,sheet.nrows):
178 sheet_dict[sheet.cell(row,name_col).value] = \
179 collections.OrderedDict()
180 sheet_dict[sheet.cell(row,name_col).value][entry_col] = \
181 sheet.cell(row,entry_col).value
182
183 self.result['substages'][sheet.name] = sheet_dict
184
185 elif sheet.name == 'WarmOptics':
186 sheet_dict = collections.OrderedDict()
187
188 # Sheet has names in col 0, values in col 1.
189 # Ignore rows 0 and 1.
190 name_col = 0
191 entry_col = 1
192
193 for row in range(2,sheet.nrows):
194 sheet_dict[sheet.cell(row,name_col).value] = \
195 collections.OrderedDict()
196 sheet_dict[sheet.cell(row,name_col).value][entry_col] = \
197 sheet.cell(row,entry_col).value
198
199 self.result['substages'][sheet.name] = sheet_dict
200
201 elif sheet.name == 'Detectors':
202 sheet_dict = collections.OrderedDict()
203
204 # Sheet has names in col 0, values in col 1.
205 # Ignore rows 0 and 1.
206 name_col = 0
207 entry_col = 1
208
209 for row in range(2,sheet.nrows):
210 sheet_dict[sheet.cell(row,name_col).value] = \
211 collections.OrderedDict()
212 sheet_dict[sheet.cell(row,name_col).value][entry_col] = \
213 sheet.cell(row,entry_col).value
214
215 self.result['substages'][sheet.name] = sheet_dict
216
217 elif sheet.name == 'SimulatorControl':
218 sheet_dict = collections.OrderedDict()
219
220 # Sheet has names in col 0, values in col 1.
221 # Ignore row 0.
222 name_col = 0
223 entry_col = 1
224
225 for row in range(1,sheet.nrows):
226 sheet_dict[sheet.cell(row,name_col).value] = \
227 collections.OrderedDict()
228 sheet_dict[sheet.cell(row,name_col).value][entry_col] = \
229 sheet.cell(row,entry_col).value
230
231 self.result['substages'][sheet.name] = sheet_dict
232
233 else:
234 print 'loadparameters skipping', sheet.name
235
236 def run(self):
237 return self.result
Note: See TracBrowser for help on using the repository browser.