c++ - 将 vector 传递给函数并返回 vector 并重铸

Question

我正在使用 MEX 将我的 C++ 代码与 MATLAB 连接起来。我的 C++ 代码要求输出的类型为 vector 。由于我是 C++ 的新手，所以我对指针的工作方式感到非常困惑。我将从 MATLAB 中获取一个输入数组

int *data_array_ptr
data_array_ptr=(int *)mxGetData(prhs[0]);
a = mxGetM(prhs[0]);
int int_data[a];
copy(*data_array_ptr, *data_array_ptr+ a, int_data);

现在，int_data 应该拥有存储在 data_array_ptr 位置的所有数据......它这样做了吗？

然后，

double *data_out_ptr;

plhs[0]= mxCreateDoubleMatrix( (mwSize)m, (mwSize)n, mxREAL); 
data_out_ptr= mxGetPr(plhs[0]);
len6=mxGetM(plhs[0]);
vector<double> data_out_data(*data_out_ptr,*data_out_ptr+len6);

这应该将空输出矩阵的内容放入一个名为 data_out_data 的 vector 中。它会这样做吗？

然后，我想将 data_out_data 和 int_data 都传递给 C++ 函数。但是，我想将 data_out_data 作为指针传递，以便 c++ 函数用数据填充 vector ，然后当函数完成时，MEX 函数将看到现在填充的 vector 并能够将其转换回 double 组可以填充plhs[0]。

所以，像

mexFunction(plhs[],prhs[]){

int *data_array_ptr
data_array_ptr=(int *)mxGetData(prhs[0]);
a = mxGetM(prhs[0]);
int int_data[a];
copy(*data_array_ptr, *data_array_ptr+ a, int_data);

 double *data_out_ptr;
plhs[0]= mxCreateDoubleMatrix( (mwSize)m, (mwSize)n, mxREAL); 
data_out_ptr= mxGetPr(plhs[0]);
len6=mxGetM(plhs[0]);
vector<double> data_out_data(*data_out_ptr,*data_out_ptr+len6);



foo(int_data, *data_out_data)

copy(data_out_data.begin(), data_out_data.end(), data_out_ptr);
}

并且在 foo 返回时，data_out_data 将被填充。我的函数没有返回参数，data_out_data 必须是 vector 类型。如何将 vector 传递给 foo，以便 foo 可以编辑数据？

谢谢!!

Answer 1

我不确定我是否正确理解了您的问题。我相信您想将一个数组从 MATLAB 传递给您的 mex 函数，然后 mex 函数调用一个对该数据数组进行操作的 C++ 函数，以及一个 vector。 . vector then 包含 C++ 函数执行的任何操作的结果，并且您希望将此数据传递回 MATLAB。

---

首先，让我们处理从 MATLAB 获取数据到您的 MEX 函数

int const *data_in = static_cast<int const *>(mxGetData(prhs[0]));

现在，data_in指向你传入的数据。顺便说一句，你确定数组包含int吗？秒？默认情况下，MATLAB 使用 double对于一切。

您的 C++ 函数是否要修改此数组？如果没有，你可以只用指针和元素的数量来调用它，而不是执行复制。

例如，如果 foo 的签名是

foo( int const *data_in, mwSize num_data_in, std::vector<double> *data_out );

你可以这样调用它

foo( data_in, mxGetNumberOfElements(prhs[0]), &data_out );

如果确实需要修改数据，和/或不能修改foo ，只需创建一个 vector 来保存数据的拷贝。

std::vector<int> data_in_vec( data_in, data_in + mxGetNumberOfElements(prhs[0]) );
foo( data_in_vec.data(), data_out );

---

至于foo函数，在调用它之前是否需要正确调整 vector 的大小？如果是这样，

std::vector<double> data_out( m * n );  // creates a vector with m * n elements
foo( data_in_vec.data(), &data_out );

如果可能，修改foo接受std::vector<double>&而不是 std::vector<double> * .然后你可以称它为

foo( data_in_vec.data(), data_out );

此外，如果有选择，我会选择 foo根据需要调整 vector 的大小，而不是要求调用者这样做。

---

现在，将数据返回给 MATLAB。

plhs[0] = mxCreateDoubleMatrix( m, n, mxReal );
std::copy( data_out.data(), data_out.data() + data_out.size(), mxGetPr(plhs[0]) );

上面一行假设 vector 的大小不大于m * n

请记住，MATLAB 以列优先格式存储矩阵，这与 C 和 C++ 不同。 foo 取决于如何函数有效，您可能必须转置返回的 vector ，在这种情况下您不能使用 std::copy要进行复制，您必须编写嵌套循环。