diff options
Diffstat (limited to 'lib/libF77/z_sqrt.c')
| -rw-r--r-- | lib/libF77/z_sqrt.c | 30 |
1 files changed, 15 insertions, 15 deletions
diff --git a/lib/libF77/z_sqrt.c b/lib/libF77/z_sqrt.c index eed38d0..c04e8f0 100644 --- a/lib/libF77/z_sqrt.c +++ b/lib/libF77/z_sqrt.c @@ -10,20 +10,20 @@ extern double f__cabs(double, double); void z_sqrt(doublecomplex *r, doublecomplex *z) #endif { -double mag; + double mag, zi = z->i, zr = z->r; -if( (mag = f__cabs(z->r, z->i)) == 0.) - r->r = r->i = 0.; -else if(z->r > 0) - { - r->r = sqrt(0.5 * (mag + z->r) ); - r->i = z->i / r->r / 2; + if( (mag = f__cabs(zr, zi)) == 0.) + r->r = r->i = 0.; + else if(zr > 0) + { + r->r = sqrt(0.5 * (mag + zr) ); + r->i = zi / r->r / 2; + } + else + { + r->i = sqrt(0.5 * (mag - zr) ); + if(zi < 0) + r->i = - r->i; + r->r = zi / r->i / 2; + } } -else - { - r->i = sqrt(0.5 * (mag - z->r) ); - if(z->i < 0) - r->i = - r->i; - r->r = z->i / r->i / 2; - } -} |
