grid2d.hpp

//grid2d.hpp
// $Id: grid2d.hpp,v 1.7 2008/01/09 13:36:46 sasha Exp $
//------------------------------------------------------------------
#ifndef __2dGrid_H
#define __2dGrid_H

//#include <zm/ZmProgress.H>

#include "vel2dmod.hpp"
#include "os.hpp"

#include "vel2dmod_global.hpp"

//==================================================================================================
struct GridParams
{
  bool  isWell;   
  int   bodyNum;  
  double vel;

  int_ar brdns;   
  int_ar n1;      
  int_ar n2;      
  void  init_clean (){
    brdns.clean();
    n1.clean();
    n2.clean();
  }
};



class Vel2dModel;  
class AbstractProgress;

//==================================================================================================
class Grid2d 
{
public:
  Grid2d(double xmn, double xmx, double zmn, double zmx, double stp=1);
  Grid2d(Grid2d*);

  virtual ~Grid2d();

  //void    set_progress_indicator (AbstractProgress* prind = NULL);

  void  generate (Vel2dModel* pmod, ObservSys* pos = NULL, double step_size = -1);

  void  re_generate (double x1, double z1, double x2, double z2,
             Vel2dModel* pmod, ObservSys* pos = NULL);
  
  double   get_cell_size();
  
  void fill_cell(int i, int j, Vel2dModel* pmod, ObservSys* pos, GridParams& gp, int &prevBodynum);

  void cell_copy(int i, int j, GridParams gp);

  void get_cell_id (double x, double z, int* ic, int* ir)
  {
    *ic=(int)((x-xmin)/step);
    *ir=(int)((z-zmin)/step);

    if (x<xmin)
      *ic = 0;
    if (x>xmax)
      *ic = nc-1;
    if (z<zmin)
      *ir = 0;
    if (z>zmax)
      *ir = nr-1;
  };

  int get_last_ir () {return (nr-1);}; 
  int get_last_ic () {return (nc-1);}; 
  GridParams&    Cell(int i, int j);

  void set_well_flag(int i, int j)
  {
    cell[i][j].isWell = true;
  };

  void  get_grid_sizes(double &xmn, double &zmn, double &xmx, double &zmx);
  void  get_cell_size(double &cs);

  GridParams  get_cell(int i, int j) { return cell[i][j]; };
  GridParams& set_cell(int i, int j) { return cell[i][j]; };

  double      get_step() { return step; };
  
  void        clean_well_info();
  
  bool        getReady() { return m_ready; }
  
private:
  
  double        xmin, xmax, zmin, zmax;
  double        step;           
  int           nr, nc;
  
  bool          m_ready;
  
  AbstractProgress *pPrInd; 
  GridParams    **cell;
  GridParams    *cell_mem;  
};

#endif