u3d_rtmod.hpp

/* u3d_rtmod.hpp */
/* $Id: u3d_rtmod.hpp,v 1.2 2008/06/02 12:02:11 sasha Exp $ */
//------------------------------------------------------------------------------
#ifndef __u3d_rtmodhpp
#define __u3d_rtmodhpp

#include <u3d_seis.hpp>

#include "u3d_rtmod_shooter.hpp"
#include "u3d_rtmod_storage.hpp"

#include "u3d_surfsource.hpp"
#include "u3d_abstract_wfield.hpp"


//==============================================================================
class U3dRTMod : public U3dRTModRayShooter
{
public:

  U3dRTMod(U3dModel *model, 
       U3dAbstractSeis *seis, 
       float stp, 
       URayReflCoefType ctype = ZOEPPRITZ, 
       bool save_ray_path = true, 
       int* waves_under_processing = NULL);
  ~U3dRTMod();

  bool calcSurfSources(int spn);
  
  bool calcSPSurfSources(int spn);
  bool calcRecSurfSources();
  
  bool calcDirectWave(int spnm, U3dAbstractWField *field);

  bool calcUpgoingWaves(int spn, U3dAbstractWField *field);

  
  bool calcDowngoingWaves(int spn, U3dAbstractWField *field);

  bool getReady() { return m_ready; };



  U3dRTModStorage* getStorage() { return &m_storage; };

  U3dAbstractSeis* getSeis() { return m_seis; };

  void calcDynamicHod(int spId, U3dWaveDynHodo &wave_hod);
  
  bool getDirWaveCalcFlag() const { return m_calc_direct_wave; }
  bool getPPUPCalcFlag() const { return m_calc_PP_UP; }
  bool getPSUPCalcFlag() const { return m_calc_PS_UP; }
  bool getPSDOWNCalcFlag() const { return m_calc_PS_DOWN; }
  
  
  /* === DEBUG === */
  int checkWaveInfo(U3dWaveInfo w);
  bool testWaveHod(U3dWaveInfo w, U3dWaveDynHodo wave_hod, int wh_size=0);
  
  void tmpFunc();
  
  //bool getSS(U3dPoint sp, U3D_WAVE_TYPE wt, int bId, U3dSurfSource*& ss_out) {
  //return U3dRTModRayShooter::getSurfSource(sp, wt, bId, ss_out);
  //}  
  
private:
  
  bool getWave(int spId, U3dCodeAr c, U3dWaveDynHodo &wh);
  
  bool getDirectWave(int spId, U3D_WAVE_TYPE wt, U3dWaveDynHodo &wh);

  bool getDirectWaveSP(int spId, U3D_WAVE_TYPE wt, U3dWaveInfo &w);
  
  bool compSS(U3dSurfSource *sp_ss, U3dSurfSource *rec_ss, int recId, 
          bool trFlag, U3dWaveInfo &w);
  
  bool compSSCell(SSPQCell sp_cell, SSPQCell rec_cell, U3dPoint cell_p, 
          U3D_WAVE_TYPE wt1, U3D_WAVE_TYPE wt2, bool transfer, 
          double &diff_f, double &diff_az);

  bool checkBoundSPRecLocation(int spId, int recId, int boundId, bool transfer);
  
  void analyseIPArray(U3dWaveInfo &w, std::vector<double> diff_ar, int debug_level=0);

  
  
  // .........................................................................
  U3dAbstractSeis   *m_seis;       
  U3dRTModStorage    m_storage;    
  bool     m_ready;                
  bool     m_calc_direct_wave;
  bool     m_calc_PP_UP;
  bool     m_calc_PS_UP;
  bool     m_calc_PS_DOWN;
  
};
  

//----------------------------------------------------------------------------
#endif /* u3d_rtmod.hpp */