s2_traces_quality_3d.hpp

/* s2_traces_quality_3d.hpp */
/* $Id$ */
#ifndef __s2_traces_quality_3d_hpp
#define __s2_traces_quality_3d_hpp

#include <map>

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

class S2TracesQuality3D
{
public:
   
  typedef 
  std::map<int, // TA_NSL 
       std::map<int,  // TA_NSS
            std::map<int, // TA_NRL
                 std::map<int, // TA_NRS 
                      std::map<PSC::Direction, S2TracesQuality::QP> 
                 > 
            > 
       > 
  > Quality;
  
  S2TracesQuality3D(const S2AbstractDBAccess &dba_if, const Trace::TimeT &max_time_range, int stride);

  ~S2TracesQuality3D();
  
  
  bool append(const Trace &tr);

  
  const S2TracesQuality::QP& getQP(int nsl, int nss, int nrl, int nrs, PSC::Direction dir, bool *rc = 0x0) const;
  
  
  const Quality& getQuality() const {return m_quality;}


  bool getWeightsDistribution(PSC::Direction avg_dir, PSC::Direction shift_dir, 
                  std::multimap<Trace::AmplT,std::pair<int, int> > &weights) const;

  
protected:
  
  Quality m_quality;

  Trace::TimeT m_max_time_range;
  
  int m_stride;
    
  S2AbstractDBAccess *mp_dba_if;
  
private:
  
  bool obtain_qp(int nsl, int nss, int nrl, int nrs, PSC::Direction dir, 
         const Trace &tr, S2TracesQuality::QP &qp);
  
  static const S2TracesQuality::QP m_default_qp;
};

#endif /* s2_traces_quality_3d.hpp */