preconditioner_jacobi.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <meltpooldg/core/scratch_data.hpp>
 
#include <any>
 
namespace MeltPoolDG
{
  template <typename OperatorType, typename VectorType>
  concept JacobiPreconditionerOperatorType = requires(const OperatorType op, VectorType &diag) {
    op.compute_inverse_diagonal_from_matrixfree(diag);
  };
 
 
  template <int dim,
            typename number,
            typename VectorType,
            JacobiPreconditionerOperatorType<VectorType> OperatorType>
  class JacobiPreconditioner
  {
  public:
    explicit JacobiPreconditioner(const OperatorType                  &operator_in,
                                  const ScratchData<dim, dim, number> &scratch_data_in,
                                  const unsigned                       dof_idx_in)
      : eq_operator(operator_in)
      , scratch_data(scratch_data_in)
      , dof_idx(dof_idx_in)
    {}
 
    void
    vmult(VectorType &dst, const VectorType &src) const
    {
      dst = src;
      dst.scale(inverse_diag);
    }
 
    void
    update(const std::any & = std::any())
    {
      eq_operator.compute_inverse_diagonal_from_matrixfree(inverse_diag);
    }
 
    void
    reinit()
    {
      scratch_data.initialize_dof_vector(inverse_diag, dof_idx);
    }
 
  private:
    VectorType inverse_diag;
 
    const OperatorType &eq_operator;
 
    const ScratchData<dim, dim, number> &scratch_data;
 
    const unsigned int dof_idx;
  };
} // namespace MeltPoolDG