Intel MKL FFT'nin nasıl kullanılacağı konusunda basit bir C++ örneği var mı?

Question

FFT ve Inverse-FFT dönüşümlerini gerçekleştirmem gerekiyor. Giriş, vektör ve çift matrisler olacaktır. İdeal olarak, çıktı bir std :: complex dizisi olmalıdır, ancak çift _Complex ile yaşayabilirim.

Basit bir örnek bulamadım, tüm Intel örnekleri bir sürü şeyi yeterince yorum yapmadan aynı anda yapıyorlar.

Sadece giriş olarak çift bir vektör (veya bir matris) alarak ve (ideal std :: kompleksi) FFT dönüştürülmüş sonucu çıkış C++ basit bir örnek istiyoruz.

Answer 1

Ben birçok şeyi test sona erdi ve nihayet ben ne istersem bu üç fonksiyon ile sona erdi ve ben basit örnekler olarak gördüğünü.

Bazı girdilere karşı test ettim ve iyi sonuçlar aldım. Yine de kapsamlı bir test yapmamıştım.

//Note after each operation status should be 0 on success 

std::vector<std::complex<float>> fft_complex(std::vector<std::complex<float>>& in){ 
    std::vector<std::complex<float>> out(in.size()); 

    DFTI_DESCRIPTOR_HANDLE descriptor; 
    MKL_LONG status; 

    status = DftiCreateDescriptor(&descriptor, DFTI_SINGLE, DFTI_COMPLEX, 1, in.size()); //Specify size and precision 
    status = DftiSetValue(descriptor, DFTI_PLACEMENT, DFTI_NOT_INPLACE); //Out of place FFT 
    status = DftiCommitDescriptor(descriptor); //Finalize the descriptor 
    status = DftiComputeForward(descriptor, in.data(), out.data()); //Compute the Forward FFT 
    status = DftiFreeDescriptor(&descriptor); //Free the descriptor 

    return out; 
} 

std::vector<std::complex<float>> fft_real(std::vector<float>& in_real){ 
    std::vector<std::complex<float>> in(in_real.size()); 

    std::copy(in_real.begin(), in_real.end(), in.begin()); 

    return fft_complex(in); 
} 

std::vector<float> ifft(std::vector<std::complex<float>>& in){ 
    std::vector<std::complex<float>> out(in.size()); 

    DFTI_DESCRIPTOR_HANDLE descriptor; 
    MKL_LONG status; 

    status = DftiCreateDescriptor(&descriptor, DFTI_SINGLE, DFTI_COMPLEX, 1, in.size()); //Specify size and precision 
    status = DftiSetValue(descriptor, DFTI_PLACEMENT, DFTI_NOT_INPLACE); //Out of place FFT 
    status = DftiSetValue(descriptor, DFTI_BACKWARD_SCALE, 1.0f/in.size()); //Scale down the output 
    status = DftiCommitDescriptor(descriptor); //Finalize the descriptor 
    status = DftiComputeBackward(descriptor, in.data(), out.data()); //Compute the Forward FFT 
    status = DftiFreeDescriptor(&descriptor); //Free the descriptor 

    std::vector<float> output(out.size()); 

    for(std::size_t i = 0; i < out.size(); ++i){ 
     output[i] = out[i].real(); 
    } 

    return output; 
}