7019
|
1 /* |
|
2 |
|
3 Copyright (C) 2007 John W. Eaton |
|
4 |
|
5 This file is part of Octave. |
|
6 |
|
7 Octave is free software; you can redistribute it and/or modify it |
|
8 under the terms of the GNU General Public License as published by the |
|
9 Free Software Foundation; either version 3 of the License, or (at your |
|
10 option) any later version. |
|
11 |
|
12 Octave is distributed in the hope that it will be useful, but WITHOUT |
|
13 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
15 for more details. |
|
16 |
|
17 You should have received a copy of the GNU General Public License |
|
18 along with Octave; see the file COPYING. If not, see |
|
19 <http://www.gnu.org/licenses/>. |
|
20 |
|
21 */ |
|
22 |
6593
|
23 #include "mex.h" |
|
24 |
|
25 void |
|
26 mexFunction (int nlhs, mxArray* plhs[], int nrhs, const mxArray* prhs[]) |
|
27 { |
6686
|
28 mwIndex i; |
|
29 mwSize n; |
6593
|
30 double *vri, *vro; |
|
31 |
|
32 if (nrhs != 1 || ! mxIsNumeric (prhs[0])) |
|
33 mexErrMsgTxt ("expects matrix"); |
|
34 |
|
35 n = mxGetNumberOfElements (prhs[0]); |
|
36 plhs[0] = (mxArray *) mxCreateNumericArray |
|
37 (mxGetNumberOfDimensions (prhs[0]), |
|
38 mxGetDimensions (prhs[0]), mxGetClassID (prhs[0]), |
|
39 mxIsComplex (prhs[0])); |
|
40 vri = mxGetPr (prhs[0]); |
|
41 vro = mxGetPr (plhs[0]); |
|
42 |
|
43 if (mxIsComplex (prhs[0])) |
|
44 { |
|
45 double *vii, *vio; |
|
46 vii = mxGetPi (prhs[0]); |
|
47 vio = mxGetPi (plhs[0]); |
|
48 |
|
49 for (i = 0; i < n; i++) |
|
50 { |
|
51 vro [i] = vri [i] * vri [i] - vii [i] * vii [i]; |
|
52 vio [i] = 2 * vri [i] * vii [i]; |
|
53 } |
|
54 } |
|
55 else |
|
56 { |
|
57 for (i = 0; i < n; i++) |
|
58 vro [i] = vri [i] * vri [i]; |
|
59 } |
|
60 } |