udb_seis.hpp

/* udb_seis.hpp */
/* $Id: udb_seis.hpp 21467 2012-02-03 15:33:40Z urij $ */
#ifndef __udb_seis_hpp
#define __udb_seis_hpp

#include <s2proc/wave_field.hpp>
#include <mix/trselect.h>
#include <string>
#include <set>
#include <utility>

class UDBSeis
{
public:
  
  typedef std::pair<size_t, size_t> Ranges;
  
  enum OSType
    {
      CMP=0,      // common middle points 
      CSP,        // common source points
      CRP,        // common receiver points 
      NMGL,       // uniq magnitograms
      EQL         // equal offsets
    };
  
  struct Type
  {
    Type() {
      comp = 'X', nmgl = 0; os_type = NMGL; L=0; 
      use_na_trs = false; min_traces_num = -1;
    }
    char comp;          
    int nmgl;           
    Int4 L;             
    OSType os_type;     
    bool use_na_trs;    
    int min_traces_num; 
  };
  
  UDBSeis();
  
  virtual ~UDBSeis();
  
  bool dbFill(const char *project, const char *name);
  
  bool setSeisType(const Type &type);
  
  Type getSeisType() const {return m_seis_type;}
  
  void close();
  
  
  int getComps(std::vector<char> &comps);
  
  int getMagnitograms(std::vector<int> &nmgls);
  
  int getOffsets(std::vector<int> &offsets);
  
  
  bool getWavefield(WaveField &wf, Ranges *pranges = 0x0) const;
      
  bool getWavefieldSize(int &wf_size) const;
  
  bool getCoords(int &ix, int &iy,                      
         double &ixl, double &iyl, double &izl  ) const; 
  

  int getSamplesCount() const;
    
  double getTbeg() const;
  
  double getTend() const;
  
  double getTimeStep() const;
  
  
  std::string getModName() {return m_mod_name;}
  
  
  bool getWavefieldHeaders(TrHeader *&headers) const;
  
  TrSet& getTrSet(){return *m_ptrset;}
  
  
  void addHdrAttr(Trace::Header::AttrId id){m_hdr_attrs_set.insert(id);}

  void clearHdrAttrsList() {m_hdr_attrs_set.clear();}

protected:
  
  TrSet *m_ptrset;
  
  std::string m_mod_name;

  Type m_seis_type;
  
  Ranges m_ids_ranges;

  std::set<Trace::Header::AttrId> m_hdr_attrs_set;
    
  ErrCode make_it(const Type &type, TrSet *trset, int id1, int id2, TrIndexTable &it);
};

#endif /* udb_seis.hpp */