TetVTUWriter.hpp

#pragma once
 
#include <wmtk/TetMesh.h>
#include <filesystem>
#include <functional>
#include <map>
#include <wmtk/Types.hpp>
 
namespace wmtk::io {
 
class TetVTUWriter
{
private:
    using Tuple = typename TetMesh::Tuple;
 
public:
    TetVTUWriter(const TetMesh& mesh);
 
    void add_vertex_positions(const std::function<VectorXd(const size_t)>& f);
 
    void add_vertex_attribute(
        const std::string& name,
        const std::function<VectorXd(const size_t)>& f);
    void add_edge_attribute(
        const std::string& name,
        const std::function<VectorXd(const size_t)>& f);
 
    void add_triangle_attribute(
        const std::string& name,
        const std::function<VectorXd(const size_t)>& f);
 
    void add_tet_attribute(const std::string& name, const std::function<VectorXd(const size_t)>& f);
 
    bool write_tets(const std::filesystem::path& filename);
 
    bool write_triangles(const std::filesystem::path& filename);
 
    bool write_edges(const std::filesystem::path& filename);
 
private:
    const TetMesh& m_mesh;
 
    MatrixXd m_V; // vertex positions
    MatrixXi m_E; // edge - vids
    MatrixXi m_F; // face - vids
    MatrixXi m_T; //  tet - vids
 
    std::map<std::string, MatrixXd> m_V_attributes;
    std::map<std::string, MatrixXd> m_E_attributes;
    std::map<std::string, MatrixXd> m_F_attributes;
    std::map<std::string, MatrixXd> m_T_attributes;
};
 
} // namespace wmtk::io