125c28e83SPiotr Jasiukajtis /*
225c28e83SPiotr Jasiukajtis * CDDL HEADER START
325c28e83SPiotr Jasiukajtis *
425c28e83SPiotr Jasiukajtis * The contents of this file are subject to the terms of the
525c28e83SPiotr Jasiukajtis * Common Development and Distribution License (the "License").
625c28e83SPiotr Jasiukajtis * You may not use this file except in compliance with the License.
725c28e83SPiotr Jasiukajtis *
825c28e83SPiotr Jasiukajtis * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
925c28e83SPiotr Jasiukajtis * or http://www.opensolaris.org/os/licensing.
1025c28e83SPiotr Jasiukajtis * See the License for the specific language governing permissions
1125c28e83SPiotr Jasiukajtis * and limitations under the License.
1225c28e83SPiotr Jasiukajtis *
1325c28e83SPiotr Jasiukajtis * When distributing Covered Code, include this CDDL HEADER in each
1425c28e83SPiotr Jasiukajtis * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
1525c28e83SPiotr Jasiukajtis * If applicable, add the following below this CDDL HEADER, with the
1625c28e83SPiotr Jasiukajtis * fields enclosed by brackets "[]" replaced with your own identifying
1725c28e83SPiotr Jasiukajtis * information: Portions Copyright [yyyy] [name of copyright owner]
1825c28e83SPiotr Jasiukajtis *
1925c28e83SPiotr Jasiukajtis * CDDL HEADER END
2025c28e83SPiotr Jasiukajtis */
2125c28e83SPiotr Jasiukajtis
2225c28e83SPiotr Jasiukajtis /*
2325c28e83SPiotr Jasiukajtis * Copyright 2011 Nexenta Systems, Inc. All rights reserved.
2425c28e83SPiotr Jasiukajtis */
2525c28e83SPiotr Jasiukajtis /*
2625c28e83SPiotr Jasiukajtis * Copyright 2006 Sun Microsystems, Inc. All rights reserved.
2725c28e83SPiotr Jasiukajtis * Use is subject to license terms.
2825c28e83SPiotr Jasiukajtis */
2925c28e83SPiotr Jasiukajtis
30*ddc0e0b5SRichard Lowe #pragma weak __csqrtf = csqrtf
3125c28e83SPiotr Jasiukajtis
3225c28e83SPiotr Jasiukajtis #include "libm.h" /* sqrt/fabsf/sqrtf */
3325c28e83SPiotr Jasiukajtis #include "complex_wrapper.h"
3425c28e83SPiotr Jasiukajtis
3525c28e83SPiotr Jasiukajtis /* INDENT OFF */
3625c28e83SPiotr Jasiukajtis static const float zero = 0.0F;
3725c28e83SPiotr Jasiukajtis /* INDENT ON */
3825c28e83SPiotr Jasiukajtis
3925c28e83SPiotr Jasiukajtis fcomplex
csqrtf(fcomplex z)4025c28e83SPiotr Jasiukajtis csqrtf(fcomplex z) {
4125c28e83SPiotr Jasiukajtis fcomplex ans;
4225c28e83SPiotr Jasiukajtis double dt, dx, dy;
4325c28e83SPiotr Jasiukajtis float x, y, t, ax, ay, w;
4425c28e83SPiotr Jasiukajtis int ix, iy, hx, hy;
4525c28e83SPiotr Jasiukajtis
4625c28e83SPiotr Jasiukajtis x = F_RE(z);
4725c28e83SPiotr Jasiukajtis y = F_IM(z);
4825c28e83SPiotr Jasiukajtis hx = THE_WORD(x);
4925c28e83SPiotr Jasiukajtis hy = THE_WORD(y);
5025c28e83SPiotr Jasiukajtis ix = hx & 0x7fffffff;
5125c28e83SPiotr Jasiukajtis iy = hy & 0x7fffffff;
5225c28e83SPiotr Jasiukajtis ay = fabsf(y);
5325c28e83SPiotr Jasiukajtis ax = fabsf(x);
5425c28e83SPiotr Jasiukajtis if (ix >= 0x7f800000 || iy >= 0x7f800000) {
5525c28e83SPiotr Jasiukajtis /* x or y is Inf or NaN */
5625c28e83SPiotr Jasiukajtis if (iy == 0x7f800000)
5725c28e83SPiotr Jasiukajtis F_IM(ans) = F_RE(ans) = ay;
5825c28e83SPiotr Jasiukajtis else if (ix == 0x7f800000) {
5925c28e83SPiotr Jasiukajtis if (hx > 0) {
6025c28e83SPiotr Jasiukajtis F_RE(ans) = ax;
6125c28e83SPiotr Jasiukajtis F_IM(ans) = ay * zero;
6225c28e83SPiotr Jasiukajtis } else {
6325c28e83SPiotr Jasiukajtis F_RE(ans) = ay * zero;
6425c28e83SPiotr Jasiukajtis F_IM(ans) = ax;
6525c28e83SPiotr Jasiukajtis }
6625c28e83SPiotr Jasiukajtis } else
6725c28e83SPiotr Jasiukajtis F_IM(ans) = F_RE(ans) = ax + ay;
6825c28e83SPiotr Jasiukajtis } else if (iy == 0) {
6925c28e83SPiotr Jasiukajtis if (hx >= 0) {
7025c28e83SPiotr Jasiukajtis F_RE(ans) = sqrtf(ax);
7125c28e83SPiotr Jasiukajtis F_IM(ans) = zero;
7225c28e83SPiotr Jasiukajtis } else {
7325c28e83SPiotr Jasiukajtis F_IM(ans) = sqrtf(ax);
7425c28e83SPiotr Jasiukajtis F_RE(ans) = zero;
7525c28e83SPiotr Jasiukajtis }
7625c28e83SPiotr Jasiukajtis } else {
7725c28e83SPiotr Jasiukajtis dx = (double) ax;
7825c28e83SPiotr Jasiukajtis dy = (double) ay;
7925c28e83SPiotr Jasiukajtis dt = sqrt(0.5 * (sqrt(dx * dx + dy * dy) + dx));
8025c28e83SPiotr Jasiukajtis t = (float) dt;
8125c28e83SPiotr Jasiukajtis w = (float) (dy / (dt + dt));
8225c28e83SPiotr Jasiukajtis if (hx >= 0) {
8325c28e83SPiotr Jasiukajtis F_RE(ans) = t;
8425c28e83SPiotr Jasiukajtis F_IM(ans) = w;
8525c28e83SPiotr Jasiukajtis } else {
8625c28e83SPiotr Jasiukajtis F_IM(ans) = t;
8725c28e83SPiotr Jasiukajtis F_RE(ans) = w;
8825c28e83SPiotr Jasiukajtis }
8925c28e83SPiotr Jasiukajtis }
9025c28e83SPiotr Jasiukajtis if (hy < 0)
9125c28e83SPiotr Jasiukajtis F_IM(ans) = -F_IM(ans);
9225c28e83SPiotr Jasiukajtis return (ans);
9325c28e83SPiotr Jasiukajtis }
94