Mercurial > hg > octave-lyh

annotate examples/mypow2.c @ 6593:3da1f4a41455

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
[project @ 2007-04-27 08:08:19 by dbateman]
author dbateman
date Fri, 27 Apr 2007 08:08:19 +0000
parents
children 2aad75fcc93a
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 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
6 int i, n;
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
7 double *vri, *vro;
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
8
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
9 if (nrhs != 1 || ! mxIsNumeric (prhs[0]))
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
10 mexErrMsgTxt ("expects matrix");
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
11
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
12 n = mxGetNumberOfElements (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
13 plhs[0] = (mxArray *) mxCreateNumericArray
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
14 (mxGetNumberOfDimensions (prhs[0]),
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
15 mxGetDimensions (prhs[0]), mxGetClassID (prhs[0]),
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
16 mxIsComplex (prhs[0]));
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
17 vri = mxGetPr (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
18 vro = mxGetPr (plhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
19
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
20 if (mxIsComplex (prhs[0]))
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
21 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
22 double *vii, *vio;
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
23 vii = mxGetPi (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
24 vio = mxGetPi (plhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
25
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
26 for (i = 0; i < n; i++)
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
27 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
28 vro [i] = vri [i] * vri [i] - vii [i] * vii [i];
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
29 vio [i] = 2 * vri [i] * vii [i];
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
30 }
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 else
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
33 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
34 for (i = 0; i < n; i++)
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
35 vro [i] = vri [i] * vri [i];
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
36 }
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
37 }