u3d_migr_shooter.hpp

/* u3d_migr_shooter.hpp */
/* $Id: u3d_migr_shooter.hpp,v 1.2 2008/09/18 14:01:50 sasha Exp $ */

#ifndef __u3d_migr_shooter_hpp
#define __u3d_migr_shooter_hpp

#include <rbm/u3d_abstract_shooter.hpp>
#include "u3d_migr_defines.hpp"

class U3dMigrRayShooter : public U3dAbstractRayShooter
{
public:
  
  U3dMigrRayShooter(U3dModel *mod3d, URayReflCoefType coeff_type = ZOEPPRITZ);
  
  U3dRay*    rayShoot(U3dPoint startP, double f, double az, 
              U3dCodeAr rCode, bool waySave);
 
  U3dBox*     getImageBox() { return new U3dBox(m_imageBox); }
  
protected:
  
  void       markCell(ushort i, ushort j, ushort k, double f, double az, double t);

  void       addRayPoint(U3dPoint p0, U3dRayPoint rp1, U3D_WAVE_TYPE wType, 
             /*double t,*/ double dt);  
 
  void       addRayPoint2D(int ind1_0, int ind2_0, int ind1_cur, int ind2_cur, 
               int ind1_step, int ind2_step, 
               double coord1_step, double coord2_step, 
               double coord1_2_mul, double coord2_1_mul, 
               double coord3_1_mul, double coord3_2_mul,
               int ind_const, U3dRayPoint rp0, int plane, 
               U3D_WAVE_TYPE wt, double end_time);
  
  void       setCurrentStartAngles(U3dPoint sp, double f, double az);
  
  bool        getCellIJK(U3dPoint point, ushort &i, ushort&j, ushort &k);
  inline void getCellI(U3dPoint p, ushort &i);   
  inline void getCellJ(U3dPoint p, ushort &j);   
  inline void getCellK(U3dPoint p, ushort &k);   
  virtual void calculateImageBox();
  
  // **************************************************************************** 
  
  //private:
  
 
  int        getFaceId(U3dRay *r);
  
  int        getFaceIMax(int faceId);

  int        getFaceJMax(int faceId);

  void       getFaceCellIJ(U3dRay *r, int faceId, int &i, int &j);
  
  void       setShadowedFlag(int i, int j, int faceId);

  float      getFaceCellT(int i, int j, int faceId);

  float      getFaceCellStF(int i, int j, int faceId);

  float      getFaceCellStAz(int i, int j, int faceId);

  
  MigrInnerCell getFaceCell(int i, int j, int faceId);
  
  // **************************************************************************** 
  
  // **************************************************************************** 
  
  MigrDataCube     m_migr_cube;       
  U3dBox           m_imageBox;
  
  float            m_stepX;           
  float            m_stepY;           
  float            m_stepZ;           
  ushort           m_i_max;           
  ushort           m_j_max;           
  ushort           m_k_max;           
  double          m_curStF;           
  double          m_curStAz;          
  U3dPoint        m_curSP;            
};


//==============================================================================
//------------------------------------------------------------------------------
inline void U3dMigrRayShooter::getCellI(U3dPoint p, ushort &i) 
{
  if (fabs(p.getX() - m_model->getModelBox().getXMax()) < U3D_EPSILON) i = m_i_max;
  else { double x = p.getX() - m_model->getModelBox().getXMin(); i = ushort (x / m_stepX); }
}

//------------------------------------------------------------------------------
inline void U3dMigrRayShooter::getCellJ(U3dPoint p, ushort &j) 
{
  if (fabs(p.getY() - m_model->getModelBox().getYMax()) < U3D_EPSILON) j = m_j_max;
  else { double y = p.getY() - m_model->getModelBox().getYMin(); j = ushort (y / m_stepY); }
}

//------------------------------------------------------------------------------
inline void U3dMigrRayShooter::getCellK(U3dPoint p, ushort &k) 
{
  if (fabs(p.getZ() - m_model->getModelBox().getZMax()) < U3D_EPSILON) k = m_k_max;
  else { double z = p.getZ() - m_model->getModelBox().getZMin(); k = ushort (z / m_stepZ); }
}
  



#endif