LinkIterable.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 LinkIterable
{
public:
    class Iterator
    {
    public:
        Iterator(LinkIterable& 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();
 
        Tuple navigate_to_link(Tuple t);
 
    private:
        LinkIterable& m_container;
        TopDimensionCofacesIterable::Iterator m_it;
        Tuple m_t;
        int8_t m_pt = 0;
        int8_t m_edge_counter = 0;
    };
 
public:
    LinkIterable(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_link; // for depth 3 iteration
};
 
} // namespace wmtk::simplex