astrox-python 1.0.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

astrox/celestial.py

@@ -74,6 +74,9 @@ class MpcOrbitalElements:
     raan_deg: float | None = None
     argument_of_periapsis_deg: float | None = None
     mean_anomaly_deg: float | None = None
+    reference_frame: str | None = None
+    """Heliocentric mean-ecliptic variant: ``MeanEclpJ2000`` (JPL) or
+    ``EclpJ2000ICRF`` (MPC, server default)."""
 
     def __post_init__(self) -> None:
         for field_name in (
@@ -95,9 +98,14 @@ class MpcOrbitalElements:
                     parameter=field_name,
                 ),
             )
+        object.__setattr__(
+            self,
+            "reference_frame",
+            _optional_string(self.reference_frame, parameter="reference_frame"),
+        )
 
-    def to_wire(self) -> dict[str, float]:
-        payload: dict[str, float] = {}
+    def to_wire(self) -> dict[str, Any]:
+        payload: dict[str, Any] = {}
         _include_if_supplied(payload, "EpochMjdTdt", self.epoch_mjd_tdt)
         _include_if_supplied(payload, "PeriTimeMjdTdt", self.periapsis_time_mjd_tdt)
         _include_if_supplied(payload, "Q", self.periapsis_distance_au)
@@ -111,6 +119,7 @@ class MpcOrbitalElements:
             self.argument_of_periapsis_deg,
         )
         _include_if_supplied(payload, "MeanAnomaly", self.mean_anomaly_deg)
+        _include_if_supplied(payload, "ReferenceFrame", self.reference_frame)
         return payload
 
 
@@ -125,6 +134,7 @@ def mpc_orbital_elements(
     raan_deg: float | None = None,
     argument_of_periapsis_deg: float | None = None,
     mean_anomaly_deg: float | None = None,
+    reference_frame: str | None = None,
 ) -> MpcOrbitalElements:
     """Build an MPC orbital-element fragment without adding physics policy."""
     return MpcOrbitalElements(
@@ -137,6 +147,7 @@ def mpc_orbital_elements(
         raan_deg=raan_deg,
         argument_of_periapsis_deg=argument_of_periapsis_deg,
         mean_anomaly_deg=mean_anomaly_deg,
+        reference_frame=reference_frame,
     )
 
 
@@ -153,7 +164,7 @@ def _elements_to_wire(
     value: MpcOrbitalElements | None,
     *,
     parameter: str,
-) -> dict[str, float] | None:
+) -> dict[str, Any] | None:
     if value is None:
         return None
     if not isinstance(value, MpcOrbitalElements):
@@ -175,6 +186,9 @@ def lambert_transfer_window(
     arrival_step_days: float | None = None,
     departure_elements: MpcOrbitalElements | None = None,
     arrival_elements: MpcOrbitalElements | None = None,
+    max_departure_delta_v_m_s: int | None = None,
+    max_arrival_delta_v_m_s: int | None = None,
+    max_time_of_flight_days: int | None = None,
 ) -> dict[str, Any]:
     """Return a Lambert transfer-window scan from ASTROX.
 
@@ -236,6 +250,30 @@ def lambert_transfer_window(
         "ArrivalElements",
         _elements_to_wire(arrival_elements, parameter="arrival_elements"),
     )
+    _include_if_supplied(
+        payload,
+        "MaxDepartureDV",
+        _optional_integer(
+            max_departure_delta_v_m_s,
+            parameter="max_departure_delta_v_m_s",
+        ),
+    )
+    _include_if_supplied(
+        payload,
+        "MaxArrivalDV",
+        _optional_integer(
+            max_arrival_delta_v_m_s,
+            parameter="max_arrival_delta_v_m_s",
+        ),
+    )
+    _include_if_supplied(
+        payload,
+        "MaxTofDays",
+        _optional_integer(
+            max_time_of_flight_days,
+            parameter="max_time_of_flight_days",
+        ),
+    )
     return _without_status_fields(
         raw.post("/celestial/transfer", json=payload),
         endpoint="/celestial/transfer",
@@ -331,8 +369,16 @@ def mpc_ephemeris(
     observer_frame: str | None = None,
     start: str | None = None,
     stop: str | None = None,
+    step_s: float | None = None,
+    target_elements: MpcOrbitalElements | None = None,
 ) -> dict[str, Any]:
-    """Return ASTROX minor-planet ephemeris output from its MPC-backed route."""
+    """Return ASTROX minor-planet ephemeris output from its MPC-backed route.
+
+    When ``target_elements`` is supplied, the server integrates those MPC
+    orbital elements directly instead of resolving ``target_name`` through the
+    MPC network query. ``step_s`` controls the output sampling cadence; zero
+    requests the server's internal integration grid.
+    """
     payload: dict[str, Any] = {
         "TargetName": _string(target_name, parameter="target_name"),
     }
@@ -351,6 +397,16 @@ def mpc_ephemeris(
         "Stop",
         _optional_string(stop, parameter="stop"),
     )
+    _include_if_supplied(
+        payload,
+        "Step",
+        _optional_number(step_s, parameter="step_s"),
+    )
+    _include_if_supplied(
+        payload,
+        "TargetElements",
+        _elements_to_wire(target_elements, parameter="target_elements"),
+    )
     return _without_status_fields(
         raw.post("/celestial/mpc", json=payload),
         endpoint="/celestial/mpc",

astrox/orbits.py

@@ -11,6 +11,7 @@ from astrox._http import raw
 __all__ = [
     "CartesianState",
     "KeplerianElements",
+    "LambertResult",
     "MeanKeplerianElements",
     "Tle",
     "cartesian_state",
@@ -517,14 +518,37 @@ def geo_ym_lambert_delta_v(
     )
 
 
+@dataclass(frozen=True, kw_only=True)
+class LambertResult:
+    """Single-case Lambert solution with the sampled transfer positions."""
+
+    departure_delta_v_m_s: tuple[float, float, float]
+    arrival_delta_v_m_s: tuple[float, float, float]
+    positions: tuple[float, ...]
+    """Flat ``(x, y, z, ...)`` transfer position sequence in meters.
+
+    The server samples ``path_point_count`` points including the departure and
+    arrival positions, so the sequence length is ``path_point_count * 3``.
+    """
+
+
 def lambert_delta_v(
     *,
     departure_state: CartesianState,
     arrival_state: CartesianState,
     time_of_flight_s: float,
     gravitational_parameter_m3_s2: float | None = None,
-) -> tuple[tuple[float, float, float], tuple[float, float, float]]:
-    """Return single-revolution Lambert delta-v between two Cartesian states."""
+    get_path_points: bool = False,
+    path_point_count: int | None = None,
+) -> tuple[tuple[float, float, float], tuple[float, float, float]] | LambertResult:
+    """Return single-revolution Lambert delta-v between two Cartesian states.
+
+    By default returns ``(departure_delta_v_m_s, arrival_delta_v_m_s)``. When
+    ``get_path_points`` is true, the server also samples the transfer orbit and
+    the return is a :class:`LambertResult` carrying the delta-v pair plus the
+    flat position sequence; ``path_point_count`` sets the sample count
+    (server default 100, including both endpoints).
+    """
     if not isinstance(departure_state, CartesianState):
         raise TypeError("departure_state must be a CartesianState instance")
     if not isinstance(arrival_state, CartesianState):
@@ -536,12 +560,20 @@ def lambert_delta_v(
         "TOF": [time_of_flight_s],
     }
     _include_if_supplied(payload, "Gm", gravitational_parameter_m3_s2)
+    if get_path_points:
+        payload["GetPathPoints"] = True
+        _include_if_supplied(payload, "NumberOfPathPoints", path_point_count)
 
     result = raw.post("/orbit/lambert", json=payload)
-    return (
-        (result["DV1"][0], result["DV1"][1], result["DV1"][2]),
-        (result["DV2"][0], result["DV2"][1], result["DV2"][2]),
-    )
+    departure_delta_v = (result["DV1"][0], result["DV1"][1], result["DV1"][2])
+    arrival_delta_v = (result["DV2"][0], result["DV2"][1], result["DV2"][2])
+    if get_path_points:
+        return LambertResult(
+            departure_delta_v_m_s=departure_delta_v,
+            arrival_delta_v_m_s=arrival_delta_v,
+            positions=tuple(result["Positions"]),
+        )
+    return departure_delta_v, arrival_delta_v
 
 
 def convert_czml_position(

astrox/propagator.py

@@ -42,6 +42,7 @@ __all__ = [
     "sgp4",
     "simple_ascent",
     "two_body",
+    "two_body_rv",
 ]
 
 
@@ -672,6 +673,36 @@ def two_body(
     return _success_path(result)
 
 
+def two_body_rv(
+    *,
+    state: CartesianState,
+    time_of_flight_s: float,
+    gravitational_parameter_m3_s2: float | None = None,
+    step_s: float | None = None,
+) -> tuple[float, ...]:
+    """Propagate a Cartesian state with two-body dynamics and return the sampled ephemeris.
+
+    Returns the flat ``[t, x, y, z, vx, vy, vz, ...]`` sequence, where ``t`` is
+    seconds relative to the initial epoch, positions are in meters, and
+    velocities are in m/s. The sequence includes the start and stop samples;
+    intermediate samples follow ``step_s`` (server default 60 s).
+    """
+    if not isinstance(state, CartesianState):
+        raise TypeError("state must be a CartesianState instance")
+
+    payload: dict[str, Any] = {
+        "RV0": state.to_wire(),
+        "TimeOfFlight": time_of_flight_s,
+    }
+    if gravitational_parameter_m3_s2 is not None:
+        payload["Gm"] = gravitational_parameter_m3_s2
+    if step_s is not None:
+        payload["Step"] = step_s
+
+    result = raw.post("/Propagator/TwoBodyRV", json=payload)
+    return tuple(result["Positions"])
+
+
 def j2(
     *,
     start: str,

{astrox_python-1.0.0.dist-info → astrox_python-1.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: astrox-python
-Version: 1.0.0
+Version: 1.2.0
 Summary: Python SDK for the ASTROX Web API.
 Project-URL: Homepage, https://github.com/Mtrya/astrox-python
 Project-URL: Repository, https://github.com/Mtrya/astrox-python.git

{astrox_python-1.0.0.dist-info → astrox_python-1.2.0.dist-info}/RECORD

@@ -3,13 +3,13 @@ astrox/_http.py,sha256=ZZDkJ6qPG4VbMDRZO7xonIib88aYS6PPTVRxRYwXU3A,15606
 astrox/access.py,sha256=pjyCher_yQNmkoojf2sWpG1HRDnLLK9gI90jM1iEoTo,5817
 astrox/cat.py,sha256=doZqy6lGDtgrVe9QlU57l1lccmzirc14RDSR41tyGUU,15057
 astrox/catalog.py,sha256=WDb3adiDO41YY6IdivjN3ri1vIQMHi6DPI3dBa8W1Gc,5255
-astrox/celestial.py,sha256=Mh8VChM-YWGAM4Jn2fs09SxfOuV4kR56j36LmKv6MP0,10840
+astrox/celestial.py,sha256=EFuKRyZ2uMsjqdRCDTdwDsA3QuzstIU98DwWj8t-Btk,12703
 astrox/conjunction.py,sha256=sXhTNAlMai27ejuJekksaAw-850nHPimo21YdFpuzQo,9108
 astrox/exceptions.py,sha256=apClRT1r-N3XY1ibXxkW3oxpsJgiHos4tC0cMJ8eBRk,2592
 astrox/libration.py,sha256=YBgcZi2pTAjL5VUlE8y6jLhVbYnwRxv-PIUxJQmFBSg,13454
 astrox/lighting.py,sha256=aWMty64rU3wv1H1Q5q-yXG5mINbIY31omyd-pjdNaVs,3876
-astrox/orbits.py,sha256=JiSkCyWqYmeV-yX5WBL9554_JNhec1Li8EdmkZmzXQc,17823
-astrox/propagator.py,sha256=kiW4kFSBy2_quZPZ2VqP49vfSYDKkJ1dna5rM5bo-9M,37334
+astrox/orbits.py,sha256=SdRE7f9W9zO4N6Xxjg5quHK--gsVh1E5qdvJigpwxLA,19223
+astrox/propagator.py,sha256=GPcg5fpra_jsof4ipxgtaeJNof2PrWBvUTnDG1aH940,38434
 astrox/rocket.py,sha256=tb6wXc4bZ0EMAd9bVzva_7k3I9S2lsYA7Q2RD6byqfk,2755
 astrox/terrain.py,sha256=2v_Ca5OnXjmMtIKNFQJ_Kbdsh42ZuIc4ZDe-0aWnpdg,6146
 astrox/astrogator/__init__.py,sha256=rgIHos9rHjQ3i-qkpLeLRwZL6zsVthc2bOVbAdwHWHU,267
@@ -33,7 +33,7 @@ astrox/coverage/number_of_assets.py,sha256=Bf1OQRu1r9lunyARyQoV4x7KetkgY_7OMNcEc
 astrox/coverage/response_time.py,sha256=yLnv8eEDL6CS0PJHC1JIKBCoOachOwo3m8dNoHH3HfQ,5164
 astrox/coverage/revisit_time.py,sha256=R5vF2egixCF9elvsGYXzpgWbD5lfcQW8uZqi1ifIDX8,5155
 astrox/coverage/simple_coverage.py,sha256=ZPj3HrP3M59rG47pz4eHt1rq-vTlk-oZ7c2ZBmFsz1U,4992
-astrox_python-1.0.0.dist-info/METADATA,sha256=zlaPEYsPsqHp6bnpp655SzjcopUFyKAAK-sC1-Z7r80,2520
-astrox_python-1.0.0.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-astrox_python-1.0.0.dist-info/licenses/LICENSE,sha256=Fbt55lCQnv_n5yGhbG_VDlowjaV_FPHBQWsJIyuBpDk,1062
-astrox_python-1.0.0.dist-info/RECORD,,
+astrox_python-1.2.0.dist-info/METADATA,sha256=wh8FRFOydoqxg-BeW5pnZZj1Pz5i_zaC_TxamnoA5mw,2520
+astrox_python-1.2.0.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+astrox_python-1.2.0.dist-info/licenses/LICENSE,sha256=Fbt55lCQnv_n5yGhbG_VDlowjaV_FPHBQWsJIyuBpDk,1062
+astrox_python-1.2.0.dist-info/RECORD,,