fftwaff.h

Go to the documentation of this file.

// include guard
#ifndef TF_FFTWAFF_H_VERSION

#define TF_FFTWAFF_H_VERSION \
  "TF_FFTWAFF_H   V1.3"

#include<complex>
#ifdef FFTWFALLBACK
#include<drfftw.h>
#else
#include<fftw3.h>
#endif
#include<aff/series.h>

namespace fourier {

  namespace fft {

    class DRFFTWAFF {
      public:
        typedef double Tsample;
        typedef std::complex<Tsample> Tcoeff;
        typedef aff::Series<Tsample> Tseries;
        typedef aff::Series<Tcoeff> Tspectrum;
#ifdef FFTWFALLBACK
        DRFFTWAFF(const unsigned int& n):
          Msize(n), Mplan_forward(0), Mplan_backward(0) { }
        DRFFTWAFF():
          Msize(0), Mplan_forward(0), Mplan_backward(0) { }
#else
        DRFFTWAFF(const unsigned int& n, const bool& deletearrays=false):
          Msize(n), Mplan_forward(0), Mplan_backward(0),
          Mseriesarray(0), Mspectrumarray(0),
          Mdeletearrays(deletearrays) { }
        DRFFTWAFF(const bool& deletearrays=false):
          Msize(0), Mplan_forward(0), Mplan_backward(0),
          Mseriesarray(0), Mspectrumarray(0),
          Mdeletearrays(deletearrays) { }
#endif
        ~DRFFTWAFF();
        Tspectrum operator()(const Tseries::Tcoc& s,
                             const bool& debug=false) const;
        Tseries operator()(const Tspectrum::Tcoc& s,
                           const bool& debug=false) const;
        Tspectrum operator()(const Tseries::Tcoc& s,
                             const Tsample& dt,
                             const bool& debug=false) const;
        Tseries operator()(const Tspectrum::Tcoc& s,
                           const Tsample& dt,
                           const bool& debug=false) const;
        Tsample scale_series(const Tsample& dt) const;
        Tsample scale_spectrum(const Tsample& dt) const;
        unsigned int size() const { return(Msize); }
        void size(const unsigned int& s) const { this->set_size(s); }

        inline
        static unsigned int spectrumsize(const unsigned int& n) 
        { return(n/2+1); }

        inline
        static unsigned int seriessize(const unsigned int& n) 
        { return(n*2-1); }

      private:
        void create_plan_forward() const;
        void create_plan_backward() const;
        void delete_plans() const;
#ifndef FFTWFALLBACK
        void create_arrays() const;
        void delete_arrays() const;
        unsigned int ssize() const 
        { return(DRFFTWAFF::spectrumsize(this->size())); }
#endif
        void set_size(const unsigned int& n) const;
        mutable unsigned int Msize;
#ifdef FFTWFALLBACK
        mutable rfftw_plan Mplan_forward;
        mutable rfftw_plan Mplan_backward;
#else
        mutable fftw_plan Mplan_forward;
        mutable fftw_plan Mplan_backward;
        mutable double *Mseriesarray;
        mutable fftw_complex *Mspectrumarray;
        mutable Tspectrum Mspectrum;
        mutable Tseries Mseries;
        bool Mdeletearrays;
#endif
    }; // class DRFFTWAFF

  } // namespace ftt

} // namespace fourier

#endif // TF_FFTWAFF_H_VERSION (includeguard)

/* ----- END OF fftwaff.h ----- */