WPFFT.h

Go to the documentation of this file.

 
#pragma once
 
#include <boost/signals2.hpp>
#include <boost/bind.hpp>
#include <mutex>
 
// Forward class declarations
class ScanProcessor;
 
class WPFFT
{
public:
    enum class FFTType
    {
        Forward = 0, 
        Backward,
        ZeroPad,
        NoZeroPad,
        NumFFTTypes
    };
 
    WPFFT();
    WPFFT(ScanProcessor* scanProcessor);
    virtual ~WPFFT();
 
    // Getters
    static  size_t  GetForwardOutputHeight  ()      { return _outputHeightFFTR2C; }
    static  size_t  GetBackwardOutputHeight ()      { return _outputHeightFFTC2R; }
    static  size_t  GetC2COutputHeight      ()      { return _outputHeightFFTC2C; }
    static  size_t  GetC2CZPOutputHeight    ()      { return _outputHeightFFTC2CZP; }
 
    virtual void    InitStorage         (int width, int height, int depth);
    virtual void    ReleaseStorage      ();
 
    virtual void    InitPlans           (int width, int height, int depth);
    virtual void    DestroyPlans        ();
 
    virtual void    OnWidthHeightChanged    (int width, int height);
 
    virtual void    Compute             () { }
    //  virtual void    Compute             (FFTType type, void* inData = NULL) { }
 
protected:
    bool            _storageInitialized;
 
    ScanProcessor*  _scanProcessor;
 
    static  size_t          _outputHeightFFTR2C;
    static  size_t          _outputHeightFFTC2R;
    static  size_t          _outputHeightFFTC2C;
    static  size_t          _outputHeightFFTC2CZP;
    
    // for averaging, need 2D-FFT - not currently used
    static  size_t          _outputHeight2DFFTR2C;
    static  size_t          _outputHeight2DFFTC2R;
 
    // The width, height, and depth of the storage elements.
    int             _storageWidth;
    int             _storageHeight;
    int             _storageDepth;
 
    static  std::mutex      _mutexSetup;
    //static    std::mutex      _mutexPlan1;
    //static    std::mutex      _mutexPlan2;
    //static    std::mutex      _mutexPlan3;
    //static    std::mutex      _mutexPlan4;
    //static    std::mutex      _mutexPlan;
 
    // The width, height, and depth of the plan elements.
    int     _planWidth;
    int     _planHeight;
    int     _planDepth;
 
    static  bool    _plansInitialized;
    static bool     _frameworkSetup;
    static int      _numFFTObjects;
 
    static  void    OutputError     (int error, char *message);
 
    virtual void    AllocateFramework   () { }
    virtual void    DestroyFramework    () { }
 
private:
    boost::signals2::connection _widthHeightConnection;
 
    void    Init();
};