TagIntersection.hpp

Go to the documentation of this file.

#pragma once
 
#include <deque>
#include <wmtk/TetMesh.hpp>
#include <wmtk/TriMesh.hpp>
#include <wmtk/io/HDF5Writer.hpp>
#include <wmtk/io/MeshReader.hpp>
#include <wmtk/simplex/Simplex.hpp>
#include <wmtk/simplex/open_star.hpp>
#include <wmtk/utils/Logger.hpp>
 
namespace wmtk {
namespace components {
class TagAttribute
{
public:
    attribute::Accessor<int64_t> m_tag_accessor; // access primitives' tag value
    PrimitiveType m_ptype; // record this tag attribute's primitivetype
    int64_t m_tag_val; // a given tagvalue used to check or tag the attributes
 
    TagAttribute(
        Mesh& m,
        const attribute::MeshAttributeHandle& attribute,
        PrimitiveType ptype,
        int64_t val)
        : m_tag_accessor(m.create_accessor<int64_t>(attribute))
        , m_ptype(ptype)
        , m_tag_val(val)
    {}
 
    TagAttribute(TagAttribute&) = delete;
 
    bool is_tagged(Mesh& m, const simplex::Simplex& s) const
    {
        if (s.primitive_type() != m_ptype) {
            return false;
        }
        return m_tag_accessor.const_scalar_attribute(s.tuple()) == m_tag_val;
    }
 
    void set_tag(Mesh& m, const simplex::Simplex& s)
    {
        if (s.primitive_type() != m_ptype) {
            return;
        }
        m_tag_accessor.scalar_attribute(s.tuple()) = m_tag_val;
    }
};
 
class TagIntersection
{
public:
    void compute_intersection(
        Mesh& m,
        const std::vector<std::tuple<attribute::MeshAttributeHandle, int64_t>>& input_tags,
        const std::vector<std::tuple<attribute::MeshAttributeHandle, int64_t>>& output_tags);
 
private:
    bool simplex_is_in_intersection(
        Mesh& m,
        const simplex::Simplex& s,
        const std::deque<TagAttribute>& input_tag_attributes);
};
 
} // namespace components
} // namespace wmtk