VolumeMap.hpp

#pragma once
 
#include "SurfaceMap.hpp"
 
#include <wmtk/history/VolumeHistory.hpp>
 
#include <Eigen/Core>
 
#include <limits>
#include <map>
#include <memory>
#include <optional>
#include <set>
#include <string>
#include <vector>
 
namespace wmtk::components::tracking {
 
struct VolumePoint
{
    static constexpr uint32_t kAnyRegion = uint32_t(-1);
    Eigen::Vector3d position = Eigen::Vector3d::Constant(std::numeric_limits<double>::quiet_NaN());
    uint32_t region = kAnyRegion;
};
 
struct VolumeMapStats
{
    size_t built = 0;
    size_t failed = 0;
    std::map<std::string, size_t> failures;
    size_t real_pairs = 0, fallback_pairs = 0;
    size_t decompositions = 0;
    size_t surface_not_kept = 0;
    size_t inverted_ops = 0;
    size_t region_changed = 0, through_failed = 0;
    size_t kept = 0, kept_bytes = 0;
};
 
struct VolumeMapOptions
{
    size_t cache_bytes = size_t(256) << 20;
};
 
class VolumeMap
{
public:
    VolumeMap(
        const history::SurfaceHistory& surface,
        history::VolumeHistory volume,
        const VolumeMapOptions& options = {});
    ~VolumeMap();
    VolumeMap(VolumeMap&&) noexcept;
 
    std::vector<VolumePoint> to_output(std::vector<VolumePoint> points) const;
    std::vector<VolumePoint> to_input(std::vector<VolumePoint> points) const;
 
    const history::VolumeHistory& history() const { return m_volume; }
    std::optional<uint32_t> region_of(std::vector<int64_t> tags) const;
    std::optional<uint32_t> region_of_groups(const std::set<std::string>& names) const;
    const SurfaceMap& surface_map() const { return m_surface; }
    VolumeMapStats stats() const;
    void build_all() const;
    std::vector<std::string> op_kinds(size_t k) const;
 
    struct OpMap;
    struct Faces;
 
private:
    std::vector<VolumePoint> map(std::vector<VolumePoint> points, bool forward) const;
    std::shared_ptr<const OpMap> op(size_t k) const;
 
    history::VolumeHistory m_volume;
    VolumeMapOptions m_options;
    SurfaceMap m_surface;
    std::vector<std::unique_ptr<Faces>> m_faces;
    std::vector<std::pair<Eigen::Vector3d, Eigen::Vector3d>> m_boxes;
    Eigen::Vector3d m_grid_lo = Eigen::Vector3d::Zero(), m_grid_hi = Eigen::Vector3d::Zero();
    double m_grid_cell = 1;
    struct Shared;
    std::unique_ptr<Shared> m_shared;
};
 
} // namespace wmtk::components::tracking