如何在Matlab mex文件中输出整数向量?

时间:2018-06-27 09:07:44

标签: c++ matlab vector mex

我有一个c ++代码,该代码生成一些整数序列。我需要为matlab编写一个mex文件,这将为我提供输出。

#include "mex.h"
#include <vector>
#include <random>


void
mexFunction(int nlhs,mxArray *plhs[],int nrhs,const mxArray *prhs[])
{
double *Input;
mxArray *mxInput,*mxOutput;
//associate inputs
mxInput = mxDuplicateArray(prhs[0]); 
Input=mxGetPr(mxInput);

// number of Steps
int NumOfSteps = (int) Input[0];
// collect diffs    
std::vector <int> result_vector;

// Initialize random generator
std::random_device rnd_device;
std::mt19937_64 rg(rnd_device());
std::normal_distribution<double> Xi(0,1);

// iterate

for (int iStep=0; iStep<NumOfSteps;iStep++) 
{
    if ((Xi(rg)-Xi(rg)) >= 0.5)
    {
       result_vector.push_back(iStep);
    }
}
// Here i should copy result_vector to output

}

我有一个想法,如何循序渐进,逐一应对结果,但我希望应该有一个更好的解决方案。

您能给我一些建议,如何在我的mex文件中将result_vector作为输出吗?

更新:

我找到了以下解决方案:

int vecN = result_vector.size();
mxOutput = plhs[0] = mxCreateDoubleMatrix(1,vecN, mxREAL);    
Output=mxGetPr(mxOutput);
std::copy(result_vector.begin(),result_vector.end(),Output);

我真的对C ++不好,这是正确的方法吗?

0 个答案:

没有答案