convec.c

matlab实用教程

#include "mex.h"
 
void convec( double *xr, double *xi, mwSize nx,
             double *yr, double *yi, mwSize ny,
             double *zr, double *zi)
{
    mwSize i,j;
  
    zr[0]=0.0;
    zi[0]=0.0;
    /* 计算复数数组的卷积 */
    for(i=0; i<nx; i++) {
    for(j=0; j<ny; j++) {
        *(zr+i+j) = *(zr+i+j) + *(xr+i) * *(yr+j) - *(xi+i) * *(yi+j);
        *(zi+i+j) = *(zi+i+j) + *(xr+i) * *(yi+j) + *(xi+i) * *(yr+j);
    }
    }
}
 
void mexFunction( int nlhs, mxArray *plhs[],
                  int nrhs, const mxArray *prhs[] )
{
    double  *xr, *xi, *yr, *yi, *zr, *zi;
    mwSize  rows, cols, nx, ny;
    
    /* 检查参数 */
    if(nrhs != 2)
      mexErrMsgTxt("Two inputs required.");
    if(nlhs > 1)
      mexErrMsgTxt("Too many output arguments.");
    /* 检查两个输入参数是否是行数组*/
    if( mxGetM(prhs[0]) != 1 || mxGetM(prhs[1]) != 1 )
      mexErrMsgTxt("Both inputs must be row vectors.");
    rows = 1; 
    /* 检查输入参数是否为复数 */
    if( !mxIsComplex(prhs[0]) || !mxIsComplex(prhs[1]) )
      mexErrMsgTxt("Inputs must be complex.\n");
  
    /* 分别获取两个每个输入参数的长度 */
    nx = mxGetN(prhs[0]);
    ny = mxGetN(prhs[1]);
 
    /* 获取两个输入参数的实部和虚部的指针 */
    xr = mxGetPr(prhs[0]);
    xi = mxGetPi(prhs[0]);
    yr = mxGetPr(prhs[1]);
    yi = mxGetPi(prhs[1]);
  
    /* 为输出参数创建数组并将输出参数指针指向这个数组 */
    cols = nx + ny - 1;
    plhs[0] = mxCreateDoubleMatrix(rows, cols, mxCOMPLEX);
    zr = mxGetPr(plhs[0]);
    zi = mxGetPi(plhs[0]);
 
    /* 调用C 计算子程序 */
    convec(xr, xi, nx, yr, yi, ny, zr, zi);
 
    return;
}