grav3D.h

#ifndef GRAV3D_H
#define GRAV3D_H
 
#include <stdio.h>
 
#include "../global/global.h"
#include "../grid/spatial_domain_props.h"
#include "../io/FnameTemplate.h"
 
#ifdef SOR
  #include "../gravity/potential_SOR_3D.h"
#endif
 
#ifdef PARIS
  #include "../gravity/potential_paris_3D.h"
#endif
 
#ifdef PARIS_GALACTIC
  #include "../gravity/potential_paris_galactic.h"
#endif
 
#ifdef HDF5
  #include <hdf5.h>
#endif
 
#define GRAV_ISOLATED_BOUNDARY_X
#define GRAV_ISOLATED_BOUNDARY_Y
#define GRAV_ISOLATED_BOUNDARY_Z
 
#define TPB_GRAV  1024
#define TPBX_GRAV 16
#define TPBY_GRAV 8
#define TPBZ_GRAV 8
 
class Grav3D
{
 public:
  Real Lbox_x;
  Real Lbox_y;
  Real Lbox_z;
 
  Real xMin;
  Real yMin;
  Real zMin;
  Real xMax;
  Real yMax;
  Real zMax;
  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;
 
  bool BC_FLAGS_SET;
  int *boundary_flags;
 
#ifdef SOR
  Potential_SOR_3D Poisson_solver;
#endif
 
#ifdef PARIS
  PotentialParis3D Poisson_solver;
#endif
 
#ifdef PARIS_GALACTIC
  #ifdef SOR
    #define PARIS_GALACTIC_TEST
  PotentialParisGalactic Poisson_solver_test;
  #else
  PotentialParisGalactic Poisson_solver;
  #endif
#endif
 
  struct Fields {
    Real *density_h;
 
    Real *potential_h;
 
    Real *potential_1_h;
 
    Real *analytic_potential_h;
 
#ifdef GRAVITY_GPU
 
    Real *density_d;
 
    Real *potential_d;
 
    Real *potential_1_d;
 
    Real *analytic_potential_d;
 
#endif  // GRAVITY_GPU
 
// Arrays for computing the potential values in isolated boundaries
#ifdef GRAV_ISOLATED_BOUNDARY_X
    Real *pot_boundary_x0;
    Real *pot_boundary_x1;
#endif
#ifdef GRAV_ISOLATED_BOUNDARY_Y
    Real *pot_boundary_y0;
    Real *pot_boundary_y1;
#endif
#ifdef GRAV_ISOLATED_BOUNDARY_Z
    Real *pot_boundary_z0;
    Real *pot_boundary_z1;
#endif
 
#ifdef GRAVITY_GPU
  #ifdef GRAV_ISOLATED_BOUNDARY_X
    Real *pot_boundary_x0_d;
    Real *pot_boundary_x1_d;
  #endif
  #ifdef GRAV_ISOLATED_BOUNDARY_Y
    Real *pot_boundary_y0_d;
    Real *pot_boundary_y1_d;
  #endif
  #ifdef GRAV_ISOLATED_BOUNDARY_Z
    Real *pot_boundary_z0_d;
    Real *pot_boundary_z1_d;
  #endif
#endif  // GRAVITY_GPU
 
  } F;
 
  Grav3D(void);
 
  void Initialize(const SpatialDomainProps &spatial_props, Real Lx, Real Ly, Real Lz, int n_ghost_pot_offset,
                  Parameters *P);
 
  void AllocateMemory_CPU(void);
  void Initialize_values_CPU();
  void FreeMemory_CPU(void);
 
  void Read_Restart_HDF5(struct Parameters *P, int nfile);
  void Write_Restart_HDF5(struct Parameters *P, int nfile, const FnameTemplate &fname_template);
 
  Real Get_Average_Density();
  Real Get_Average_Density_function(int g_start, int g_end);
 
  void Set_Boundary_Flags(int *flags);
 
#ifdef SOR
  void Copy_Isolated_Boundary_To_GPU_buffer(Real *isolated_boundary_h, Real *isolated_boundary_d, int boundary_size);
  void Copy_Isolated_Boundaries_To_GPU(struct Parameters *P);
#endif
 
#ifdef GRAVITY_GPU
  void AllocateMemory_GPU(void);
  void FreeMemory_GPU(void);
#endif
};
 
#endif  // GRAV3D_H