Czy jest jakiś prosty przykład C++ na temat korzystania z Intel MKL FFT?

Question

Potrzebuję wykonać transformacje FFT i Inverse-FFT. Wejściem byłby wektor i macierze podwójne. Idealnie, wyjście powinno być tablicą std :: complex, ale mogę żyć z podwójnym _Complex.

Nie znalazłem żadnego prostego przykładu, wszystkie przykłady Intela robią wiele rzeczy na raz bez wystarczającej ilości komentarzy.

Po prostu chcę prosty przykład w C++ biorąc wektor (lub macierz) podwójne jako wejście i wyprowadzając wynik transformacji FFT (najlepiej ze std :: complex).

Answer 1

Skończyło się na testowaniu kilku rzeczy i ostatecznie skończyłem z tymi trzema funkcjami, które robią to, co chcę i że rozważałem proste przykłady.

Przetestowałem to na niektórych wejściach i miałem dobre wyniki. Nie przeprowadziłem jednak obszernych testów.

//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; 
}