Linting, meta_data initialization, adding TIM and SFM

This commit is contained in:
t2 2018-11-07 16:51:21 +01:00
parent fa80340330
commit 01a9d76b31
1 changed files with 9 additions and 5 deletions

View File

@ -110,20 +110,24 @@ def generate_wavelengths(lxv_spc, fxv_spc, npt_spc):
wavenumbers.append(arr) wavenumbers.append(arr)
return wavenumbers return wavenumbers
def find_key(buff,key):
def find_key(buff, key):
hit = buff.find(key) + 8 hit = buff.find(key) + 8
value = unpack_from('2000s', buff, hit)[0] value = unpack_from('2000s', buff, hit)[0]
value = value[:value.find(b'\x00')] value = value[:value.find(b'\x00')]
return value return value
def get_metadata(buff): def get_metadata(buff):
all_keys=['INS','SRC','DAT','SNM'] #Further keys can be added here e.g. 'TIM', 'SFM'... meta = {}
all_keys = ['INS', 'SRC', 'DAT', 'SNM', 'TIM', 'SFM']
for k in all_keys: for k in all_keys:
keystr= k keystr = k
value=find_key(buff,keystr.encode('utf-8')) value = find_key(buff, keystr.encode('utf-8'))
meta[k] = value meta[k] = value
return meta return meta
def filter_spc_params(end_spc, spc_param_list, npt_all): def filter_spc_params(end_spc, spc_param_list, npt_all):
def indexes_of_valid_series(arr): def indexes_of_valid_series(arr):
return list(arr).index(min(filter(lambda x: x > 0, arr))) return list(arr).index(min(filter(lambda x: x > 0, arr)))