rt_storage.hpp

// rt_storage.hpp
// $Id: rt_storage.hpp,v 1.5 2006/10/05 07:49:01 guser4 Exp $
//-----------------------------------------------------------------------------
#ifndef __rtStorageH
#define __rtStorageH

#include "vel2dmod_raytr_global.hpp"

#ifndef GE_BUILD_RAYTR
#include <vel2dmod/vel2dmod_global.hpp>
#else
#include <vel2dmod_global.hpp>
#endif



//== Structures ================================================================
typedef struct
{
  int       bNum;   /* bounds[0]=-1 means shot point */
  int           segNum;
  WAVE_TYPE wType;  /* PRESSURE or SHEAR */
  bool      trFlag; /* TRUE-transfer, FALSE-reflection from the bound */  
}   CodeItem;



typedef objects_ar<CodeItem> CodeAr;

/*
#define AnyAr   CodeAr
#define AnyType CodeItem
#include <mix/AnyAr.H>
*/



//=============================================================================
struct RayIntSt
{
  int          bNum1, bNum2;    
  int          boundNum;        
  double       ix, iz;          
  double       coeff;           
  double       t;               
  double       f1;              
  double       f2;              
  double       iAngle;          
  double       iCurv;           
};

//=============================================================================
struct RayWIntSt
{
  double       x, z;
  double       f;
  double       t;
  double       ampl;
};

typedef objects_ar<RayIntSt> RayIntStAr;

/*
#define AnyType     RayIntSt
#define AnyAr       RayIntStAr
#include <mix/AnyAr.H>
*/

//=============================================================================

class RayInfo
{
 public:

  RayInfo(double shotX, double shotZ, double startAngle);  
                                                           
  ~RayInfo();

  void        AddPoint(double x, double z);
  
  void        AddRayAngle(double f);         
  
  void        GetStartParams(double &x, double &z, double &f);
  
  void        GetRayCode(CodeAr &rCode);
  
  int         GetPointsNum();  
  
  void        GetPoints(double *x, double *z); 
  
  void        GetLastPoint(double &x, double &z);
  
  void        GetRayAngles(double *f);

  double      GetLastAngle();
  
  int         GetRayCodeNum(); 

  void        SetRayCode(CodeAr rCode);

  void        AddRayInt(double x, double z,double f1, double f2, double time, int bn1, 
            int bn2, int boundN, double coeff, double iAngle, double iCurv);  
                                        
  void        AddRayWInt(double x, double z, double f, double t);                                                                     

  int         GetRayIntNum();
  
  void        GetRayInt(double* x, double* z, int* bn1, int* bn2, int* boundN, 
            double* coeff, int &n, double *t = NULL, double *f1 = NULL,
            double *f2 = NULL, double *iAngle = NULL, double *iCurv = NULL); 
  
  bool        GetRayNInt(int N, double &x, double &z, int &bn1, int &bn2, int &boundN, double &coeff, double &t, 
             double &f1, double &f2, double &iAngle, double &iCurv);
  void        GetRayIntN(int n, double &x, double &z);
  
  bool        GetBoundIntInfo(int boundNum, double &x, double &z, int &bn1, int &bn2, double &coeff, double &t, 
                  double &f1, double &f2, double &iAngle, double &iCurv);
    
  void        GetRayIntN(int n, int &bn);

  void        GetRayInt1(double &x, double &z, int &bn1, int &bn2, int &boundN, double &coeff, 
             double &t, double &f1, double &f2, double &iA, double &iCurv);
  
  void        GetRayIBNums(int_ar *iBNums);  
               
  int         GetRayWIntNum();  
  
  void        GetRayWInt(double *x, double *z, double *f, double *t, int &n);  

  void        GetRayWInt1(double &x, double &z, double &f, double &t);         

  double      GetSD();            
  
  void        SetSD(double SD);   

  void        SetResParams(double resAmpl, double resAngle, double resTime, 
               double resX, double resZ);

  void        GetResParams(double &resAmpl, double &resAngle, double &resTime, 
               double &resX, double &resZ);

  void        CalcRayPz(double f, double az, double &pf, double &paz, WAVE_TYPE vPS);

  void        SetOOMFlag();  
  bool        GetOOMFlag();  

 private:
  
  double      spX, spZ;
  double      stAngle;  
  double      sd;

  double      resAmpl, resAngle, resTime, resX, resZ;

  int         *codeBN;
  WAVE_TYPE   *codeVPS;
  bool        *codeRTR;
  int         codeNum;

  int         pNum;
  double_ar    *rx;
  double_ar    *rz;
  double_ar    *rf;
  RayIntStAr  RayInt; 

  //  RayWIntStAr RayWInt;

  double wx, wz, wf, wt, wampl;
  bool outOfModel;
  
};

#endif