StructureTensorParticleSim.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
 
#include <opencv2/core.hpp>
 
#include "vc/segmentation/ChainSegmentationAlgorithm.hpp"
#include "vc/segmentation/stps/Particle.hpp"
#include "vc/segmentation/stps/ParticleChain.hpp"
 
namespace volcart::segmentation
{
class StructureTensorParticleSim : public ChainSegmentationAlgorithm
{
public:
    StructureTensorParticleSim() = default;
    ~StructureTensorParticleSim() override = default;
    void setPropagationScaleFactor(double g) { propagationScaleFactor_ = g; }
 
    void setMaterialThickness(double m) { materialThickness_ = m; }
 
    void setRKStepSize(double s) { rkStepSize_ = s; }
    PointSet compute() override;
    std::size_t progressIterations() const override;
 
private:
    double propagationScaleFactor_{0.5};
    double springConstantK_{-0.5};
    double materialThickness_{100};
    int radius_{5};
 
    ParticleChain currentChain_;
 
    double rkStepSize_{0.5};
 
    ForceChain calc_prop_forces_(ParticleChain c);
 
    ForceChain calc_spring_forces_(ParticleChain c);
 
    void add_chain_to_result_();
 
    bool chain_stopped_();
};
}  // namespace volcart::segmentation