OperatorAdapter.hpp

#pragma once

#include <dune/typetree/childextraction.hh>

namespace AMDiS
{
  template <class LocalOperator, class RowTreePath, class ColTreePath = RowTreePath>
  class LocalOperatorAdapter
  {
  public:
    LocalOperatorAdapter(LocalOperator const& lop, RowTreePath rowTp, ColTreePath colTp = {})
      : lop_(lop)
      , rowTp_(rowTp)
      , colTp_(colTp)
    {}


    template <class Element>
    void bind(Element const& element)
    {
      lop_.bind(element);
    }

    void unbind()
    {
      lop_.unbind();
    }


    template <class ContextGeo, class RowNode, class ColNode, class Mat>
    void assemble(ContextGeo const& cg, RowNode const& row, ColNode const& col, Mat& mat) const
    {
      using Dune::TypeTree::child;
      lop_.assemble(cg,child(row,rowTp_),child(col,colTp_),mat);
    }


    template <class ContextGeo, class Node, class Vec>
    void assemble(ContextGeo const& cg, Node const& node, Vec& vec) const
    {
      using Dune::TypeTree::child;
      lop_.assemble(cg,child(node,rowTp_),vec);
    }

  private:
    LocalOperator lop_;
    RowTreePath rowTp_;
    ColTreePath colTp_;
  };


  template <class Operator, class RowTreePath, class ColTreePath = RowTreePath>
  class OperatorAdapter
  {
  public:
    OperatorAdapter(Operator op, RowTreePath rowTp, ColTreePath colTp = {})
      : op_(std::move(op))
      , rowTp_(rowTp)
      , colTp_(colTp)
    {}

    template <class GridView>
    void update(GridView const& gv)
    {
      op_.update(gv);
    }

    friend auto localOperator(OperatorAdapter const& adapter)
    {
      return LocalOperatorAdapter{localOperator(adapter.op_),adapter.rowTp_,adapter.colTp_};
    }

  private:
    Operator op_;
    RowTreePath rowTp_;
    ColTreePath colTp_;
  };

} // end namespace AMDiS