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