EulerGradientVectorFlow.hpp

Go to the documentation of this file.

#ifndef EULER_GRADIENT_VECTOR_FLOW_HPP_
#define EULER_GRADIENT_VECTOR_FLOW_HPP_
 
#include "FAST/ProcessObject.hpp"
 
namespace fast {
 
class Image;
 
class FAST_EXPORT  EulerGradientVectorFlow : public ProcessObject {
    FAST_OBJECT(EulerGradientVectorFlow)
    public:
        void setIterations(uint iterations);
        void setMuConstant(float mu);
        float getMuConstant() const;
        void set16bitStorageFormat();
        void set32bitStorageFormat();
    private:
        EulerGradientVectorFlow();
        void execute();
        void execute2DGVF(std::shared_ptr<Image> input, std::shared_ptr<Image> output, uint iterations);
        void execute3DGVF(std::shared_ptr<Image> input, std::shared_ptr<Image> output, uint iterations);
        void execute3DGVFNo3DWrite(std::shared_ptr<Image> input, std::shared_ptr<Image> output, uint iterations);
 
        float mMu;
        uint mIterations;
        bool mUse16bitFormat;
};
 
} // end namespace fast
 
#endif