1 | /* Complex square root of float value. |
2 | Copyright (C) 1997-2016 Free Software Foundation, Inc. |
3 | This file is part of the GNU C Library. |
4 | Based on an algorithm by Stephen L. Moshier <moshier@world.std.com>. |
5 | Contributed by Ulrich Drepper <drepper@cygnus.com>, 1997. |
6 | |
7 | The GNU C Library is free software; you can redistribute it and/or |
8 | modify it under the terms of the GNU Lesser General Public |
9 | License as published by the Free Software Foundation; either |
10 | version 2.1 of the License, or (at your option) any later version. |
11 | |
12 | The GNU C Library is distributed in the hope that it will be useful, |
13 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 | Lesser General Public License for more details. |
16 | |
17 | You should have received a copy of the GNU Lesser General Public |
18 | License along with the GNU C Library; if not, see |
19 | <http://www.gnu.org/licenses/>. */ |
20 | |
21 | #include <complex.h> |
22 | #include <math.h> |
23 | #include <math_private.h> |
24 | #include <float.h> |
25 | |
26 | __complex__ float |
27 | __csqrtf (__complex__ float x) |
28 | { |
29 | __complex__ float res; |
30 | int rcls = fpclassify (__real__ x); |
31 | int icls = fpclassify (__imag__ x); |
32 | |
33 | if (__glibc_unlikely (rcls <= FP_INFINITE || icls <= FP_INFINITE)) |
34 | { |
35 | if (icls == FP_INFINITE) |
36 | { |
37 | __real__ res = HUGE_VALF; |
38 | __imag__ res = __imag__ x; |
39 | } |
40 | else if (rcls == FP_INFINITE) |
41 | { |
42 | if (__real__ x < 0.0) |
43 | { |
44 | __real__ res = icls == FP_NAN ? __nanf ("" ) : 0; |
45 | __imag__ res = __copysignf (HUGE_VALF, __imag__ x); |
46 | } |
47 | else |
48 | { |
49 | __real__ res = __real__ x; |
50 | __imag__ res = (icls == FP_NAN |
51 | ? __nanf ("" ) : __copysignf (0.0, __imag__ x)); |
52 | } |
53 | } |
54 | else |
55 | { |
56 | __real__ res = __nanf ("" ); |
57 | __imag__ res = __nanf ("" ); |
58 | } |
59 | } |
60 | else |
61 | { |
62 | if (__glibc_unlikely (icls == FP_ZERO)) |
63 | { |
64 | if (__real__ x < 0.0) |
65 | { |
66 | __real__ res = 0.0; |
67 | __imag__ res = __copysignf (__ieee754_sqrtf (-__real__ x), |
68 | __imag__ x); |
69 | } |
70 | else |
71 | { |
72 | __real__ res = fabsf (__ieee754_sqrtf (__real__ x)); |
73 | __imag__ res = __copysignf (0.0, __imag__ x); |
74 | } |
75 | } |
76 | else if (__glibc_unlikely (rcls == FP_ZERO)) |
77 | { |
78 | float r; |
79 | if (fabsf (__imag__ x) >= 2.0f * FLT_MIN) |
80 | r = __ieee754_sqrtf (0.5f * fabsf (__imag__ x)); |
81 | else |
82 | r = 0.5f * __ieee754_sqrtf (2.0f * fabsf (__imag__ x)); |
83 | |
84 | __real__ res = r; |
85 | __imag__ res = __copysignf (r, __imag__ x); |
86 | } |
87 | else |
88 | { |
89 | float d, r, s; |
90 | int scale = 0; |
91 | |
92 | if (fabsf (__real__ x) > FLT_MAX / 4.0f) |
93 | { |
94 | scale = 1; |
95 | __real__ x = __scalbnf (__real__ x, -2 * scale); |
96 | __imag__ x = __scalbnf (__imag__ x, -2 * scale); |
97 | } |
98 | else if (fabsf (__imag__ x) > FLT_MAX / 4.0f) |
99 | { |
100 | scale = 1; |
101 | if (fabsf (__real__ x) >= 4.0f * FLT_MIN) |
102 | __real__ x = __scalbnf (__real__ x, -2 * scale); |
103 | else |
104 | __real__ x = 0.0f; |
105 | __imag__ x = __scalbnf (__imag__ x, -2 * scale); |
106 | } |
107 | else if (fabsf (__real__ x) < 2.0f * FLT_MIN |
108 | && fabsf (__imag__ x) < 2.0f * FLT_MIN) |
109 | { |
110 | scale = -((FLT_MANT_DIG + 1) / 2); |
111 | __real__ x = __scalbnf (__real__ x, -2 * scale); |
112 | __imag__ x = __scalbnf (__imag__ x, -2 * scale); |
113 | } |
114 | |
115 | d = __ieee754_hypotf (__real__ x, __imag__ x); |
116 | /* Use the identity 2 Re res Im res = Im x |
117 | to avoid cancellation error in d +/- Re x. */ |
118 | if (__real__ x > 0) |
119 | { |
120 | r = __ieee754_sqrtf (0.5f * (d + __real__ x)); |
121 | if (scale == 1 && fabsf (__imag__ x) < 1.0f) |
122 | { |
123 | /* Avoid possible intermediate underflow. */ |
124 | s = __imag__ x / r; |
125 | r = __scalbnf (r, scale); |
126 | scale = 0; |
127 | } |
128 | else |
129 | s = 0.5f * (__imag__ x / r); |
130 | } |
131 | else |
132 | { |
133 | s = __ieee754_sqrtf (0.5f * (d - __real__ x)); |
134 | if (scale == 1 && fabsf (__imag__ x) < 1.0f) |
135 | { |
136 | /* Avoid possible intermediate underflow. */ |
137 | r = fabsf (__imag__ x / s); |
138 | s = __scalbnf (s, scale); |
139 | scale = 0; |
140 | } |
141 | else |
142 | r = fabsf (0.5f * (__imag__ x / s)); |
143 | } |
144 | |
145 | if (scale) |
146 | { |
147 | r = __scalbnf (r, scale); |
148 | s = __scalbnf (s, scale); |
149 | } |
150 | |
151 | math_check_force_underflow (r); |
152 | math_check_force_underflow (s); |
153 | |
154 | __real__ res = r; |
155 | __imag__ res = __copysignf (s, __imag__ x); |
156 | } |
157 | } |
158 | |
159 | return res; |
160 | } |
161 | #ifndef __csqrtf |
162 | weak_alias (__csqrtf, csqrtf) |
163 | #endif |
164 | |