我有一个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 ++不好,这是正确的方法吗?