PolyMatrix.hpp

/* PolyMatrix.hpp */
/* $Id: PolyMatrix.hpp 21638 2012-06-01 12:07:34Z hoot $ */
#ifndef __PolyMatrix_hpp
#define __PolyMatrix_hpp

#include <vector>
#include <set>

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

//===================================================================================================
//=========================* Abstract interface for SP/RP based matrix construction *================
//===================================================================================================


class PolyMatrix
{
public:
  
  PolyMatrix();
  virtual ~PolyMatrix();

  virtual bool SetMatrixElement(size_t i, size_t j, double& val) = 0;
  virtual bool SetVectorElement(size_t i, double& val) = 0;
  virtual bool IsReady() = 0;
  virtual size_t GetPointsCount(std::vector<int> &ID_num) = 0;
};


//===================================================================================================
//======================* Class for universal based matrix construction *============================
//===================================================================================================

class PolyMatrixPSC : public PolyMatrix
{
public:
  
  PolyMatrixPSC(const std::map<int, std::multimap<int, Trace> > &sum_rows, float max_shift);
  virtual ~PolyMatrixPSC();

  bool SetMatrixElement(size_t i, size_t j, double& val);
  bool SetVectorElement(size_t i, double& val);

  bool IsReady();
  size_t GetPointsCount(std::vector<int> &ID_num);

protected:
  
  bool set_ID_Row();
  std::set<int> m_point_set;
  std::map<int, std::multimap<int, Trace> > m_sum_rows;
  float m_max_shift;

private:
  bool m_ready;

};

#endif /* PolyMatrix.hpp */