s2_traces_quality.hpp

/* s2_traces_quality.hpp */
/* $Id$ */
#ifndef __s2_traces_quality_hpp
#define __s2_traces_quality_hpp

#include <map>

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

class S2TracesQuality
{
public:

  struct QP
  {
    QP() : max_pos(0.), min_neg(0.) {/* Nothing to do. */}
    
    QP& operator/=(const int &val) {max_pos /= val; min_neg /= val; return *this;}
    QP& operator+=(const QP &that) {max_pos += that.max_pos; min_neg += that.min_neg; return *this;}
    
    Trace::AmplT max_pos; 
    Trace::AmplT min_neg; 
  };
  
  typedef std::map<int, std::map<int, std::map<PSC::Direction, QP> > > Quality;
  
  S2TracesQuality(const S2AbstractDBAccess &dba_if, 
          const Trace::TimeT &max_time_range, 
          const Trace::TimeT &start_time,
          const Trace::TimeT &time_length);
  
  ~S2TracesQuality();

  bool fill();

 
  const QP& getQP(int nsp, int nrp, PSC::Direction dir, bool *rc = 0x0) const;
  
  const Quality& getQuality() const {return m_quality;}
  
  bool getWeightsDistribution(PSC::Direction dir, std::map<int, Trace::AmplT>& weights) const;

  static bool calc_qp(const Trace &tr1, const Trace &tr2, 
              const Trace::TimeT &max_time_range, QP &qp);


protected:
  
  Quality m_quality;

  Trace::TimeT m_max_time_range;
  
  Trace::TimeT m_start_time;

  Trace::TimeT m_time_length;

  S2AbstractDBAccess *mp_dba_if;
  
private:  
  
  bool add_correlations(int comp, PSC::Direction dir);

  bool obtain_qp(int nsp, int nrp, PSC::Direction dir, const Trace &tr, QP &qp) const;
     
  QP merge_qps(const QP &qp1, bool rc1, const QP &qp2, bool rc2);

  static QP m_tmp_qp;
};

#endif /* s2_traces_quality.hpp */