psc_ts_container.hpp

/* psc_ts_container.hpp */
/* $Id$ */
#ifndef __psc_ts_container_hpp
#define __psc_ts_container_hpp

#include <vector>

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

namespace psc
{
  class TsContainer
  {
  public:
    
    struct Entry
    {
      Entry();
      
      Entry(const Trace::TimeT &i_dt, const float i_w);
      
      bool actual() const;
      
      void setInactual();
      
      Trace::TimeT dt; 
      Trace::AmplT w;  
    };
    
    typedef std::vector<Entry> Entries;
    
    typedef std::vector<Entries> Storage;
    
    
    TsContainer();
    
    virtual ~TsContainer();
    
    virtual bool init(const dba::Seismic &dba,
              const unsigned int corr_dir_mask,
              const unsigned int max_stride,
              const Trace::OffsetT offsets_range[2] = NULL);
    
    
    virtual const Entry& operator()(const int nsp, 
                    const int nrp, 
                    const unsigned int corr_dir, 
                    const unsigned int stride) const;
    
    virtual Entry& operator()(const int nsp, 
                  const int nrp, 
                  const unsigned int corr_dir, 
                  const unsigned int stride);
    
    const std::vector<int>& getIds(const Trace::Header::AttrId &attr_id) const;
    
    unsigned int getCorrDirMask() const {return m_corr_dir_mask;}
        
    unsigned int getMaxStride() const {return m_max_stride;}
    
    
    virtual void applyTimeShift(const Trace::Header::AttrId &attr_id, const int id, const Trace::TimeT dt);
    
    void removeIncline(const Trace::Header::AttrId &attr_id);

    
    void test_to_file(const std::string &fname) const;
    
  protected:
    
    Storage m_storage;              
    std::vector<int> m_nsp_ids_vec; 
    std::vector<int> m_nrp_ids_vec; 
    unsigned int m_corr_dir_mask;   
    unsigned int m_max_stride;      
    virtual void progress(const int cur, const int max) const;
    
    bool get_ids_from_db(const dba::Seismic &dba,
             const Trace::Header::AttrId &attr_id,
             std::vector<int> &ids_vec) const;
    
    Entry tr2entry(Trace &tr, 
           const unsigned int dir, 
           const unsigned int stride) const; 
    
    bool id_of_corr_dir(const unsigned int dir, size_t &id) const;
    
    bool get_entry_ids(const int nsp, 
               const int nrp, 
               const unsigned int corr_dir, 
               const unsigned int stride,
               size_t &storage_id,
               size_t &entries_id) const;
    
    Entry def_inactual_entry;
  };
  
}; // namespace psc

#endif /* psc_ts_container.hpp */