// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_LOCALFUNCTIONS_BREZZIDOUGLASMARINI2_SIMPLEX2D_LOCALCOEFFICIENTS_HH
#define DUNE_LOCALFUNCTIONS_BREZZIDOUGLASMARINI2_SIMPLEX2D_LOCALCOEFFICIENTS_HH

#include <cstddef>
#include <vector>

#include "../../common/localkey.hh"

namespace Dune
{

  /**
   * \brief Layout map for Brezzi-Douglas-Marini-2 elements on triangles.
   *
   * \ingroup LocalLayoutImplementation
   * \nosubgrouping
   * \implements Dune::LocalCoefficientsVirtualImp
   */
  class BDM2Simplex2DLocalCoefficients
  {

  public:
    //! \brief Standard constructor
    BDM2Simplex2DLocalCoefficients() : li(12)
    {
      for (std::size_t i = 0; i < 3; ++i)
      {
        li[3 * i] = LocalKey(i,1,0);
        li[3 * i + 1] = LocalKey(i,1,1);
        li[3 * i + 2] = LocalKey(i,1,2);
      }

      // last DOFs are associated with the cell (codim=0)
      li[9]  = LocalKey(0,0,0);
      li[10] = LocalKey(0,0,1);
      li[11] = LocalKey(0,0,2);
    }

    //! \brief number of coefficients
    std::size_t size() const
    {
      return 12;
    }

    //! \brief get i'th index
    const LocalKey& localKey(std::size_t i) const
    {
      return li[i];
    }

  private:
    std::vector<LocalKey> li;
  };
} // end namespace Dune
#endif // DUNE_LOCALFUNCTIONS_BREZZIDOUGLASMARINI2_SIMPLEX2D_LOCALCOEFFICIENTS_HH
