ENH: Plotting both power on and off drag curves in a single plot.

CHANGELOG.md

@@ -44,15 +44,15 @@ straightforward as possible.
 - ENH Precalculate Barometric Height [#511](https://github.com/RocketPy-Team/RocketPy/pull/511)
 - MNT: Encapsulate quaternion conversions [#537](https://github.com/RocketPy-Team/RocketPy/pull/537)
 - MNT: improve the low pass filter and document an example [#538](https://github.com/RocketPy-Team/RocketPy/pull/538)
-
+- ENH: Plotting both power on and off drag curves in a single plot [#547](https://github.com/RocketPy-Team/RocketPy/pull/547)
 
 ### Fixed
 
 - ENH: Parachute trigger doesn't work if "Apogee" is used instead of "apogee" [#489](https://github.com/RocketPy-Team/RocketPy/pull/489)
 - BUG: fin_flutter_analysis doesn't find any fin set [#510](https://github.com/RocketPy-Team/RocketPy/pull/510)
 - FIX: EmptyMotor is breaking the Rocket.draw() method [#516](https://github.com/RocketPy-Team/RocketPy/pull/516)
 - BUG: 3D trajectory plot not labeling axes [#533](https://github.com/RocketPy-Team/RocketPy/pull/533)
-- 
+
 
 ## [v1.1.5] - 2024-01-21
 

rocketpy/plots/rocket_plots.py

@@ -86,27 +86,31 @@ def stability_margin(self):
 
         return None
 
-    def power_on_drag(self):
-        """Plots power on drag of the rocket as a function of time.
-
-        Returns
-        -------
-        None
-        """
-
-        self.rocket.power_on_drag()
-
-        return None
-
-    def power_off_drag(self):
-        """Plots power off drag of the rocket as a function of time.
+    def power_on_and_off_drag(self):
+        """Plots power off and on drag curve of the rocket as a function of time.
 
         Returns
         -------
         None
         """
+        x_power_drag_off = self.rocket.power_off_drag.x_array
+        y_power_drag_off = self.rocket.power_off_drag.y_array
+        x_power_drag_on = self.rocket.power_on_drag.x_array
+        y_power_drag_on = self.rocket.power_on_drag.y_array
+
+        fig, ax = plt.subplots()
+        ax.plot(x_power_drag_on, y_power_drag_on, label="Power on Drag")
+        ax.plot(
+            x_power_drag_off, y_power_drag_off, label="Power off Drag", linestyle="--"
+        )
 
-        self.rocket.power_off_drag()
+        ax.set_title("Power on and off Drag")
+        ax.set_ylabel("Drag Coefficient")
+        ax.set_xlabel("Mach")
+        ax.axvspan(0.8, 1.2, alpha=0.3, color="gray", label="Transonic Region")
+        ax.legend(loc="best", shadow=True)
+        plt.grid(True)
+        plt.show()
 
         return None
 
@@ -524,8 +528,7 @@ def all(self):
         # Drag Plots
         print("Drag Plots")
         print("-" * 20)  # Separator for Drag Plots
-        self.power_on_drag()
-        self.power_off_drag()
+        self.power_on_and_off_drag()
 
         # Stability Plots
         print("\nStability Plots")