我正在使用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读取数据的错误,但我无法发现它!
答案 0 :(得分:1)
你好像假设连续的malloc调用是连续的,当然它们不太可能(你可能只是#34;幸运的是#OS;在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]);