您可以编写一些使用C
兼容类型调用函数的桥接C++
函数。这是未经测试的代码,只是为了让您了解您可以做什么:
namespace Fft {
void transform(std::vector<double> &real, std::vector<double> &imag);
}
extern "C" {
struct complex_vector
{
void* real;
void* imag;
};
complex_vector create_complex_vector(size_t size)
{
complex_vector cv;
cv.real = new std::vector<double>(size);
cv.imag = new std::vector<double>(size);
return cv;
}
void complex_vector_push_back(complex_vector cv, double real, double imag)
{
reinterpret_cast<std::vector<double>*>(cv.real)->push_back(real);
reinterpret_cast<std::vector<double>*>(cv.imag)->push_back(imag);
}
void complex_vector_reserve(complex_vector cv, size_t size)
{
reinterpret_cast<std::vector<double>*>(cv.real)->reserve(size);
reinterpret_cast<std::vector<double>*>(cv.imag)->reserve(size);
}
void complex_vector_resize(complex_vector cv, size_t size)
{
reinterpret_cast<std::vector<double>*>(cv.real)->resize(size);
reinterpret_cast<std::vector<double>*>(cv.imag)->resize(size);
}
void fill_complex_vector(complex_vector cv, double* real, double* imag)
{
auto v_real = reinterpret_cast<std::vector<double>*>(cv.real)->data();
auto v_imag = reinterpret_cast<std::vector<double>*>(cv.imag)->data();
auto v_size = reinterpret_cast<std::vector<double>*>(cv.real)->size();
std::copy(real, real + v_size, v_real);
std::copy(imag, imag + v_size, v_imag);
}
void fft_transform(complex_vector cv)
{
Fft::transform(*reinterpret_cast<std::vector<double>*>(cv.real), *reinterpret_cast<std::vector<double>*>(cv.imag));
}
double* complex_vector_real_array(complex_vector cv)
{
return reinterpret_cast<std::vector<double>*>(cv.real)->data();
}
double* complex_vector_imag_array(complex_vector cv)
{
return reinterpret_cast<std::vector<double>*>(cv.imag)->data();
}
size_t complex_vector_size(complex_vector cv)
{
return reinterpret_cast<std::vector<double>*>(cv.imag)->size();
}
void destroy_complex_vector(complex_vector cv)
{
delete reinterpret_cast<std::vector<double>*>(cv.real);
delete reinterpret_cast<std::vector<double>*>(cv.imag);
}
} // extern "C"
如果您将其编译为C++
块extern "C" {}
将使其生成,那么您可以从C
.
您可以编写一个C
有点像这样的程序,例如:
complex_vector cv = create_complex_vector(1024);
// fill the vector
for(int i = 0; i < complex_vector_size(cv); ++i)
complex_vector_push_back(cv, 0.2 * i, 0.4 * i);
// call the transform
fft_transform(cv);
double* real = complex_vector_real_array(cv);
double* imag = complex_vector_imag_array(cv);
size_t size = complex_vector_size(cv);
// use your transformed data here ...
destroy_complex_vector(cv);
注意:完全未经测试的代码。