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/q1tbnp/dof_traits.hpp>
#include <kernel/space/q1tbnp/evaluator.hpp>
#include <kernel/space/q1tbnp/node_functional.hpp>
 
namespace FEAT
{
  namespace Space
  {
    namespace Q1TBNP
    {
      namespace Intern
      {
        template<typename Shape_>
        struct LocalDegree;
 
        template<int dim_>
        struct LocalDegree<Shape::Hypercube<dim_>>
        {
          static constexpr int value = 2;
        };
      } // namespace Intern
 
      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 = true;
 
        static constexpr int local_degree = Intern::LocalDegree<ShapeType>::value;
 
        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, ShapeType::dimension-1>::count + 2*ShapeType::dimension - 3;
 
          typedef StandardScalarEvalTraits<EvalPolicy, num_loc_dofs, DataType_> Traits;
 
        public:
          typedef Q1TBNP::Evaluator<Element, TrafoEvaluator_, Traits> Type;
        };
 
        typedef DofMappingUniform<Element, Q1TBNP::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
        {
        private:
          static constexpr int codim = ShapeType::dimension - shape_dim_;
 
        public:
          typedef Q1TBNP::NodeFunctional<Element, ShapeType, codim, DataType_> Type;
        };
 
      public:
        explicit Element(TrafoType& trafo) :
          BaseClass(trafo)
        {
        }
 
        virtual ~Element()
        {
        }
 
        Index get_num_dofs() const
        {
          // number of DOFs = number of facets in the mesh
          return this->get_mesh().get_num_entities(ShapeType::dimension - 1)
            + Index(ShapeType::dimension == 3 ? 3 : 1) * this->get_mesh().get_num_elements();
        }
 
        static String name()
        {
          return "Q1TBNP";
        }
      }; // class Element
    } // namespace Q1TBNP
  } // namespace Space
} // namespace FEAT