inverse matrix in mexFunction

조회 수: 10 (최근 30일)
Jane Jean
Jane Jean 2012년 4월 4일
void inverseMatrix(int dim, double *matrix, double *invMatrix)
{
// matrix and invMatrix are columnwise.
int *IPIV, LWORK, INFO, i;
double *WORK;
mexPrintf("entered inverseMatrix");
IPIV = mxMalloc((dim+1)*sizeof(int));
LWORK = dim*dim;
WORK = mxMalloc(LWORK*sizeof(double));
for (i=0;i<dim*dim;i++){
invMatrix[i] = matrix[i];
}
mexPrintf("before dgetrf");
dgetrf(&dim, &dim, invMatrix, &dim, IPIV, &INFO);
mexPrintf("before dgetri");
dgetri(&dim, invMatrix, &dim, IPIV, WORK, &LWORK, &INFO);
mxFree(IPIV);
mxFree(WORK);
return;
}
I am trying to use dgetrf and dgetri to inverse a matrix in C but Matlab crashes after successfully giving the correct answer 2 times (I did an interation to try the stability of the c code).

채택된 답변

Jan
Jan 2012년 4월 4일
The copy of the data to the output would be faster using memcpy:
plhs[0] = mxCreateDoubleMatrix(outputRowLen, outputColLen, mxREAL);
memcpy(mxGetPr(plhs[0]), Sigma_Psi_inv_t,
outputRowLen*outputColLen * sizeof(double));
But you can use the data of the output array directly:
plhs[0] = mxCreateDoubleMatrix(rowSigma_Psi, colSigma_Psi, mxREAL);
mexPrintf("entering inverseMatrix");
inverseMatrix(rowSigma_Psi, arraySigma_Psi, mxGetPr(plhs[0]));
Do not use int for LAPACK calls in an 64 bit environment. Use mwSignedIndex instead, which is a 64 bit integer.
  댓글 수: 5
Jane Jean
Jane Jean 2012년 4월 4일
I found the problem now. The code that I pasted below did not work as I forgot to change the 'int' in sizeof to 'size_t'. Thank you guys for pointing out the errors!!! ;) You saved my day!
Jan
Jan 2012년 4월 4일
mwSignedIndex is defined in tmwtypes.h, which is included by mex.h. So "#include mex.h" should be enough.

댓글을 달려면 로그인하십시오.

추가 답변 (3개)

Titus Edelhofer
Titus Edelhofer 2012년 4월 4일
Hi Jane,
the code looks fine to me. Maybe it's the calling mex file causing the problem?
Titus
  댓글 수: 1
Jane Jean
Jane Jean 2012년 4월 4일
Thanks for the reply. I've pasted the code in my mexFunction below.

댓글을 달려면 로그인하십시오.


Jane Jean
Jane Jean 2012년 4월 4일
Sigma_Psi_inv_t = mxMalloc(rowSigma_Psi*colSigma_Psi*sizeof(double));
if(!Sigma_Psi_inv_t){
mexErrMsgTxt("Memory allocation error in least-squares initialization.");
}
mexPrintf("entering inverseMatrix");
inverseMatrix(rowSigma_Psi, arraySigma_Psi, Sigma_Psi_inv_t);
outputRowLen = rowSigma_Psi;
outputColLen = colSigma_Psi;
plhs[0] = mxCreateDoubleMatrix(outputRowLen, outputColLen, mxREAL);
arrayOutput = mxGetPr(plhs[0]);
for (i=0; i<outputRowLen; i++){
for(j=0; j<outputColLen; j++){
arrayOutput[i + j*outputRowLen] = Sigma_Psi_inv_t[i*outputColLen +j];
}
}
mxFree(Sigma_Psi_inv_t);
  댓글 수: 1
Jane Jean
Jane Jean 2012년 4월 4일
I still can't find the error of the code. I have tried mexPrintf but after a few iterations, Matlab just crashed without printing out any error message.
Is there anyway to efficiently debug to error?

댓글을 달려면 로그인하십시오.


Jane Jean
Jane Jean 2012년 4월 4일
void inverseMatrix(size_t dim, double *matrix, double *invMatrix)
{
// matrix and invMatrix are columnwise.
int *IPIV, LWORK, INFO=0, i;
double *WORK;
mexPrintf("entered inverseMatrix");
IPIV = (int*)mxMalloc((dim+1)*sizeof(int));
LWORK = dim*dim;
WORK = (double*)mxMalloc(LWORK*sizeof(double));
for (i=0;i<dim*dim;i++){
invMatrix[i] = matrix[i];
}
mexPrintf("before dgetrf");
dgetrf(&dim, &dim, invMatrix, &dim, IPIV, &INFO);
mexPrintf("before dgetri");
dgetri(&dim, invMatrix, &dim, IPIV, WORK, &LWORK, &INFO);
mxFree(IPIV);
mxFree(WORK);
return;
}
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
mxArray *xData;
double *xValues, *outArray, *invMatrix;
int i,j;
int rowLen, colLen;
size_t row;
xData = prhs[0];
xValues = mxGetPr(xData);
rowLen = mxGetM(xData);
colLen = mxGetN(xData);
row = rowLen;
plhs[0] = mxCreateDoubleMatrix(rowLen, colLen, mxREAL);
outArray = mxGetPr(plhs[0]);
inverseMatrix(row, xValues, mxGetPr(plhs[0]));
return;
}
to compile and run.
for i=1:50
blaslib = fullfile('C:\Program Files\MATLAB\R2011b\extern\lib\win64\microsoft\libmwblas.lib');
lapacklib = fullfile('C:\Program Files\MATLAB\R2011b\extern\lib\win64\microsoft\libmwlapack.lib');
mex('-largeArrayDims', 'test_id.c','mathFunction64.c', blaslib, lapacklib)
A=[0,1,2,3;4,5,6,7;8,9,10,11;12,13,14,15]
[N] = test_id(A);
end;

카테고리

Help CenterFile Exchange에서 Write C Functions Callable from MATLAB (MEX Files)에 대해 자세히 알아보기

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by