s2_ldb_access.hpp

/* s2_ldb_access.hpp */
/* $Id$ */
#ifndef __s2_ldb_access_hpp
#define __s2_ldb_access_hpp

#include <s2proc/s2_abstract_db_access.hpp>
#include <mix/trselect.h>

class S2LDBAccess : public S2AbstractDBAccess
{
public:
  
  S2LDBAccess(TrSet &trset);
  
  virtual ~S2LDBAccess();
  
  virtual S2AbstractDBAccess* clone() const {return new S2LDBAccess(*this);}
  
  
  virtual int getSamplesNum() const;
  
  virtual Trace::TimeT getTimeStep() const;
  
  
  virtual int getTracesNum() const;
  
  virtual bool getTrace(int id, Trace &tr) const;
  
  
  virtual bool getWaveField(const Trace::Header::AttrId &attr, 
                int id, 
                WaveField &wf) const;
  
  virtual bool getWaveField(PointType ptype, 
                int id1, 
                int id2, 
                WaveField &wf) const;
  
  virtual bool getWaveField(PointType3 ptype, 
                int id1, 
                int id2, 
                int id3,
                WaveField &wf) const;
  
  
  virtual bool getWaveField(int nsl, int nss, int nrl, int nrs, WaveField &wf) const;
  
  
  virtual bool getShotsIds(std::set<int> &ids) const;
  
  virtual bool getRecvsIds(std::set<int> &ids) const;
  
  virtual bool getCMPsIds(std::set<int> &ids) const;

  virtual bool getLIds(std::set<int> &ids) const;
  
  virtual bool getCompsIds(std::set<int> &ids) const;
    
  virtual bool getShotLinesIds(std::set<int> &ids) const;
  
  virtual bool getShotsOnLineIds(int line_id, std::set<int> &ids) const;
  
  virtual bool getRecvLinesIds(std::set<int> &ids) const;
  
  virtual bool getRecvsOnLineIds(int line_id, std::set<int> &ids) const;
  
  virtual bool getPSCStrides(std::set<int> &strides) const;
  
  virtual bool getPSCDirs(std::set<int> &dirs) const;
  

  virtual bool appendTrace(const Trace &tr);
  
protected:
  
  TrSet *m_ptrset;

private:
  
  struct P2D
  {
    P2D(int i_id1, int i_id2) : id1(i_id1), id2(i_id2) {}
    
    int id1;
    int id2;
    
    bool operator<(const P2D &other) const;
    bool operator==(const P2D &other) const;
  };
  
  typedef std::multimap<P2D, int> P2DMap;
  P2DMap m_p2d_sp_rp_map;
  P2DMap m_p2d_sp_dp_map;
  P2DMap m_p2d_rp_dp_map;
  P2DMap m_p2d_sp_L_map;
  P2DMap m_p2d_rp_L_map;
  
  void make_p2d_map(TrSet &trset, PointType ptype, P2DMap &p2d_map);
  
  bool get_seismogram(PointType ptype, int comp_id, int id, WaveField &wf) const;
};

#endif /* s2_ldb_access.hpp */