CESets.hpp

Go to the documentation of this file.

#ifndef MESHKIT_CD_SETS_HPP
#define MESHKIT_CD_SETS_HPP

#include <set>
#include <string>
#include <vector>

#include "meshkit/MKCore.hpp"

#include <iBase.h>
#include <iMesh.h>

namespace MeshKit {

class CESets
{
public:
  struct tag_data
  {
    tag_data(iMesh::TagHandle tag, char *value)
      : tag(tag), value(value)
    {}

    iMesh::TagHandle tag;
    char *value;
  };

  typedef std::vector<tag_data> tag_type;
  typedef std::set<iMesh::EntitySetHandle> set_type;
  typedef tag_type::iterator       tag_iterator;
  typedef tag_type::const_iterator const_tag_iterator;
  typedef set_type::iterator       set_iterator;
  typedef set_type::const_iterator const_set_iterator;

  CESets(MKCore *mkcore) : mesh_(mkcore->imesh_instance())
  {}

  ~CESets();

  iMesh * imesh_instance() const { return mesh_; }

  void add_set(iMesh::EntitySetHandle set)
  {
    sets_.insert(set);
  }

  template <typename T>
  void add_sets(T begin, T end)
  {
    sets_.insert(begin, end);
  }

  void add_tag(iMesh::TagHandle tag_handle, const char *value = NULL);
  void add_tag(const std::string &tag_name, const char *value = NULL);
  void update_tagged_sets();

  void clear()
  {
    sets_.clear();
  }

  tag_type & tags()             { return tags_; }
  const tag_type & tags() const { return tags_; }
  set_type & sets()             { return sets_; }
  const set_type & sets() const { return sets_; }

  tag_iterator tbegin()             { return tags_.begin(); }
  const_tag_iterator tbegin() const { return tags_.begin(); }
  tag_iterator tend()               { return tags_.end(); }
  const_tag_iterator tend() const   { return tags_.end(); }
  set_iterator sbegin()             { return sets_.begin(); }
  const_set_iterator sbegin() const { return sets_.begin(); }
  set_iterator send()               { return sets_.end(); }
  const_set_iterator send() const   { return sets_.end(); }

private:
  iMesh *mesh_;
  tag_type tags_;
  set_type sets_;
};

void link_expand_sets(const CESets &ce_sets, iMesh::TagHandle local_tag);

void process_ce_sets(iMesh *mesh,
                     const std::set<iMesh::EntitySetHandle> &cesets,
                     iMesh::TagHandle local_tag);

void tag_copy_sets(iMesh *mesh, iMesh::TagHandle copyTag,
                   const std::set<iMesh::EntitySetHandle> &copySets,
                   iMesh::TagHandle tag, const char *tag_val);

void tag_copy_sets(const CESets &ce_sets, iMesh::TagHandle local_tag,
                   iMesh::TagHandle copy_tag);

} // namespace MeshKit
#endif