shape_convert_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/index_calculator.hpp>
#include <kernel/geometry/mesh_part.hpp>
#include <kernel/geometry/intern/shape_convert_index.hpp>
#include <kernel/geometry/intern/shape_convert_target.hpp>
#include <kernel/geometry/intern/shape_convert_traits.hpp>
#include <kernel/geometry/intern/shape_convert_vertex.hpp>
 
namespace FEAT
{
  namespace Geometry
  {
    template<typename Mesh_>
#ifndef DOXYGEN
    class ShapeConvertFactory;
#else
    class ShapeConvertFactory :
      public Factory<Mesh_>
    {
    public:
      typedef Mesh_ MeshType;
      typedef ... OtherMeshType;
 
      explicit ShapeConvertFactory(const OtherMeshType& other_mesh);
    };
#endif // DOXYGEN
 
    template<
      typename Shape_,
      int num_coords_,
      typename Coord_>
    class ShapeConvertFactory<ConformalMesh<Shape_, num_coords_, Coord_> > :
      public Factory<ConformalMesh<Shape_, num_coords_, Coord_> >
    {
    public:
      typedef Factory<ConformalMesh<Shape_, num_coords_, Coord_> > BaseClass;
      typedef Shape_ ShapeType;
      typedef typename Intern::OtherShape<Shape_>::Type OtherShapeType;
      typedef ConformalMesh<Shape_, num_coords_, Coord_> MeshType;
      typedef ConformalMesh<OtherShapeType, num_coords_, Coord_> OtherMeshType;
      typedef typename MeshType::VertexSetType VertexSetType;
      typedef typename MeshType::IndexSetHolderType IndexSetHolderType;
 
      static constexpr int shape_dim = ShapeType::dimension;
 
    protected:
      const OtherMeshType& _other_mesh;
      Index _num_entities_in[shape_dim + 1];
      Index _num_entities_out[shape_dim + 1];
 
    public:
      explicit ShapeConvertFactory(const OtherMeshType& other_mesh) :
        _other_mesh(other_mesh)
      {
        // get number of entities in input mesh
        for(int i(0); i <= shape_dim; ++i)
        {
          _num_entities_out[i] = _num_entities_in[i] = _other_mesh.get_num_entities(i);
        }
 
        // calculate number of entities in output mesh
        Intern::EntityCountWrapper<Intern::ShapeConvertTraits, OtherShapeType>::query(_num_entities_out);
      }
 
      virtual Index get_num_entities(int dim) override
      {
        return _num_entities_out[dim];
      }
 
      virtual void fill_vertex_set(VertexSetType& vertex_set) override
      {
        // call wrapper
        Intern::ShapeConvertVertexWrapper<OtherShapeType, VertexSetType>
          ::refine(vertex_set, _other_mesh.get_vertex_set(), _other_mesh.get_index_set_holder());
      }
 
      virtual void fill_index_sets(IndexSetHolderType& index_set_holder) override
      {
        // call wrapper to build essential index sets
        Intern::ShapeConvertIndexWrapper<ShapeType>
          ::refine(index_set_holder, _num_entities_in, _other_mesh.get_index_set_holder());
 
        // build redundant index sets
        RedundantIndexSetBuilder<ShapeType>::compute(index_set_holder);
      }
    }; // class ShapeConvertFactory<ConformalMesh<...>>
 
    template<
      typename Shape_,
      typename Coord_>
    class ShapeConvertFactory<MeshPart<ConformalMesh<Shape_, Shape_::dimension, Coord_> > > :
      public Factory<MeshPart<ConformalMesh<Shape_, Shape_::dimension, Coord_> > >
    {
    public:
      typedef Factory<MeshPart<ConformalMesh<Shape_, Shape_::dimension, Coord_> > > BaseClass;
      typedef Shape_ ShapeType;
      typedef typename Intern::OtherShape<Shape_>::Type OtherShapeType;
      typedef MeshPart<ConformalMesh<Shape_, Shape_::dimension, Coord_> > MeshType;
      typedef MeshPart<ConformalMesh<OtherShapeType, Shape_::dimension, Coord_> > OtherMeshType;
      typedef typename MeshType::VertexSetType VertexSetType;
      typedef typename MeshType::IndexSetHolderType IndexSetHolderType;
      typedef typename MeshType::TargetSetHolderType TargetSetHolderType;
 
      static constexpr int shape_dim = ShapeType::dimension;
 
    protected:
      const OtherMeshType& _other_mesh;
      Index _num_entities_in[shape_dim + 1];
      Index _num_entities_out[shape_dim + 1];
 
    public:
      explicit ShapeConvertFactory(const OtherMeshType& other_mesh) :
        _other_mesh(other_mesh)
      {
        // get number of entities in input mesh
        for(int i(0); i <= shape_dim; ++i)
        {
          _num_entities_out[i] = _num_entities_in[i] = _other_mesh.get_num_entities(i);
        }
 
        // calculate number of entities in output mesh
        Intern::EntityCountWrapper<Intern::ShapeConvertTraits, OtherShapeType>::query(_num_entities_out);
 
      }
 
      virtual Index get_num_entities(int dim)
      {
        return _num_entities_out[dim];
      }
 
      virtual int get_num_coords()
      {
        return _other_mesh.get_vertex_set().get_num_coords();
      }
 
      virtual int get_vertex_stride()
      {
        return _other_mesh.get_vertex_set().get_stride();
      }
 
      virtual void fill_vertex_set(VertexSetType& vertex_set)
      {
        // call wrapper
        Intern::ShapeConvertVertexWrapper<OtherShapeType, VertexSetType>
          ::refine(vertex_set, _other_mesh.get_vertex_set(), _other_mesh.get_index_set_holder());
      }
 
      virtual void fill_index_sets(IndexSetHolderType& index_set_holder)
      {
        // call wrapper to build essential index sets
        Intern::ShapeConvertIndexWrapper<ShapeType>
          ::refine(index_set_holder, _num_entities_in, _other_mesh.get_index_set_holder());
 
        // build redundant index sets
        RedundantIndexSetBuilder<ShapeType>::compute(index_set_holder);
      }
 
      virtual void fill_target_sets(TargetSetHolderType& /*target_set_holder*/)
      {
        // TODO
      }
    }; // class ShapeConvertFactory<MeshPart<ConformalMesh<...>>>
  } // namespace Geometry
} // namespace FEAT