u_ag_storage.hpp

/* u_ag_storage.hpp */
/* $Id: u_ag_storage.hpp,v 1.2 2008/05/20 09:05:23 urij Exp $ */
//------------------------------------------------------------------------------
#ifndef __u_ag_storage_hpp
#define __u_ag_storage_hpp


#include <mth/u3d_box.hpp>
#include <mth/u3d_point.hpp>
#include <rbm/u3d_model_defines.hpp>

//==============================================================================
void Transform3D(U3dModAreaPos pos, U3dPoint &p);
void Transform3DBack(U3dModAreaPos pos, U3dPoint &p);


//==============================================================================
class UAGStorage 
{
public:
  
  UAGStorage();
  
  UAGStorage(U3dPoints sp_ar, U3dPoints rec_ar, U3dModAreaPos modelOrigin);
  
  virtual ~UAGStorage();
  
  int        getSPNum()  { return m_sp_ar.size(); }
  
  int        getRecNum() { return m_rec_ar.size(); }
  
  /* Returns pointer to the observation system object */
  //U3dObservSys* getOS(int spId);
  
  bool       getSP(int spId, U3dPoint &sp);
  
  bool       getRec(int recId, U3dPoint &recp);
  
  bool       setSP(U3dPoint sp, int id=-1);
  
  bool       setRec(U3dPoint recp, int id=-1);
  
  


  bool       getReady() { return m_ready; }

  
  virtual bool loadSPTraces(int spId) { return false; }
  
  double     getAmpl(int recId, double time, double f, double az);
  
  double     getAmplPure(int recId, double time, double f, double az);
  double     getAmplParabolic(int recId, double time, double f, double az);
  double     getAmplWindowSumm(int recId, double time, double f, double az);
  
  
  double     getA(int recId, int sample_num, double f, double az);
  

  
  int        getTraceLength() { return m_trLength; }
  
  double     getTraceStep() { return m_trStep; }
  
  U3dBox getAGBox();
  

  
  /* ================ DEBUG ================ */
  float** getTraceXComp() { return m_seisDataXComp; }
  float** getTraceYComp() { return m_seisDataYComp; }
  float** getTraceZComp() { return m_seisDataZComp; }
  
  void    setReady(bool flag) { m_ready = flag; }
  /* ======================================= */
  


  
protected:
  
  /* Sorts receivers array according to their depth growth */
  //void   sortRec(U3dPointsAr &recs);
  
  void       agModelCoordRecalc(U3dPoints sp_ar, U3dPoints recs_ar);
  
  bool       m_ready;              
  int        m_compNum;            
  int        m_trLength;           
  double     m_trStep;             
  int        m_currentSP;
  
  U3dModAreaPos m_modelOrigin;     /*< always in UNIVERS coordinate system */
  
  float**    m_seisDataXComp;     
  float**    m_seisDataYComp;     
  float**    m_seisDataZComp;     
  U3dPoints  m_sp_ar;             
  U3dPoints  m_rec_ar;            
};


#endif