element.hpp

// FEAT3: Finite Element Analysis Toolbox, Version 3
// Copyright (C) 2010 by Stefan Turek & the FEAT group
// FEAT3 is released under the GNU General Public License version 3,
// see the file 'copyright.txt' in the top level directory for details.
 
#pragma once
 
// includes, FEAT
#include <kernel/space/element_base.hpp>
#include <kernel/space/dof_assignment_base.hpp>
#include <kernel/space/dof_mapping_common.hpp>
#include <kernel/space/hermite3/dof_traits.hpp>
#include <kernel/space/hermite3/evaluator.hpp>
#include <kernel/space/hermite3/node_functional.hpp>
 
namespace FEAT
{
  namespace Space
  {
    namespace Hermite3
    {
      template<typename Trafo_>
      class Element :
        public ElementBase<Trafo_>
      {
      public:
        typedef ElementBase<Trafo_> BaseClass;
        typedef Trafo_ TrafoType;
        typedef typename TrafoType::MeshType MeshType;
        typedef typename TrafoType::ShapeType ShapeType;
 
        static constexpr int num_vert_dofs = DofTraits<ShapeType, 0>::count;
        static constexpr int num_cell_dofs = DofTraits<ShapeType, ShapeType::dimension>::count;
 
        static constexpr bool have_node_func = true;
 
        static constexpr int local_degree = 3;
 
        template<
          typename TrafoEvaluator_,
          typename DataType_ = typename TrafoEvaluator_::DataType>
        class Evaluator
        {
        private:
          typedef typename TrafoEvaluator_::EvalPolicy EvalPolicy;
 
          static constexpr int num_loc_dofs = Shape::FaceTraits<ShapeType, 0>::count * num_vert_dofs + num_cell_dofs;
 
          typedef StandardScalarEvalTraits<EvalPolicy, num_loc_dofs, DataType_> Traits;
 
        public:
          typedef Hermite3::Evaluator<Element, TrafoEvaluator_, Traits> Type;
        };
 
        typedef DofMappingUniform<Element, DofTraits, ShapeType> DofMappingType;
 
        template<
          int shape_dim_,
          typename DataType_ = Real>
        class DofAssignment
        {
        public:
          typedef DofAssignmentUniform<Element, shape_dim_, DataType_, DofTraits, ShapeType> Type;
        };
 
        template<
          int shape_dim_,
          typename DataType_ = Real>
        class NodeFunctional
        {
        public:
          typedef Hermite3::NodeFunctional<Element, ShapeType, shape_dim_, DataType_> Type;
        };
 
      public:
        explicit Element(TrafoType& trafo)
          : BaseClass(trafo)
        {
        }
 
        virtual ~Element()
        {
        }
 
        Index get_num_dofs() const
        {
          return
            // number of vertex dofs
            Index(num_vert_dofs) * this->get_mesh().get_num_entities(0) +
            // number of cell dofs
            Index(num_cell_dofs) * this->get_mesh().get_num_entities(ShapeType::dimension);
        }
 
        static String name()
        {
          return "Hermite3";
        }
      }; // class Element<...>
    } // namespace Hermite3
  } // namespace Space
} // namespace FEAT