ENH: Add orbital flight simulation capabilities

Author: phmbressan

pyproject.toml

@@ -33,13 +33,19 @@ build-backend = "setuptools.build_meta"
 
 [tool.setuptools]
 packages = { find = { where = ["."], include = ["rocketpy*"] } }
+[tool.setuptools.package-data]
+"rocketpy.plots" = ["assets/*.stl"]
+"rocketpy.environment" = ["data/*.npz"]
 
 
 [tool.setuptools.dynamic]
 dependencies = { file = ["requirements.txt"] }
 
 
 [project.optional-dependencies]
+performance = [
+    "numba>=0.60",
+]
 tests = [
     "pytest",
     "pytest-coverage",
@@ -68,7 +74,17 @@ monte-carlo = [
     "contextily>=1.0.0; python_version < '3.14'",
 ]
 
-all = ["rocketpy[env-analysis]", "rocketpy[monte-carlo]"]
+animation = [
+    "pyvista[jupyter]>=0.45",
+    "imageio",
+    "imageio-ffmpeg>=0.5"
+]
+
+all = [
+    "rocketpy[env-analysis]",
+    "rocketpy[monte-carlo]",
+    "rocketpy[animation]",
+]
 
 
 [tool.coverage.report]

rocketpy/__init__.py

@@ -1,10 +1,41 @@
 from ._logging import enable_logging, logger, set_log_level
 from .control import _Controller
-from .environment import Environment, EnvironmentAnalysis
+from .environment import (
+    Atmosphere,
+    AnalyticalEphemeris,
+    AtmosphericState,
+    CelestialBody,
+    DefaultGravity,
+    Environment,
+    EnvironmentAnalysis,
+    ExponentialAtmosphere,
+    Gravity,
+    HarrisPriesterAtmosphere,
+    LayeredAtmosphere,
+    NRLMSISE00,
+    SphericalGravity,
+    SphericalHarmonicGravity,
+    SpiceEphemeris,
+    VacuumAtmosphere,
+    VerticalGravity,
+    ZeroGravity,
+    ZonalGravity,
+)
 from .mathutils import (
+    EarthDatum,
+    Epoch,
+    FlightState,
     Function,
+    NUMBA_AVAILABLE,
+    OrbitalElements,
     PiecewiseFunction,
+    ReferenceFrame,
+    WGS84,
+    VectorFunction,
     funcify_method,
+    gcrf_to_rtn_matrix,
+    itrf_to_topocentric,
+    numbify,
     reset_funcified_methods,
 )
 from .motors import (
@@ -51,10 +82,16 @@
 from .sensitivity import SensitivityModel
 from .sensors import Accelerometer, Barometer, GnssReceiver, Gyroscope
 from .simulation import (
+    EarthRadiationPressure,
     Event,
     Flight,
+    FlightOrbit,
     MonteCarlo,
     MultivariateRejectionSampler,
+    PlanetaryRadiationPressure,
+    RelativisticCorrection,
+    SolarRadiationPressure,
+    ThirdBodyGravity,
 )
 from .stochastic import (
     CustomSampler,

rocketpy/environment/__init__.py

@@ -3,7 +3,48 @@
 considered private and should be used with caution.
 """
 
+from .atmosphere import (
+    Atmosphere,
+    AtmosphericState,
+    ExponentialAtmosphere,
+    FunctionAtmosphere,
+    HarrisPriesterAtmosphere,
+    LayeredAtmosphere,
+    NRLMSISE00,
+    VacuumAtmosphere,
+)
+from .celestial_body import AnalyticalEphemeris, CelestialBody, SpiceEphemeris
 from .environment import Environment
 from .environment_analysis import EnvironmentAnalysis
+from .gravity import (
+    DefaultGravity,
+    Gravity,
+    SphericalGravity,
+    SphericalHarmonicGravity,
+    VerticalGravity,
+    ZeroGravity,
+    ZonalGravity,
+)
 
-__all__ = ["Environment", "EnvironmentAnalysis"]
+__all__ = [
+    "AnalyticalEphemeris",
+    "Atmosphere",
+    "AtmosphericState",
+    "CelestialBody",
+    "DefaultGravity",
+    "Environment",
+    "EnvironmentAnalysis",
+    "ExponentialAtmosphere",
+    "FunctionAtmosphere",
+    "Gravity",
+    "HarrisPriesterAtmosphere",
+    "LayeredAtmosphere",
+    "NRLMSISE00",
+    "SphericalGravity",
+    "SphericalHarmonicGravity",
+    "SpiceEphemeris",
+    "VacuumAtmosphere",
+    "VerticalGravity",
+    "ZeroGravity",
+    "ZonalGravity",
+]