wraps

GHA_triaxial/gha1_num.py

@@ -10,6 +10,7 @@ from typing import Callable, Tuple, List
 from numpy.typing import NDArray
 
 from GHA_triaxial.utils import alpha_ell2para, pq_ell
+from utils_angle import wrap_0_2pi
 
 
 def buildODE(ell: EllipsoidTriaxial) -> Callable:
@@ -75,8 +76,7 @@ def gha1_num(ell: EllipsoidTriaxial, point: NDArray, alpha0: float, s: float, nu
 
     alpha1 = arctan2(P, Q)
 
-    if alpha1 < 0:
-        alpha1 += 2 * np.pi
+    alpha1 = wrap_0_2pi(alpha1)
 
     _, _, h = ell.cart2geod(point1, "ligas3")
     if h > 1e-5: