fftwaffar.h

Go to the documentation of this file.

// include guard
#ifndef TF_FFTWAFFAR_H_VERSION

#define TF_FFTWAFFAR_H_VERSION \
  "TF_FFTWAFFAR_H   V1.2"

#include<complex>
#include<fftw3.h>
#include<aff/array.h>
#include<aff/series.h>

namespace fourier {

  namespace fft {

    class DRFFTWAFFArrayEngine {
      public:
        typedef double Tsample;
        typedef std::complex<Tsample> Tcoeff;
        typedef aff::Array<Tsample> TAseries;
        typedef aff::Array<Tcoeff> TAspectrum;
        DRFFTWAFFArrayEngine(const int& nseis,
                             const int& nsamp);
        DRFFTWAFFArrayEngine(const TAseries& series,
                             const TAspectrum& spec);
        DRFFTWAFFArrayEngine(const DRFFTWAFFArrayEngine& engine);
        DRFFTWAFFArrayEngine();
        DRFFTWAFFArrayEngine& operator=(const DRFFTWAFFArrayEngine& e);
        ~DRFFTWAFFArrayEngine();
        void r2c();
        void c2r();
        Tsample scale_series(const Tsample& dt) const;
        Tsample scale_spectrum(const Tsample& dt) const;
        TAseries series() const { return Mseriesarray; }
        TAspectrum spectrum() const { return Mspectrumarray; }
        unsigned int nseries() const;
        unsigned int nsamples() const;
        unsigned int nfrequencies() const;
        TAseries series(const unsigned int& i) const;
        TAspectrum spectrum(const unsigned int& i) const;

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

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

      private:
        void checkconsistency();
        void createplanr2c();
        void createplanc2r();
        void delete_plans();
        TAseries Mseriesarray;
        TAspectrum Mspectrumarray;
        fftw_plan Mplanr2c;
        fftw_plan Mplanc2r;
    }; // class DRFFTWAFFArrayEngine

  } // namespace fft

} // namespace fourier

#endif // TF_FFTWAFFAR_H_VERSION (includeguard)

/* ----- END OF fftwaffar.h ----- */