Merge remote-tracking branch 'origin/master' into windows
[scilab.git] / scilab / modules / elementary_functions / sci_gateway / cpp / sci_sqrt.cpp
1 /*
2  * Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
3  * Copyright (C) 2015 - Scilab Enterprises - Antoine ELIAS
4  *
5  * Copyright (C) 2012 - 2016 - Scilab Enterprises
6  *
7  * This file is hereby licensed under the terms of the GNU GPL v2.0,
8  * pursuant to article 5.3.4 of the CeCILL v.2.1.
9  * This file was originally licensed under the terms of the CeCILL v2.1,
10  * and continues to be available under such terms.
11  * For more information, see the COPYING file which you should have received
12  * along with this program.
13  *
14  */
15 /*--------------------------------------------------------------------------*/
16 #include "elem_func_gw.hxx"
17 #include "double.hxx"
18 #include "overload.hxx"
19
20 extern "C"
21 {
22 #include "Scierror.h"
23 #include "localization.h"
24 #include "basic_functions.h"
25 }
26
27 const char fname[] = "sqrt";
28
29 types::Function::ReturnValue sci_sqrt(types::typed_list &in, int _iRetCount, types::typed_list &out)
30 {
31     if (in.size() != 1)
32     {
33         Scierror(77, _("%s: Wrong number of input argument(s): %d expected.\n"), fname, 1);
34         return types::Function::Error;
35     }
36
37     if (_iRetCount > 1)
38     {
39         Scierror(78, _("%s: Wrong number of output argument(s): %d expected.\n"), fname, 1);
40         return types::Function::Error;
41     }
42
43     if (in[0]->isDouble() == false)
44     {
45         std::string stFuncName = "%" + in[0]->getShortTypeStr() + "_sqrt";
46         return Overload::call(stFuncName, in, _iRetCount, out);
47     }
48
49     types::Double* input = in[0]->getAs<types::Double>();
50     bool isComplex = input->isComplex();
51     int size = input->getSize();
52     double* pR = input->get();
53
54     //check neg on input values for real matrix
55     if (isComplex == false)
56     {
57         for (int i = 0; i < size; ++i)
58         {
59             if (pR[i] < 0)
60             {
61                 isComplex = true;
62                 break;
63             }
64         }
65     }
66
67     types::Double* output = new types::Double(input->getDims(), input->getDimsArray(), isComplex);
68     double* pOR = output->get();
69
70     if (isComplex)
71     {
72         double* pOI = output->getImg();
73         if (input->isComplex())
74         {
75             double* pI = input->getImg();
76             for (int i = 0; i < size; ++i)
77             {
78                 zsqrts(pR[i], pI[i], &pOR[i], &pOI[i]);
79             }
80         }
81         else
82         {
83             for (int i = 0; i < size; ++i)
84             {
85                 zsqrts(pR[i], 0, &pOR[i], &pOI[i]);
86             }
87         }
88     }
89     else
90     {
91         for (int i = 0; i < size; ++i)
92         {
93             pOR[i] = std::sqrt(pR[i]);
94         }
95     }
96
97     out.push_back(output);
98     return types::Function::OK;
99 }
100 /*--------------------------------------------------------------------------*/