ChainSegmentationAlgorithm.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
 
#include "vc/core/types/BoundingBox.hpp"
#include "vc/core/types/Mixins.hpp"
#include "vc/core/types/OrderedPointSet.hpp"
#include "vc/core/types/Volume.hpp"
 
namespace volcart::segmentation
{
class ChainSegmentationAlgorithm : public IterationsProgress
{
public:
    using Pointer = std::shared_ptr<ChainSegmentationAlgorithm>;
 
    virtual ~ChainSegmentationAlgorithm() = default;
 
    using Chain = std::vector<cv::Vec3d>;
    using PointSet = volcart::OrderedPointSet<cv::Vec3d>;
    using Bounds = volcart::BoundingBox<double, 3>;
    enum class Status { Success, Failure, ReturnedEarly };
 
    void setVolume(Volume::Pointer vol)
    {
        vol_ = std::move(vol);
        bb_ = vol_->bounds();
    }
 
    void setBounds(Bounds b) { bb_ = std::move(b); }
 
    void setChain(Chain c) { startingChain_ = std::move(c); }
    void setNumberOfSteps(std::size_t n) { numSteps_ = n; }
 
    void setStepSize(double s) { stepSize_ = s; }
    virtual auto compute() -> PointSet = 0;
 
    auto getStatus() const -> Status { return status_; }
    [[nodiscard]] auto getPointSet() const -> const PointSet&
    {
        return result_;
    }
 
    auto getPointSet() -> PointSet& { return result_; }
    auto progressIterations() const -> std::size_t override
    {
        return numSteps_;
    }
 
protected:
    ChainSegmentationAlgorithm() = default;
    Volume::Pointer vol_;
    Chain startingChain_;
    Bounds bb_;
    std::size_t numSteps_{0};
    double stepSize_{1.0};
    PointSet result_;
    Status status_{Status::Success};
};
}  // namespace volcart::segmentation