ENH: add time_overshoot option to rocketpy.stochastic_flight

Since the new StochasticAirBrake class is defined, we need the 'time_overshoot' option in the Flight class to ensure that the time step defined in the simulation is the controller sampling rate. The MonteCarlo class has had to be modified as well to include this option.

Author: kevin-alcaniz

rocketpy/simulation/monte_carlo.py

@@ -458,6 +458,7 @@ def __run_single_simulation(self):
             heading=self.flight._randomize_heading(),
             initial_solution=self.flight.initial_solution,
             terminate_on_apogee=self.flight.terminate_on_apogee,
+            time_overshoot=self.flight.time_overshoot,
         )
 
     def __evaluate_flight_inputs(self, sim_idx):

rocketpy/stochastic/stochastic_flight.py

@@ -29,6 +29,9 @@ class StochasticFlight(StochasticModel):
     terminate_on_apogee : bool
         Whether or not the flight should terminate on apogee. This attribute
         can not be randomized.
+    time_overshoot : bool
+        If False, the simulation will run at the time step defined by the controller
+        sampling rate. Be aware that this will make the simulation run much slower.
     """
 
     def __init__(
@@ -39,6 +42,7 @@ def __init__(
         heading=None,
         initial_solution=None,
         terminate_on_apogee=None,
+        time_overshoot=None,
     ):
         """Initializes the Stochastic Flight class.
 
@@ -63,11 +67,18 @@ def __init__(
         terminate_on_apogee : bool, optional
             Whether or not the flight should terminate on apogee. This attribute
             can not be randomized.
+        time_overshoot : bool
+            If False, the simulation will run at the time step defined by the controller
+            sampling rate. Be aware that this will make the simulation run much slower.
         """
         if terminate_on_apogee is not None:
             assert isinstance(terminate_on_apogee, bool), (
                 "`terminate_on_apogee` must be a boolean"
             )
+        if time_overshoot is not None:
+            assert isinstance(time_overshoot, bool), (
+                "`time_overshoot` must be a boolean"
+            )
         super().__init__(
             flight,
             rail_length=rail_length,
@@ -77,6 +88,7 @@ def __init__(
 
         self.initial_solution = initial_solution
         self.terminate_on_apogee = terminate_on_apogee
+        self.time_overshoot = time_overshoot
 
     def _validate_initial_solution(self, initial_solution):
         if initial_solution is not None:
@@ -128,4 +140,5 @@ def create_object(self):
             heading=generated_dict["heading"],
             initial_solution=self.initial_solution,
             terminate_on_apogee=self.terminate_on_apogee,
+            time_overshoot=self.time_overshoot,
         )