pk2d.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_PK2DLOCALFINITEELEMENT_HH
#define DUNE_PK2DLOCALFINITEELEMENT_HH
 
#include <cstddef>
 
#include <dune/geometry/type.hh>
 
#include <dune/localfunctions/common/localfiniteelementtraits.hh>
#include <dune/localfunctions/common/localtoglobaladaptors.hh>
#include <dune/localfunctions/lagrange/lagrangesimplex.hh>
 
namespace Dune
{
 
  template<class D, class R, unsigned int k>
  using Pk2DLocalFiniteElement = LagrangeSimplexLocalFiniteElement<D,R,2,k>;
 
 
 
  template<class Geometry, class RF, std::size_t k>
  class Pk2DFiniteElement {
    typedef typename Geometry::ctype DF;
    typedef Impl::LagrangeSimplexLocalBasis<DF,RF,2,k> LocalBasis;
    typedef Impl::LagrangeSimplexLocalInterpolation<LocalBasis> LocalInterpolation;
 
  public:
    struct Traits {
      typedef ScalarLocalToGlobalBasisAdaptor<LocalBasis, Geometry> Basis;
      typedef LocalToGlobalInterpolationAdaptor<
          LocalInterpolation,
          typename Basis::Traits
          > Interpolation;
      typedef Impl::LagrangeSimplexLocalCoefficients<2,k> Coefficients;
    };
 
  private:
    static const GeometryType gt;
    static const LocalBasis localBasis;
    static const LocalInterpolation localInterpolation;
 
    typename Traits::Basis basis_;
    typename Traits::Interpolation interpolation_;
    typename Traits::Coefficients coefficients_;
 
  public:
 
    template<class VertexOrder>
    Pk2DFiniteElement(const Geometry &geometry,
                      const VertexOrder& vertexOrder) :
      basis_(localBasis, geometry), interpolation_(localInterpolation),
      coefficients_(vertexOrder.begin(0, 0))
    { }
 
    const typename Traits::Basis& basis() const { return basis_; }
    const typename Traits::Interpolation& interpolation() const
    { return interpolation_; }
    const typename Traits::Coefficients& coefficients() const
    { return coefficients_; }
    const GeometryType &type() const { return gt; }
  };
 
  template<class Geometry, class RF, std::size_t k>
  const GeometryType
  Pk2DFiniteElement<Geometry, RF, k>::gt(GeometryTypes::simplex(2));
 
  template<class Geometry, class RF, std::size_t k>
  const typename Pk2DFiniteElement<Geometry, RF, k>::LocalBasis
  Pk2DFiniteElement<Geometry, RF, k>::localBasis = LocalBasis();
 
  template<class Geometry, class RF, std::size_t k>
  const typename Pk2DFiniteElement<Geometry, RF, k>::LocalInterpolation
  Pk2DFiniteElement<Geometry, RF, k>::localInterpolation =
    LocalInterpolation();
 
 
  template<class Geometry, class RF, std::size_t k>
  struct Pk2DFiniteElementFactory {
    typedef Pk2DFiniteElement<Geometry, RF, k> FiniteElement;
 
 
    template<class VertexOrder>
    const FiniteElement make(const Geometry& geometry,
                             const VertexOrder& vertexOrder)
    { return FiniteElement(geometry, vertexOrder); }
  };
}
 
#endif