编辑Mex和Armadillo的难度

时间:2012-11-23 05:57:04

标签: matlab mex armadillo

我在编写示例mex和犰狳程序时遇到困难,并且想知道是否有人可以帮助我。我使用的是Mac OS,并且在安装方面取得了成功。

代码:

#include "mex.h"
#include "math.h"
#include<armadillo>


using namespace arma;

void matlab2arma(mat& A, const mxArray *mxdata){
// delete [] A.mem; // don't do this!
access::rw(A.mem)=mxGetPr(mxdata);
access::rw(A.n_rows)=mxGetM(mxdata); // transposed!
access::rw(A.n_cols)=mxGetN(mxdata);
access::rw(A.n_elem)=A.n_rows*A.n_cols;
};

void freeVar(mat& A, const double *ptr){
access::rw(A.mem)=ptr;
access::rw(A.n_rows)=1; // transposed!
access::rw(A.n_cols)=1;
access::rw(A.n_elem)=1;
};

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
if (nrhs != 2)
mexErrMsgTxt("Incorrect number of input arguments");
if (nlhs != 1)
mexErrMsgTxt("Incorrect number of output arguments");

mat D1(1,1);
const double* D1mem=access::rw(D1.mem);
matlab2arma(D1,prhs[0]); // First create the matrix, then change it to point to the matlab data.

mat D2(1,1);
const double* D2mem=access::rw(D2.mem);
matlab2arma(D2,prhs[1]);

// check if the input corresponds to what you are expecting
if( D1.n_rows != D2.n_rows )
mexErrMsgTxt("Columns of D1 and D2 must be of equal length!");

if( D1.n_cols != D2.n_cols )
mexErrMsgTxt("Rows of D1 and D2 must be of equal length!");

plhs[0] = mxCreateDoubleMatrix(D1.n_rows, D1.n_cols, mxREAL);
mat output(1,1);
const double* outputmem=access::rw(output.mem);
matlab2arma(output,plhs[0]);

output=D1+D2;
// output.print();

freeVar(D1,D1mem); // Change back the pointers!!
freeVar(D2,D2mem);
freeVar(output,outputmem);
return;
}

2 个答案:

答案 0 :(得分:1)

它使用以下编译命令在我的系统(Ubuntu 12.04,64位,MatlabR2013a,g ++)上工作:

mex mexTest.cpp -llapack -larmadillo -lblas

其中mexTest.cpp包含您提供的代码段。请注意,为了在Matlab下正确编译Armadillo,需要正确设置库路径。这些可以在mexopts.sh中更新,或者通过重定向$ matlabroot / sys / os / XXXXX中的符号链接来更新,其中XXXXX可能会根据您的系统(32/64位和OS)而有所不同。

答案 1 :(得分:0)

如果您正确设置了犰狳,只需执行以下操作就足够了(代码也能为我成功):

mex your_code.cpp -larmadillo