ws_selector.hpp

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

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

namespace WSel
{
  class Selector
  {
  public:
    
    struct Parameters
    {
      size_t time_base_samples_num;              
      size_t time_base_shift_samples_num;        
      size_t slide_base_traces_num;              
      Trace::TimeT ampl_est_win_b;               
      Trace::TimeT ampl_est_win_e;               
      Trace::TimeT defect_ampls_part;            
      Trace::TimeT energy_filter_rigidity;       
      bool apply_smoothing;                      
      Trace::TimeT ampl_smothing_order;          
    };
    
    Selector(const Parameters &params);
    
    WaveField getWave(const WaveField &iwf) const;
    
  protected:
    
    Parameters m_params;
    
    void get_wave_form(const WaveField &iwf, Trace &wform) 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;
  };
  
};

#endif /* ws_selector.hpp */