u3d_rtmod_shooter.hpp

/* u3d_rtmod_shooter.hpp */
/* $Id: u3d_rtmod_shooter.hpp,v 1.19 2008/03/31 07:58:19 guser1 Exp $ */
//------------------------------------------------------------------------------

#ifndef __u3d_rtmod_shooter_hpp
#define __u3d_rtmod_shooter_hpp

#include <u3d_abstract_shooter.hpp>
#include <u3d_raydyn_calculator.hpp>
#include "u3d_rtmod_defines.hpp"
#include "u3d_surfsource.hpp"

#include <iostream>
#include <fstream>
#include <iomanip>
using namespace std;

//==============================================================================
class U3dRTModRayShooter : public U3dAbstractRayShooter
{
public:
  
  U3dRTModRayShooter(U3dModel *mod3d, float stp, U3dReflCoefType ctype = ZOEPPRITZ, bool save_ray_path = false);
  
  
  /* ===================== surface sources calculation ===================== */
    
  bool getSurfSource(U3dPoint sp                  , 
             U3D_WAVE_TYPE wt             , 
             int bId                      , 
             U3dSurfSource*& ss_out       );  
  
  bool initialSurfCovering(U3dPoint p             , 
               U3dCodeAr c            , 
               int debug_level = 0    );
  
  bool shootSurfFirstRay(U3dPoint p         , 
             U3dCodeAr code     , 
             double &in_f       , 
             double &in_az      , 
             double df          , 
             double daz         );
  
  bool coverSurfFace(U3dPoint p             ,
             U3dCodeAr c            , 
             int debug_level = 0    );
  
  U3dRay*    rayShoot(U3dPoint startP             , 
              double f                    , 
              double az                   , 
              U3dCodeAr rCode             , 
              bool waySave                );

  
  /* ===================== direct SP-REC shooting ===================== */
  
  bool       shootPoint(U3dRayPoint start_p       , 
            U3dPoint recp             , 
            double &devF              , 
            double &devAz             , 
            U3dCodeAr rCode           , 
            int bodyId                ,
            U3dRayPoint &wp           ,
            int debug_level = 0       );
  
  double     SPBodyRecShoot(U3dPoint startP       , 
                U3dPoint recp         , 
                double f              , 
                double az             , 
                U3dCodeAr rCode       , 
                int bodyId            , 
                U3dRayPoint &wp       );
  
  U3dRay*    oneBodySP2RecRayShoot(U3dPoint startP      , 
                   U3dPoint recp        , 
                   double stF           , 
                   double stAz          ,
                   U3D_WAVE_TYPE wt       , 
                   int bodyId           );
  
  
  /* ===================== dynamic ===================== */
  
  void       calcDynamic(U3dRayPoint sp, U3dPoint recp, U3dCodeAr c, double &ampl, U3dPoints &ray_points);
  
  U3dRay*    getDynRay(U3dRayPoint sp, U3dPoint recp, U3dCodeAr c, double &dist, U3dPoint &lp, int debug_level = 0);
  
  bool       calcGeomDivergence(U3dRay* ray, U3dPoint recp, double &geom_div);
  
  U3dRay*    rayShootDyn(U3dRayPoint start_p     , 
             U3dPoint recp           , 
             U3dCodeAr rCode         , 
             bool waySave            ,
             int debug_level=0);
  
  
  /* ===================== service ===================== */
  
  bool       getReady() { return m_ready; }
  
  bool       initSS(int bId, U3dCodeAr wc);
  
  void       cleanSS();
  
  bool       m_save_ray_path;
  
  void       reloadSS();
  
protected:

  void       shadowPerimeter();
  
  /* Get the nearest to the receiver point of the (rp_prev -> rp0) segment of ray trajectory */
  U3dRayPoint getNearestRayPoint(U3dRayPoint rp_prev, U3dRayPoint rp0, U3dPoint recp, double vel);
  
  //bool       initSS(int bId, U3dCodeAr wc);
  //void       cleanSS();

  int        getFaceId(U3dRay *r);
      
  int        getFaceIMax(int faceId);

  int        getFaceJMax(int faceId);

  void       getFaceCellIJ(U3dPoint p, 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);
  
  void       formFaceActiveZones(int faceId);

  U3dSurfSource    *m_ss;
  bool              m_ready;
  float             m_ss_stp;
  
  static int        m_faceId;
  
};


//==============================================================================
#endif