VectorBase.hpp

#pragma once

#include <iterator>

#include <dune/common/rangeutilities.hh>
#include <dune/functions/functionspacebases/flatmultiindex.hh>

#include <amdis/common/FakeContainer.hpp>
#include <amdis/utility/MappedRangeView.hpp>

namespace AMDiS
{
  template <class Derived>
  class VectorBase
  {
  private:
    Derived const& asDerived() const
    {
      return static_cast<Derived const&>(*this);
    }

    Derived& asDerived()
    {
      return static_cast<Derived&>(*this);
    }

  protected:
    VectorBase() = default;

  public:
    void finish() { /* do nothing */ }
    void synchronize() { /* do nothing */ }

    template <class MultiIndex, class ValueType, class Assign>
    void insert(MultiIndex const& idx, ValueType const& value, Assign assign)
    {
      assign(value, asDerived().at(idx));
    }

    template <class IndexRange, class OutputIterator>
    void gather(IndexRange const& localInd, OutputIterator buffer) const
    {
      for (auto const& idx : localInd)
        *buffer++ = asDerived().at(idx);
    }


    template <class IndexRange, class LocalVec, class Assign>
    void scatter(IndexRange const& localInd, LocalVec const& vec, FakeContainer<bool,true>, Assign assign)
    {
      auto vec_it = std::begin(vec);
      for (auto const& idx : localInd)
        assign(*vec_it++, asDerived().at(idx));
    }

    template <class IndexRange, class LocalVec, class MaskRange, class Assign>
    void scatter(IndexRange const& localInd, LocalVec const& vec, MaskRange const& mask, Assign assign)
    {
      auto vec_it = std::begin(vec);
      auto mask_it = std::begin(mask);
      auto ind_it = std::begin(localInd);
      for (; vec_it != std::end(vec); ++vec_it, ++mask_it, ++ind_it) {
        if (*mask_it)
          assign(*vec_it, asDerived().at(*ind_it));
      }
    }


    template <class IndexRange, class Func>
    void forEach(IndexRange const& localInd, Func&& f) const
    {
      for (auto const& idx : localInd)
        f(idx, asDerived().at(idx));
    }

    template <class IndexRange, class Func>
    void forEach(IndexRange const& localInd, Func&& f)
    {
      for (auto const& idx : localInd)
        f(idx, asDerived().at(idx));
    }


    template <class Func>
    void forEach(Func&& f) 
    {
      forEach(mappedRangeView(Dune::range(asDerived().size()),
        [](auto i) { return Dune::Functions::FlatMultiIndex<decltype(i)>{i}; }), FWD(f));
    }

    template <class Func>
    void forEach(Func&& f) const 
    {
      forEach(mappedRangeView(Dune::range(asDerived().size()),
        [](auto i) { return Dune::Functions::FlatMultiIndex<decltype(i)>{i}; }), FWD(f));
    }
  };

} // end namespace AMDiS