fkflt.hpp

/* $Id: fkflt.hpp 21344 2011-12-15 14:03:25Z leha $ Kolosov 20_03_2010*/
#ifndef __fkflt_hpp
#define __fkflt_hpp

#ifdef GE_BUILD
#include <wave_field.hpp>
#include <interpol.h>
#else
#include <s2proc/wave_field.hpp>
#include <mth/interpol.h>
#endif

class FKFlt
{
public:


  struct Params
  {
    // set default parameters    
    Params()
    {
      suppression = 0.05;
      vel1 = -1.0;
      vel2 = 1.0;
      taper = 0;
      curvature = 0;
      time_step = 0.0;
      rgFiltKind = 0;
      };

    int    rgFiltKind;             
    float  suppression; 
    float  vel1;        
    float  vel2;        
    int    taper;       
    int    curvature;      
    float  time_step;
    std::vector< float > hcab;    
  }; 
  
  FKFlt();
  
  FKFlt(const Params &params);
  
  virtual ~FKFlt();

  void setParams(const Params &params);
 
  Params getParams();

  bool calc_filter(WaveField &filter);
  
  bool calc(WaveField &wf);
    
  bool operator()(WaveField &wf) {return calc(wf);}
  
 protected:


  bool calcMain(WaveField &filter, bool filtergo );

  int CheckParam();

  void calcVelParam( float vel1, float vel2, bool &upv1, bool &upv2, bool &invertv1, bool &invertv2);
 
  bool rangeCos( float vel, float cur_k, float cur_w, float delta, bool upv);

  double calcFiltrCos( float cur_k, float cur_w, float delta, float vel, bool upv);

  void saveKoef( WaveField &filter, size_t st_k, size_t st_w, bool invertv, double stkoef, bool filtergo, bool upv);
  
private:

  Params my_filter;  

};


#endif /* __fkflt_hpp */