vel2dmod_service.hpp

// vel2dmod_service.hpp
// $Id: vel2dmod_service.hpp,v 1.17 2007/11/30 09:12:06 guser1 Exp $
//-----------------------------------------------------------------------------
#ifndef __2dModelServiceH
#define __2dModelServiceH

#include "vel2dmod.hpp"
#include "vel2dmod_global.hpp"

//------------------------------------------------------------------------------
struct BoundSegment
{
  int kn1, kn2;
  int initial_kn1, initial_kn2;
};

typedef objects_ar<BoundSegment*> BoundSegmentsAr;

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

class Bound 
{  

#ifndef GE_BUILD
  friend class Vel2dModelLDB;
#else
  friend class Vel2dModelGEDB;
#endif

 public:

  Bound(double *xx, double *zz, int nn, bool *flag, double rc, double sp,
    double *xx1 = NULL, double *zz1 = NULL, int nn1 = 0);
  
  Bound(double *xx, double *zz, int nn, double rc, double sp);
  
  Bound(double *xx, double *zz, double *dx, double *dz, double *curv, bool *flag, int nn,
    double *init_x = NULL, double *init_z = NULL, bool *init_flags = NULL, int init_num = 0);
  
  Bound(double *xx, double *zz, double *dx, double *dz, double *curv, bool *flag, int nn, int app_type,
    double *init_x = NULL, double *init_z = NULL, bool *init_flags = NULL, int init_num = 0);
    
  Bound(Bound*);

  ~Bound();
  
  int getApproxType() {
    return m_approx_type;
  }
  
  bool        BoundInt(double x1, double z1, double x2, double z2,
               double eps, double &resultX, double &resultZ, bool flag=true,
               bool finitLine=true, int *nn=NULL);

  void        GetEndPoints(double &x1, double &z1, double &x2, double &z2);

  void        Segmentation();
  int         GetSegNum();
  int         GetSegNum(int n);
  int         GetSegNum(double x, double z);
  
  int         InitialPNum();
  void        GetInitialPoints(double *x, double *z, bool *flags);
  void        GetRC_SP(double &rc, double &sp);

  void        ReInterpolate(double *x, double *z, bool *flag, int n, double r_c = 0, double s_p = 0, bool *im_flag=NULL);
  
  void        GetZLimits(double &z_min, double &z_max);
  
  void        GetFirstPoint(double &x, double &z)   { if ((initial_x!=NULL)&&(initial_z!=NULL)&&(initial_pnum>0)&&filled)
                                                         { x=initial_x[0]; z=initial_z[0]; } else {x=0; z=0;} }
    
  void        GetLastPoint(double &x, double &z)   { if ((initial_x!=NULL)&&(initial_z!=NULL)&&(initial_pnum>0)&&filled)
                                                         { x=initial_x[initial_pnum-1]; z=initial_z[initial_pnum-1]; } 
                                                     else {x=0; z=0;} }
 
  ParSpline*  spline();

  int        num;

  int        bNum1;
  int        bNum2;

  bool       filled;

 private:

  void        CalcZLimits();
  double      GetPeak(double *x, double *z, bool top);
  void        SetRC_SP(double rc, double sp);

  BoundSegmentsAr  Segments;
  double     *ixs;  
  double     *izs;
  int        *inns;
  
  // first/last numbers of the nodes of Curve, between which the respective Bound is situated
  //int        knotNum1;
  //int        knotNum2;
  //int        initial_kn1;
  //int        initial_kn2;

  double     *initial_x;
  double     *initial_z;
  bool       *initial_flags;
  int        initial_pnum;
  double     RC, SP;

  ParSpline  *psp;
  double     zMin;
  double     zMax;

  int m_approx_type;
};

typedef Bound* pBound;

// dynamic Bound arrays definition
typedef objects_ar<pBound> BoundsAr;



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

class BodyParam
{
public:
  
  BodyParam(int vMN, double *x, double *z, double *vp, double *vs, double *pkz,
        double *skz, double *den, double ma_xmin, double ma_xmax,
        double an_kp, double an_ks, double R);
  
  BodyParam(double vp, double vs, double pkz, double skz, double den, double ma_xmin, double ma_xmax,
        double an_kp, double an_ks, double R);
  
  BodyParam(int vMN, double *x, double *vp,  double *vs, double *pkz,
        double *skz, double *den, double ma_xmin, double ma_xmax,
        double an_kp, double an_ks, double R);
  
  BodyParam(BodyParam*);
  
  ~BodyParam();

  double    GetV0(double x, WAVE_TYPE wt);

  double    GetKx(double x, WAVE_TYPE wt);

  bool      ParamsConst();

  double    GetCVel(WAVE_TYPE wt);

  double    GetCKz(WAVE_TYPE wt);

  double    GetCDen();

  void      GetParams(int &pn, double *&px, double *&pz, double *&pPVel, double *&pSVel,
              double *&pPKz, double *&pSKz, double *&pDen);

  bool      filled;

  bool      paramsConst;

  void      RebuildParams(int vMN, double *new_x, double *new_vp,  double *new_vs, double *new_kzp,
              double *new_kzs, double *new_den);
  
  void      SortParams();

  void      GetMASizes(double &xmn, double &xmx) { xmn=x_min; xmx=x_max; };

  Spline*   GetSpVS();
  Spline*   GetSpVP();
  
  int       GetParamNum() { return paramNum; };

  double   GetAnCoef(WAVE_TYPE wt, double ang);
  
  double   GetAn(WAVE_TYPE wt);

  double    GetAnKP() { return m_an_kp; }
  
  double    GetAnKS() { return m_an_ks; }
  
  double    GetRParam() { return m_R; }
  
  void SetAnKP(double an_kp) { m_an_kp = an_kp; }
  
  void SetAnKS(double an_ks) { m_an_ks = an_ks; }

  void SetRParam(double R) { m_R = R; }
  
 private:

  void      Replace(int N, double *M, int i, int j);

  void      RefreshParams(int vMN, double *new_x, double *new_vp,  double *new_vs, double *new_kzp,
              double *new_kzs, double *new_den);
  
  Spline    *SpVS;  
  Spline    *SpVP;

  int       paramNum;

  double    *x;
  double    *z;
  double    *vp;
  double    *vs;
  double    *kzP;
  double    *kzS;
  double    *den;

  double    cKzP;        
  double    cKzS;        
  double    cDen;        
  double    cVelP;       
  double    cVelS;
  
  double    *vLine;

  double x_min, x_max; 
  
  double m_an_kp;        
  double m_an_ks;        
  double m_R;            
};



//==============================================================================
class Body              //------------------------------------------------------
{                       // BODY
public:                 //    ----------
                        //    |BodyParam| a set of pointers to the Bounds
                        //    ----------
                        //------------------------------------------------------

  Body(int bAm); 
  Body(Body*);
  ~Body();
  
  void    AddBoundPtr(Bound *pB);

  bool    TopBorder(double x, double &z);

  bool    BotBorder(double x, double &z);

  bool    BodyBorder(int bNum, double x, double &z);

  void    GetBodyBrd(int &num, int *&brds);

  void    SetConstParams(double vp, double vs, double kzp, double kzs, double den,
             double an_kp = 1, double an_ks = 1, double R = 1);

  void    SetParams(int pNum, double *x, double *vp, double *vs, double *kzp, double *kzs, double *den,
            double an_kp = 1, double an_ks = 1, double R = 1);

  int GetTopBN();
  
  int GetBotBN();
  
  bool GetLateralBounds(int &left_bN, int &right_bN);

  void GetZLimits(double &z_min, double &z_max);

  void GetLZLimits(double &z_min, double &z_max);

  void GetRZlimits(double &z_min, double &z_max);
  
  BodyParam* Param;       
  int num;                
  int boundNum;           
  int *bounds;            
  int *bPos;              
  bool *n_direct;         
  bool filled;
  bool ready;
  int presParams;
  
  BoundsAr Bounds;        
};


typedef Body* pBody;
typedef objects_ar<pBody> BodiesAr;

//--------------------------------------------------------
struct ParamsInfo
{
  double x,z;
  double vp, vs;
  double pkz, skz;
  double density;
};

typedef objects_ar<ParamsInfo*> ParamsInfoAr;

struct BodyInfo
{
  ParamsInfoAr  *Params;     // parameters
  double_ar      tbx;        // top border coordinats
  double_ar      tbz;
  int_ar         flag;
  double         an_kp;      
  double         an_ks;      
  double         R;          
};

typedef objects_ar<BodyInfo*> BodyInfoAr;


#endif