polarity_wave_detector.hpp

/* polarity_wave_detector.hpp */
/* $Id$ */
#ifndef __polarity_wave_detector_hpp
#define __polarity_wave_detector_hpp

#include <lapackpp/laslv.h>
#include <s2proc/abstract_wave_detector.hpp>

class PolarityWaveDetector : public AbstractWaveDetector
{
public:
  
  PolarityWaveDetector(size_t time_base, size_t sb_size, size_t sb_step = 1);

  virtual ~PolarityWaveDetector();
  
  bool appendTracePointer(const Trace &tr) const;
  
  bool setTFB(size_t app_tfb);
  
  void clear();
  
  static float calcRatio(const float& E1, const float& E2);
  
protected:
  
  virtual bool calc(Trace &tr) const;
  
  typedef struct PolarityParams {
    double R1;
    double R2;
    PolarityParams() : R1(0.), R2(0.) {}
    PolarityParams(const double& r1, const double& r2) : R1(r1), R2(r2) {}
  };
  typedef std::vector<PolarityParams> PolarityParamsAr;
  
  static void interpolatePolParams(const size_t&, const size_t&, 
                   const PolarityWaveDetector::PolarityParams&, 
                   const PolarityWaveDetector::PolarityParams&, 
                   std::map<size_t, PolarityWaveDetector::PolarityParams>&);
  
  bool formDetectorData(Trace &tr) const;
  
  bool formParamsAr(PolarityParamsAr&) const;
  
  bool getWinPolarityParams(PolarityWaveDetector::PolarityParams& params, const size_t& s0, const size_t& s1) const;
  
  bool getPolarityParams(PolarityWaveDetector::PolarityParams& params) const;
  
  bool getMatrix(LaGenMatDouble& matrix) const;
  
  double getMatrixItem(size_t tr1, size_t tr2) const;
  
  double getAvValue(size_t tr) const;
  
  
  mutable std::vector<const Trace*> m_traces; 
  size_t m_tfb;                               
  size_t m_time_base;                         
  size_t m_sbase;                             
  size_t m_sb_step;                           
  bool m_ready;                               
};


#endif /* polarity_wave_detector.hpp */