A = mxCreateDoubleMatrix(M, N, mxREAL);mxDestroyArray(A);

最新推荐文章于 2021-03-17 18:46:42 发布

原创最新推荐文章于 2021-03-17 18:46:42 发布 · 5.5k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#matlab #function #input #matrix #include

mex学习笔记专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一个使用MATLAB API进行矩阵QR分解的MEX文件实现。该程序首先显示输入矩阵，然后调用MATLAB的内置qr函数进行分解，并输出得到的Q和R矩阵。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <string.h>
#include "mex.h"
void DisplayMatrix(char *Name, double *Data, int M, int N)
{ /* Display matrix data */
	int m, n;
	mexPrintf("%s = \n", Name);
	for(m = 0; m < M; m++, mexPrintf("\n"))
		for(n = 0; n < N; n++)
			mexPrintf("%8.4f ", Data[m + M*n]);
}

void CallQR(double *Data, int M, int N)
{ /* Perform QR factorization by calling the MATLAB function */
	mxArray *Q, *R, *A;
	mxArray *ppLhs[2];
	DisplayMatrix("Input", Data, M, N);
	A = mxCreateDoubleMatrix(M, N, mxREAL); /* Put input in an mxArray */
	memcpy(mxGetPr(A), Data, sizeof(double)*M*N);
	mexCallMATLAB(2, ppLhs, 1, &A, "qr"); /* Call MATLAB's qr function */
	Q = ppLhs[0];
	R = ppLhs[1];
	DisplayMatrix("Q", mxGetPr(Q), M, N);
	DisplayMatrix("R", mxGetPr(R), M, N);
	mxDestroyArray(R); /* No longer need these */
	mxDestroyArray(Q);
	mxDestroyArray(A);
}

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
#define M_IN prhs[0]
	if(nrhs != 1 || mxGetNumberOfDimensions(M_IN) != 2 || !mxIsDouble(M_IN))
		mexErrMsgTxt("Invalid input.");
	CallQR(mxGetPr(M_IN), mxGetM(M_IN), mxGetN(M_IN));
}