使用fftw3库进行dct

时间:2016-05-30 16:59:32

标签: c fftw dct

我正在使用离散余弦变换测试库。

#include <fftw3.h>
void dump_vector(int n, double* vec) {
    for(int i = 0; i < n; i++)
        printf("%f ", vec[i]);
    printf("\n");
}
int main()
{
    double a[] = {0.5, 0.6, 0.7, 0.8};
    double b[] = {0, 0, 0, 0};
    printf("Original vector\n");
    dump_vector(4, a);
    fftw_plan plan = fftw_plan_r2r_1d(4, a, a, FFTW_REDFT10, FFTW_ESTIMATE);
    fftw_execute(plan);
    printf("DCT\n");
    dump_vector(4, a);
    fftw_plan plani = fftw_plan_r2r_1d(4, a, a, FFTW_REDFT10, FFTW_ESTIMATE);
    fftw_execute(plani);
    printf("IDCT\n");
    dump_vector(4, a);
    return 0;
}

我希望获得相同的a,或者可能是近似值,但我的输出如下:

Original vector
0.500000 0.600000 0.700000 0.800000 
DCT
5.200000 -0.630864 0.000000 -0.044834 
IDCT
9.048603 9.208347 8.182682 5.179908

1 个答案:

答案 0 :(得分:2)

请参阅the documentation of fftw about real to real transforms

  

FFTW_REDFT10计算REDFT10变换,即DCT-II(有时称为“DCT”)。 (逻辑N = 2 * n,反向是FFTW_REDFT01。)

因此,标志FFTW_REDFT01必须用于逆变换而不是FFTW_REDFT10

此外,FFTW不会重新调整变换的输出。因此,输出必须由nn*n除以,其中n是向量的长度。 (我会在几分钟内测试一下......)

编辑:缩放系数既不是n也不是n*n2*n ...