xtimesy.c

来自「matlab实用教程」· C语言 代码 · 共 52 行

#include "mex.h"

void xtimesy(double x, double *y, double *z, mwSize m, mwSize n)
{
  mwSize i,j,count=0;
  
  for (i=0; i<n; i++) {
    for (j=0; j<m; j++) {
      *(z+count) = x * *(y+count);
      count++;
    }
  }
}

void mexFunction( int nlhs, mxArray *plhs[],
                int nrhs, const mxArray *prhs[])
{
  double *y,*z;
  double  x;
  mwSize mrows,ncols;
  
  /*  检查输入参数的个数 */
  if(nrhs!=2) 
    mexErrMsgTxt("Two inputs required."); // mexErrMsgTxt语句会退出MEX文件的执行
  if(nlhs!=1) 
    mexErrMsgTxt("One output required.");
  
  /* 第一个输入参数必须是标量 */
  if( !mxIsDouble(prhs[0]) || mxIsComplex(prhs[0]) ||
      mxGetN(prhs[0])*mxGetM(prhs[0])!=1 ) {
    mexErrMsgTxt("Input x must be a scalar.");
  }
  
  /*  获取标量参数 x */
  x = mxGetScalar(prhs[0]);
  
  /*  创建指针指向输入参数矩阵 y */
  y = mxGetPr(prhs[1]);
  
  /*  获取y的维数 */
  mrows = mxGetM(prhs[1]);
  ncols = mxGetN(prhs[1]);
  
  /*  输出参数指针指向输出矩阵 */
  plhs[0] = mxCreateDoubleMatrix(mrows,ncols, mxREAL);
  
  /*  创建C语言的指针指向输出矩阵的副本 */
  z = mxGetPr(plhs[0]);
  
  xtimesy(x,y,z,mrows,ncols);  
}