lfeprebasismixin.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_LFEPREBASISMIXIN_HH
#define DUNE_FUNCTIONS_FUNCTIONSPACEBASES_LFEPREBASISMIXIN_HH
 
#include <cassert>
#include <type_traits>
 
#include <dune/common/exceptions.hh>
 
#include <dune/functions/functionspacebases/leafprebasismappermixin.hh>
#include <dune/functions/functionspacebases/nodes.hh>
 
#include <dune/grid/common/mcmgmapper.hh>
 
namespace Dune::Functions {
 
template <class GV, class LFE>
class LFEPreBasisMixin :
  public LeafPreBasisMapperMixin< GV >
{
  using Base = LeafPreBasisMapperMixin< GV >;
 
public:
 
  using GridView = GV;
 
  class Node;
 
  template <class LFE_ = LFE,
    std::enable_if_t<std::is_default_constructible_v<LFE_>, int> = 0>
  LFEPreBasisMixin (const GridView& gv, MCMGLayout layout)
    : Base(gv, layout)
    , lfe_{}
  {}
 
  template <class LFE_>
  LFEPreBasisMixin (const GridView& gv, LFE_&& lfe, MCMGLayout layout)
    : Base(gv, layout)
    , lfe_(std::forward<LFE_>(lfe))
  {}
 
  Node makeNode () const
  {
    return Node(lfe_);
  }
 
private:
  LFE lfe_;
};
 
// deduction guide
template <class GV, class LFE>
LFEPreBasisMixin(const GV&, const LFE&, MCMGLayout)
  -> LFEPreBasisMixin<GV,LFE>;
 
 
 
template <class GV, class LFE>
class LFEPreBasisMixin<GV,LFE>::Node
  : public LeafBasisNode
{
  static constexpr int dim = GV::dimension;
 
public:
  using size_type = std::size_t;
  using Element = typename GV::template Codim<0>::Entity;
  using FiniteElement = LFE;
 
  explicit Node (const LFE& lfe)
    : lfe_{&lfe}
    , element_{nullptr}
  {}
 
  const Element& element () const
  {
    assert(!!element_);
    return *element_;
  }
 
  const FiniteElement& finiteElement () const
  {
    assert(!!lfe_);
    return *lfe_;
  }
 
  void bind (const Element& e)
  {
    element_ = &e;
    this->setSize(lfe_->size());
  }
 
protected:
  const FiniteElement* lfe_;
  const Element* element_;
};
 
 
} // end namespace Dune::Functions
 
 
#endif // DUNE_FUNCTIONS_FUNCTIONSPACEBASES_LFEPREBASISMIXIN_HH