我可以返回的矩阵只有第一个元素正确,所有其他元素均为零。我尝试按照此链接中的建议进行操作;但是,它不适合我。一些帮助,将不胜感激。
输出:
>> Matlab_mex_Testing
B1 =
1.0000 0 0
0 1.0000 0
0 0 1.0000
0 0 0
B =
1 0 0
0 0 0
0 0 0
0 0 0
Matlab代码:
Dir2 = '/home/dkumar/Mex_Codes_DKU/MexCode_Working/Mex_CPP_N_ARMADILLO_Codes_DKU_makefile_Working';
% MEX
cd(Dir2);
[y, B] = normpdfDKU(1/2,0,1);
%Matlab declared
B1 = eye(4,3)
% Identical matrix returned from MEX
B
CPP:
#include "mex.h"
#include <math.h>
#include "/home/dkumar/armadillo-4.600.3/include/armadillo"
using namespace arma;
using namespace std;
#define pi (3.141592653589793)
extern void _main();
const int numInputArgs = 3;
const int numOutputArgs = 2;
// Function declarations.
// -----------------------------------------------------------------
double getMatlabScalar (const mxArray* ptr);
double& createMatlabScalar (mxArray*& ptr);
int TestingLibraries() ; // declared and defined in In Include_4_TSNNLS.c and Include_4_TSNNLS.h
// Function definitions.
// -----------------------------------------------------------------
void mexFunction (int nlhs, mxArray *plhs[],
int nrhs, const mxArray *prhs[]) {
// Check to see if we have the correct number of input and output
// arguments.
if (nrhs != numInputArgs)
mexErrMsgTxt(" DKU-1: Incorrect number of input arguments");
if (nlhs != numOutputArgs)
mexErrMsgTxt("DKU-2: Incorrect number of output arguments");
// Get the inputs.
double x = getMatlabScalar(prhs[0]);
double mu = getMatlabScalar(prhs[1]);
double v = getMatlabScalar(prhs[2]);
// Create the output. It is also a double-precision scalar.
double& p = createMatlabScalar(plhs[0]);
// Compute the value of the univariate Normal at x.
p = exp(-(x-mu)*(x-mu)/(2*v)) / sqrt(2*pi*v);
// CREATE ARMA::mat and print
mat B = eye<mat>(4,3);
//Print B
B.print();
// Trying to return B as second output
mwSize sz[2];
sz[0] = B.n_rows ; // Matlab is row first
sz[1] = B.n_cols ;
//mxArray* pOUT = mxCreateNumericArray(2, sz, mxDOUBLE_CLASS, mxREAL);
plhs[1] = mxCreateNumericArray(2, sz, mxDOUBLE_CLASS, mxREAL);
//Get a pointer to pOUT
double* p2 = (double*)mxGetData(plhs[1]);
*p2 = B[0];
}
double getMatlabScalar (const mxArray* ptr) {
// Make sure the input argument is a scalar in double-precision.
if (!mxIsDouble(ptr) || mxGetNumberOfElements(ptr) != 1)
mexErrMsgTxt("The input argument must be a double-precision scalar");
return *mxGetPr(ptr);
}
double& createMatlabScalar (mxArray*& ptr) {
ptr = mxCreateDoubleMatrix(1,1,mxREAL);
return *mxGetPr(ptr);
}
罪魁祸首就在这里:
*p2 = B[0];
这等效于仅复制的第一个值B
并将其放入的第一个插槽p2
,即输出的第一行,第一列。您需要复制整个矩阵。
还要记住,MATLAB中的矩阵是列专业的,因此您需要了解索引到的方式p2
。但是,Armadillo指出,如果仅使用单个线性索引访问矩阵,则您将以列优先格式访问元素。因此,不要这样做,而是这样*p2 = B[0];
做:
for (int i = 0; i < B.n_elem; i++)
p2[i] = B[i];
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句