too much sorry future me -.-

stimuli/AbstractStimulus.py

@@ -8,22 +8,25 @@ class AbstractStimulus:
         raise NotImplementedError("This is an abstract class!")
 
     def get_stimulus_start_ms(self):
-        raise NotImplementedError("This is an abstract class!")
+        return self.get_stimulus_start_s() * 1000
 
     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_ms(self):
+        return self.get_stimulus_duration_s() * 1000
+
     def get_stimulus_duration_s(self):
-        return self.get_stimulus_duration_ms() / 1000
+        raise NotImplementedError("This is an abstract class!")
 
     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
+        return self.get_stimulus_start_s() + self.get_stimulus_duration_s()
 
     def get_amplitude(self):
         raise NotImplementedError("This is an abstract class!")
+
+    def as_array(self, time_start, total_time, step_size):
+        raise NotImplementedError("This is an abstract class!")

stimuli/StepStimulus.py

@@ -5,8 +5,9 @@ from stimuli.AbstractStimulus import AbstractStimulus
 class StepStimulus(AbstractStimulus):
 
     def __init__(self, start, duration, value, base_value=0, seconds=True):
-        self.start = 0
-        self.duration = 0
+        if duration < 0:
+            raise ValueError("Duration cannot be negative")
+
         self.base_value = base_value
         self.value = value
         if seconds:
@@ -22,10 +23,10 @@ class StepStimulus(AbstractStimulus):
         else:
             return self.base_value
 
-    def get_stimulus_start_ms(self):
+    def get_stimulus_start_s(self):
         return self.start
 
-    def get_stimulus_duration_ms(self):
+    def get_stimulus_duration_s(self):
         return self.duration
 
     def get_amplitude(self):

stimuli/StimulusSequence.py

@@ -0,0 +1,20 @@
+
+from stimuli.AbstractStimulus import AbstractStimulus
+
+
+class StimulusSequence(AbstractStimulus):
+
+    def __init__(self, stimulus_list):
+        self.stimuli = stimulus_list
+
+    def value_at_time_in_s(self, time_point):
+        pass
+
+    def get_stimulus_start_ms(self):
+        pass
+
+    def get_stimulus_duration_ms(self):
+        pass
+
+    def get_amplitude(self):
+        pass