MergeMesh.hpp

Go to the documentation of this file.

#ifndef MESHKIT_MERGE_MESH_HPP
#define MESHKIT_MERGE_MESH_HPP

#include "MKException.hpp"

#include "meshkit/Types.hpp"
#include "meshkit/Error.hpp"
#include "meshkit/MeshScheme.hpp"
#include "meshkit/ModelEnt.hpp"

#include <cassert>
#include <string.h>
#include <vector>
#include <set>

#include "iMesh.h"

#ifdef HAVE_MOAB
#include "MBiMesh.hpp"
#include "MBInterface.hpp"
#include "MBRange.hpp"
#include "MBAdaptiveKDTree.hpp"
#endif

namespace MeshKit {

  class MKCore;


class MergeMesh : public MeshScheme
{
public:
  /* \brief Constructor
   *
   * Create a new MergeMesh instance
   * \param impl the iMesh instance handle for the mesh
   */


  MergeMesh(MKCore *mkcore, const MEntVector &me_vec);
  
  /* \brief Destructor
   */
  virtual ~MergeMesh();


    static const char* name();

    static bool can_mesh(iBase_EntityType dim);

    static bool can_mesh(ModelEnt *me);

    static const moab::EntityType* output_types();

    virtual const moab::EntityType* mesh_types_arr() const;

    virtual bool add_modelent(ModelEnt *model_ent);

    virtual void setup_this();

    virtual void execute_this();

  /* \brief Return the imesh instance
   */
  iMesh* impl() const {return imeshImpl;}



    /* \brief Merge vertices in elements passed in
     */
  void merge_entities(iBase_EntityHandle *elems,
                      int elems_size,
                      const double merge_tol,
                      const int do_merge = true,
                      const int update_sets = false,
                      iBase_TagHandle merge_tag = 0);
  
    /* \brief Perform the actual merge between entities
     */
  void perform_merge(iBase_TagHandle merged_to);
  void set_merge_params(double mergeTol = 1e-4, int update_sets = 0,
                          int do_merge = 1, iBase_TagHandle mergeTag = NULL);
private:
  
  iMesh *imeshImpl;
  double mergeTol, mergeTolSq;
  iBase_TagHandle mergeTag;
  int updateSets;
  int doMerge;

#ifdef HAVE_MOAB

    //- given a kdtree, set tag on vertices in leaf nodes with vertices
    //- to which they should be merged
  MBErrorCode find_merged_to(MBAdaptiveKDTree & tree, MBEntityHandle &tree_root, MBTag merged_to);

  
  MBErrorCode merge_entities(MBRange &elems,
                             const int do_merge,
                             const int update_sets,
                             MBTag merge_tag);
  
    //- perform the actual merge
  MBErrorCode perform_merge(MBTag merged_to);

  MBInterface *mbImpl;

    //- the tag pointing to the entity to which an entity will be merged
  MBTag mbMergeTag;

    //- entities which will go away after the merge
  MBRange deadEnts;
#endif
};

  inline const char* MergeMesh::name()
  {
    return "MergeMesh";
  }

  inline bool MergeMesh::can_mesh(iBase_EntityType)
  {
    return false;
  }

  inline bool MergeMesh::can_mesh(ModelEnt *)
  {
    return true;
  }

  inline const moab::EntityType* MergeMesh::mesh_types_arr() const
  {
    return output_types();
  }

} // namespace

#endif