grav3D.h

#ifndef GRAV3D_H
#define GRAV3D_H

#include<stdio.h>
#include"../global.h"

#ifdef PFFT
#include"potential_PFFT_3D.h"
#endif

#ifdef CUFFT
#include"potential_CUFFT_3D.h"
#endif

#ifdef SOR
#include"potential_SOR_3D.h"
#endif


#ifdef HDF5
#include<hdf5.h>
#endif


class Grav3D
{
  public:

  Real Lbox_x;
  Real Lbox_y;
  Real Lbox_z;

  Real xMin;
  Real yMin;
  Real zMin;
  int nx_total;
  int ny_total;
  int nz_total;

  int nx_local;
  int ny_local;
  int nz_local;

  Real dx;
  Real dy;
  Real dz;

  #ifdef COSMOLOGY
  Real current_a;
  #endif

  Real dens_avrg ;


  int n_cells;
  int n_cells_potential;


  bool INITIAL;

  Real dt_prev;
  Real dt_now;

  Real Gconst;

  bool TRANSFER_POTENTIAL_BOUNDARIES;
  
  #ifdef PFFT
  Potential_PFFT_3D Poisson_solver;
  #endif
  
  #ifdef CUFFT
  Potential_CUFFT_3D Poisson_solver;
  #endif

  #ifdef SOR
  Potential_SOR_3D Poisson_solver;
  #endif


  struct Fields
  {
    Real *density_h;

    Real *potential_h;

    Real *potential_1_h;

  } F;
  
  Grav3D(void);

  void Initialize( Real x_min, Real y_min, Real z_min, Real Lx, Real Ly, Real Lz, int nx_total, int ny_total, int nz_total, int nx_real, int ny_real, int nz_real, Real dx_real, Real dy_real, Real dz_real, int n_ghost_pot_offset, struct parameters *P);

  void AllocateMemory_CPU(void);
  void Initialize_values_CPU();
  void FreeMemory_CPU(void);
  
  Real Get_Average_Density( );
  Real Get_Average_Density_function( int g_start, int g_end );


};


















#endif //GRAV3D_H