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/bernstein2/dof_traits.hpp>
#include <kernel/space/bernstein2/evaluator.hpp>
#include <kernel/space/bernstein2/node_functional.hpp>
 
namespace FEAT
{
  namespace Space
  {
    namespace Bernstein2
    {
      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 bool have_node_func = 1;
 
        static constexpr int local_degree = 2;
 
        typedef DofMappingUniform<Element, Bernstein2::DofTraits, ShapeType> DofMappingType;
 
        template<
          typename TrafoEvaluator_,
          typename DataType_ = typename TrafoEvaluator_::DataType>
        class Evaluator
        {
        private:
          typedef typename TrafoEvaluator_::EvalPolicy EvalPolicy;
 
          static constexpr int num_loc_dofs = DofMappingType::dof_count;
 
          typedef StandardScalarEvalTraits<EvalPolicy, num_loc_dofs, DataType_> Traits;
 
        public:
          typedef Bernstein2::Evaluator<Element, TrafoEvaluator_, Traits> Type;
        };
 
        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
        {
          typedef typename Shape::FaceTraits<ShapeType, shape_dim_>::ShapeType FaceType;
        public:
          typedef Bernstein2::NodeFunctional<Element, FaceType, DataType_> Type;
        };
 
      public:
        explicit Element(TrafoType& trafo)
          : BaseClass(trafo)
        {
        }
 
        virtual ~Element()
        {
        }
 
        Index get_num_dofs() const
        {
          DofMappingType dof_map(*this);
          return dof_map.get_num_global_dofs();
        }
 
        static String name()
        {
          return "Bernstein2";
        }
      }; // class Element
    } // namespace Bernstein2
  } // namespace Space
} // namespace FEAT