amr_indicators.hpp

Go to the documentation of this file.

#pragma once
 
#include <deal.II/base/exceptions.h>
#include <deal.II/base/vectorization.h>
 
#include <deal.II/grid/tria.h>
 
#include <deal.II/lac/la_parallel_vector.h>
#include <deal.II/lac/vector.h>
 
#include <deal.II/matrix_free/matrix_free.h>
 
#include <meltpooldg/utilities/fe_integrator.hpp>
#include <meltpooldg/utilities/matrix_free_util.hpp>
 
#include <memory>
#include <utility>
#include <vector>
 
namespace MeltPoolDG::AMR
{
  template <int dim, typename number>
  class AMRIndicatorBase
  {
  public:
    virtual dealii::Vector<number>
    compute_indicator(const dealii::Triangulation<dim> &tria) = 0;
 
    virtual ~AMRIndicatorBase() = default;
  };
 
  template <int dim, typename number, template <typename> class BinaryOp = std::plus>
  class BinaryOpIndicatorComposite : public AMRIndicatorBase<dim, number>
  {
    using VectorType = dealii::Vector<number>;
 
  public:
    void
    add_indicator(std::unique_ptr<AMRIndicatorBase<dim, number>> &&indicator, number weight = 1.);
 
    VectorType
    compute_indicator(const dealii::Triangulation<dim> &tria) override;
 
    bool
    empty() const;
 
  private:
    struct IndicatorAndWeight
    {
      std::unique_ptr<AMRIndicatorBase<dim, number>> indicator;
 
      number weight;
    };
 
    std::vector<IndicatorAndWeight> indicators_and_weights;
  };
 
  template <int dim, typename number, int n_dof_components, int n_indicator_components>
  class SSEDIndicator : public AMRIndicatorBase<dim, number>
  {
    using VectorizedArrayType = dealii::VectorizedArray<number>;
 
    using VectorType = dealii::LinearAlgebra::distributed::Vector<number>;
 
    using ExtractIndicatorDofValuesFunction =
      std::function<dealii::Tensor<1, n_indicator_components, VectorizedArrayType>(
        const dealii::Tensor<1, n_dof_components, VectorizedArrayType> &)>;
 
  public:
    SSEDIndicator(const MatrixFreeContext<dim, number>    matrix_free_context,
                  const VectorType                       &solution,
                  const ExtractIndicatorDofValuesFunction extract_indicator_dof_values,
                  const unsigned                          fe_index,
                  const unsigned                          fe_degree);
 
    dealii::Vector<number>
    compute_indicator(const dealii::Triangulation<dim> &tria) override;
 
  protected:
    void
    local_apply_cell(const dealii::MatrixFree<dim, number> &,
                     dealii::AlignedVector<VectorizedArrayType> &dst,
                     const VectorType                           &src,
                     const std::pair<unsigned, unsigned>        &cell_range);
 
  private:
    const MatrixFreeContext<dim, number> matrix_free_context;
 
    const VectorType &solution;
 
    ExtractIndicatorDofValuesFunction extract_indicator_dof_values;
 
    const unsigned fe_index;
 
    dealii::AlignedVector<VectorizedArrayType> linearized_matrix;
  };
 
  template <int dim, typename number, int n_mf_components = dim, int n_relevant_components = dim>
  class JumpIndicator : public AMRIndicatorBase<dim, number>
  {
  public:
    using VectorizedArrayType = dealii::VectorizedArray<number>;
 
    using VectorType = dealii::LinearAlgebra::distributed::Vector<number>;
    using ExtractIndicatorDofValuesFunction =
      std::function<dealii::Tensor<1, n_relevant_components, VectorizedArrayType>(
        const dealii::Tensor<1, n_mf_components, VectorizedArrayType> &)>;
 
    JumpIndicator(const MatrixFreeContext<dim, number>    matrix_free_context,
                  const VectorType                       &solution,
                  const ExtractIndicatorDofValuesFunction extract_indicator_dof_values,
                  const unsigned                          fe_index = 0);
 
    dealii::Vector<number>
    compute_indicator(const dealii::Triangulation<dim> &tria) override;
 
  protected:
    void
    local_apply_face(const dealii::MatrixFree<dim, number> &,
                     dealii::AlignedVector<VectorizedArrayType> &dst,
                     const VectorType                           &src,
                     const std::pair<unsigned, unsigned>        &face_range) const;
 
  private:
    const MatrixFreeContext<dim, number> matrix_free_context;
 
    const VectorType &solution;
 
    ExtractIndicatorDofValuesFunction extract_indicator_dof_values;
 
    const unsigned fe_index;
  };
} // namespace MeltPoolDG::AMR