dofhandler_wrapper.h

Go to the documentation of this file.

#ifndef DOPE_DOFHANDLER_H_
#define DOPE_DOFHANDLER_H_

#include <deal.II/dofs/dof_handler.h>
#include <deal.II/hp/dof_handler.h>
#include <deal.II/fe/fe_system.h>

namespace DOpEWrapper
{

  template<int dim,
      template<int DIM, int spacedim> class DOFHANDLER = dealii::DoFHandler>
    class DoFHandler : public DOFHANDLER<dim, dim>
    {
      public:
        DoFHandler(const dealii::Triangulation<dim, dim> &tria) :
            DOFHANDLER<dim, dim>(tria)
        {
        }

        const DOFHANDLER<dim, dim>&
        GetDEALDoFHandler() const
        {
          return *this;
        }

        static bool
        NeedIndexSetter();

    };

  //Template specialization DOFHANDLER = dealii::DoFHandler<dim>
  template<int dim>
    class DoFHandler<dim, dealii::DoFHandler> : public dealii::DoFHandler<dim>
    {
      public:
        DoFHandler(const dealii::Triangulation<dim, dim> &tria) :
            dealii::DoFHandler<dim>(tria)
        {
        }
        static bool
        NeedIndexSetter()
        {
          return false;
        }
        const dealii::DoFHandler<dim>&
        GetDEALDoFHandler() const
        {
          return *this;
        }

    };

  //Template specialization DOFHANDLER = dealii::hp::DoFHandler<dim>
  template<int dim>
    class DoFHandler<dim, dealii::hp::DoFHandler> : public dealii::hp::DoFHandler<
        dim>
    {
      public:
        DoFHandler(const dealii::Triangulation<dim, dim> &tria) :
            dealii::hp::DoFHandler<dim>(tria)
        {
        }
        static bool
        NeedIndexSetter()
        {
          return true;
        }
        const dealii::hp::DoFHandler<dim>&
        GetDEALDoFHandler() const
        {
          return *this;
        }
    };

// //Template specialization DOFHANDLER = dealii::MGDoFHandler<dim>
//  template<int dim>
//    class DoFHandler<dim, dealii::MGDoFHandler > : public dealii::MGDoFHandler<
//        dim>
//    {
//      public:
//        DoFHandler(const dealii::Triangulation<dim, dim> &tria)
//            : dealii::MGDoFHandler<dim>(tria)
//        {
//        }
//        static bool
//        NeedIndexSetter()
//        {
//          return true;
//        }
//        const dealii::MGDoFHandler<dim>&
//        GetDEALDoFHandler() const
//        {
//          return *this;
//        }
//    };

  template<>
    class DoFHandler<0, dealii::DoFHandler>
    {
      private:
        unsigned int dofs_;

      public:
        template<int dim>
          DoFHandler(const dealii::Triangulation<dim, dim> &/*tria*/)
          {
          }
        template<int dim>
          void
          distribute_dofs(const dealii::FESystem<dim> &fe,
                          const unsigned int /*offset*/=0)
          {
            dofs_ = fe.element_multiplicity(0);
          }
        void
        clear()
        {
        }
        unsigned int
        n_dofs() const
        {
          return dofs_;
        }
        static bool
        NeedIndexSetter()
        {
          return false;
        }
    };

  template<>
    class DoFHandler<0, dealii::hp::DoFHandler>
    {
      private:
        unsigned int dofs_;

      public:
        template<int dim>
          DoFHandler(const dealii::Triangulation<dim, dim> &/*tria*/)
          {
          }
        template<int dim>
          void
          distribute_dofs(const dealii::hp::FECollection<dim> &fe_collection,
                          const unsigned int /*offset*/ = 0)
          {
            dofs_ = fe_collection[0].element_multiplicity(0);
          }
        void
        clear()
        {
        }
        unsigned int
        n_dofs() const
        {
          return dofs_;
        }
        static bool
        NeedIndexSetter()
        {
          return false;
        }
    };
}

#endif