ClosedStarIterable.hpp

Go to the documentation of this file.

#pragma once
 
#include <wmtk/Mesh.hpp>
#include <wmtk/simplex/IdSimplex.hpp>
#include <wmtk/simplex/Simplex.hpp>
#include <wmtk/simplex/internal/VisitedArray.hpp>
 
#include "TopDimensionCofacesIterable.hpp"
 
namespace wmtk::simplex {
 
class ClosedStarIterable
{
public:
    enum IteratorPhase { Faces, OpenStar, Link };
 
    class Iterator
    {
    public:
        Iterator(ClosedStarIterable& container, const Tuple& t = Tuple());
        Iterator& operator++();
        bool operator!=(const Iterator& other) const;
        IdSimplex operator*();
        const IdSimplex operator*() const;
 
    private:
        int64_t depth();
 
        void init();
 
        Iterator& step_depth_3();
 
        Iterator& step_edge_mesh();
        Iterator& step_tri_mesh();
        Iterator& step_tet_mesh();
 
        Tuple navigate_to_link(Tuple t);
 
        bool step_faces();
        bool step_faces_edge();
        bool step_faces_triangle();
        bool step_faces_tetrahedron();
 
    private:
        ClosedStarIterable& m_container;
        const Mesh& m_mesh;
 
        TopDimensionCofacesIterable::Iterator m_it;
        Tuple m_t;
        int8_t m_pt = -1;
        int8_t m_face_counter = 0;
        IteratorPhase m_phase = IteratorPhase::Faces;
    };
 
public:
    ClosedStarIterable(const Mesh& mesh, const Simplex& simplex);
 
    Iterator begin() { return Iterator(*this, m_simplex.tuple()); }
    Iterator end() { return Iterator(*this); }
 
private:
    const Mesh& m_mesh;
    const Simplex m_simplex;
    TopDimensionCofacesIterable m_tdc_itrbl;
    TopDimensionCofacesIterable::Iterator m_it_end;
 
    std::array<simplex::internal::VisitedArray<simplex::IdSimplex>, 2>
        m_visited_cofaces; // for depth 3 iteration
    std::array<simplex::internal::VisitedArray<simplex::IdSimplex>, 2>
        m_visited_link; // for depth 3 iteration
};
 
} // namespace wmtk::simplex