Add the aberrated orbit to Ephemeris

B612-Asteroid-Institute · Aug 28, 2023 · 2f4cb2a · 2f4cb2a
1 parent ad63e16
commit 2f4cb2a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/adam_core/orbits/ephemeris.py b/adam_core/orbits/ephemeris.py
@@ -1,5 +1,6 @@
 from quivr import StringColumn, Table
 
+from ..coordinates.cartesian import CartesianCoordinates
 from ..coordinates.spherical import SphericalCoordinates
 
 
@@ -8,3 +9,4 @@ class Ephemeris(Table):
     orbit_id = StringColumn(nullable=False)
     object_id = StringColumn()
     coordinates = SphericalCoordinates.as_column()
+    aberrated_orbit = CartesianCoordinates.as_column()
diff --git a/adam_core/propagator/pyoorb.py b/adam_core/propagator/pyoorb.py
@@ -466,6 +466,23 @@ def _generate_ephemeris(
                     origin=Origin.from_kwargs(code=codes),
                     frame="equatorial",
                 ),
+                aberrated_orbit=CartesianCoordinates.from_kwargs(
+                    x=ephemeris[:, 24],
+                    y=ephemeris[:, 25],
+                    z=ephemeris[:, 26],
+                    vx=ephemeris[:, 27],
+                    vy=ephemeris[:, 28],
+                    vz=ephemeris[:, 29],
+                    time=Times.from_astropy(
+                        Time(
+                            ephemeris[:, 0],
+                            scale="utc",
+                            format="mjd",
+                        )
+                    ),
+                    origin=Origin.from_kwargs(code=["SUN" for i in range(len(codes))]),
+                    frame="ecliptic",
+                ),
             )
             ephemeris_list.append(ephemeris)