surface_io.hpp

#pragma once
 
#include <wmtk/MeshData.hpp>
 
#include <Eigen/Core>
 
#include <array>
#include <cstdint>
#include <map>
#include <string>
#include <vector>
 
namespace wmtk::components::tracking {
 
struct SurfaceData
{
    std::vector<Eigen::Vector3d> vertices;
    std::vector<std::array<uint64_t, 3>> faces;
    std::map<std::string, std::vector<std::vector<double>>> face_fields;
    std::map<std::string, std::vector<std::vector<double>>> corner_fields;
 
    struct FieldType
    {
        size_t type = 0;
        std::vector<std::string> labels;
    };
    std::map<std::string, FieldType> field_types;
};
 
bool is_corner_field(const std::string& name);
 
SurfaceData surface_from_mesh(const MeshData& mesh, const std::string& what, bool strict);
 
MeshData mesh_from_surface(const SurfaceData& s);
 
SurfaceData read_surface_msh(const std::string& path);
 
SurfaceData read_surface(const std::vector<std::string>& paths);
 
void write_surface_msh(const std::string& path, const SurfaceData& s);
 
void add_surface(
    MeshData& mesh,
    const SurfaceData& s,
    const std::vector<int64_t>& rows,
    const std::string& what);
 
void write_surface_into_msh(
    const std::string& tet_mesh,
    const std::string& path,
    const SurfaceData& s,
    const std::vector<int64_t>& rows);
 
void write_surface_obj(
    const std::string& path,
    const SurfaceData& s,
    const std::string& uv_field,
    const std::string& texture);
 
} // namespace wmtk::components::tracking