MNT: reimplement EmptyMotor class and refactor imports

Author: Gui-FernandesBR

rocketpy/motors/__init__.py

@@ -1,7 +1,8 @@
+from .empty_motor import EmptyMotor
 from .fluid import Fluid
 from .hybrid_motor import HybridMotor
 from .liquid_motor import LiquidMotor
-from .motor import EmptyMotor, GenericMotor, Motor
+from .motor import GenericMotor, Motor
 from .solid_motor import SolidMotor
 from .tank import (
     LevelBasedTank,

rocketpy/motors/empty_motor.py

@@ -0,0 +1,74 @@
+from rocketpy.mathutils import Function
+from rocketpy.motors.motor import Motor
+
+
+class EmptyMotor(Motor):
+    """Class that represents an empty motor with no mass and no thrust."""
+
+    def __init__(self):
+        """Initializes an empty motor with no mass and no thrust."""
+
+        super().__init__(
+            thrust_source=0,
+            dry_inertia=(0, 0, 0),
+            nozzle_radius=0,
+            center_of_dry_mass_position=0,
+            dry_mass=0,
+            nozzle_position=0,
+            burn_time=1,
+            reshape_thrust_curve=False,
+            interpolation_method="linear",
+            coordinate_system_orientation="nozzle_to_combustion_chamber",
+        )
+
+        # Mass properties
+        self.propellant_mass = Function(0, "Time (s)", "Propellant Mass (kg)")
+        self.total_mass = Function(0, "Time (s)", "Total Mass (kg)")
+        self.total_mass_flow_rate = Function(
+            0, "Time (s)", "Mass Depletion Rate (kg/s)"
+        )
+        self.center_of_mass = Function(0, "Time (s)", "Center of Mass (kg)")
+
+        # Inertia properties
+        self.I_11 = Function(0)
+        self.I_22 = Function(0)
+        self.I_33 = Function(0)
+        self.I_12 = Function(0)
+        self.I_13 = Function(0)
+        self.I_23 = Function(0)
+
+    @property
+    def center_of_propellant_mass(self):
+        return Function(0, "Time (s)", "Center of Propellant Mass (kg)")
+
+    @property
+    def exhaust_velocity(self):
+        return Function(0, "Time (s)", "Exhaust Velocity (m/s)")
+
+    @property
+    def propellant_initial_mass(self):
+        return 0
+
+    @property
+    def propellant_I_11(self):
+        return Function(0, "Time (s)", "Propellant I_11 (kg m²)")
+
+    @property
+    def propellant_I_12(self):
+        return Function(0, "Time (s)", "Propellant I_12 (kg m²)")
+
+    @property
+    def propellant_I_13(self):
+        return Function(0, "Time (s)", "Propellant I_13 (kg m²)")
+
+    @property
+    def propellant_I_22(self):
+        return Function(0, "Time (s)", "Propellant I_22 (kg m²)")
+
+    @property
+    def propellant_I_23(self):
+        return Function(0, "Time (s)", "Propellant I_23 (kg m²)")
+
+    @property
+    def propellant_I_33(self):
+        return Function(0, "Time (s)", "Propellant I_33 (kg m²)")

rocketpy/motors/motor.py

@@ -1162,7 +1162,7 @@ def all_info(self):
         self.prints.all()
         self.plots.all()
 
-
+# TODO: move this class to a separate file, needs a breaking change warning
 class GenericMotor(Motor):
     """Class that represents a simple motor defined mainly by its thrust curve.
     There is no distinction between the propellant types (e.g. Solid, Liquid).
@@ -1601,56 +1601,3 @@ def from_dict(cls, data):
             nozzle_position=data["nozzle_position"],
             interpolation_method=data["interpolate"],
         )
-
-
-class EmptyMotor:
-    """Class that represents an empty motor with no mass and no thrust."""
-
-    # TODO: This is a temporary solution. It should be replaced by a class that
-    # inherits from the abstract Motor class. Currently cannot be done easily.
-    # pylint: disable=too-many-statements
-    def __init__(self):
-        """Initializes an empty motor with no mass and no thrust.
-
-        Notes
-        -----
-        This class is a temporary solution to the problem of having a motor
-        with no mass and no thrust. It should be replaced by a class that
-        inherits from the abstract Motor class. Currently cannot be done easily.
-        """
-        self._csys = 1
-        self.dry_mass = 0
-        self.nozzle_radius = 0
-        self.thrust = Function(0, "Time (s)", "Thrust (N)")
-        self.propellant_mass = Function(0, "Time (s)", "Propellant Mass (kg)")
-        self.propellant_initial_mass = 0
-        self.total_mass = Function(0, "Time (s)", "Total Mass (kg)")
-        self.total_mass_flow_rate = Function(
-            0, "Time (s)", "Mass Depletion Rate (kg/s)"
-        )
-        self.burn_out_time = 1
-        self.nozzle_position = 0
-        self.nozzle_radius = 0
-        self.center_of_dry_mass_position = 0
-        self.center_of_propellant_mass = Function(
-            0, "Time (s)", "Center of Propellant Mass (kg)"
-        )
-        self.center_of_mass = Function(0, "Time (s)", "Center of Mass (kg)")
-        self.dry_I_11 = 0
-        self.dry_I_22 = 0
-        self.dry_I_33 = 0
-        self.dry_I_12 = 0
-        self.dry_I_13 = 0
-        self.dry_I_23 = 0
-        self.propellant_I_11 = Function(0, "Time (s)", "Propellant I_11 (kg m²)")
-        self.propellant_I_22 = Function(0, "Time (s)", "Propellant I_22 (kg m²)")
-        self.propellant_I_33 = Function(0, "Time (s)", "Propellant I_33 (kg m²)")
-        self.propellant_I_12 = Function(0, "Time (s)", "Propellant I_12 (kg m²)")
-        self.propellant_I_13 = Function(0, "Time (s)", "Propellant I_13 (kg m²)")
-        self.propellant_I_23 = Function(0, "Time (s)", "Propellant I_23 (kg m²)")
-        self.I_11 = Function(0)
-        self.I_22 = Function(0)
-        self.I_33 = Function(0)
-        self.I_12 = Function(0)
-        self.I_13 = Function(0)
-        self.I_23 = Function(0)

rocketpy/rocket/rocket.py

@@ -6,7 +6,7 @@
 from rocketpy.control.controller import _Controller
 from rocketpy.mathutils.function import Function
 from rocketpy.mathutils.vector_matrix import Matrix, Vector
-from rocketpy.motors.motor import EmptyMotor
+from rocketpy.motors.empty_motor import EmptyMotor
 from rocketpy.plots.rocket_plots import _RocketPlots
 from rocketpy.prints.rocket_prints import _RocketPrints
 from rocketpy.rocket.aero_surface import (

rocketpy/stochastic/stochastic_rocket.py

@@ -4,7 +4,8 @@
 from random import choice
 
 from rocketpy.mathutils.vector_matrix import Vector
-from rocketpy.motors.motor import EmptyMotor, GenericMotor, Motor
+from rocketpy.motors.empty_motor import EmptyMotor
+from rocketpy.motors.motor import GenericMotor, Motor
 from rocketpy.motors.solid_motor import SolidMotor
 from rocketpy.rocket.aero_surface import (
     EllipticalFins,

tests/fixtures/motor/solid_motor_fixtures.py

@@ -122,7 +122,7 @@ def dimensionless_cesaroni_m1670(kg, m):  # old name: dimensionless_motor
 @pytest.fixture
 def dummy_empty_motor():
     # Create a motor with ZERO thrust and ZERO mass to keep the rocket's speed constant
-    # TODO: why don t we use these same values to create EmptyMotor class?
+    # TODO: Maybe we should simple use the new `EmptyMotor` class?
     return SolidMotor(
         thrust_source=1e-300,
         burn_time=1e-10,

tests/unit/test_rocket.py

@@ -5,7 +5,8 @@
 
 from rocketpy import Function, NoseCone, Rocket, SolidMotor
 from rocketpy.mathutils.vector_matrix import Vector
-from rocketpy.motors.motor import EmptyMotor, Motor
+from rocketpy.motors.empty_motor import EmptyMotor
+from rocketpy.motors.motor import Motor
 
 
 @patch("matplotlib.pyplot.show")