Skip to content
Snippets Groups Projects
Commit 331d153b authored by MoritzThomasHuebner's avatar MoritzThomasHuebner
Browse files

Changed the default argument for parameter_conversion to be a lambda...

Changed the default argument for parameter_conversion to be a lambda expression that is an identity for parameter conversion. This allowed to simplify some more things further down.
parent 5d1ec4de
No related branches found
No related tags found
1 merge request!124Simplify wg redundant code
......@@ -7,7 +7,8 @@ import numpy as np
class WaveformGenerator(object):
def __init__(self, duration=None, sampling_frequency=None, start_time=0, frequency_domain_source_model=None,
time_domain_source_model=None, parameters=None, parameter_conversion=None,
time_domain_source_model=None, parameters=None,
parameter_conversion=lambda parameters, search_keys: (parameters, []),
non_standard_sampling_parameter_keys=None,
waveform_arguments=None):
""" A waveform generator
......@@ -108,7 +109,8 @@ class WaveformGenerator(object):
def _calculate_strain(self, model, model_data_points, transformation_function, transformed_model,
transformed_model_data_points):
added_keys = self._setup_conversion()
self.parameters, added_keys = self.parameter_conversion(self.parameters,
self.non_standard_sampling_parameter_keys)
if model is not None:
model_strain = self._strain_from_model(model_data_points, model)
elif transformed_model is not None:
......@@ -119,13 +121,6 @@ class WaveformGenerator(object):
self._remove_added_keys(added_keys)
return model_strain
def _setup_conversion(self):
added_keys = []
if self.parameter_conversion is not None:
self.parameters, added_keys = self.parameter_conversion(self.parameters,
self.non_standard_sampling_parameter_keys)
return added_keys
def _strain_from_model(self, model_data_points, model):
self.__full_source_model_keyword_arguments.update(self.parameters)
return model(model_data_points, **self.__full_source_model_keyword_arguments)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment