Fix problem with disturbance module if no sensor is present

pymoskito/examples/car/model.py

@@ -46,7 +46,7 @@ def state_function(self, t, x, args):
         """
 
         x1, x2, x3, x4, x5 = x
-        v, w = 1, 1
+        v, w = args[0]
 
         return np.array([v * np.cos(x3),
                          v * np.sin(x3),

pymoskito/examples/tanksystem/model.py

@@ -79,7 +79,7 @@ def event_level_2_high(self, t, x):
         return x[1] - self.settings["hT"]
 
     def calc_output(self, input_vector):
-        return [input_vector[0], input_vector[1] + np.random.random(1)[0] / 100]
+        return input_vector
 
 
 pm.register_simulation_module(pm.Model, TwoTankSystem)

pymoskito/generic_simulation_modules.py

@@ -627,10 +627,10 @@ class GaussianNoise(Disturbance):
                                    ("mean", 0)])
 
     def __init__(self, settings):
-        settings.update([("input signal", "Sensor")])
+        settings.update([("input signal", "Model_Output")])
         Disturbance.__init__(self, settings)
 
-    def _disturb(self, value):
+    def _disturb(self, t, signal):
         return np.random.normal(self._settings['mean'],
                                 self._settings['sigma'],
-                                value.output_dim)
+                                signal.shape)

pymoskito/simulation_modules.py

@@ -593,18 +593,21 @@ def __init__(self, settings):
         SimulationModule.__init__(self, settings)
 
     def calc_output(self, input_dict):
-        return self._disturb(input_dict[self._settings["input signal"]])
+        return self._disturb(
+            input_dict["time"],
+            input_dict.get(self._settings["input signal"], None))
 
     @abstractmethod
-    def _disturb(self, value):
+    def _disturb(self, t, signal):
         """
         Placeholder for disturbance calculations.
 
         If the noise is to be dependent on the measured signal use its `value`
         to create the noise.
 
         Args:
-            value (array-like float): Values from the source selected by the
+            t (float): Current simulation time.
+            signal (array-like float): Values from the source selected by the
                 ``input_signal`` property.
         Returns:
             array-like float: Noise that will be mixed with a signal later on.