-
Notifications
You must be signed in to change notification settings - Fork 0
/
spectra_collection_functions.py
75 lines (58 loc) · 2.23 KB
/
spectra_collection_functions.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
import numpy as np
import joblib
# --------------------------------------------------------
# ---------------- Save and read collection dumps --------
# --------------------------------------------------------
def read_pkl_spectra(file_path, read_rows=None, read_cols=None):
data_read = joblib.load(file_path)
if read_cols is not None:
data_read = data_read[:, read_cols]
if read_rows is not None:
data_read = data_read[read_rows, :]
return data_read
def save_pkl_spectra(data, file_path):
joblib.dump(data, file_path)
# --------------------------------------------------------
# ---------------- Spectra collection class --------------
# --------------------------------------------------------
class CollectionParameters:
def __init__(self, filename):
self.filename_full = filename
self.filename_split = '.'.join(filename.split('.')[:-1]).split('_')
def __get_str_pos__(self, search_str):
try:
idx = self.filename_split.index(search_str)
except ValueError:
idx = None
return idx
def __get_value__(self, search_str):
idx_value = self.__get_str_pos__(search_str)
if idx_value is not None:
value_str = self.filename_split[idx_value + 1]
try:
return int(value_str)
except ValueError:
return float(value_str)
else:
return None
def get_ccd(self):
return self.filename_split[2][3] # the fourth character is the number of the ccd
def get_wvl_range(self):
wvl_start = float(self.filename_split[3])
wvl_end = float(self.filename_split[4])
return wvl_start, wvl_end
def get_wvl_values(self):
min, max = self.get_wvl_range()
return np.arange(min, max, self.get_wvl_step())
def get_interp_method(self):
return self.filename_split[5]
def get_wvl_step(self):
return self.__get_value__('wvlstep')
def get_snr_limit(self):
return self.__get_value__('snr')
def get_teff_step(self):
return self.__get_value__('teff')
def get_logg_step(self):
return self.__get_value__('logg')
def get_feh_step(self):
return self.__get_value__('feh')