MEX:如何将矩阵从C ++ / C返回到MATLAB

加里玛·辛格(Garima Singh)

我可以返回的矩阵只有第一个元素正确,所有其他元素均为零。我尝试按照此链接中的建议进行操作但是,它不适合我。一些帮助,将不胜感激。

输出:

>> 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] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章