seis.hpp

// seis.hpp
// $Id: seis.hpp,v 1.19 2008/06/30 08:04:40 guser1 Exp $
//---------------------------------------------------------------------
#ifndef __ASeisH
#define __ASeisH

#ifndef GE_BUILD_RAYTR
#include <vel2dmod/os.hpp>
#include <mth/coord2d.hpp>  // for coordinate system transformation
#else
#include <os.hpp>
#include <coord2d.hpp> // for coordinate system transformation
#endif

enum OS_TYPE {OFFSET_VSP, WALKAWAY_VSP};

//---------------------------------------------------------------------
struct MigrGridArea 
{
  double    x_min;
  double    x_max;
  double    z_min;
  double    z_max;
  double    step_x;
  double    step_z;
};

struct Ampls
{
  double **ampl_grid;
  int i_max;
  int j_max;
  int epsilon;
};

//=============================================================================
class AbstractSeis {

public : 
  AbstractSeis();
  virtual ~AbstractSeis();
  
  double getAmpl(int spnum, int recnum, double time, double angle);
  double getA(int spnum, int recnum, int sample_num, double angle);
  
  int    getSPNum()  { return spNum; };
  
  int    getRecNum() { return recNum; };
  
  ObservSys* getOS(int spN);
  
  bool   getSP(int spN, double &sp_x, double &sp_z);
  
  bool   getRec(int recN, double &rec_x, double &rec_z);
  
  bool   getReady();
  
  void   sortRec(double_ar &xrecs, double_ar &yrecs, double_ar &zrecs);

  double* getTrace(int sp_n, int rec_n, int &length)
  {
    double *trace = new double[trLength];
    for (int i=0; i<trLength; i++)
      trace[i] = seisDataXComp[sp_n][rec_n][i];
    length = trLength;
    return trace; 
  };
  
  void getTraceParam(int &tr_length, double &tr_step)
  {
    tr_length = trLength;
    tr_step = trStep;
  };
  
  bool saveAmpl(double **ampl_grid, char *OutMod, MigrGridArea mg_area);
  double** interpolation(Ampls stAmpls, double factor);
  void smoothing(Ampls stAmpls);
  void detalization(Ampls stAmpls, double reinforcement);
  // UNUSED
  void contrastIncreasing(Ampls stAmpls, double reinforcement);
  void medianFilter();
  
  virtual bool saveResult(double **ampl_grid, char *OutMod, MigrGridArea mg_area) { return false; };
  void cutDown(Ampls stAmpls, double x_min, double step_x);
  void cleaning(Ampls stAmpls);
  // DEBUG
  bool saveResultToFile(Ampls stAmpls);
  
  //void   RecalculateAmplGrid(double **ampl_grid, int **ipc_num_grid, int i_max, int j_max);  
  
  void   recalcSPRecCoordinates(double_ar xsps,  double_ar ysps,  double_ar zsps, 
                double_ar &xrecs, double_ar &yrecs, double_ar &zrecs, 
                OS_TYPE ostype, double mx, double my, double maz, bool well_project=true);
  
  void   setReady() { ready = true; }
  
protected:
  
  double     getAzimuth(double x, double y);
  
  void       deleteAllData();
  
  bool       ready;             
  bool       flag1d;            
  int        recNum;            
  int        spNum;             
  int        trLength;          
  double     trStep;            
  double     ***seisDataXComp;  
  double     ***seisDataZComp;  
  double     *spX;              
  double     *spZ;              
  double     *recX;             
  double     *recZ;             
};


#endif