psc_ts_container_3d.hpp

/* psc_ts_container_3d.hpp */
/* $Id$ */
#ifndef __psc_ts_container_3d_hpp
#define __psc_ts_container_3d_hpp

#include <string>

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

namespace psc
{
  class TsContainer3D : public TsContainer
  {
  public:
    
    TsContainer3D();
    
    virtual ~TsContainer3D();
    
    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);
        
    virtual void applyTimeShift(const Trace::Header::AttrId &attr_id, const int id, const Trace::TimeT dt);
    
    static void test_apriory_statics_apply(TsContainer3D &ts);
    
    static void test_to_file(const TsContainer3D &ts, const std::string &fname);
    
  protected:
    
    std::vector<int> m_nss_ids_vec; 
    std::vector<int> m_nsl_ids_vec; 
    std::vector<int> m_nrs_ids_vec; 
    std::vector<int> m_nrl_ids_vec; 
    std::map<int, std::pair<int, int> > m_nsp_to_nss_nsl_map;
    std::map<int, std::pair<int, int> > m_nrp_to_nrs_nrl_map;
    
    bool get_entry_ids(const int nss, 
               const int nsl,
               const int nrs,
               const int nrl,
               const unsigned int corr_dir, 
               const unsigned int stride,
               size_t &storage_id,
               size_t &entries_id) const;
    
    Entry& get_entry(const int nss, 
             const int nsl,
             const int nrs,
             const int nrl,
             const unsigned int corr_dir, 
             const unsigned int stride);
  };
}; // namespace psc

#endif /* psc_ts_container_3d.hpp */