LocalViewPair.hpp

#pragma once

#include <amdis/functions/LocalView.hpp>

namespace AMDiS
{
  template <class RB, class CB>
  class LocalViewPair
  {
    using RowLocalView = LocalView<RB>;
    using ColLocalView = LocalView<CB>;

  public:
    LocalViewPair(RB const& rowBasis, CB const& colBasis)
      : rowLocalView_(rowBasis)
      , colLocalView_(colBasis)
    {}

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

    void unbind ()
    {
      rowLocalView_.unbind();
      colLocalView_.unbind();
    }

    RowLocalView const& row() const
    {
      return rowLocalView_;
    }

    ColLocalView const& col() const
    {
      return colLocalView_;
    }

  protected:
    RowLocalView rowLocalView_;
    ColLocalView colLocalView_;
  };

  template <class B>
  class LocalViewPair<B,B>
  {
    using LocalView = AMDiS::LocalView<B>;

  public:
    LocalViewPair(B const& rowBasis, B const& colBasis)
      : same_(&rowBasis == &colBasis)
      , rowLocalView_(rowBasis)
      , colLocalView_(same_
          ? Dune::stackobject_to_shared_ptr(rowLocalView_)
          : std::make_shared<LocalView>(colBasis))
    {}

    ~LocalViewPair() = default;
    LocalViewPair(LocalViewPair const&) = delete;
    LocalViewPair& operator=(LocalViewPair const&) = delete;
    LocalViewPair(LocalViewPair&&) = delete;
    LocalViewPair& operator=(LocalViewPair&&) = delete;

    template <class Element>
    void bind (Element const& element)
    {
      rowLocalView_.bind(element);
      if (!same_)
        colLocalView_->bind(element);
    }

    void unbind ()
    {
      rowLocalView_.unbind();
      if (!same_)
        colLocalView_->unbind();
    }

    LocalView const& row() const
    {
      return rowLocalView_;
    }

    LocalView const& col() const
    {
      return *colLocalView_;
    }

  protected:
    bool same_ = false;
    LocalView rowLocalView_;
    std::shared_ptr<LocalView> colLocalView_;
  };

} // end namespace AMDiS