Mercurial > octave-nkf
view examples/code/mypow2.c @ 20297:5c42ff6f0eb1 stable
Clean up MEX example code.
* myfeval.c: Use mxIsChar rather than deprecated mxIsString.
* mypow2.c: Validate that input is a double matrix.
* myprop.c: Use space after '!' operator to conform to Octave conventions.
* myset.c: Use mexPutVariable instead of missing mxSetName and deprecated
mexPutArray. Find existing variable EITHER in global workspace OR in
caller workspace. Don't check both.
* mystruct.c: Clarify input validation message.
author | Rik <rik@octave.org> |
---|---|
date | Mon, 15 Jun 2015 10:24:13 -0700 |
parents | c8240a60dd01 |
children |
line wrap: on
line source
#include "mex.h" void mexFunction (int nlhs, mxArray* plhs[], int nrhs, const mxArray* prhs[]) { mwSize n; mwIndex i; double *vri, *vro; if (nrhs != 1 || ! mxIsDouble (prhs[0])) mexErrMsgTxt ("ARG1 must be a double matrix"); n = mxGetNumberOfElements (prhs[0]); plhs[0] = mxCreateNumericArray (mxGetNumberOfDimensions (prhs[0]), mxGetDimensions (prhs[0]), mxGetClassID (prhs[0]), mxIsComplex (prhs[0])); vri = mxGetPr (prhs[0]); vro = mxGetPr (plhs[0]); if (mxIsComplex (prhs[0])) { double *vii, *vio; vii = mxGetPi (prhs[0]); vio = mxGetPi (plhs[0]); for (i = 0; i < n; i++) { vro[i] = vri[i] * vri[i] - vii[i] * vii[i]; vio[i] = 2 * vri[i] * vii[i]; } } else { for (i = 0; i < n; i++) vro[i] = vri[i] * vri[i]; } }