volume_io.hpp

#pragma once
 
#include <wmtk/MeshData.hpp>
 
#include <Eigen/Core>
 
#include <array>
#include <cstdint>
#include <memory>
#include <set>
#include <string>
#include <vector>
 
namespace wmtk::components::tracking {
 
struct TetMeshData
{
    std::vector<Eigen::Vector3d> vertices;
    std::vector<std::array<uint64_t, 4>> tets;
    std::vector<uint32_t> groups;
    std::vector<std::set<std::string>> group_sets;
};
TetMeshData tet_mesh_data(const MeshData& m);
MeshData read_tet_mesh(const std::string& path);
TetMeshData read_tet_msh(const std::string& path);
 
class TetLocator
{
public:
    explicit TetLocator(std::shared_ptr<const TetMeshData> m);
    const TetMeshData& mesh() const { return *m_mesh; }
    std::pair<int64_t, Eigen::Vector4d> locate(const Eigen::Vector3d& p) const;
    std::pair<int64_t, Eigen::Vector4d> locate(const Eigen::Vector3d& p, uint32_t group) const;
    bool in_group(const Eigen::Vector3d& p, uint32_t group) const;
 
private:
    std::shared_ptr<const TetMeshData> m_mesh;
    Eigen::Vector3d m_lo = Eigen::Vector3d::Zero();
    double m_h = 1;
    double m_eps = 0;
    std::array<int64_t, 3> m_n = {{1, 1, 1}};
    std::vector<std::vector<uint32_t>> m_cells;
};
 
} // namespace wmtk::components::tracking