VolumetricMask.hpp

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <unordered_set>
 
#include "vc/core/types/PointSet.hpp"
#include "vc/core/util/HashFunctions.hpp"
 
namespace volcart
{
 
class VolumetricMask
{
public:
    using Voxel = cv::Vec3i;
 
private:
    using MaskSet = std::unordered_set<Voxel, Vec3iHash>;
 
public:
    using iterator = MaskSet::iterator;
    using const_iterator = MaskSet::const_iterator;
 
    using Pointer = std::shared_ptr<VolumetricMask>;
 
    template <typename... Args>
    static auto New(Args... args) -> Pointer
    {
        return std::make_shared<VolumetricMask>(std::forward<Args>(args)...);
    }
 
    VolumetricMask() = default;
 
    template <class Container>
    explicit VolumetricMask(const Container& ps)
    {
        mask_.reserve(ps.size());
        mask_.insert(std::begin(ps), std::end(ps));
    }
 
    void setIn(const Voxel& v);
    void setOut(const Voxel& v);
 
    template <class Container>
    void setIn(const Container& ps)
    {
        mask_.reserve(mask_.size() + ps.size());
        mask_.insert(std::begin(ps), std::end(ps));
    }
 
    template <class Container>
    void setOut(const Container& ps)
    {
        for (const auto& p : ps) {
            setOut(p);
        }
    }
 
    [[nodiscard]] auto isIn(const Voxel& v) const -> bool;
    [[nodiscard]] auto isOut(const Voxel& v) const -> bool;
 
    [[nodiscard]] auto isIn(const cv::Vec3d& v) const -> bool;
    [[nodiscard]] auto isOut(const cv::Vec3d& v) const -> bool;
 
    auto begin() noexcept -> iterator;
    [[nodiscard]] auto begin() const noexcept -> const_iterator;
    [[nodiscard]] auto cbegin() const noexcept -> const_iterator;
 
    auto end() noexcept -> iterator;
    [[nodiscard]] auto end() const noexcept -> const_iterator;
    [[nodiscard]] auto cend() const noexcept -> const_iterator;
 
    void clear();
 
    [[nodiscard]] auto empty() const -> bool;
 
    [[nodiscard]] auto as_vector() const -> std::vector<Voxel>;
 
private:
    MaskSet mask_;
};
 
}  // namespace volcart