basic_structs.h

Go to the documentation of this file.

 
#pragma once
 
#include "../global/global.h"
#include "../global/global_cuda.h"
#include "../utils/gpu.hpp"
 
namespace hydro_utilities
{
// =====================================================================================================================
// Here are some basic structs that can be used in various places when needed
// =====================================================================================================================
template <typename T>
struct VectorXYZ {
  T arr_[3];
 
 
  __device__ __host__ T* data() { return arr_; }
 
  __device__ __host__ T& operator[](std::size_t i) { return arr_[i]; }
  __device__ __host__ const T& operator[](std::size_t i) const { return arr_[i]; }
 
  __device__ __host__ T& x() noexcept { return arr_[0]; }
  __device__ __host__ const T& x() const noexcept { return arr_[0]; }
  __device__ __host__ T& y() noexcept { return arr_[1]; }
  __device__ __host__ const T& y() const noexcept { return arr_[1]; }
  __device__ __host__ T& z() noexcept { return arr_[2]; }
  __device__ __host__ const T& z() const noexcept { return arr_[2]; }
};
// =====================================================================================================================
 
// =====================================================================================================================
struct Conserved {
  // Hydro variables
  Real density, energy;
  VectorXYZ<Real> momentum;
 
#ifdef MHD
  // These are all cell centered values
  VectorXYZ<Real> magnetic;
#endif  // MHD
 
#ifdef DE
  Real gas_energy;
#endif  // DE
 
#ifdef SCALAR
  Real scalar[grid_enum::nscalars];
#endif  // SCALAR
 
  Conserved() = default;
  Conserved(Real const in_density, VectorXYZ<Real> const& in_momentum, Real const in_energy,
            VectorXYZ<Real> const& in_magnetic = {0, 0, 0}, Real const in_gas_energy = 0.0)
      : density(in_density), momentum(in_momentum), energy(in_energy)
  {
#ifdef MHD
    magnetic = in_magnetic;
#endif  // mhd
 
#ifdef DE
    gas_energy = in_gas_energy;
#endif  // DE
  };
};
// =====================================================================================================================
 
// =====================================================================================================================
struct Primitive {
  // Hydro variable
  Real density, pressure;
  VectorXYZ<Real> velocity;
 
#ifdef MHD
  // These are all cell centered values
  VectorXYZ<Real> magnetic;
#endif  // MHD
 
#ifdef DE
  Real gas_energy;
#endif  // DE
 
#ifdef SCALAR
  Real scalar[grid_enum::nscalars];
#endif  // SCALAR
 
  Primitive() = default;
  Primitive(Real const in_density, VectorXYZ<Real> const& in_velocity, Real const in_pressure,
            VectorXYZ<Real> const& in_magnetic = {0, 0, 0}, Real const in_gas_energy = 0.0)
      : density(in_density), velocity(in_velocity), pressure(in_pressure)
  {
#ifdef MHD
    magnetic = in_magnetic;
#endif  // mhd
 
#ifdef DE
    gas_energy = in_gas_energy;
#endif  // DE
  };
};
// =====================================================================================================================
}  // namespace hydro_utilities
 
namespace reconstruction
{
struct InterfaceState {
  // Hydro variables
  Real density, energy;
  Real pressure;
  hydro_utilities::VectorXYZ<Real> velocity, momentum;
 
#ifdef MHD
  // These are all cell centered values
  Real total_pressure;
  hydro_utilities::VectorXYZ<Real> magnetic;
#endif  // MHD
 
#ifdef DE
  Real gas_energy;
#endif  // DE
 
#ifdef SCALAR
  Real scalar[grid_enum::nscalars];
#endif  // SCALAR
 
  // Define the constructors
  InterfaceState() = default;
  InterfaceState(Real const in_density, hydro_utilities::VectorXYZ<Real> const in_velocity, Real const in_energy,
                 Real const in_pressure, hydro_utilities::VectorXYZ<Real> const in_magnetic = {0, 0, 0},
                 Real const in_total_pressure = 0.0)
      : density(in_density), velocity(in_velocity), energy(in_energy), pressure(in_pressure)
  {
    momentum.x() = velocity.x() * density;
    momentum.y() = velocity.y() * density;
    momentum.z() = velocity.z() * density;
#ifdef MHD
    magnetic       = in_magnetic;
    total_pressure = in_total_pressure;
#endif  // MHD
#ifdef DE
    gas_energy = 0.0;
#endif  // DE
  };
};
}  // namespace reconstruction