/* origin: FreeBSD /usr/src/lib/msun/src/e_atan2.c */ /* * ==================================================== * Copyright (C) 1993 by Sun Microsystems, Inc. All rights reserved. * * Developed at SunSoft, a Sun Microsystems, Inc. business. * Permission to use, copy, modify, and distribute this * software is freely granted, provided that this notice * is preserved. * ==================================================== * */ /* atan2(y,x) * Method : * 1. Reduce y to positive by atan2(y,x)=-atan2(-y,x). * 2. Reduce x to positive by (if x and y are unexceptional): * ARG (x+iy) = arctan(y/x) ... if x > 0, * ARG (x+iy) = pi - arctan[y/(-x)] ... if x < 0, * * Special cases: * * ATAN2((anything), NaN ) is NaN; * ATAN2(NAN , (anything) ) is NaN; * ATAN2(+-0, +(anything but NaN)) is +-0 ; * ATAN2(+-0, -(anything but NaN)) is +-pi ; * ATAN2(+-(anything but 0 and NaN), 0) is +-pi/2; * ATAN2(+-(anything but INF and NaN), +INF) is +-0 ; * ATAN2(+-(anything but INF and NaN), -INF) is +-pi; * ATAN2(+-INF,+INF ) is +-pi/4 ; * ATAN2(+-INF,-INF ) is +-3pi/4; * ATAN2(+-INF, (anything but,0,NaN, and INF)) is +-pi/2; * * Constants: * The hexadecimal values are the intended ones for the following * constants. The decimal values may be used, provided that the * compiler will convert from decimal to binary accurately enough * to produce the hexadecimal values shown. */ use super::atan; use super::fabs; const PI: f64 = 3.1415926535897931160E+00; /* 0x400921FB, 0x54442D18 */ const PI_LO: f64 = 1.2246467991473531772E-16; /* 0x3CA1A626, 0x33145C07 */ /// Arctangent of y/x (f64) /// /// Computes the inverse tangent (arc tangent) of `y/x`. /// Produces the correct result even for angles near pi/2 or -pi/2 (that is, when `x` is near 0). /// Returns a value in radians, in the range of -pi to pi. #[cfg_attr(all(test, assert_no_panic), no_panic::no_panic)] pub fn atan2(y: f64, x: f64) -> f64 { if x.is_nan() || y.is_nan() { return x + y; } let mut ix = (x.to_bits() >> 32) as u32; let lx = x.to_bits() as u32; let mut iy = (y.to_bits() >> 32) as u32; let ly = y.to_bits() as u32; if ((ix.wrapping_sub(0x3ff00000)) | lx) == 0 { /* x = 1.0 */ return atan(y); } let m = ((iy >> 31) & 1) | ((ix >> 30) & 2); /* 2*sign(x)+sign(y) */ ix &= 0x7fffffff; iy &= 0x7fffffff; /* when y = 0 */ if (iy | ly) == 0 { return match m { 0 | 1 => y, /* atan(+-0,+anything)=+-0 */ 2 => PI, /* atan(+0,-anything) = PI */ _ => -PI, /* atan(-0,-anything) =-PI */ }; } /* when x = 0 */ if (ix | lx) == 0 { return if m & 1 != 0 { -PI / 2.0 } else { PI / 2.0 }; } /* when x is INF */ if ix == 0x7ff00000 { if iy == 0x7ff00000 { return match m { 0 => PI / 4.0, /* atan(+INF,+INF) */ 1 => -PI / 4.0, /* atan(-INF,+INF) */ 2 => 3.0 * PI / 4.0, /* atan(+INF,-INF) */ _ => -3.0 * PI / 4.0, /* atan(-INF,-INF) */ }; } else { return match m { 0 => 0.0, /* atan(+...,+INF) */ 1 => -0.0, /* atan(-...,+INF) */ 2 => PI, /* atan(+...,-INF) */ _ => -PI, /* atan(-...,-INF) */ }; } } /* |y/x| > 0x1p64 */ if ix.wrapping_add(64 << 20) < iy || iy == 0x7ff00000 { return if m & 1 != 0 { -PI / 2.0 } else { PI / 2.0 }; } /* z = atan(|y/x|) without spurious underflow */ let z = if (m & 2 != 0) && iy.wrapping_add(64 << 20) < ix { /* |y/x| < 0x1p-64, x<0 */ 0.0 } else { atan(fabs(y / x)) }; match m { 0 => z, /* atan(+,+) */ 1 => -z, /* atan(-,+) */ 2 => PI - (z - PI_LO), /* atan(+,-) */ _ => (z - PI_LO) - PI, /* atan(-,-) */ } } #[test] fn sanity_check() { assert_eq!(atan2(0.0, 1.0), 0.0); assert_eq!(atan2(0.0, -1.0), PI); assert_eq!(atan2(-0.0, -1.0), -PI); assert_eq!(atan2(3.0, 2.0), atan(3.0 / 2.0)); assert_eq!(atan2(2.0, -1.0), atan(2.0 / -1.0) + PI); assert_eq!(atan2(-2.0, -1.0), atan(-2.0 / -1.0) - PI); }