Merge pull request #356 from RocketPy-Team/bug/no-chute-lag-warning

BUG: Parachute lag warning when lag is 0 #2

Author: giovaniceotto

rocketpy/Flight.py

@@ -547,7 +547,7 @@ def __init__(
             Default is 90, which points in the x direction.
         initialSolution : array, Flight, optional
             Initial solution array to be used. Format is
-            initialSolution = []
+            initialSolution = [
                 self.tInitial,
                 xInit, yInit, zInit,
                 vxInit, vyInit, vzInit,
@@ -729,13 +729,15 @@ def __init__(
                         self.parachutes.remove(parachute)
                         # Create flight phase for time after detection and before inflation
                         # Must only be created if parachute has any lag
+                        i = 1
                         if parachute.lag != 0:
                             self.flightPhases.addPhase(
                                 node.t,
                                 phase.derivative,
                                 clear=True,
-                                index=phase_index + 1,
+                                index=phase_index + i,
                             )
+                            i += 1
                         # Create flight phase for time after inflation
                         callbacks = [
                             lambda self, parachuteCdS=parachute.CdS: setattr(
@@ -747,7 +749,7 @@ def __init__(
                             self.uDotParachute,
                             callbacks,
                             clear=False,
-                            index=phase_index + 2,
+                            index=phase_index + i,
                         )
                         # Prepare to leave loops and start new flight phase
                         phase.timeNodes.flushAfter(node_index)
@@ -1024,12 +1026,16 @@ def __init__(
                                         # Remove parachute from flight parachutes
                                         self.parachutes.remove(parachute)
                                         # Create flight phase for time after detection and before inflation
-                                        self.flightPhases.addPhase(
-                                            overshootableNode.t,
-                                            phase.derivative,
-                                            clear=True,
-                                            index=phase_index + 1,
-                                        )
+                                        # Must only be created if parachute has any lag
+                                        i = 1
+                                        if parachute.lag != 0:
+                                            self.flightPhases.addPhase(
+                                                overshootableNode.t,
+                                                phase.derivative,
+                                                clear=True,
+                                                index=phase_index + i,
+                                            )
+                                            i += 1
                                         # Create flight phase for time after inflation
                                         callbacks = [
                                             lambda self, parachuteCdS=parachute.CdS: setattr(
@@ -1041,7 +1047,7 @@ def __init__(
                                             self.uDotParachute,
                                             callbacks,
                                             clear=False,
-                                            index=phase_index + 2,
+                                            index=phase_index + i,
                                         )
                                         # Rollback history
                                         self.t = overshootableNode.t