q1.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
 
#ifndef DUNE_Q1_LOCALFINITEELEMENT_HH
#define DUNE_Q1_LOCALFINITEELEMENT_HH
 
#include <dune/geometry/type.hh>
 
#include <dune/localfunctions/common/localfiniteelementtraits.hh>
#include <dune/localfunctions/common/localtoglobaladaptors.hh>
#include <dune/localfunctions/lagrange/lagrangecube.hh>
 
namespace Dune
{
 
  // The test test-q1.cc triggers compiler bugs in gcc-6 and earlier when Q1LocalFiniteElement
  // is simply redefined as LagrangeCubeLocalFiniteElement: it grabs more and more main memory,
  // and eventually stalls the machine.  Since I didn't find the real cause for this let's just
  // keep the relevant parts of Q1LocalFiniteElement until we can retire the problematic gcc versions.
#if !defined __GNUC__ || __GNUC__ > 6
  template<class D, class R, int dim>
  using Q1LocalFiniteElement = LagrangeCubeLocalFiniteElement<D,R,dim,1>;
#else
  template <int dim>
  class Q1LocalCoefficients
  {
  public:
    Q1LocalCoefficients () : li(1<<dim)
    {
      for (std::size_t i=0; i<(1<<dim); i++)
        li[i] = LocalKey(i,dim,0);
    }
 
    std::size_t size () const
    {
      return 1<<dim;
    }
 
    const LocalKey& localKey (std::size_t i) const
    {
      return li[i];
    }
 
  private:
    std::vector<LocalKey> li;
  };
 
  template<class D, class R, int dim>
  class Q1LocalFiniteElement
  {
  public:
    // user-defined default constructor is required for clang 3.8,
    // see https://gitlab.dune-project.org/core/dune-localfunctions/merge_requests/60
    Q1LocalFiniteElement() {}
 
    typedef LocalFiniteElementTraits<Impl::LagrangeCubeLocalBasis<D,R,dim,1>,Q1LocalCoefficients<dim>,
        Impl::LagrangeCubeLocalInterpolation<Impl::LagrangeCubeLocalBasis<D,R,dim,1> > > Traits;
 
    const typename Traits::LocalBasisType& localBasis () const
    {
      return basis;
    }
 
    const typename Traits::LocalCoefficientsType& localCoefficients () const
    {
      return coefficients;
    }
 
    const typename Traits::LocalInterpolationType& localInterpolation () const
    {
      return interpolation;
    }
 
    unsigned int size () const
    {
      return basis.size();
    }
 
    static constexpr GeometryType type ()
    {
      return GeometryTypes::cube(dim);
    }
 
  private:
    Impl::LagrangeCubeLocalBasis<D,R,dim,1> basis;
    Q1LocalCoefficients<dim> coefficients;
    Impl::LagrangeCubeLocalInterpolation<Impl::LagrangeCubeLocalBasis<D,R,dim,1> > interpolation;
  };
#endif
 
 
  template<class Geometry, class RF>
  class Q1FiniteElementFactory :
    public ScalarLocalToGlobalFiniteElementAdaptorFactory<
#if !defined __GNUC__ || __GNUC__ > 6
        LagrangeCubeLocalFiniteElement<
            typename Geometry::ctype, RF, Geometry::mydimension, 1
            >,
#else
        Q1LocalFiniteElement<
            typename Geometry::ctype, RF, Geometry::mydimension
            >,
#endif
        Geometry
        >
  {
#if !defined __GNUC__ || __GNUC__ > 6
    typedef LagrangeCubeLocalFiniteElement<
        typename Geometry::ctype, RF, Geometry::mydimension, 1
        > LFE;
#else
    typedef Q1LocalFiniteElement<
        typename Geometry::ctype, RF, Geometry::mydimension
        > LFE;
#endif
    typedef ScalarLocalToGlobalFiniteElementAdaptorFactory<LFE, Geometry> Base;
 
    static const LFE lfe;
 
  public:
    Q1FiniteElementFactory() : Base(lfe) {}
  };
 
  template<class Geometry, class RF>
  const typename Q1FiniteElementFactory<Geometry, RF>::LFE
  Q1FiniteElementFactory<Geometry, RF>::lfe;
}
 
#endif