Source code for pybhpt.redshift

from cybhpt_full import RedshiftCoefficients as RedshiftCoefficientsCython
from cybhpt_full import SphericalHarmonicCoupling as SphericalHarmonicCouplingCython


[docs]
class RedshiftCoefficients:
    def __init__(self, gauge, geo):
        self.base = RedshiftCoefficientsCython(gauge, geo.base)


[docs]
    def __call__(self, Ni, ai, bi, ci, di, jr = 0, jz = 0):
        return self.base(Ni, ai, bi, ci, di, jr, jz)


    

[docs]
class SphericalHarmonicCoupling:
    def __init__(self, lmax, m):
        self.base = SphericalHarmonicCouplingCython(lmax, m)
        self.azimuthalmode = self.base.azimuthalmode


[docs]
    def zcouplingcoefficient(self, n, i, l):
        return self.base.zcouplingcoefficient(n, i, l)



[docs]
    def dzcouplingcoefficient(self, n, i, l):
        return self.base.dzcouplingcoefficient(n, i, l)