pybhpt 0.9.5__cp310-cp310-macosx_14_0_arm64.whl → 0.9.8__cp310-cp310-macosx_14_0_arm64.whl

pybhpt/geo.py

@@ -499,25 +499,256 @@ class KerrGeodesic:
         """
         return np.dot(np.array([n, k, m]), (self.frequencies))
     
-    def minotime(self, t):
+    def psi_radial(self, la):
+        """
+        Function that returns the radial true anomaly for a given Mino time value.
+
+        Parameters
+        ----------
+        la : float or numpy.ndarray
+            The Mino time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The radial true anomaly for the given Mino time value(s).
+        """
+        if isinstance(la, np.ndarray) or isinstance(la, list):
+            return self.base.psi_radial_vec(np.array(la))
+        else:
+            return self.base.psi_radial(la)
+        
+    def psi_polar(self, la):
+        """
+        Function that returns the polar Darwin phase for a given Mino time value.
+
+        Parameters
+        ----------
+        la : float or numpy.ndarray
+            The Mino time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The polar Darwin phase for the given Mino time value(s).
+        """
+        if isinstance(la, np.ndarray) or isinstance(la, list):
+            return self.base.psi_polar_vec(np.array(la))
+        else:
+            return self.base.psi_polar(la)
+        
+    def psi_radial_of_t(self, t):
+        """
+        Function that returns the radial true anomaly for a given time value.
+
+        Parameters
+        ----------
+        t : float or numpy.ndarray
+            The time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The radial true anomaly for the given time value(s).
+        """
+        if isinstance(t, np.ndarray) or isinstance(t, list):
+            return self.base.psi_radial_time_vec(np.array(t))
+        else:
+            return self.base.psi_radial_time(t)
+
+    def psi_polar_of_t(self, t):
+        """
+        Function that returns the polar Darwin phase for a given time value.
+
+        Parameters
+        ----------
+        t : float or numpy.ndarray
+            The time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The polar Darwin phase for the given time value(s).
+        """
+        if isinstance(t, np.ndarray) or isinstance(t, list):
+            return self.base.psi_polar_time_vec(np.array(t))
+        else:
+            return self.base.psi_polar_time(t)
+
+    def time_position(self, la):
+        """
+        Function that returns the time position for a given Mino time value.
+
+        Parameters
+        ----------
+        la : float or numpy.ndarray
+            The Mino time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The time position for the given Mino time value(s).
+        """
+        if isinstance(la, np.ndarray) or isinstance(la, list):
+            return self.base.time_position_vec(np.array(la))
+        else:
+            return self.base.time_position(la)
+
+    def radial_position(self, la):
+        """
+        Function that returns the radial position for a given Mino time value.
+
+        Parameters
+        ----------
+        la : float or numpy.ndarray
+            The Mino time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The radial position for the given Mino time value(s).
+        """
+        if isinstance(la, np.ndarray) or isinstance(la, list):
+            return self.base.radial_position_vec(np.array(la))
+        else:
+            return self.base.radial_position(la)
+        
+    def polar_position(self, la):
+        """
+        Function that returns the polar position for a given Mino time value.
+
+        Parameters
+        ----------
+        la : float or numpy.ndarray
+            The Mino time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The polar position for the given Mino time value(s).
+        """
+        if isinstance(la, np.ndarray) or isinstance(la, list):
+            return self.base.polar_position_vec(np.array(la))
+        else:
+            return self.base.polar_position(la)
+    
+    def azimuthal_position(self, la):
+        """
+        Function that returns the azimuthal position for a given Mino time value.
+
+        Parameters
+        ----------
+        la : float or numpy.ndarray
+            The Mino time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The azimuthal position for the given Mino time value(s).
+        """
+        if isinstance(la, np.ndarray) or isinstance(la, list):
+            return self.base.azimuthal_position_vec(np.array(la))
+        else:
+            return self.base.azimuthal_position(la)
+    
+    def radial_position_of_t(self, t):
+        """
+        Function that returns the radial position for a given Mino time value.
+
+        Parameters
+        ----------
+        t : float or numpy.ndarray
+            The Mino time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The radial position for the given Mino time value(s).
+        """
+        if isinstance(t, np.ndarray) or isinstance(t, list):
+            return self.base.radial_position_time_vec(np.array(t))
+        else:
+            return self.base.radial_position_time(t)
+        
+    def polar_position_of_t(self, t):
+        """
+        Function that returns the polar position for a given time value.
+
+        Parameters
+        ----------
+        t : float or numpy.ndarray
+            The time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The polar position for the given time value(s).
+        """
+        if isinstance(t, np.ndarray) or isinstance(t, list):
+            return self.base.polar_position_time_vec(np.array(t))
+        else:
+            return self.base.polar_position_time(t)
+    
+    def azimuthal_position_of_t(self, t):
+        """
+        Function that returns the azimuthal position for a given time value.
+
+        Parameters
+        ----------
+        t : float or numpy.ndarray
+            The time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The azimuthal position for the given time value(s).
+        """
+        if isinstance(t, np.ndarray) or isinstance(t, list):
+            return self.base.azimuthal_position_time_vec(np.array(t))
+        else:
+            return self.base.azimuthal_position_time(t)
+    
+    def mino_of_t(self, t):
         """
         Function that returns the Mino time for a given Boyer-Lindquist time.
 
         Parameters
         ----------
-        t : float
-            The Boyer-Lindquist time.
+        t : float or numpy.ndarray
+            The Boyer-Lindquist time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
         
-        Returns
+        Returns 
         -------
-        float
-            The Mino time for the given Boyer-Lindquist time.
+        numpy.ndarray
+            The Mino time for the given Boyer-Lindquist time value(s).
         """
-        return self.base.mino_time(t)
+        if isinstance(t, np.ndarray) or isinstance(t, list):
+            return self.base.mino_time_vec(np.array(t))
+        else:
+            return self.base.mino_time(t)
     
-    def __call__(self, la):
+    def position_of_t(self, t):
         """
-        Function that returns the position vector for a given Mino time value.
+        Function that returns the position vector (r, theta, phi) for a given Boyer-Lindquist time value.
+        
+        Parameters
+        ----------
+        t : float or numpy.ndarray
+            The Boyer-Lindquist time value(s). If a numpy array is provided, the function will return a numpy array of the same shape.
+        
+        Returns 
+        -------
+        numpy.ndarray
+            The position vector (r, theta, phi) for the given Boyer-Lindquist time value(s).
+        """
+        if isinstance(t, np.ndarray) or isinstance(t, list):
+            return self.base.position_time_vec(np.array(t))
+        else:
+            return self.base.position_time(t)
+        
+    def position(self, la):
+        """
+        Function that returns the position vector (t, r, theta, phi) for a given Mino time value.
         Parameters
         ----------
         la : float or numpy.ndarray
@@ -525,9 +756,12 @@ class KerrGeodesic:
         Returns 
         -------
         numpy.ndarray
-            The position vector for the given Mino time value(s).
+            The position vector (t, r, theta, phi) for the given Mino time value(s).
         """
         if isinstance(la, np.ndarray) or isinstance(la, list):
             return self.base.position_vec(np.array(la))
         else:
-            return self.base.position(la)
+            return self.base.position(la)
+    
+    def __call__(self, la):
+        return self.position(la)

pybhpt/swsh.py

@@ -48,6 +48,9 @@ def Yslm(s, l, m, th, ph = None):
     array_like
         The values of the spherical harmonic at the specified angles.
     """
+    assert isinstance(s, int) and isinstance(l, int) and isinstance(m, int), "s, l, and m must be integers"
+    if ph is not None:
+        return Yslm(s, l, m, th)*np.exp(1.j*m*ph)
     if np.abs(s) > l:
         return 0.*th
     if s == 0:
@@ -377,7 +380,7 @@ class SpinWeightedSpheroidalHarmonic(SWSHSeriesBase):
         if self.spheroidicity == 0.:
             self.eval = self.Yslm
             self.eigenvalue = Yslm_eigenvalue(self.s, self.l)
-            self.coeffs = np.zeros(self.l - self.lmin)
+            self.coeffs = np.zeros(self.l - self.lmin + 1)
             self.coeffs[-1] = 1.
         else:
             self.eval = self.Sslm

{pybhpt-0.9.5.dist-info → pybhpt-0.9.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pybhpt
-Version: 0.9.5
+Version: 0.9.8
 Summary: Black Hole Perturbation Theory and Self-Force Algorithms in Python
 Author-Email: Zach Nasipak <znasipak@gmail.com>
 License: GPL

{pybhpt-0.9.5.dist-info → pybhpt-0.9.8.dist-info}/RECORD

@@ -1,16 +1,16 @@
-cybhpt_full.cpython-310-darwin.so,sha256=493ENi9_M0E6c5zz7-2Nn-alFS24D06wrtXFcZU_0XY,1784976
-pybhpt/geo.py,sha256=H-rTM6zy-9EvLaMaeyMfwbdwgAS53d_Z6dmBCp3eya4,15473
+cybhpt_full.cpython-310-darwin.so,sha256=hLPhDQ2nSQlVL0WnxVZ_ZdqkkGLx5JD04QVrQf5Tvcg,1818688
+pybhpt-0.9.8.dist-info/RECORD,,
+pybhpt-0.9.8.dist-info/WHEEL,sha256=neR5pitvb4mxbnmS2Hj1G0KqZayFx4hgZm2essb8zGY,141
+pybhpt-0.9.8.dist-info/METADATA,sha256=NIzqGzLYfDWkON7jWb24fRxC7DbZzIrcdMYBF2iOWlA,6367
+pybhpt-0.9.8.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pybhpt/geo.py,sha256=VqbarbR1CC7ARD82FI8vmoNnklCZUoraRVJOii3vBlE,23803
 pybhpt/redshift.py,sha256=ozDDbkv34FQ_Op3FotmHJ-J5Vfy2oF4_Ri1HRujnzAY,791
 pybhpt/hertz.py,sha256=9atjscxbR7eszsY6BuJvcy0PqlgMFwYI23kUIn4mUwc,13565
 pybhpt/radial.py,sha256=t9Lx6Cyuprxqk2iowbvHMvsoL-6UO-YnZmO1G1wdyMU,15597
 pybhpt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pybhpt/swsh.py,sha256=w3XQoadHY-aiZZYQo5sUjweimM3HJ44AWuFcA2uuffs,14641
+pybhpt/swsh.py,sha256=sQR0S-tHXYaAIuuW9kHK99U39htXt5j8oGWPZzIqwt0,14826
 pybhpt/metric.py,sha256=y8ip24rFto7m7rZb0-fuYMrYUmbCzInaWQVYJRI3m3o,11971
 pybhpt/teuk.py,sha256=8rZ-2fyzAfTbt9T4-lkUL8wUmPpORv0VkJCfkpEBssc,13616
 pybhpt/flux.py,sha256=2ahStjbsswsWMNPfDiOlQIrzulJZrhaqoDENC0bCLE4,4299
 pybhpt/.dylibs/libgsl.28.dylib,sha256=57NeYzXRx_Fvr8ChQggQOBqCgqn5xXxWyFIiGMm3-_M,2241424
 pybhpt/.dylibs/libgslcblas.0.dylib,sha256=FuXGrQKlT1nkAMaRH6O1B_CnRFFX4pqVcK36KX_-56A,239760
-pybhpt-0.9.5.dist-info/RECORD,,
-pybhpt-0.9.5.dist-info/WHEEL,sha256=neR5pitvb4mxbnmS2Hj1G0KqZayFx4hgZm2essb8zGY,141
-pybhpt-0.9.5.dist-info/METADATA,sha256=OXj1Mg_Dtv15cYn28Uo0FffhH9LkR9Xp8NcIUWBxMK0,6367
-pybhpt-0.9.5.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149