MNT: speedup Flight._calculate_pressure_signal method

- Initializing a Function object is complicated. Instead, let's sum two Functions
RocketPy-Team · Mar 26, 2024 · 6deb233 · 6deb233
1 parent 9b47dfe
commit 6deb233
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/rocketpy/simulation/flight.py b/rocketpy/simulation/flight.py
@@ -3074,17 +3074,13 @@ def _calculate_pressure_signal(self):
                 "Pressure - Without Noise (Pa)",
                 "linear",
             )
-            parachute.noisy_pressure_signal_function = Function(
-                parachute.noisy_pressure_signal,
-                "Time (s)",
-                "Pressure - With Noise (Pa)",
-                "linear",
-            )
             parachute.noise_signal_function = Function(
                 parachute.noise_signal, "Time (s)", "Pressure Noise (Pa)", "linear"
             )
-
-        return None
+            parachute.noisy_pressure_signal_function = (
+                parachute.clean_pressure_signal_function
+                + parachute.noise_signal_function
+            )
 
     def post_process(self, interpolation="spline", extrapolation="natural"):
         """This method is **deprecated** and is only kept here for backwards