cc68a136 |
1 | /* @(#)e_sqrt.c 5.1 93/09/24 */ |
2 | /* |
3 | * ==================================================== |
4 | * Copyright (C) 1993 by Sun Microsystems, Inc. All rights reserved. |
5 | * |
6 | * Developed at SunPro, a Sun Microsystems, Inc. business. |
7 | * Permission to use, copy, modify, and distribute this |
8 | * software is freely granted, provided that this notice |
9 | * is preserved. |
10 | * ==================================================== |
11 | */ |
12 | |
13 | #if defined(LIBM_SCCS) && !defined(lint) |
14 | static char rcsid[] = "$NetBSD: e_sqrt.c,v 1.8 1995/05/10 20:46:17 jtc Exp $"; |
15 | #endif |
16 | |
17 | /* __ieee754_sqrt(x) |
18 | * Return correctly rounded sqrt. |
19 | * ------------------------------------------ |
20 | * | Use the hardware sqrt if you have one | |
21 | * ------------------------------------------ |
22 | * Method: |
23 | * Bit by bit method using integer arithmetic. (Slow, but portable) |
24 | * 1. Normalization |
25 | * Scale x to y in [1,4) with even powers of 2: |
26 | * find an integer k such that 1 <= (y=x*2^(2k)) < 4, then |
27 | * sqrt(x) = 2^k * sqrt(y) |
28 | * 2. Bit by bit computation |
29 | * Let q = sqrt(y) truncated to i bit after binary point (q = 1), |
30 | * i 0 |
31 | * i+1 2 |
32 | * s = 2*q , and y = 2 * ( y - q ). (1) |
33 | * i i i i |
34 | * |
35 | * To compute q from q , one checks whether |
36 | * i+1 i |
37 | * |
38 | * -(i+1) 2 |
39 | * (q + 2 ) <= y. (2) |
40 | * i |
41 | * -(i+1) |
42 | * If (2) is false, then q = q ; otherwise q = q + 2 . |
43 | * i+1 i i+1 i |
44 | * |
45 | * With some algebric manipulation, it is not difficult to see |
46 | * that (2) is equivalent to |
47 | * -(i+1) |
48 | * s + 2 <= y (3) |
49 | * i i |
50 | * |
51 | * The advantage of (3) is that s and y can be computed by |
52 | * i i |
53 | * the following recurrence formula: |
54 | * if (3) is false |
55 | * |
56 | * s = s , y = y ; (4) |
57 | * i+1 i i+1 i |
58 | * |
59 | * otherwise, |
60 | * -i -(i+1) |
61 | * s = s + 2 , y = y - s - 2 (5) |
62 | * i+1 i i+1 i i |
63 | * |
64 | * One may easily use induction to prove (4) and (5). |
65 | * Note. Since the left hand side of (3) contain only i+2 bits, |
66 | * it does not necessary to do a full (53-bit) comparison |
67 | * in (3). |
68 | * 3. Final rounding |
69 | * After generating the 53 bits result, we compute one more bit. |
70 | * Together with the remainder, we can decide whether the |
71 | * result is exact, bigger than 1/2ulp, or less than 1/2ulp |
72 | * (it will never equal to 1/2ulp). |
73 | * The rounding mode can be detected by checking whether |
74 | * huge + tiny is equal to huge, and whether huge - tiny is |
75 | * equal to huge for some floating point number "huge" and "tiny". |
76 | * |
77 | * Special cases: |
78 | * sqrt(+-0) = +-0 ... exact |
79 | * sqrt(inf) = inf |
80 | * sqrt(-ve) = NaN ... with invalid signal |
81 | * sqrt(NaN) = NaN ... with invalid signal for signaling NaN |
82 | * |
83 | * Other methods : see the appended file at the end of the program below. |
84 | *--------------- |
85 | */ |
86 | |
87 | #include "math.h" |
88 | #include "math_private.h" |
89 | |
90 | #ifdef __STDC__ |
91 | static const double one = 1.0, tiny=1.0e-300; |
92 | #else |
93 | static double one = 1.0, tiny=1.0e-300; |
94 | #endif |
95 | |
96 | #ifdef __STDC__ |
97 | double __ieee754_sqrt(double x) |
98 | #else |
99 | double __ieee754_sqrt(x) |
100 | double x; |
101 | #endif |
102 | { |
103 | double z; |
104 | int32_t sign = (int)0x80000000; |
105 | int32_t ix0,s0,q,m,t,i; |
106 | u_int32_t r,t1,s1,ix1,q1; |
107 | |
108 | EXTRACT_WORDS(ix0,ix1,x); |
109 | |
110 | /* take care of Inf and NaN */ |
111 | if((ix0&0x7ff00000)==0x7ff00000) { |
112 | return x*x+x; /* sqrt(NaN)=NaN, sqrt(+inf)=+inf |
113 | sqrt(-inf)=sNaN */ |
114 | } |
115 | /* take care of zero */ |
116 | if(ix0<=0) { |
117 | if(((ix0&(~sign))|ix1)==0) return x;/* sqrt(+-0) = +-0 */ |
118 | else if(ix0<0) |
119 | return (x-x)/(x-x); /* sqrt(-ve) = sNaN */ |
120 | } |
121 | /* normalize x */ |
122 | m = (ix0>>20); |
123 | if(m==0) { /* subnormal x */ |
124 | while(ix0==0) { |
125 | m -= 21; |
126 | ix0 |= (ix1>>11); ix1 <<= 21; |
127 | } |
128 | for(i=0;(ix0&0x00100000)==0;i++) ix0<<=1; |
129 | m -= i-1; |
130 | ix0 |= (ix1>>(32-i)); |
131 | ix1 <<= i; |
132 | } |
133 | m -= 1023; /* unbias exponent */ |
134 | ix0 = (ix0&0x000fffff)|0x00100000; |
135 | if(m&1){ /* odd m, double x to make it even */ |
136 | ix0 += ix0 + ((ix1&sign)>>31); |
137 | ix1 += ix1; |
138 | } |
139 | m >>= 1; /* m = [m/2] */ |
140 | |
141 | /* generate sqrt(x) bit by bit */ |
142 | ix0 += ix0 + ((ix1&sign)>>31); |
143 | ix1 += ix1; |
144 | q = q1 = s0 = s1 = 0; /* [q,q1] = sqrt(x) */ |
145 | r = 0x00200000; /* r = moving bit from right to left */ |
146 | |
147 | while(r!=0) { |
148 | t = s0+r; |
149 | if(t<=ix0) { |
150 | s0 = t+r; |
151 | ix0 -= t; |
152 | q += r; |
153 | } |
154 | ix0 += ix0 + ((ix1&sign)>>31); |
155 | ix1 += ix1; |
156 | r>>=1; |
157 | } |
158 | |
159 | r = sign; |
160 | while(r!=0) { |
161 | t1 = s1+r; |
162 | t = s0; |
163 | if((t<ix0)||((t==ix0)&&(t1<=ix1))) { |
164 | s1 = t1+r; |
165 | if(((t1&sign)==sign)&&(s1&sign)==0) s0 += 1; |
166 | ix0 -= t; |
167 | if (ix1 < t1) ix0 -= 1; |
168 | ix1 -= t1; |
169 | q1 += r; |
170 | } |
171 | ix0 += ix0 + ((ix1&sign)>>31); |
172 | ix1 += ix1; |
173 | r>>=1; |
174 | } |
175 | |
176 | /* use floating add to find out rounding direction */ |
177 | if((ix0|ix1)!=0) { |
178 | z = one-tiny; /* trigger inexact flag */ |
179 | if (z>=one) { |
180 | z = one+tiny; |
181 | if (q1==(u_int32_t)0xffffffff) { q1=0; q += 1;} |
182 | else if (z>one) { |
183 | if (q1==(u_int32_t)0xfffffffe) q+=1; |
184 | q1+=2; |
185 | } else |
186 | q1 += (q1&1); |
187 | } |
188 | } |
189 | ix0 = (q>>1)+0x3fe00000; |
190 | ix1 = q1>>1; |
191 | if ((q&1)==1) ix1 |= sign; |
192 | ix0 += (m <<20); |
193 | INSERT_WORDS(z,ix0,ix1); |
194 | return z; |
195 | } |
196 | |
197 | /* |
198 | Other methods (use floating-point arithmetic) |
199 | ------------- |
200 | (This is a copy of a drafted paper by Prof W. Kahan |
201 | and K.C. Ng, written in May, 1986) |
202 | |
203 | Two algorithms are given here to implement sqrt(x) |
204 | (IEEE double precision arithmetic) in software. |
205 | Both supply sqrt(x) correctly rounded. The first algorithm (in |
206 | Section A) uses newton iterations and involves four divisions. |
207 | The second one uses reciproot iterations to avoid division, but |
208 | requires more multiplications. Both algorithms need the ability |
209 | to chop results of arithmetic operations instead of round them, |
210 | and the INEXACT flag to indicate when an arithmetic operation |
211 | is executed exactly with no roundoff error, all part of the |
212 | standard (IEEE 754-1985). The ability to perform shift, add, |
213 | subtract and logical AND operations upon 32-bit words is needed |
214 | too, though not part of the standard. |
215 | |
216 | A. sqrt(x) by Newton Iteration |
217 | |
218 | (1) Initial approximation |
219 | |
220 | Let x0 and x1 be the leading and the trailing 32-bit words of |
221 | a floating point number x (in IEEE double format) respectively |
222 | |
223 | 1 11 52 ...widths |
224 | ------------------------------------------------------ |
225 | x: |s| e | f | |
226 | ------------------------------------------------------ |
227 | msb lsb msb lsb ...order |
228 | |
229 | |
230 | ------------------------ ------------------------ |
231 | x0: |s| e | f1 | x1: | f2 | |
232 | ------------------------ ------------------------ |
233 | |
234 | By performing shifts and subtracts on x0 and x1 (both regarded |
235 | as integers), we obtain an 8-bit approximation of sqrt(x) as |
236 | follows. |
237 | |
238 | k := (x0>>1) + 0x1ff80000; |
239 | y0 := k - T1[31&(k>>15)]. ... y ~ sqrt(x) to 8 bits |
240 | Here k is a 32-bit integer and T1[] is an integer array containing |
241 | correction terms. Now magically the floating value of y (y's |
242 | leading 32-bit word is y0, the value of its trailing word is 0) |
243 | approximates sqrt(x) to almost 8-bit. |
244 | |
245 | Value of T1: |
246 | static int T1[32]= { |
247 | 0, 1024, 3062, 5746, 9193, 13348, 18162, 23592, |
248 | 29598, 36145, 43202, 50740, 58733, 67158, 75992, 85215, |
249 | 83599, 71378, 60428, 50647, 41945, 34246, 27478, 21581, |
250 | 16499, 12183, 8588, 5674, 3403, 1742, 661, 130,}; |
251 | |
252 | (2) Iterative refinement |
253 | |
254 | Apply Heron's rule three times to y, we have y approximates |
255 | sqrt(x) to within 1 ulp (Unit in the Last Place): |
256 | |
257 | y := (y+x/y)/2 ... almost 17 sig. bits |
258 | y := (y+x/y)/2 ... almost 35 sig. bits |
259 | y := y-(y-x/y)/2 ... within 1 ulp |
260 | |
261 | |
262 | Remark 1. |
263 | Another way to improve y to within 1 ulp is: |
264 | |
265 | y := (y+x/y) ... almost 17 sig. bits to 2*sqrt(x) |
266 | y := y - 0x00100006 ... almost 18 sig. bits to sqrt(x) |
267 | |
268 | 2 |
269 | (x-y )*y |
270 | y := y + 2* ---------- ...within 1 ulp |
271 | 2 |
272 | 3y + x |
273 | |
274 | |
275 | This formula has one division fewer than the one above; however, |
276 | it requires more multiplications and additions. Also x must be |
277 | scaled in advance to avoid spurious overflow in evaluating the |
278 | expression 3y*y+x. Hence it is not recommended uless division |
279 | is slow. If division is very slow, then one should use the |
280 | reciproot algorithm given in section B. |
281 | |
282 | (3) Final adjustment |
283 | |
284 | By twiddling y's last bit it is possible to force y to be |
285 | correctly rounded according to the prevailing rounding mode |
286 | as follows. Let r and i be copies of the rounding mode and |
287 | inexact flag before entering the square root program. Also we |
288 | use the expression y+-ulp for the next representable floating |
289 | numbers (up and down) of y. Note that y+-ulp = either fixed |
290 | point y+-1, or multiply y by nextafter(1,+-inf) in chopped |
291 | mode. |
292 | |
293 | I := FALSE; ... reset INEXACT flag I |
294 | R := RZ; ... set rounding mode to round-toward-zero |
295 | z := x/y; ... chopped quotient, possibly inexact |
296 | If(not I) then { ... if the quotient is exact |
297 | if(z=y) { |
298 | I := i; ... restore inexact flag |
299 | R := r; ... restore rounded mode |
300 | return sqrt(x):=y. |
301 | } else { |
302 | z := z - ulp; ... special rounding |
303 | } |
304 | } |
305 | i := TRUE; ... sqrt(x) is inexact |
306 | If (r=RN) then z=z+ulp ... rounded-to-nearest |
307 | If (r=RP) then { ... round-toward-+inf |
308 | y = y+ulp; z=z+ulp; |
309 | } |
310 | y := y+z; ... chopped sum |
311 | y0:=y0-0x00100000; ... y := y/2 is correctly rounded. |
312 | I := i; ... restore inexact flag |
313 | R := r; ... restore rounded mode |
314 | return sqrt(x):=y. |
315 | |
316 | (4) Special cases |
317 | |
318 | Square root of +inf, +-0, or NaN is itself; |
319 | Square root of a negative number is NaN with invalid signal. |
320 | |
321 | |
322 | B. sqrt(x) by Reciproot Iteration |
323 | |
324 | (1) Initial approximation |
325 | |
326 | Let x0 and x1 be the leading and the trailing 32-bit words of |
327 | a floating point number x (in IEEE double format) respectively |
328 | (see section A). By performing shifs and subtracts on x0 and y0, |
329 | we obtain a 7.8-bit approximation of 1/sqrt(x) as follows. |
330 | |
331 | k := 0x5fe80000 - (x0>>1); |
332 | y0:= k - T2[63&(k>>14)]. ... y ~ 1/sqrt(x) to 7.8 bits |
333 | |
334 | Here k is a 32-bit integer and T2[] is an integer array |
335 | containing correction terms. Now magically the floating |
336 | value of y (y's leading 32-bit word is y0, the value of |
337 | its trailing word y1 is set to zero) approximates 1/sqrt(x) |
338 | to almost 7.8-bit. |
339 | |
340 | Value of T2: |
341 | static int T2[64]= { |
342 | 0x1500, 0x2ef8, 0x4d67, 0x6b02, 0x87be, 0xa395, 0xbe7a, 0xd866, |
343 | 0xf14a, 0x1091b,0x11fcd,0x13552,0x14999,0x15c98,0x16e34,0x17e5f, |
344 | 0x18d03,0x19a01,0x1a545,0x1ae8a,0x1b5c4,0x1bb01,0x1bfde,0x1c28d, |
345 | 0x1c2de,0x1c0db,0x1ba73,0x1b11c,0x1a4b5,0x1953d,0x18266,0x16be0, |
346 | 0x1683e,0x179d8,0x18a4d,0x19992,0x1a789,0x1b445,0x1bf61,0x1c989, |
347 | 0x1d16d,0x1d77b,0x1dddf,0x1e2ad,0x1e5bf,0x1e6e8,0x1e654,0x1e3cd, |
348 | 0x1df2a,0x1d635,0x1cb16,0x1be2c,0x1ae4e,0x19bde,0x1868e,0x16e2e, |
349 | 0x1527f,0x1334a,0x11051,0xe951, 0xbe01, 0x8e0d, 0x5924, 0x1edd,}; |
350 | |
351 | (2) Iterative refinement |
352 | |
353 | Apply Reciproot iteration three times to y and multiply the |
354 | result by x to get an approximation z that matches sqrt(x) |
355 | to about 1 ulp. To be exact, we will have |
356 | -1ulp < sqrt(x)-z<1.0625ulp. |
357 | |
358 | ... set rounding mode to Round-to-nearest |
359 | y := y*(1.5-0.5*x*y*y) ... almost 15 sig. bits to 1/sqrt(x) |
360 | y := y*((1.5-2^-30)+0.5*x*y*y)... about 29 sig. bits to 1/sqrt(x) |
361 | ... special arrangement for better accuracy |
362 | z := x*y ... 29 bits to sqrt(x), with z*y<1 |
363 | z := z + 0.5*z*(1-z*y) ... about 1 ulp to sqrt(x) |
364 | |
365 | Remark 2. The constant 1.5-2^-30 is chosen to bias the error so that |
366 | (a) the term z*y in the final iteration is always less than 1; |
367 | (b) the error in the final result is biased upward so that |
368 | -1 ulp < sqrt(x) - z < 1.0625 ulp |
369 | instead of |sqrt(x)-z|<1.03125ulp. |
370 | |
371 | (3) Final adjustment |
372 | |
373 | By twiddling y's last bit it is possible to force y to be |
374 | correctly rounded according to the prevailing rounding mode |
375 | as follows. Let r and i be copies of the rounding mode and |
376 | inexact flag before entering the square root program. Also we |
377 | use the expression y+-ulp for the next representable floating |
378 | numbers (up and down) of y. Note that y+-ulp = either fixed |
379 | point y+-1, or multiply y by nextafter(1,+-inf) in chopped |
380 | mode. |
381 | |
382 | R := RZ; ... set rounding mode to round-toward-zero |
383 | switch(r) { |
384 | case RN: ... round-to-nearest |
385 | if(x<= z*(z-ulp)...chopped) z = z - ulp; else |
386 | if(x<= z*(z+ulp)...chopped) z = z; else z = z+ulp; |
387 | break; |
388 | case RZ:case RM: ... round-to-zero or round-to--inf |
389 | R:=RP; ... reset rounding mod to round-to-+inf |
390 | if(x<z*z ... rounded up) z = z - ulp; else |
391 | if(x>=(z+ulp)*(z+ulp) ...rounded up) z = z+ulp; |
392 | break; |
393 | case RP: ... round-to-+inf |
394 | if(x>(z+ulp)*(z+ulp)...chopped) z = z+2*ulp; else |
395 | if(x>z*z ...chopped) z = z+ulp; |
396 | break; |
397 | } |
398 | |
399 | Remark 3. The above comparisons can be done in fixed point. For |
400 | example, to compare x and w=z*z chopped, it suffices to compare |
401 | x1 and w1 (the trailing parts of x and w), regarding them as |
402 | two's complement integers. |
403 | |
404 | ...Is z an exact square root? |
405 | To determine whether z is an exact square root of x, let z1 be the |
406 | trailing part of z, and also let x0 and x1 be the leading and |
407 | trailing parts of x. |
408 | |
409 | If ((z1&0x03ffffff)!=0) ... not exact if trailing 26 bits of z!=0 |
410 | I := 1; ... Raise Inexact flag: z is not exact |
411 | else { |
412 | j := 1 - [(x0>>20)&1] ... j = logb(x) mod 2 |
413 | k := z1 >> 26; ... get z's 25-th and 26-th |
414 | fraction bits |
415 | I := i or (k&j) or ((k&(j+j+1))!=(x1&3)); |
416 | } |
417 | R:= r ... restore rounded mode |
418 | return sqrt(x):=z. |
419 | |
420 | If multiplication is cheaper then the foregoing red tape, the |
421 | Inexact flag can be evaluated by |
422 | |
423 | I := i; |
424 | I := (z*z!=x) or I. |
425 | |
426 | Note that z*z can overwrite I; this value must be sensed if it is |
427 | True. |
428 | |
429 | Remark 4. If z*z = x exactly, then bit 25 to bit 0 of z1 must be |
430 | zero. |
431 | |
432 | -------------------- |
433 | z1: | f2 | |
434 | -------------------- |
435 | bit 31 bit 0 |
436 | |
437 | Further more, bit 27 and 26 of z1, bit 0 and 1 of x1, and the odd |
438 | or even of logb(x) have the following relations: |
439 | |
440 | ------------------------------------------------- |
441 | bit 27,26 of z1 bit 1,0 of x1 logb(x) |
442 | ------------------------------------------------- |
443 | 00 00 odd and even |
444 | 01 01 even |
445 | 10 10 odd |
446 | 10 00 even |
447 | 11 01 even |
448 | ------------------------------------------------- |
449 | |
450 | (4) Special cases (see (4) of Section A). |
451 | |
452 | */ |
453 | |