cdft2d_fftw.hpp

/* cdft2d.hpp */
/* $Id: cdft2d_fftw.hpp 21438 2012-01-19 12:37:29Z leha $ Kolosov  07_02_2009*/
#ifndef __cdft2d_hpp
#define __cdft2d_hpp

#ifdef GE_BUILD
#include <cdft2d.hpp>
#include <wave_field.hpp>
#include <fftw3.h>
#else
#include <s2proc/cdft2d.hpp>
#include <s2proc/wave_field.hpp>
#include <fftw3.h>
#endif


void calcAruwf(std::vector< std::vector<float> > &data,
           std::vector< std::vector<float> > &w2DARU,
           float step_time, float agc_gap);


void calcAruwf(std::vector< std::vector<float> > &in_data,
           std::vector< std::vector<double> > &out_data,
           std::vector< std::vector<float> > &w2DARU,
           float step_time, float agc_gap);


void calcAruwf(WaveField &in_data,
           std::vector< std::vector<float> > &out_data,
           std::vector< std::vector<float> > &w2DARU,
           float step_time, float agc_gap);


void calcAruLight(WaveField &in_data,
           std::vector< std::vector<float> > &out_data,
           std::vector< std::vector<float> > &w2DARU,
          float agc_gap);

class CDFT2D_FFTW : public CDFT2D
{
public:
  
  CDFT2D_FFTW(int direction = 1);  
  
  virtual ~CDFT2D_FFTW();
  
  bool operator ()(WaveField &tReal, WaveField &tIm) const 
  {return calc(tReal, tIm);}
  
  virtual bool calc( WaveField &tReal, WaveField &tIm ) const; 
  
  virtual bool calc( Trace &tReal, Trace &tIm ) const; 
  
  void setDirection(int direction) {m_direction = direction;}
  
  int getDirection() const {return m_direction;}

  void setRotateFlag(bool rotate) {m_brotate = rotate;}
  
  bool getRotateFlag() const {return m_brotate;}
  
protected:
  
  void rotate( WaveField &matr ) const; 

  void rotate( Trace &vec ) const; 
  
  void norm( WaveField &matr ) const;

  void norm( Trace &vector ) const;

  void set_proper_freq(Trace &tr) const;
  
  void set_proper_freq(WaveField &field) const;
  
  virtual void create_1d_plan(int N0, fftw_plan &plan, fftw_complex *in, fftw_complex *out) const;
  
  virtual void create_2d_plan(int N0, int N1, fftw_plan &plan, fftw_complex *in, fftw_complex *out) const;
  
  virtual void destroy_plan(fftw_plan &plan) const;
  
private:

  bool m_brotate;

};





#endif /* cdft2d.hpp */