def __get_detector_parameters(parameters, data_path, detector_type):
if isinstance(parameters, etree._ElementTree):
print('Detector parameters received from external source.')
return parameters
elif data_path is None:
return ut.GetData(data_path_name='detector/' + detector_type +
'_default.xml').data
elif isinstance(data_path, str):
return ut.GetData(data_path_name=data_path).data
else:
raise ValueError(
'Expected data type for <data_path> is str or None, '
'in which case default values will be loaded')
def __init__(self, param, rate_type, data_path):
self.param = param
if not isinstance(self.param, etree._ElementTree):
self.param = getdata.GetData(data_path_name=data_path).data
assert isinstance(self.param, etree._ElementTree)
self.__set_impurity_mass_scaling_dictionary()
self.__projectile_parameters()
self.__set_atomic_dictionary()
self.__set_rates_path(rate_type)
self.__set_charge_state_lib()
def __get_atomic_mass(self):
data_path_name = os.path.join(
'atomic_data',
self.param.getroot().find('body').find('beamlet_species').text,
'supplementary_data', 'default',
self.param.getroot().find('body').find('beamlet_species').text +
'_m.txt')
mass_str = getdata.GetData(data_path_name=data_path_name,
data_format="array").data
try:
self.mass = float(mass_str)
except ValueError:
print('Unexpected data in file: ' + data_path_name +
'(Expecting single float!)')
raise ValueError
def _load_synthetic_signals(self, path):
self.signals = ut.GetData(data_path_name=path,
data_format="array").data
def load_rate_data(path, tag_name):
return getdata.GetData(data_path_name=path,
data_key=[tag_name],
data_format='array').data
def read_observation_profile(data_path):
obs_profile = getdata.GetData(data_path_name=data_path).data
assert isinstance(obs_profile, pandas.DataFrame)
print('obs_profile read from file: ' + data_path)
return obs_profile
def read_observation_param(data_path):
obs_param = getdata.GetData(data_path_name=data_path).data
assert isinstance(obs_param, etree._ElementTree)
print('obs_param read from file: ' + data_path)
return obs_param