dspl/sqrt__cmplx_8c_source.html

/*

* Copyright (c) 2015-2023 Sergey Bakhurin

* Digital Signal Processing Library [http://dsplib.org]

*

* This file is part of DSPL.

*

* is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

*

* DSPL is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.    See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with Foobar.    If not, see <http://www.gnu.org/licenses/>.

*/


#include <stdio.h>

#include <stdlib.h>

#include "dspl.h"


#ifdef DOXYGEN_ENGLISH


#endif

#ifdef DOXYGEN_RUSSIAN


#endif

int DSPL_API sqrt_cmplx(complex_t* x, int n, complex_t *y)

{

    int k;

    double r, zr, at;

    complex_t t;

    if(!x || !y)

        return ERROR_PTR;

    if(n < 1)

        return ERROR_SIZE;


    for(k = 0; k < n; k++)

    {

        r = ABS(x[k]);

        if(r == 0.0)

        {

            RE(y[k]) = 0.0;

            IM(y[k]) = 0.0;

        }

        else

        {

            RE(t) = RE(x[k]) + r;

            IM(t) = IM(x[k]);

            at = ABS(t);

            if(at == 0.0)

            {

                RE(y[k]) = 0.0;

                IM(y[k]) = sqrt(r);

            }

            else

            {

                zr = 1.0 / ABS(t);

                r = sqrt(r);

                RE(y[k]) = RE(t) * zr * r;

                IM(y[k]) = IM(t) * zr * r;

            }

        }

    }

    return RES_OK;

}