state_views_mixins.hpp

Go to the documentation of this file.

#pragma once
 
#include <deal.II/base/exception_macros.h>
#include <deal.II/base/tensor.h>
#include <deal.II/base/vectorization.h>
 
#include <meltpooldg/compressible_flow/data_types.hpp>
#include <meltpooldg/utilities/concepts.hpp>
 
namespace MeltPoolDG::CompressibleFlow
{
  template <int dim, ArithmeticType ValueType, typename Derived>
  struct DofValueMixin
  {
    using Idx = ConservedVariableIndex<dim>;
 
    decltype(auto)
    density() const
    {
      return this->value()[Idx::density];
    }
 
    decltype(auto)
    momentum(const unsigned int component) const
    {
      AssertIndexRange(component, dim);
      return this->value()[Idx::momentum + component];
    }
 
    decltype(auto)
    momentum() const
    {
      dealii::Tensor<1, dim, ValueType> momentum;
      for (unsigned int d = 0; d < dim; ++d)
        momentum[d] = this->momentum(d);
      return momentum;
    }
 
    decltype(auto)
    velocity(const unsigned int component) const
    {
      AssertIndexRange(component, dim);
      return this->value()[Idx::momentum + component] / this->value()[Idx::density];
    }
 
    dealii::Tensor<1, dim, ValueType>
    velocity() const
    {
      dealii::Tensor<1, dim, ValueType> velocity;
      for (unsigned int d = 0; d < dim; ++d)
        velocity[d] = this->velocity(d);
      return velocity;
    }
 
    decltype(auto)
    total_energy() const
    {
      return this->value()[Idx::energy];
    }
 
    decltype(auto)
    specific_total_energy() const
    {
      return this->value()[Idx::energy] / this->value()[Idx::density];
    }
 
  private:
    decltype(auto)
    value() const
    {
      return static_cast<const Derived &>(*this).value();
    }
  };
 
  template <int dim, ArithmeticType ValueType, typename Derived>
  struct DofGradientMixin
  {
    using Idx = ConservedVariableIndex<dim>;
 
    decltype(auto)
    grad_density() const
    {
      return this->gradient_value()[Idx::density];
    }
 
    decltype(auto)
    grad_momentum(const unsigned int component) const
    {
      AssertIndexRange(component, dim);
      return this->gradient_value()[Idx::momentum + component];
    }
 
    decltype(auto)
    grad_momentum() const
    {
      dealii::Tensor<1, dim, dealii::Tensor<1, dim, ValueType>> grad_momentum;
      for (unsigned int d = 0; d < dim; ++d)
        grad_momentum[d] = this->grad_momentum(d);
      return grad_momentum;
    }
 
    decltype(auto)
    grad_velocity(const unsigned int component,
                  const ValueType   &density,
                  const ValueType   &velocity_component) const
    {
      AssertIndexRange(component, dim);
      return (this->gradient_value()[Idx::momentum + component] -
              velocity_component * this->gradient_value()[Idx::density]) /
             density;
    }
 
    decltype(auto)
    grad_velocity() const
    {
      dealii::Tensor<1, dim, dealii::Tensor<1, dim, ValueType>> grad_velocity;
      for (unsigned int d = 0; d < dim; ++d)
        grad_velocity[d] = this->grad_velocity(d, derived().density(), derived().velocity(d));
      return grad_velocity;
    }
 
    decltype(auto)
    grad_total_energy() const
    {
      return this->gradient_value()[Idx::energy];
    }
 
    decltype(auto)
    grad_specific_total_energy(const ValueType &density,
                               const ValueType &specific_total_energy) const
    {
      return (this->gradient_value()[Idx::energy] -
              specific_total_energy * this->gradient_value()[Idx::density]) /
             density;
    }
 
  private:
    decltype(auto)
    gradient_value() const
    {
      return static_cast<const Derived &>(*this).gradient_value();
    }
 
    decltype(auto)
    value_view() const
    {
      return static_cast<const Derived &>(*this).value_view();
    }
 
    const Derived &
    derived() const
    {
      return static_cast<const Derived &>(*this);
    }
  };
 
  template <int dim, typename Derived>
  struct FluxMixin
  {
    using Idx = ConservedVariableIndex<dim>;
 
    decltype(auto)
    density_flux() const
    {
      return get_value()[Idx::density];
    }
 
    decltype(auto)
    momentum_flux(const unsigned int component) const
    {
      return get_value()[Idx::momentum + component];
    }
 
    decltype(auto)
    energy_flux() const
    {
      return get_value()[Idx::energy];
    }
 
  private:
    decltype(auto)
    get_value() const
    {
      return static_cast<const Derived &>(*this).value();
    }
  };
} // namespace MeltPoolDG::CompressibleFlow