VolumeMask.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <vector>
 
#include <Eigen/SparseCore>
#include <opencv2/core.hpp>
 
#include "vc/core/types/NDArray.hpp"
 
namespace volcart
{
 
class VolumeMask
{
public:
    enum class State { Unsegmented = 0, Segmented };
 
    using Subvolume = NDArray<State>;
 
    VolumeMask(std::size_t width, std::size_t height, std::size_t numSlices);
 
    void setVoxelState(const cv::Vec3i& xyz, State state);
 
    State getVoxelState(const cv::Vec3i& xyz);
 
    void setSubvolumeState(
        const cv::Vec3i& origin, const cv::Vec3i& dims, State state);
 
    Subvolume getSubvolumeState(const cv::Vec3i& origin, const cv::Vec3i& dims);
 
private:
    Eigen::SparseMatrix<int, Eigen::RowMajor> states_;
 
    std::size_t sliceWidth_;
 
    std::size_t sliceHeight_;
};
 
}  // namespace volcart