TopDimensionCofacesIterable.hpp

Go to the documentation of this file.

#pragma once
 
#include <queue>
 
#include <wmtk/Mesh.hpp>
#include <wmtk/simplex/Simplex.hpp>
#include <wmtk/simplex/SimplexCollection.hpp>
#include <wmtk/simplex/internal/VisitedArray.hpp>
#include <wmtk/utils/DynamicArray.hpp>
 
namespace wmtk::simplex {
 
class TopDimensionCofacesIterable
{
public:
    enum IteratorPhase { Forward = 0, Intermediate = 1, Backward = 2, End = 3 };
 
    class Iterator
    {
    public:
        Iterator(TopDimensionCofacesIterable& container, const Tuple& t = Tuple());
        Iterator& operator++();
        bool operator!=(const Iterator& other) const;
        Tuple& operator*();
        const Tuple& operator*() const;
 
        bool is_intermediate() const;
 
    private:
        PrimitiveType pt(int64_t depth) const;
        int64_t depth();
 
        void init(int64_t depth);
 
        Iterator& step_depth_0();
        Iterator& step_depth_1();
        Iterator& step_depth_2();
        Iterator& step_depth_3();
 
        void add_neighbors_to_queue();
 
    private:
        TopDimensionCofacesIterable& m_container;
 
        Tuple m_t; // the tuple that iterates through the mesh
        IteratorPhase m_phase = IteratorPhase::Forward; // for depth 1 and 2 iteration
        bool m_is_intermediate = false;
    };
 
public:
    TopDimensionCofacesIterable(
        const Mesh& mesh,
        const Simplex& simplex,
        const bool retrieve_intermediate_tuple = false);
 
    Iterator begin() { return Iterator(*this, m_simplex.tuple()); }
    Iterator end() { return Iterator(*this); }
 
private:
    const Mesh& m_mesh;
    const Simplex m_simplex;
 
    bool m_retrieve_intermediate_tuple = false;
 
    wmtk::utils::DynamicArray<Tuple> m_q;
    int64_t m_q_front = 0;
    simplex::internal::VisitedArray<int64_t> m_visited; // for depth 3 iteration
};
 
} // namespace wmtk::simplex