LocalResliceParticleSim.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <iostream>
 
#include "vc/core/types/OrderedPointSet.hpp"
#include "vc/core/types/VolumePkg.hpp"
#include "vc/segmentation/ChainSegmentationAlgorithm.hpp"
#include "vc/segmentation/lrps/Common.hpp"
#include "vc/segmentation/lrps/FittedCurve.hpp"
 
namespace volcart::segmentation
{
class LocalResliceSegmentation : public ChainSegmentationAlgorithm
{
public:
    using Pointer = std::shared_ptr<LocalResliceSegmentation>;
 
    LocalResliceSegmentation() = default;
 
    ~LocalResliceSegmentation() override = default;
 
    template <typename... Args>
    static auto New(Args... args) -> Pointer
    {
        return std::make_shared<LocalResliceSegmentation>(
            std::forward<Args>(args)...);
    }
 
    void setTargetZIndex(int z) { endIndex_ = z; }
 
    void setOptimizationIterations(int n) { numIters_ = n; }
 
    void setAlpha(double a) { alpha_ = a; }
 
    void setK1(double k) { k1_ = k; }
 
    void setK2(double k) { k2_ = k; }
 
    void setBeta(double b) { beta_ = b; }
 
    void setDelta(double d) { delta_ = d; }
 
    void setMaterialThickness(double m) { materialThickness_ = m; }
 
    void setResliceSize(int s) { resliceSize_ = s; }
 
    void setDistanceWeightFactor(int f) { peakDistanceWeight_ = f; }
 
    void setConsiderPrevious(bool b) { considerPrevious_ = b; }
 
    auto compute() -> PointSet override;
 
    void setVisualize(bool b) { visualize_ = b; }
 
    void setDumpVis(bool b) { dumpVis_ = b; }
 
    [[nodiscard]] auto progressIterations() const -> std::size_t override;
 
private:
    auto estimate_normal_at_index_(const FittedCurve& currentCurve, int index)
        -> cv::Vec3d;
 
    [[nodiscard]] auto draw_particle_on_slice_(
        const FittedCurve& curve,
        int sliceIndex,
        int particleIndex = -1,
        bool showSpline = false) const -> cv::Mat;
 
    auto create_final_pointset_(const std::vector<std::vector<Voxel>>& points)
        -> PointSet;
 
    constexpr static double DEFAULT_MIN_ENERGY_GRADIENT = 1e-7;
 
    int endIndex_{0};
    double alpha_{1.0 / 3.0};
    double k1_{0.5};
    double k2_{0.5};
    double beta_{1.0 / 3.0};
    double delta_{1.0 / 3.0};
    int peakDistanceWeight_{50};
    bool considerPrevious_{false};
    bool dumpVis_{false};
    bool visualize_{false};
    int numIters_{15};
    double materialThickness_{100};
    int resliceSize_{32};
};
}  // namespace volcart::segmentation