ManExtractTetMesh.h

#pragma once
 
#include <wmtk/components/topological_offset/Parameters.h>
#include <wmtk/components/topological_offset/TopoOffsetTetMesh.h>
#include <wmtk/simplex/Simplex.hpp>
#include "Parameters.h"
 
using namespace wmtk::components;
 
 
namespace wmtk::components::manifold_extraction {
 
 
class ManExtractTetMesh : public topological_offset::TopoOffsetTetMesh
{
public:
    int m_vtu_counter = 0;
    int m_surfvtu_counter = 0;
 
    Parameters& m_man_params;
 
    ManExtractTetMesh(
        Parameters& _m_man_params,
        topological_offset::Parameters& _m_top_params,
        int _num_threads = 0)
        : topological_offset::TopoOffsetTetMesh(_m_top_params, _num_threads)
        , m_man_params(_m_man_params)
    {}
 
    ~ManExtractTetMesh() {}
 
public:
    std::pair<size_t, size_t> label_non_manifold(); // return # of non manifold edges, verts
 
    bool edge_is_manifold(const Tuple& t) const;
 
    void edge_dfs_helper(std::set<size_t>& visited_tids, const Tuple& t) const;
 
    bool vertex_is_manifold(const Tuple& t) const;
 
    void vertex_dfs_helper(
        std::set<size_t>& visited_tids,
        const Tuple& t,
        const bool include,
        const std::vector<simplex::Face>& b_out_faces) const;
 
    bool is_boundary_vertex(const Tuple& v) const;
 
    std::vector<simplex::Face> get_boundary_faces_for_out_tets(size_t vid) const;
 
    void extract_surface_mesh(MatrixXd& V, MatrixXi& F);
 
    void write_surface(const std::string& path);
 
private: // helpers
    bool is_surface_vertex(const Tuple& v) const;
    bool is_surface_edge(const Tuple& e) const;
    bool is_surface_face(const Tuple& f) const;
    bool is_interior_tet(const Tuple& t) const;
    bool is_interior_tet(const size_t& t_id) const;
};
 
} // namespace wmtk::components::manifold_extraction