2

我实现了一个 fftw (fftw.org) 示例来使用快速傅里叶变换......这是代码......

我加载了一个从 uint8_t 转换为 double 的图像（此代码工作正常......）。

string bmpFileNameImage = "files/testDummyFFTWWithWisdom/onechannel_image.bmp"; 
BMPImage bmpImage(bmpFileNameImage);
vector<double>pixelColors; 
vector<uint8_t> image = bmpImage.copyBits();

toDouble(image,pixelColors,256,256, 1); 
int width = bmpImage.width();
int height = bmpImage.height();

我使用智慧文件来提高性能

FILE * file = fopen("wisdom.fftw", "r");
if (file) {
    fftw_import_wisdom_from_file(file);
    fclose(file);
} 

///*  fftw variables  */
fftw_complex *out;

double *wisdomInput = (double *) fftw_malloc(sizeof(double)*width*2*(height/2 +1 ));

const fftw_plan forward =fftw_plan_dft_r2c_2d(width,height, wisdomInput,reinterpret_cast<fftw_complex *>(wisdomInput),FFTW_PATIENT);
const fftw_plan inverse = fftw_plan_dft_c2r_2d(width, height,reinterpret_cast<fftw_complex *>(wisdomInput),wisdomInput, FFTW_PATIENT);

file = fopen("wisdom.fftw", "w");

if (file) {
    fftw_export_wisdom_to_file(file);
    fclose(file);
}

最后，我执行了 fftw 库......我收到第一个函数（）的访问冲突错误，fftw_execute_dft_r2c我不知道为什么......我阅读了本教程： http ://www.fftw.org/fftw3_doc/ Multi_002dDimensional-DFTs-of-Real-Data.html#Multi_002dDimensional-DFTs-of-Real-Data。我用 (ny/2+1) 做了一个 malloc，它是如何解释的...... 我不明白为什么它不起作用......我正在测试不同的尺寸......

out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * width *(height / 2 + 1));
double *result =(double *)fftw_malloc(width * (height+2)  * sizeof(double));

fftw_execute_dft_r2c(forward,&pixelColors[0],out);
fftw_execute_dft_c2r(inverse,out,result);

问候。

4

3 回答 3

1

这是更正后的代码。它有几个错误：

  • 它正在读取错误的wise.fftw 文件（来自一些旧测试......）。现在，它总是创建一个新的 fftw_plan 和一个新文件。
  • 我误解了 fftw 库是如何使用就地和就地参数工作的。我必须更改 malloc 以获得“就地”的正确填充（我在 malloc 函数中添加了 +2）。
  • 为了恢复图像，我必须除以它的大小 ((width+2) * height) 它是如何在这个链接中解释的。

`

/* load image */
string bmpFileNameImage = "files/polyp.bmp";

BMPImage bmpImage(bmpFileNameImage);

int width = bmpImage.width();
int height = bmpImage.height();

vector<double> pixelColors;
vector<uint8_t> image = bmpImage.copyBits();
//get one channel from the image
Uint8ToDouble(image,pixelColors,bmpImage.width(),bmpImage.height(),1);

//We don't reuse old wisdom.fftw... It can be corrupt 
/*
FILE * file = fopen("wisdom.fftw", "r");
if (file) {
    fftw_import_wisdom_from_file(file);
    fclose(file);
} */

double *wisdomInput = (double *) fftw_malloc(sizeof(double)*height*(width+2));

const fftw_plan forward =fftw_plan_dft_r2c_2d(width,height,wisdomInput,reinterpret_cast<fftw_complex *>(wisdomInput),FFTW_PATIENT);
const fftw_plan inverse = fftw_plan_dft_c2r_2d(width,height,reinterpret_cast<fftw_complex *>(wisdomInput),wisdomInput, FFTW_PATIENT);
double *bitsColors =(double *)fftw_malloc((width) * height * sizeof(double));

for (int y = 0; y < height; y++) {
    for (int x = 0; x < width+2; x++) {
        if (x < width) {
            int currentIndex = ((y * width) + (x));
            bitsColors[currentIndex] = (static_cast<double>(result[y * (width+2) + x])) / (height*width);
        }
    }
}
fftw_free (wisdomInput);
fftw_free (out);
fftw_free (result);
fftw_free (bitsColors);

fftw_destroy_plan(forward);
fftw_destroy_plan(inverse);
fftw_cleanup();
}

`

于 2013-06-26T16:48:18.220 回答
0 
fftw_execute_dft_r2c(forward,&pixelColors[0],out);

你在这里做什么 ？数组已经有一个指针。将其更改为fftw_execute_dft_r2c(forward,pixelColors[0],out);现在应该可以使用。

于 2013-06-25T09:43:04.023 回答
0

也许问题在这里（http://www.fftw.org/doc/New_002darray-Execute-Functions.html）：

[...] 满足以下条件：

  • 如果计划最初被创建为就地或异地，则输入和输出数组是相同的（就地）或不同的（异地）。

在计划中，您正在使用就地转换参数（分配错误，顺便说一句，因为：

double *wisdomInput = (double *) fftw_malloc(sizeof(double)*width*2*(height/2 +1 ));

应该：

double *wisdomInput = (double *) fftw_malloc(sizeof(fftw_complex)*width*2*(height/2 +1 ));

也适合输出）。

但是您正在使用不合适fftw_execute_dft_r2c的参数调用函数。

于 2013-06-25T11:04:31.650 回答