Preconditioner.hpp

#pragma once

#include <amdis/linearalgebra/mtl/PreconditionerInterface.hpp>

namespace AMDiS
{
  template <class Mat, class Vec, template <class> class PreconImpl>
  class Preconditioner
      : public PreconditionerInterface<Mat, Vec>
  {
    using M = typename Mat::BaseMatrix;
    using X = typename Vec::BaseVector;
    using Y = typename Vec::BaseVector;

    using Self = Preconditioner;
    using Super = PreconditionerInterface<Mat, Vec>;
    using Precon = PreconImpl<M>;

  public:
    struct Creator : CreatorInterfaceName<Super>
    {
      std::unique_ptr<Super> createWithString(std::string) override
      {
        return std::make_unique<Self>();
      }
    };

  public:
    void init(M const& A) override
    {
      precon_ = std::make_shared<Precon>(A);
    }

    void exit() override
    {
      precon_.reset();
    }

    void solve(X const& x, Y& y) const override
    {
      test_exit_dbg(bool(precon_), "No preconditioner initialized!");
      precon_->solve(x, y);
    }

    void adjoint_solve(X const& x, Y& y) const override
    {
      test_exit_dbg(bool(precon_), "No preconditioner initialized!");
      precon_->adjoint_solve(x, y);
    }

  private:
    std::shared_ptr<Precon> precon_;
  };

} // namespace AMDiS