diff options
Diffstat (limited to 'lib/libF77/c_sqrt.c')
| -rw-r--r-- | lib/libF77/c_sqrt.c | 37 |
1 files changed, 19 insertions, 18 deletions
diff --git a/lib/libF77/c_sqrt.c b/lib/libF77/c_sqrt.c index 3b7342f..8481ee4 100644 --- a/lib/libF77/c_sqrt.c +++ b/lib/libF77/c_sqrt.c @@ -12,23 +12,24 @@ extern double f__cabs(double, double); void c_sqrt(complex *r, complex *z) #endif { -double mag, t; + double mag, t; + double 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 = t = sqrt(0.5 * (mag + z->r) ); - t = z->i / t; - r->i = 0.5 * t; + if( (mag = f__cabs(zr, zi)) == 0.) + r->r = r->i = 0.; + else if(zr > 0) + { + r->r = t = sqrt(0.5 * (mag + zr) ); + t = zi / t; + r->i = 0.5 * t; + } + else + { + t = sqrt(0.5 * (mag - zr) ); + if(zi < 0) + t = -t; + r->i = t; + t = zi / t; + r->r = 0.5 * t; + } } -else - { - t = sqrt(0.5 * (mag - z->r) ); - if(z->i < 0) - t = -t; - r->i = t; - t = z->i / t; - r->r = 0.5 * t; - } -} |
