diff --git a/stimuli/AbstractStimulus.py b/stimuli/AbstractStimulus.py index aa38dc8..649f7d6 100644 --- a/stimuli/AbstractStimulus.py +++ b/stimuli/AbstractStimulus.py @@ -6,3 +6,24 @@ class AbstractStimulus: def value_at_time_in_s(self, time_point): raise NotImplementedError("This is an abstract class!") + + def get_stimulus_start_ms(self): + raise NotImplementedError("This is an abstract class!") + + def get_stimulus_start_s(self): + return self.get_stimulus_start_ms() / 1000 + + def get_stimulus_duration_ms(self): + raise NotImplementedError("This is an abstract class!") + + def get_stimulus_duration_s(self): + return self.get_stimulus_duration_ms() / 1000 + + def get_stimulus_end_ms(self): + return self.get_stimulus_start_ms() + self.get_stimulus_duration_ms() + + def get_stimulus_end_s(self): + return self.get_stimulus_end_ms() / 1000 + + def get_amplitude(self): + raise NotImplementedError("This is an abstract class!") diff --git a/stimuli/StepStimulus.py b/stimuli/StepStimulus.py index f0622ec..cae3bf9 100644 --- a/stimuli/StepStimulus.py +++ b/stimuli/StepStimulus.py @@ -21,3 +21,12 @@ class StepStimulus(AbstractStimulus): return self.value else: return self.base_value + + def get_stimulus_start_ms(self): + return self.start + + def get_stimulus_duration_ms(self): + return self.duration + + def get_amplitude(self): + return self.value - self.base_value