C语言 FFTW 计划分段错误

Question

我正在使用FFTW3对多列数据（即多声道音频，我希望每个声道进行转换）执行fft。这在 OSX 上工作正常，但将代码移植到 linux 会给我一个 seg 错误。

const int fftwFlags = FFTW_PRESERVE_INPUT|FFTW_PATIENT;
struct fft {
    fftw_complex **complexSig;
    double **realSig;
    fftw_plan forwardR2C;
    int fftLen;
    int numChan;
}
void create FFT(struct fft *fft) {
int bufLen = 1024;
int numChan = 4;
fft->fftLen = bufLen;
fft->numChan = numChan;
fft->realSig = fftw_malloc(sizeof(double *) * numChan);
for(int i = 0; i < numChan; i++) {
    fft->realSig[i] = fftw_malloc(sizeof(double) * bufLen);
}
fft->complexSig = fftw_malloc(sizeof(fftw_complex *) * numChan);
for(int i = 0; i < numChan; i++) {
    fft->complexSig[i] = fftw_malloc(sizeof(fftw_complex) * bufLen);
}
fft->forwardR2C = fftw_plan_many_dft_r2c(1, &fft->fftLen, fft->numChan, *fft->realSig, &fft->fftLen, 1, fft->fftLen, *fft->complexSig, &fft->fftLen, 1, fft->fftLen, fftwFlags);
}

Valgrind 显示 FFTW 计划器正在尝试访问此数组的末尾（8 个字节，一个样本），从而导致分段错误。当将分配给 realSig 的内存量增加到 bufLen * 2 时，不存在此错误。

我确信这是我告诉 FFTW 读取我的数据的方式的错误，但我无法发现它！

Answer 1

您似乎假设连续的malloc调用将是连续的，当然它们不太可能（您可能只是在OS X上"幸运"）。您可以通过进行一次大分配来轻松解决此问题，例如

void createFFT(struct fft *fft)
{
    const int bufLen = 1024;
    const int numChan = 4;
    fft->fftLen = bufLen;
    fft->numChan = numChan;
    fft->realSig = fftw_malloc(sizeof(double *) * numChan);
                                     // array of numChan pointers
    fft->realSig[0] = fftw_malloc(sizeof(double) * numChan  * bufLen);
                                     // one large contiguous block of size `numChan * bufLen`
    for(int i = 1; i < numChan; i++) // init pointers
    {
        fft->realSig[i] = fft->realSig[i - 1] + bufLen;
    }
    // ...
}

注意：完成后，您只需要：

fftw_free(fft->realSig[0]);