ThinnedFloodFillSegmentation.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <cstdint>
#include <limits>
 
#include "vc/core/types/PointSet.hpp"
#include "vc/core/types/VolumetricMask.hpp"
#include "vc/segmentation/RegionGrowingSegmentationAlgorithmBaseClass.hpp"
 
namespace volcart::segmentation
{
class ThinnedFloodFillSegmentation
    : public RegionGrowingSegmentationAlgorithmBaseClass
{
public:
    using VoxelMask = volcart::PointSet<cv::Vec3i>;
 
    Signal<PointSet> pointsetUpdated;
    Signal<VoxelMask> maskUpdated;
 
    ThinnedFloodFillSegmentation() = default;
    ~ThinnedFloodFillSegmentation() override = default;
    ThinnedFloodFillSegmentation(const ThinnedFloodFillSegmentation&) = default;
    ThinnedFloodFillSegmentation& operator=(
        const ThinnedFloodFillSegmentation&) = default;
    ThinnedFloodFillSegmentation(ThinnedFloodFillSegmentation&&) = default;
    ThinnedFloodFillSegmentation& operator=(ThinnedFloodFillSegmentation&&) =
        default;
 
    void setFFLowThreshold(std::uint16_t t);
 
    void setFFHighThreshold(std::uint16_t t);
 
    void setDistanceTransformThreshold(float t);
 
    void setClosingKernelSize(int s);
 
    void setMeasureVertical(bool b);
 
    void setSpurLengthThreshold(int length);
 
    void setMaxRadius(std::size_t radius);
 
    PointSet compute() override;
 
    VoxelMask getMask() const;
 
    void setDumpVis(bool b);
 
private:
    std::uint16_t low_{14135};
    std::uint16_t high_{65535};
    float dtt_{0};
    int kernel_{5};
    bool dumpVis_{false};
    std::size_t spurLength_{6};
    bool measureVertically_{false};
    std::size_t maxRadius_{std::numeric_limits<std::size_t>::max()};
    VoxelMask volMask_;
};
}  // namespace volcart::segmentation