Mercurial > hg > octave-nkf

annotate examples/code/mypow2.c @ 20828:a3b9ee5c040a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Replace bsxfun with broadcasting for performance with complex inputs (bug #38628). cumtrapz.m, quadgk.m, trapz.m, center.m, zscore.m: Replace bsxfun with broadcasting for performance where inputs might be complex.
author Rik <rik@octave.org>
date Mon, 12 Oct 2015 21:28:32 -0700
parents 5c42ff6f0eb1
children
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
16867
be41c30bcb44 Re-write documentation and all examples of dynamically linked functions.
Rik <rik@octave.org>
parents: 9932
diff changeset
4 mexFunction (int nlhs, mxArray* plhs[],
be41c30bcb44 Re-write documentation and all examples of dynamically linked functions.
Rik <rik@octave.org>
parents: 9932
diff changeset
5 int nrhs, const mxArray* prhs[])
6593
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
6 {
16867
be41c30bcb44 Re-write documentation and all examples of dynamically linked functions.
Rik <rik@octave.org>
parents: 9932
diff changeset
7 mwSize n;
6686
2aad75fcc93a [project @ 2007-06-03 20:58:28 by dbateman]
dbateman
parents: 6593
diff changeset
8 mwIndex i;
6593
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
9 double *vri, *vro;
17791
224e76250443 Use GNU style coding conventions for code in examples/
Rik <rik@octave.org>
parents: 16867
diff changeset
10
20472
5c42ff6f0eb1 Clean up MEX example code.
Rik <rik@octave.org>
parents: 19225
diff changeset
11 if (nrhs != 1 || ! mxIsDouble (prhs[0]))
5c42ff6f0eb1 Clean up MEX example code.
Rik <rik@octave.org>
parents: 19225
diff changeset
12 mexErrMsgTxt ("ARG1 must be a double matrix");
6593
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
13
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
14 n = mxGetNumberOfElements (prhs[0]);
17791
224e76250443 Use GNU style coding conventions for code in examples/
Rik <rik@octave.org>
parents: 16867
diff changeset
15 plhs[0] = mxCreateNumericArray (mxGetNumberOfDimensions (prhs[0]),
224e76250443 Use GNU style coding conventions for code in examples/
Rik <rik@octave.org>
parents: 16867
diff changeset
16 mxGetDimensions (prhs[0]),
224e76250443 Use GNU style coding conventions for code in examples/
Rik <rik@octave.org>
parents: 16867
diff changeset
17 mxGetClassID (prhs[0]),
224e76250443 Use GNU style coding conventions for code in examples/
Rik <rik@octave.org>
parents: 16867
diff changeset
18 mxIsComplex (prhs[0]));
6593
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
19 vri = mxGetPr (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
20 vro = mxGetPr (plhs[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 if (mxIsComplex (prhs[0]))
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
23 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
24 double *vii, *vio;
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
25 vii = mxGetPi (prhs[0]);
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
26 vio = mxGetPi (plhs[0]);
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 for (i = 0; i < n; i++)
9932
6cb30a539481 untabify files in examples directory
John W. Eaton <jwe@octave.org>
parents: 9053
diff changeset
29 {
16867
be41c30bcb44 Re-write documentation and all examples of dynamically linked functions.
Rik <rik@octave.org>
parents: 9932
diff changeset
30 vro[i] = vri[i] * vri[i] - vii[i] * vii[i];
be41c30bcb44 Re-write documentation and all examples of dynamically linked functions.
Rik <rik@octave.org>
parents: 9932
diff changeset
31 vio[i] = 2 * vri[i] * vii[i];
9932
6cb30a539481 untabify files in examples directory
John W. Eaton <jwe@octave.org>
parents: 9053
diff changeset
32 }
6593
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 else
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
35 {
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
36 for (i = 0; i < n; i++)
16867
be41c30bcb44 Re-write documentation and all examples of dynamically linked functions.
Rik <rik@octave.org>
parents: 9932
diff changeset
37 vro[i] = vri[i] * vri[i];
6593
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
38 }
3da1f4a41455 [project @ 2007-04-27 08:08:19 by dbateman]
dbateman
parents:
diff changeset
39 }