RegionGrowingSegmentationAlgorithmBaseClass.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
 
#include "vc/core/types/Mixins.hpp"
#include "vc/core/types/PointSet.hpp"
#include "vc/core/types/Volume.hpp"
 
namespace volcart::segmentation
{
class RegionGrowingSegmentationAlgorithmBaseClass : public IterationsProgress
{
public:
    virtual ~RegionGrowingSegmentationAlgorithmBaseClass() = default;
    RegionGrowingSegmentationAlgorithmBaseClass(
        const RegionGrowingSegmentationAlgorithmBaseClass&) = default;
    RegionGrowingSegmentationAlgorithmBaseClass& operator=(
        const RegionGrowingSegmentationAlgorithmBaseClass&) = default;
    RegionGrowingSegmentationAlgorithmBaseClass(
        RegionGrowingSegmentationAlgorithmBaseClass&&) = default;
    RegionGrowingSegmentationAlgorithmBaseClass& operator=(
        RegionGrowingSegmentationAlgorithmBaseClass&&) = default;
 
    using SeedPoints = std::vector<cv::Vec3d>;
 
    using PointSet = volcart::PointSet<cv::Vec3d>;
 
    enum class Status { Success, Failure, ReturnedEarly };
 
    void setVolume(Volume::Pointer vol) { vol_ = std::move(vol); }
 
    void setSeedPoints(SeedPoints p) { startingPoints_ = std::move(p); }
 
    void setIterations(std::size_t i) { iterations_ = i; }
 
    virtual PointSet compute() = 0;
 
    Status getStatus() const { return status_; }
 
    const PointSet& getPointSet() const { return result_; }
 
    PointSet& getPointSet() { return result_; }
 
    std::size_t progressIterations() const override { return iterations_; }
 
protected:
    RegionGrowingSegmentationAlgorithmBaseClass() = default;
 
    Volume::Pointer vol_;
    SeedPoints startingPoints_;
    PointSet result_;
    std::size_t iterations_{0};
    Status status_{Status::Success};
};
}  // namespace volcart::segmentation