From 2b5752c2828caed09b134d6e066485a392b682a5 Mon Sep 17 00:00:00 2001
From: "Sebastian M. Ernst" <ernst@pleiszenburg.de>
Date: Fri, 23 Feb 2024 10:51:09 +0100
Subject: [PATCH] jit t_p cleanup

---
 src/hapsira/twobody/states.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/hapsira/twobody/states.py b/src/hapsira/twobody/states.py
index c7b9ce71e..f80e65e00 100644
--- a/src/hapsira/twobody/states.py
+++ b/src/hapsira/twobody/states.py
@@ -13,7 +13,7 @@
     mean_motion_vf,
     period_vf,
 )
-from hapsira.twobody.elements import t_p
+from hapsira.core.propagation.farnocchia import delta_t_from_nu_vf, FARNOCCHIA_DELTA
 
 
 u_km3s2 = u.km**3 / u.s**2
@@ -85,11 +85,16 @@ def r_a(self):
     @cached_property
     def t_p(self):
         """Elapsed time since latest perifocal passage."""
-        return t_p(
-            self.to_classical().nu,
-            self.to_classical().ecc,
-            self.attractor.k,
-            self.r_p,
+        self_classical = self.to_classical()
+        return (
+            delta_t_from_nu_vf(
+                self_classical.nu.to_value(u.rad),
+                self_classical.ecc.value,
+                self.attractor.k.to_value(u_km3s2),
+                self.r_p.to_value(u.km),
+                FARNOCCHIA_DELTA,
+            )
+            * u.s
         )
 
     def to_tuple(self):