ENH: refactor motor prints classes to inherit from _MotorPrints

Author: Gui-FernandesBR

rocketpy/prints/hybrid_motor_prints.py

@@ -1,7 +1,9 @@
 import numpy as np
 
+from .motor_prints import _MotorPrints
 
-class _HybridMotorPrints:
+
+class _HybridMotorPrints(_MotorPrints):
     """Class that holds prints methods for HybridMotor class.
 
     Attributes
@@ -26,6 +28,7 @@ def __init__(
         -------
         None
         """
+        super().__init__(hybrid_motor)
         self.hybrid_motor = hybrid_motor
 
     def nozzle_details(self):
@@ -63,28 +66,6 @@ def grain_details(self):
         print(f"Grain Volume: {self.hybrid_motor.solid.grain_initial_volume:.3f} m3")
         print(f"Grain Mass: {self.hybrid_motor.solid.grain_initial_mass:.3f} kg\n")
 
-    def motor_details(self):
-        """Prints out all data available about the HybridMotor.
-
-        Returns
-        -------
-        None
-        """
-        print("Motor Details")
-        print(f"Total Burning Time: {self.hybrid_motor.burn_duration} s")
-        print(
-            f"Total Propellant Mass: {self.hybrid_motor.propellant_initial_mass:.3f} kg"
-        )
-        print(f"Structural Mass Ratio: {self.hybrid_motor.structural_mass_ratio:.3f}")
-        avg = self.hybrid_motor.exhaust_velocity.average(*self.hybrid_motor.burn_time)
-        print(f"Average Propellant Exhaust Velocity: {avg:.3f} m/s")
-        print(f"Average Thrust: {self.hybrid_motor.average_thrust:.3f} N")
-        print(
-            f"Maximum Thrust: {self.hybrid_motor.max_thrust} N at "
-            f"{self.hybrid_motor.max_thrust_time} s after ignition."
-        )
-        print(f"Total Impulse: {self.hybrid_motor.total_impulse:.3f} Ns\n")
-
     def all(self):
         """Prints out all data available about the HybridMotor.
 

rocketpy/prints/liquid_motor_prints.py

@@ -1,4 +1,7 @@
-class _LiquidMotorPrints:
+from .motor_prints import _MotorPrints
+
+
+class _LiquidMotorPrints(_MotorPrints):
     """Class that holds prints methods for LiquidMotor class.
 
     Attributes
@@ -23,6 +26,7 @@ def __init__(
         -------
         None
         """
+        super().__init__(liquid_motor)
         self.liquid_motor = liquid_motor
 
     def nozzle_details(self):
@@ -35,28 +39,6 @@ def nozzle_details(self):
         print("Nozzle Details")
         print("Nozzle Radius: " + str(self.liquid_motor.nozzle_radius) + " m\n")
 
-    def motor_details(self):
-        """Prints out all data available about the motor.
-
-        Returns
-        -------
-        None
-        """
-        print("Motor Details")
-        print(f"Total Burning Time: {self.liquid_motor.burn_duration} s")
-        print(
-            f"Total Propellant Mass: {self.liquid_motor.propellant_initial_mass:.3f} kg"
-        )
-        print(f"Structural Mass Ratio: {self.liquid_motor.structural_mass_ratio:.3f}")
-        avg = self.liquid_motor.exhaust_velocity.average(*self.liquid_motor.burn_time)
-        print(f"Average Propellant Exhaust Velocity: {avg:.3f} m/s")
-        print(f"Average Thrust: {self.liquid_motor.average_thrust:.3f} N")
-        print(
-            f"Maximum Thrust: {self.liquid_motor.max_thrust} N at "
-            f"{self.liquid_motor.max_thrust_time} s after ignition."
-        )
-        print(f"Total Impulse: {self.liquid_motor.total_impulse:.3f} Ns\n")
-
     def all(self):
         """Prints out all data available about the LiquidMotor.
 

rocketpy/prints/solid_motor_prints.py

@@ -1,4 +1,7 @@
-class _SolidMotorPrints:
+from .motor_prints import _MotorPrints
+
+
+class _SolidMotorPrints(_MotorPrints):
     """Class that holds prints methods for SolidMotor class.
 
     Attributes
@@ -23,6 +26,7 @@ def __init__(
         -------
         None
         """
+        super().__init__(solid_motor)
         self.solid_motor = solid_motor
 
     def nozzle_details(self):
@@ -53,28 +57,6 @@ def grain_details(self):
         print(f"Grain Volume: {self.solid_motor.grain_initial_volume:.3f} m3")
         print(f"Grain Mass: {self.solid_motor.grain_initial_mass:.3f} kg\n")
 
-    def motor_details(self):
-        """Prints out all data available about the SolidMotor.
-
-        Returns
-        -------
-        None
-        """
-        print("Motor Details")
-        print("Total Burning Time: " + str(self.solid_motor.burn_duration) + " s")
-        print(
-            f"Total Propellant Mass: {self.solid_motor.propellant_initial_mass:.3f} kg"
-        )
-        print(f"Structural Mass Ratio: {self.solid_motor.structural_mass_ratio:.3f}")
-        average = self.solid_motor.exhaust_velocity.average(*self.solid_motor.burn_time)
-        print(f"Average Propellant Exhaust Velocity: {average:.3f} m/s")
-        print(f"Average Thrust: {self.solid_motor.average_thrust:.3f} N")
-        print(
-            f"Maximum Thrust: {self.solid_motor.max_thrust} N "
-            f"at {self.solid_motor.max_thrust_time} s after ignition."
-        )
-        print(f"Total Impulse: {self.solid_motor.total_impulse:.3f} Ns\n")
-
     def all(self):
         """Prints out all data available about the SolidMotor.