u3d_abstract_wfield.hpp

/* u3d_abstract_wfield.hpp */
/* $Id: u3d_abstract_wfield.hpp,v 1.1 2008/05/13 12:19:18 sad Exp $ */
//------------------------------------------------------------------------------
#ifndef __u3d_abstract_wfieldhpp
#define __u3d_abstract_wfieldhpp

#include <u3d_os.hpp>
#include <u3d_raytr_defines.hpp>

#include "u3d_rtmod_defines.hpp"

//------------------------------------------------------------------------------
struct U3dTrace {
  float *trbody_x;  
  float *trbody_y;  
  float *trbody_z;  
};

//==============================================================================
class U3dSignal 
{
public:
  
  U3dSignal(U3dWaveletType sid, float freq);
  virtual ~U3dSignal() {};
  
  virtual float getForm(float t);

  float getLength() { return m_length; };
  
  float getT0() { return m_t0; }
  
  virtual void setData(float length, float step, int samples_num, float* data) {};
  
protected:
  
  U3dWaveletType   m_sigType;      
  float            m_t0;       
  float            m_freq;         
  float            m_length;       
};

//==============================================================================
class U3dVSPSignal : public U3dSignal
{
   
public:
  
  U3dVSPSignal(U3dWaveletType sid, float freq);
  virtual ~U3dVSPSignal();
 
  void setData(float length, float step, int samples_num, float* data);
  
  float getForm(float t);

  bool getReady() { return m_ready; }
  
private:
  
  bool m_ready;       
  float m_dt;
  float* m_data;      
  int m_sample_num;   
};

//------------------------------------------------------------------------------
class U3dAbstractWField
{
public:
  
  U3dAbstractWField(int spn                        , 
            U3dObservSys* os               , 
            float tmax                     , 
            float dt                       , 
            U3dWaveletType signal_type     , 
            float signal_freq              , 
            bool save_rp = false           );
  
  virtual ~U3dAbstractWField();

  void   addWave(int spn, U3dWaveDynHodo wh);
  
  int    getRecNum() { return m_recnum; }
  
  int    getSPN() { return m_spn; }
  
  int    getTrLen() { return m_trlen; }
  
  virtual bool saveAllHods(const char* file_name) = 0;
  
  //bool    analyseRayCode(U3dCodeAr code, int ciId, int &type);
  
  bool    addRayPath(U3dWaveHodoCell whc, int rp_id);
  
  virtual bool saveAllRays(char* file_name, int pid);
  
protected:
  
  static int m_ray_path_id;
  
  static double m_max_ampl;
  
  bool   getHodName(U3dComplexWaveHod hod, char *name);
  
  int    getSampleNum(double t) { return std::max(0, int(t/m_dt)); } 
  
  float  getSampleTime(int sn)  { return sn*m_dt; }

  void   addWHCell(U3dSPHods &sp_hods, U3dWaveHodoCell whi);
  
  
  void   recalcShearDirection(double &f, double &az);
  
  
  bool        m_ready;              
  float       m_dt;                 
  float       m_tmax;               
  int         m_trlen;              
  U3dObservSys* m_os;               
  int         m_spn;                
  int         m_recnum;             
  U3dTrace   *m_traces;             
  U3dSignal  *m_signal;             
  U3dSPHodsAr m_sp_hods;            
  bool        m_save_rays_paths;    
  struct U3dRayPathSt {
    int ray_type;
    U3dRayPath ray_path;
  };
  
  U3dRayPathSt m_rp_st;             
};



void mergeRPFiles(const char* const * filenames, int size, const char* resultFileName, int actualNum) throw (std::ios_base::failure,
                                                         std::logic_error);


void deleteRPFiles(const char* const * filenames, int size, int actualNum) throw (std::ios_base::failure);

//------------------------------------------------------------------------------
#endif /* u3d_abstract_wfield.hpp */