p1localbasis.hh

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_P1_LOCALBASIS_HH
#define DUNE_P1_LOCALBASIS_HH
 
#include <array>
#include <numeric>
 
#include <dune/common/fmatrix.hh>
 
#include <dune/localfunctions/common/localbasis.hh>
 
namespace Dune
{
  template<class D, class R, int dim>
  class P1LocalBasis
  {
  public:
    typedef LocalBasisTraits<D,dim,Dune::FieldVector<D,dim>,R,1,Dune::FieldVector<R,1>,
        Dune::FieldMatrix<R,1,dim> > Traits;
 
    unsigned int size () const
    {
      return dim+1;
    }
 
    inline void evaluateFunction (const typename Traits::DomainType& in,
                                  std::vector<typename Traits::RangeType>& out) const
    {
      out.resize(size());
      out[0] = 1.0;
      for (size_t i=0; i<dim; i++) {
        out[0]  -= in[i];
        out[i+1] = in[i];
      }
    }
 
    inline void
    evaluateJacobian (const typename Traits::DomainType& in,         // position
                      std::vector<typename Traits::JacobianType>& out) const      // return value
    {
      out.resize(size());
 
      for (int i=0; i<dim; i++)
        out[0][0][i] = -1;
 
      for (int i=0; i<dim; i++)
        for (int j=0; j<dim; j++)
          out[i+1][0][j] = (i==j);
 
    }
 
    inline void partial(const std::array<unsigned int,dim>& order,
                        const typename Traits::DomainType& in,
                        std::vector<typename Traits::RangeType>& out) const
    {
      auto totalOrder = std::accumulate(order.begin(), order.end(), 0);
 
      if (totalOrder==0)
        evaluateFunction(in, out);
      else if (totalOrder==1)
      {
        auto direction = std::find(order.begin(), order.end(), 1);
        out.resize(size());
 
        out[0] = -1;
        for (int i=0; i<dim; i++)
          out[i+1] = (i==(direction-order.begin()));
      }
      else  // all higher order derivatives are zero
      {
        out.resize(size());
 
        for (int i=0; i<dim+1; i++)
          out[i] = 0;
      }
    }
 
    unsigned int order () const
    {
      return 1;
    }
  };
}
#endif