1

我有一个用傅立叶变换实现的数值计算问题。我正在使用 fftw 来实现 FFT,但被 FFT 处理的信号非常奇怪。一半的信号来自一个阵列,另一半来自另一个阵列。在当前代码中,我只是复制数据

#include <iostream>
#include <vector>
#include <complex>
#include <fftw3.h>

using namespace std;

void main(void)
{
  std::vector< std::complex<double> > V1(4);
  std::vector< std::complex<double> > V2(4);
  std::vector< std::complex<double> > V(8);

  for (int i=0; i<4; i++) V[i] = V1[i] = std::complex<double>(i, i);
  for (int i=0; i<4; i++) V[i+4] = V2[i] = std::complex<double>(i+4, i+4);

  fftw_complex* in = reinterpret_cast<fftw_complex*>(&V[0]);
  fftw_plan plan = fftw_plan_dft_1d(8, in, in, FFTW_FORWARD, FFTW_MEASURE);
  fftw_execute(plan);
}

这段代码工作得很好。但我想让大数组引用 V1 和 V2 而不是复制数据。我尝试以下代码但它不起作用,它说输入数组是**(双指针?)

std::vector< std::complex<double> > V1(4);
std::vector< std::complex<double> > V2(4);
fftw_complex *in[8];
for (int i=0; i<4; i++) in[i] = reinterpret_cast<fftw_complex*>(&V1[i]);
for (int i=0; i<4; i++) in[i+4] = reinterpret_cast<fftw_complex*>(&V2[i]);

for (int i=0; i<4; i++) V1[i] = std::complex<double>(i, i);
for (int i=0; i<4; i++) V2[i] = std::complex<double>(i+4, i+4);

fftw_plan plan = fftw_plan_dft_1d(8, in, in, FFTW_FORWARD, FFTW_MEASURE);
fftw_execute(plan);

无论如何都要将“参考”数组作为fftw的输入吗?二维数组会发生什么,是否可以让 2d-fftw 的输入数组引用两个单独的二维数组?

4

1 回答 1

1

如果我理解正确,您需要两个单独的矢量对象:

vector<fftw_complex> V1, V2;

并将它们各自的内容组合成另一个

vector<fftw_complex> V;

没有任何复制。V1这是不可能的,因为和一起占用的内存V2不是连续的。V但是, (对于 any )占用的内存vector必须是连续的。如果您想避免在给出的示例中进行复制,可以使用指针来完成,如下所示:

void main(void)
{
    std::complex<double>* V1;
    std::complex<double>* V2;
    std::vector< std::complex<double> > V(8);

    V1=&V[0];V2=&V[4];

    for (int i=0; i<4; i++) V1[i] = std::complex<double>(i, i);
    for (int i=0; i<4; i++) V2[i] = std::complex<double>(i+4, i+4);

    fftw_complex* in = reinterpret_cast<fftw_complex*>(&V[0]);
    fftw_plan plan = fftw_plan_dft_1d(8, in, in, FFTW_FORWARD, FFTW_MEASURE);
    fftw_execute(plan);
 }

当然,这在您的程序中可能可行,也可能不可行。您的第二个代码段有一个主要问题:

fftw_complex *in[8];

定义了一个由 8 个指向复数的指针组成的数组,而不是一个由 8 个复数组成的数组,因此它不适合作为输入fftw_plan_dft_1d,即使这样做,结果也没有意义,因为您正在向它提供一系列内存地址而不是您的数据。

于 2013-09-19T03:47:33.310 回答