diagonal_precond.hpp

// FEAT3: Finite Element Analysis Toolbox, Version 3
// Copyright (C) 2010 by Stefan Turek & the FEAT group
// FEAT3 is released under the GNU General Public License version 3,
// see the file 'copyright.txt' in the top level directory for details.
 
#pragma once
 
// includes, FEAT
#include <kernel/solver/base.hpp>
 
namespace FEAT
{
  namespace Solver
  {
    template<typename Vector_, typename Filter_>
    class DiagonalPrecond :
      public SolverBase<Vector_>
    {
    public:
      typedef SolverBase<Vector_> BaseClass;
      typedef Vector_ VectorType;
      typedef Filter_ FilterType;
 
    protected:
      const VectorType& _diag;
      const FilterType& _filter;
 
    public:
      explicit DiagonalPrecond(const VectorType& diag, const FilterType& filter) :
        _diag(diag),
        _filter(filter)
      {
      }
 
      explicit DiagonalPrecond(const String& section_name, const PropertyMap* section,
      const VectorType& diag, const FilterType& filter) :
        BaseClass(section_name, section),
        _diag(diag),
        _filter(filter)
      {
      }
 
      virtual String name() const override
      {
        return "Diagonal";
      }
 
      virtual Status apply(VectorType& vec_cor, const VectorType& vec_def) override
      {
        vec_cor.component_product(_diag, vec_def);
        this->_filter.filter_cor(vec_cor);
        return Status::success;
      }
    }; // class DiagonalPrecond<...>
 
    template<typename Vector_, typename Filter_>
    inline std::shared_ptr<DiagonalPrecond<Vector_, Filter_>> new_diagonal_precond(
      const Vector_& diag, const Filter_& filter)
    {
      return std::make_shared<DiagonalPrecond<Vector_, Filter_>>(diag, filter);
    }
 
    template<typename Vector_, typename Filter_>
    inline std::shared_ptr<DiagonalPrecond<Vector_, Filter_>> new_diagonal_precond(
      const String& section_name, const PropertyMap* section,
      const Vector_& diag, const Filter_& filter)
    {
      return std::make_shared<DiagonalPrecond<Vector_, Filter_>>(section_name, section, diag, filter);
    }
  } // namespace Solver
} // namespace FEAT