B d-@sFdZddlZddZddZddZd d Zd d Zdddd d gZdS) z Coordinate transformations. NcCst|d|d|dS)ae Calculates the magnitude in spherical coordinates from Cartesian coordinates. Parameters ---------- x : {numpy.array, float} X-coordinate. y : {numpy.array, float} Y-coordinate. z : {numpy.array, float} Z-coordinate. Returns ------- rho : {numpy.array, float} The radial amplitude. )numpysqrt)xyzr^/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/coordinates.pycartesian_to_spherical_rhosr cCs0t|trt|n|}t||}|dtjS)a[ Calculates the azimuthal angle in spherical coordinates from Cartesian coordinates. The azimuthal angle is in [0,2*pi]. Parameters ---------- x : {numpy.array, float} X-coordinate. y : {numpy.array, float} Y-coordinate. Returns ------- phi : {numpy.array, float} The azimuthal angle. r) isinstanceintfloatrZarctan2pi)rrphirrr cartesian_to_spherical_azimuthal+s rcCsRt|||}t|r,|r(t||SdSttj||t||dkdSdS)a Calculates the polar angle in spherical coordinates from Cartesian coordinates. The polar angle is in [0,pi]. Parameters ---------- x : {numpy.array, float} X-coordinate. y : {numpy.array, float} Y-coordinate. z : {numpy.array, float} Z-coordinate. Returns ------- theta : {numpy.array, float} The polar angle. gr)outwhereN)r rZisscalarZarccosdivideZ ones_like)rrrrhorrr cartesian_to_spherical_polar?s   rcCs,t|||}t||}t|||}|||fS)a  Maps cartesian coordinates (x,y,z) to spherical coordinates (rho,phi,theta) where phi is in [0,2*pi] and theta is in [0,pi]. Parameters ---------- x : {numpy.array, float} X-coordinate. y : {numpy.array, float} Y-coordinate. z : {numpy.array, float} Z-coordinate. Returns ------- rho : {numpy.array, float} The radial amplitude. phi : {numpy.array, float} The azimuthal angle. theta : {numpy.array, float} The polar angle. )r rr)rrrrrthetarrr cartesian_to_sphericalYs   rcCsH|t|t|}|t|t|}|t|}|||fS)a  Maps spherical coordinates (rho,phi,theta) to cartesian coordinates (x,y,z) where phi is in [0,2*pi] and theta is in [0,pi]. Parameters ---------- rho : {numpy.array, float} The radial amplitude. phi : {numpy.array, float} The azimuthal angle. theta : {numpy.array, float} The polar angle. Returns ------- x : {numpy.array, float} X-coordinate. y : {numpy.array, float} Y-coordinate. z : {numpy.array, float} Z-coordinate. )rcossin)rrrrrrrrr spherical_to_cartesianusr)__doc__rr rrrr__all__rrrr s