rws_selector.hpp

/* rws_selector.hpp */
/* $Id: rws_selector.hpp 21267 2011-11-23 15:19:39Z guser1 $ */
#ifndef __rws_selector_hpp
#define __rws_selector_hpp

#ifdef GE_BUILD
#include <rws_sim.hpp>
#else
#include <s2proc/rws_sim.hpp>
#endif

namespace RWSel
{
  class Selector
  {
  public:
    
    struct Parameters
    {
      bool weight_sel_apply;                //26.02.12

      size_t time_base_samples_num;        
      size_t time_base_shift_samples_num;  
      size_t slide_base_traces_num;        
      float Tb;                            
      Trace::TimeT ampl_est_win_b;         
      Trace::TimeT ampl_est_win_e;         
      bool win_sel_apply;                  
      Trace::TimeT winsel_b;               
      Trace::TimeT winsel_e;               
      bool fPolFil;                        
      Trace::TimeT defect_ampls_part;      
      Trace::TimeT energy_filter_rigidity; 
      bool apply_smoothing;                
      Trace::TimeT ampl_smothing_order;    
      bool fk_fil_apply;                     // 26.02.12
      int fkfil_type;                      
      Trace::TimeT fk_vel1, fk_vel2;       
      Trace::TimeT reject_level;           
      bool agc_apply;                      
      Trace::TimeT agc_gate_size;          
      size_t fk_slope_width;               
      size_t fk_slope_curv;                
    };

    struct MedFilter                    // 14.02.12
    {
      //int fMedFil;
      double *spFilOp;
      int ideg;
      int cntTr2;
      int cntSp;
    };

    struct TimeSelWinPar               // 18.02.12
    {
      int iLeftSlope_b;     
      int iLeftSlope_e;     
      int iRightSlope_b;    
      int iRightSlope_e;    
    };
    
    Selector(const Parameters &params);

    void FKfil(WaveField &iwf) const;

    void FKfilMute(WaveField &iwf, double tavg, std::vector<double> &ts,
           double tleft, double tright, size_t ihb, size_t ihe) const;
    
    WaveField getWave(const WaveField &iwf) const;
    
  protected:
    
    Parameters m_params;

    void Apply2DFFT(double **sp2Dm, int dir, int n1, int n2) const;

    //void FKFilWeights(double **weFil, int n1, int n2, float Dt, float dh,
    //        float df, float dk) const;   // 28.02.12
    void FKFilWeights(double **weFil, int n1, int n2,
              float df, float dk) const;   // 09.03.12
    // 09.03.12
  //void GetWeightPar(int fdim, int kdim, int js1, int js2, int jds, int *jb1, 
    void GetWeightPar(int fdim, int kdim, int js1, int js2, int *jb1, 
      int *je1, int *jb2, int *je2, int *lw1, int *lw2, int dir) const;

    float DownCos(int j,int je1,int lw1) const;
    float UpCos(int j,int je2,int lw2) const;

    //void get_wave_form(const WaveField &iwf, Trace &wform) const; 12.02.12
    void get_wave_form(const WaveField &iwf, Trace &wform,
               int ibw, int iew) const;
        
    Trace::AmplT get_ampl_coeff(const Trace &tr, const Trace &wform,
                Trace::TimeT win_b, Trace::TimeT win_e) const; 
    
    bool smooth_ampls(S2Corr::DVector &ampls, int pdeg) const;
    
    int get_smooth_poly_order(size_t n, float smth_order) const;
  
    void get_slide_base_ids(const WaveField &iwf, size_t slide_base_traces_num, size_t id,
                size_t &trb_id, size_t &tre_id) const;

    void GetTimeSelWinPar(const WaveField &iwf, TimeSelWinPar &tswinPar) const;

    int WeightSelTrWin(Trace &wform, TimeSelWinPar &tswinPar) const;

    int ConvTrSpOpTr(Trace &wform, MedFilter &medFil) const;
  };
  
};

#endif /* rws_selector.hpp */