ComputeVolumetricMask.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <cstdint>
#include <limits>
 
#include "vc/core/types/Mixins.hpp"
#include "vc/core/types/PointSet.hpp"
#include "vc/core/types/Volume.hpp"
#include "vc/core/types/VolumetricMask.hpp"
 
namespace volcart::segmentation
{
 
class ComputeVolumetricMask : public IterationsProgress
{
public:
    using PointSet = volcart::PointSet<cv::Vec3d>;
 
    void setPointSet(const PointSet& ps);
 
    void setVolume(const Volume::Pointer& v);
 
    void setLowThreshold(std::uint16_t t);
 
    void setHighThreshold(std::uint16_t t);
 
    void setEnableClosing(bool b);
 
    void setClosingKernelSize(int s);
 
    void setMeasureVertical(bool b);
 
    void setMaxRadius(std::size_t radius);
 
    VolumetricMask::Pointer compute();
 
    VolumetricMask::Pointer getMask() const;
 
    std::size_t progressIterations() const override;
 
private:
    PointSet input_;
    Volume::Pointer vol_;
    std::uint16_t low_{14135};
    std::uint16_t high_{65535};
    bool enableClosing_{true};
    int kernel_{5};
    bool measureVertically_{false};
    std::size_t maxRadius_{std::numeric_limits<std::size_t>::max()};
    VolumetricMask::Pointer mask_;
};
 
}  // namespace volcart::segmentation