pk1d.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
 
// Pk in one dimension with k as runtime variable
 
#ifndef DUNE_PDELAB_FINITEELEMENTMAP_PK1D_HH
#define DUNE_PDELAB_FINITEELEMENTMAP_PK1D_HH
 
#include<dune/localfunctions/lagrange.hh>
#include<dune/localfunctions/lagrange/equidistantpoints.hh>
 
#include<dune/pdelab/finiteelementmap/finiteelementmap.hh>
 
namespace Dune {
 
  namespace PDELab {
 
    template<class D, class R>
    class Pk1dLocalFiniteElementMap
      : public Dune::PDELab::SimpleLocalFiniteElementMap<LagrangeLocalFiniteElement<EquidistantPointSet,1,D,R>,1>
    {
    public:
 
      Pk1dLocalFiniteElementMap (std::size_t k)
        : Dune::PDELab::SimpleLocalFiniteElementMap<LagrangeLocalFiniteElement<EquidistantPointSet,1,D,R>,1>(LagrangeLocalFiniteElement<EquidistantPointSet,1,D,R>(GeometryTypes::cube(1),k))
        , _k(k)
      {}
 
      static constexpr bool fixedSize()
      {
        return true;
      }
 
      bool hasDOFs(int codim) const
      {
        switch (codim)
          {
          case 0:
            return _k != 1;
          case 1:
            return _k > 0;
          }
        return false;
      }
 
      std::size_t size(GeometryType gt) const
      {
        if (gt.isVertex())
          return _k > 0 ? 1 : 0;
        if (gt.isLine())
          return _k > 0 ? _k - 1 : 1;
        return 0;
      }
 
      std::size_t maxLocalSize() const
      {
        return _k + 1;
      }
 
    private:
      const std::size_t _k;
    };
  }
}
#endif // DUNE_PDELAB_FINITEELEMENTMAP_PK1D_HH