fftimplementation.h

Go to the documentation of this file.

/*****************************************************************************
 * Copyright 2015 Haye Hinrichsen, Christoph Wick
 *
 * This file is part of Entropy Piano Tuner.
 *
 * Entropy Piano Tuner is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 *
 * Entropy Piano Tuner is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * Entropy Piano Tuner. If not, see http://www.gnu.org/licenses/.
 *****************************************************************************/

//=============================================================================
//           FFTW3 implementation for fast Fourier transformations
//=============================================================================

#ifndef FFT_IMPLEMENTATION_H
#define FFT_IMPLEMENTATION_H

#include "fftadapter.h"


#include <fftw3.h>
#include <mutex>

#include "../system/prerequisites.h"


class FFT_Implementation : public FFTAdapter
{
public:

    FFT_Implementation();
    ~FFT_Implementation();

    void calculateFFT  (const FFTRealVector &in, FFTComplexVector &out);
    void calculateFFT  (const FFTComplexVector &in, FFTRealVector &out);

    void optimize (FFTRealVector &in);
    void optimize (FFTComplexVector &in);


private:

    // R and C mean: real and complex
    // CR means: complex to real
    // RC means: real to complex

    double       *mRvec1;               
    double       *mRvec2;               
    fftw_complex *mCvec1;               
    fftw_complex *mCvec2;               
    size_t       mNRC;                  
    size_t       mNCR;                  

    fftw_plan mPlanRC;                  
    fftw_plan mPlanCR;                  
    static std::mutex mPlanMutex;       

    void updatePlan (const FFTRealVector &in, unsigned flags);
    void updatePlan (const FFTComplexVector &in, unsigned flags);
};

#endif // FFT_IMPLEMENTATION_H