kai/pounder_test
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

atan2: store sign bits and greater of |x| and |y|

Browse Source
This commit is contained in:
Matt Huszagh 2020-12-17 09:47:39 -08:00
parent cb38c3e3bd
commit 1f28949bc5
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
dsp/src/trig.rs
View File

@ -1,4 +1,4 @@
use super::{abs, shift_round, Complex}; use super::{shift_round, Complex};
use core::f64::consts::PI; use core::f64::consts::PI;
include!(concat!(env!("OUT_DIR"), "/cossin_table.rs")); include!(concat!(env!("OUT_DIR"), "/cossin_table.rs"));
@ -22,11 +22,18 @@ include!(concat!(env!("OUT_DIR"), "/cossin_table.rs"));
/// corresponds to an angle of -pi and i32::MAX corresponds to an /// corresponds to an angle of -pi and i32::MAX corresponds to an
/// angle of +pi. /// angle of +pi.
pub fn atan2(y: i32, x: i32) -> i32 { pub fn atan2(y: i32, x: i32) -> i32 {
let y = y >> 16; let mut y = y >> 16;
let x = x >> 16; let mut x = x >> 16;
let ux = abs(x); let sign = ((y >> 14) & 2) | ((x >> 15) & 1);
let uy = abs(y); if sign & 1 == 1 {
x *= -1;
}
if sign & 2 == 2 {
y *= -1;
}
let y_greater = y > x;
// Uses the general procedure described in the following // Uses the general procedure described in the following
// Mathematics stack exchange answer: // Mathematics stack exchange answer:
@ -41,7 +48,7 @@ pub fn atan2(y: i32, x: i32) -> i32 {
// //
// which is taken from Rajan 2006: Efficient Approximations for // which is taken from Rajan 2006: Efficient Approximations for
// the Arctangent Function. // the Arctangent Function.
let (min, max) = if ux < uy { (ux, uy) } else { (uy, ux) }; let (min, max) = if y_greater { (x, y) } else { (y, x) };
if max == 0 { if max == 0 {
return 0; return 0;
@ -64,15 +71,15 @@ pub fn atan2(y: i32, x: i32) -> i32 {
ratio * K1 - K2 * shift_round(ratio * ratio, ATAN_ARGUMENT_BITS) ratio * K1 - K2 * shift_round(ratio * ratio, ATAN_ARGUMENT_BITS)
}; };
if uy > ux { if y_greater {
angle = (i32::MAX >> 1) - angle; angle = (i32::MAX >> 1) - angle;
} }
if x < 0 { if sign & 1 == 1 {
angle = i32::MAX - angle; angle = i32::MAX - angle;
} }
if y < 0 { if sign & 2 == 2 {
angle *= -1; angle *= -1;
} }