Mercurial > hg > octave-lyh

annotate examples/mypow2.c @ 6686:2aad75fcc93a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
[project @ 2007-06-03 20:58:28 by dbateman]
author dbateman
date Sun, 03 Jun 2007 20:58:29 +0000
parents 3da1f4a41455
children 4270ded9ddc6
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
6593
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
1 #include "mex.h"
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
2
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
3 void
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
4 mexFunction (int nlhs, mxArray* plhs[], int nrhs, const mxArray* prhs[])
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
5 {
6686
2aad75fcc93a [project @ 2007-06-03 20:58:28 by dbateman]
dbateman
parents: 6593
diff changeset
6 mwIndex i;
2aad75fcc93a [project @ 2007-06-03 20:58:28 by dbateman]
dbateman
parents: 6593
diff changeset
7 mwSize n;
6593
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
8 double *vri, *vro;
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
9
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
10 if (nrhs != 1 || ! mxIsNumeric (prhs[0]))
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
11 mexErrMsgTxt ("expects matrix");
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
12
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
13 n = mxGetNumberOfElements (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
14 plhs[0] = (mxArray *) mxCreateNumericArray
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
15 (mxGetNumberOfDimensions (prhs[0]),
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
16 mxGetDimensions (prhs[0]), mxGetClassID (prhs[0]),
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
17 mxIsComplex (prhs[0]));
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
18 vri = mxGetPr (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
19 vro = mxGetPr (plhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
20
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
21 if (mxIsComplex (prhs[0]))
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
22 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
23 double *vii, *vio;
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
24 vii = mxGetPi (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
25 vio = mxGetPi (plhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
26
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
27 for (i = 0; i < n; i++)
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
28 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
29 vro [i] = vri [i] * vri [i] - vii [i] * vii [i];
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
30 vio [i] = 2 * vri [i] * vii [i];
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
31 }
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
32 }
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
33 else
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
34 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
35 for (i = 0; i < n; i++)
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
36 vro [i] = vri [i] * vri [i];
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
37 }
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
38 }