leafprebasismappermixin.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_FUNCTIONS_FUNCTIONSPACEBASES_LEAFPREBASISMAPPERMIXIN_HH
#define DUNE_FUNCTIONS_FUNCTIONSPACEBASES_LEAFPREBASISMAPPERMIXIN_HH
 
#include <dune/common/rangeutilities.hh>
 
#include <dune/functions/functionspacebases/leafprebasismixin.hh>
 
#include <dune/geometry/referenceelements.hh>
 
#include <dune/grid/common/mcmgmapper.hh>
 
 
 
namespace Dune::Functions {
 
 
 
// Helper function returning a random access range
// of global indices associated to the element. The ordering
// is derived from the LocalCoefficients object.
// Having this as a member of MCMGMapper would be nice.
// But this would introduce the LocalCoefficient in dune-grid.
// This would introduce at least a weak 'conceptual' dependency problem.
template<class GridView, class LocalCoefficients>
auto subIndexRange(const Dune::MultipleCodimMultipleGeomTypeMapper<GridView>& mapper, const typename GridView::template Codim<0>::Entity& element, const LocalCoefficients& localCoefficients)
{
  // Here we make use of the 'hidden' (poorly documented) MCMGMapper feature to support
  // multiple DOFs per subentity. However, we do not take care for any reordering.
  return Dune::transformedRangeView(Dune::range(localCoefficients.size()), [&](auto localIndex) {
    auto localKey = localCoefficients.localKey(localIndex);
    return mapper.subIndex(element, localKey.subEntity(), localKey.codim()) + localKey.index();
  });
}
 
 
 
template<typename GV>
class LeafPreBasisMapperMixin
    : public LeafPreBasisMixin<LeafPreBasisMapperMixin<GV>>
{
  static const int gridDim = GV::dimension;
 
public:
 
  using GridView = GV;
 
  using size_type = std::size_t;
 
  LeafPreBasisMapperMixin(const GridView& gv, Dune::MCMGLayout layout) :
    gridView_(gv),
    mapper_(gridView_, std::move(layout))
  {}
 
  void initializeIndices()
  {
    // Determine upper bound for node size by traversing
    // the layout for all element types in the grid
    maxNodeSize_ = 0;
    for(const GeometryType& elementType : gridView_.indexSet().types(0))
    {
      auto referenceElement = Dune::referenceElement<double, gridDim>(elementType);
      for(auto codim : Dune::range(gridDim+1))
        for(auto i : Dune::range(referenceElement.size(codim)))
          maxNodeSize_ += mapper_.layout()(referenceElement.type(i, codim), gridDim);
    }
  }
 
  const GridView& gridView() const
  {
    return gridView_;
  }
 
  void update(const GridView& gv)
  {
    gridView_ = gv;
    mapper_.update(gridView_);
  }
 
  size_type dimension() const
  {
    return mapper_.size();
  }
 
  size_type maxNodeSize() const
  {
    return maxNodeSize_;
  }
 
  template<class Node, class It>
  It indices(const Node& node, It it) const
  {
    for(const auto& globalIndex : subIndexRange(mapper_, node.element(), node.finiteElement().localCoefficients()))
    {
      *it = {{ (size_type)globalIndex }};
      ++it;
    }
    return it;
  }
 
protected:
  GridView gridView_;
  Dune::MultipleCodimMultipleGeomTypeMapper<GridView> mapper_;
  std::size_t maxNodeSize_;
};
 
 
 
} // end namespace Dune::Functions
 
#endif // DUNE_FUNCTIONS_FUNCTIONSPACEBASES_LEAFPREBASISMAPPERMIXIN_HH