rt_wavecalc.hpp

// rt_wavecalc.H
// $Id: rt_wavecalc.hpp,v 1.2 2006/10/05 07:49:01 guser4 Exp $
//-----------------------------------------------------------------------------
#ifndef __rtWaveCalcH
#define __rtWaveCalcH

#include "rt_shooter.hpp"
#include <zm/ZmProgress.H>
#include <mth/interpol.h>

//============================================================================
//-- Structures --------------------------------------------------------------

struct TrInfo
{
  float *am;
  float dp;
  int num;
};

// Dynamic array of Trace info sructures
typedef objects_ar<TrInfo*> TrInfoAr;

//=============================================================================
class RTHod 
{
public:
  RTHod(int N);
  ~RTHod();
  void InitHod(int N, double *hodZ1, double *hodT1, double *fi1, double *ix1, double *iz1, double *coeff1, 
           double *iA1, double *inF1, double *outF1, int *recNum1);
  
  void Remove(int i, int n);
  void SetValues(int i, double hodZ1, double hodT1, double fi1, double ix1, double iz1, double coeff1, 
         double iA1, double inF1, double outF1, int recNum1);
  void SetRTHod(int i, RTHod *Hod);
  void OverturnHod();
  void SortRTHod(int i_beg, int i_end);
  void Print(int i, char *print=NULL);
  
  int *recNum;
  int hodNum;
  double *hodZ, *hodT, *fi, *ix, *iz, *coeff, *iA, *inF, *outF;
private:
  bool ready;
};


//=============================================================================
class RTWaveCalc : public RayShooter
{
public:

  RTWaveCalc(Vel2dModelLDB *VelMod2d, ObservSys *OSys=NULL);  
  ~RTWaveCalc();

  bool DirectWaveCalc(WAVE_TYPE wt, bool &stop_pressed_flag);

  bool LSWaveCalc(LSList *ls, LSCode *lsc, bool reflect, WAVE_TYPE wt, bool &stop_pressed_flag, int segment_num = -1);

  void CalcWaveParams(LSList *init_List, CodeAr rCode, RTHod *&HODO, int &prog_val, bool &stop_pressed_flag);
  
  double GetCoeff(double start_angle, CodeAr rCode); //double ix, double iz, double inF, double outF, CodeAr rCode);
  
  void DirectWaveParamsCalc(WAVE_TYPE wType, RTHod *&HODO, bool &stop_pressed_flag, bool load_LSList);
  
  void GetDirectWaveTimes(WAVE_TYPE wType, double eps, RTHod *&HODO, bool &stop_pressed_flag);
  
  void InitLSLoading(CodeAr rCode, LSList *&init_List, int &count, int &prog_val);
                 
  void InitLSForming(CodeAr rCode, LSList *&init_List, int &prog_val, bool &stop_pressed_flag, bool dir_wave=FALSE);
                                   
  void WaveCalculation(CodeAr rCode, double *&hodZ, double *&hodT, double *&fi,
               double *&ix, double *&iz, double *&coeff, double *&iA, double *&inF, double *&outF,
               int *&recNum, int &hodNum, bool &stop_pressed_flag, bool load_LSList = TRUE);
  
  void CalcComplicatedWaveParams(CodeAr rCode, RTHod *&HODO, bool &stop_pressed_flag, LSList *&init_List, int count, int &prog_val);
  
  void LS_LSWaveCalc(LSList *in_ls, LSList *&out_ls, WAVE_TYPE wt, bool tr_flag, int end_bound_num, int &prog_val, bool &stop_pressed_flag);
  
private:
  
  bool CheckLS(LSList *ls, int &bn);
  void AddCodeLine(int bn, bool rt, bool ps, LSCode *lsc);
  bool CheckBN(int bn, bool direction);
    
  void SortRecInfo(RecInfoAr &rInfoAr);
  
  void RecInterpol(RecInfoAr &recInfos, int i, CodeAr rCode, double *&fi, double *&coeff, RTHod *&HODO, int &i0, bool amplitude);
  
  void stop_wave_calc(bool &stop_pressed_flag, LSList *&init_list, LSList *&end_list);  
  
};


//==============================================================================
//-- Global functions ----------------------------------------------------------
void TrTime2Depth(double t0, double dt, float *a, long N, double h0, double hb, float dh, float *b, long &M);
void TrDXReCalc(float dx, float_ar *xs, int &stN, TrInfoAr *InTr, TrInfoAr *OutAr);

#endif