reference_cell_factory.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/geometry/conformal_mesh.hpp>
#include <kernel/geometry/intern/macro_index_mapping.hpp>
 
namespace FEAT
{
  namespace Geometry
  {
    namespace Intern
    {
      template<typename Shape_>
      struct RefCellVertexer;
    }
 
    template<typename Shape_, typename CoordType_ = Real>
    class ReferenceCellFactory :
      public Factory< ConformalMesh<Shape_, Shape_::dimension, CoordType_> >
    {
    public:
      typedef ConformalMesh<Shape_, Shape_::dimension, CoordType_> MeshType;
      typedef typename MeshType::ShapeType ShapeType;
      typedef typename MeshType::VertexSetType VertexSetType;
      typedef typename MeshType::IndexSetHolderType IndexSetHolderType;
 
      static constexpr int shape_dim = ShapeType::dimension;
 
    public:
      virtual Index get_num_entities(int dim) override
      {
        return Index(Intern::DynamicNumFaces<Shape_>::value(dim));
      }
 
      virtual void fill_vertex_set(VertexSetType& vertex_set) override
      {
        Intern::RefCellVertexer<ShapeType>::fill(vertex_set);
      }
 
      virtual void fill_index_sets(IndexSetHolderType& index_set_holder) override
      {
        Intern::MacroIndexWrapper<Shape_>::build(index_set_holder);
      }
    }; // class ReferenceCellFactory<...>
 
    namespace Intern
    {
      template<int dim_>
      struct RefCellVertexer< Shape::Simplex<dim_> >
      {
        template<typename VertexSet_>
        static void fill(VertexSet_& vtx)
        {
          typedef typename VertexSet_::CoordType CoordType;
          // loop over all vertices
          for(int i(0); i < (dim_+1); ++i)
          {
            // loop over all coords
            for(int j(0); j < dim_; ++j)
            {
              vtx[Index(i)][j] = CoordType(j+1 == i ? 1 : 0);
            }
          }
        }
      };
 
      template<int dim_>
      struct RefCellVertexer< Shape::Hypercube<dim_> >
      {
        template<typename VertexSet_>
        static void fill(VertexSet_& vtx)
        {
          typedef typename VertexSet_::CoordType CoordType;
          // loop over all vertices
          for(int i(0); i < (1 << dim_); ++i)
          {
            // loop over all coords
            for(int j(0); j < dim_; ++j)
            {
              vtx[Index(i)][j] = CoordType((((i >> j) & 1) << 1) - 1);
            }
          }
        }
      };
    }
  } // namespace Geometry
} // namespace FEAT