TupleTag.hpp

Go to the documentation of this file.

#pragma once
#include <set>
#include <wmtk/Mesh.hpp>
#include <wmtk/Tuple.hpp>
#include <wmtk/attribute/Attribute.hpp>
namespace wmtk::multimesh::utils::internal {
class TupleTag
{
public:
    TupleTag(Mesh& mesh, const std::set<int64_t>& critical_points);
    const Mesh& mesh() const { return m_mesh; }
    Mesh& mesh() { return m_mesh; }
    void initialize();
    std::set<int64_t> run();
    bool is_critical_vertex(const Tuple& v) const;
 
    bool vertex_is_root(const Tuple& v) const;
    int64_t vertex_get_root(const Tuple& v) const;
    void vertex_set_root(const Tuple& v, int64_t root);
 
    void vertex_sets_unify(const Tuple& v1, const Tuple& v2);
    void run(const Tuple& e);
 
    int64_t get_vertex_tag(const Tuple& tuple) const;
    int64_t get_edge_tag(const Tuple& tuple) const;
    void set_vertex_tag(const Tuple& tuple, int64_t tag);
    void set_edge_tag(const Tuple& tuple, int64_t tag);
    int64_t vid(const Tuple& tuple) const;
    Tuple v_tuple(int64_t vid) const;
 
    Mesh& m_mesh;
    std::set<int64_t> m_critical_points;
    wmtk::attribute::Accessor<int64_t> m_vertex_tag_acc;
    wmtk::attribute::Accessor<int64_t> m_edge_tag_acc;
 
 
    // this is shoved in this class to reduce the number of times we have to friend mesh - this is
    // called by wmtk::multimesh::utils::extract_child_mesh_from_tag
    template <typename T>
    static std::shared_ptr<Mesh> extract_and_register_child_mesh_from_tag_handle(
        Mesh& m,
        const wmtk::attribute::TypedAttributeHandle<T>& tag_handle,
        const T& tag_value, bool child_is_free);
};
} // namespace wmtk::multimesh::utils::internal