Merge pull request #12 from esa/kepler_equation

kepler equation
esa · Jul 9, 2024 · 000a418 · 000a418
2 parents 152df44 + 6944691
commit 000a418
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/dsgp4/__init__.py b/dsgp4/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '1.0.0'
+__version__ = '1.0.1'
 
 import torch
 torch.set_default_dtype(torch.float64)

diff --git a/dsgp4/sgp4.py b/dsgp4/sgp4.py
@@ -130,7 +130,10 @@ def sgp4(satellite, tsince):
         tem5   = (u - aynl * coseo1 + axnl * sineo1 - eo1) / tem5
         tem5=torch.where(tem5>=0.95, 0.95, tem5)
         tem5=torch.where(tem5<=-0.95, -0.95, tem5)
+        #we need to break if abs value of tem5 is less than 1e-12:
         eo1    = eo1 + tem5
+        if torch.all(torch.abs(tem5) < 1e-12):
+            break
 
     #  short period preliminary quantities
     ecose = axnl*coseo1 + aynl*sineo1

diff --git a/dsgp4/sgp4_batched.py b/dsgp4/sgp4_batched.py
@@ -139,6 +139,8 @@ def sgp4_batched(satellite_batch, tsince):
         tem5=torch.where(tem5>=0.95, 0.95, tem5)
         tem5=torch.where(tem5<=-0.95, -0.95, tem5)
         eo1    = eo1 + tem5
+        if torch.all(torch.abs(tem5) < 1e-12):
+            break
 
     # short period preliminary quantities
     ecose = axnl*coseo1 + aynl*sineo1