Special case Keplerian angles for b=0, to get rid of NaN angles

jobovy · Nov 13, 2017 · e5c2d8b · e5c2d8b
1 parent 443b82e
commit e5c2d8b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/galpy/actionAngle_src/actionAngleIsochrone.py b/galpy/actionAngle_src/actionAngleIsochrone.py
@@ -247,8 +247,11 @@ def _actionsFreqsAngles(self,*args,**kwargs):
             c= -self.amp/2./E-self.b
             e2= 1.-L2/self.amp/c*(1.+self.b/c)
             e= nu.sqrt(e2)
-            s= 1.+nu.sqrt(1.+(R**2.+z**2.)/self.b**2.)
-            coseta= 1/e*(1.-self.b/c*(s-2.))
+            if self.b == 0.:
+                coseta= 1/e*(1.-nu.sqrt(R**2.+z**2.)/c)
+            else:
+                s= 1.+nu.sqrt(1.+(R**2.+z**2.)/self.b**2.)
+                coseta= 1/e*(1.-self.b/c*(s-2.))
             pindx= (coseta > 1.)*(coseta < (1.+10.**-7.))
             coseta[pindx]= 1.
             pindx= (coseta < -1.)*(coseta > (-1.-10.**-7.))