BlockMatching.hpp

Go to the documentation of this file.

#pragma once
 
#include <FAST/ProcessObject.hpp>
#include <deque>
 
namespace fast {
 
class Image;
 
class FAST_EXPORT BlockMatching : public ProcessObject {
    FAST_OBJECT(BlockMatching)
    public:
        enum class MatchingMetric {
            NORMALIZED_CROSS_CORRELATION,
            SUM_OF_SQUARED_DIFFERENCES,
            SUM_OF_ABSOLUTE_DIFFERENCES,
        };
 
        static MatchingMetric stringToMetric(std::string name) {
            std::map<std::string, MatchingMetric> map = {
                {"NCC", MatchingMetric::NORMALIZED_CROSS_CORRELATION},
                {"SAD", MatchingMetric::SUM_OF_ABSOLUTE_DIFFERENCES},
                {"SSD", MatchingMetric::SUM_OF_SQUARED_DIFFERENCES},
            };
            return map.at(name);
        }
 
        void setMatchingMetric(MatchingMetric type);
        void setBlockSize(int size);
        void setSearchSize(int size);
        void setIntensityThreshold(float value);
        void setTimeLag(int timeLag);
        void setForwardBackwardTracking(bool forwardBackward);
        void setRegionOfInterest(Vector2i offset, Vector2i size);
        void loadAttributes() override;
    private:
        BlockMatching();
        void execute() override;
 
        MatchingMetric m_type = MatchingMetric::SUM_OF_ABSOLUTE_DIFFERENCES;
        int m_blockSizeHalf = 5;
        int m_searchSizeHalf = 5;
        float m_intensityThreshold = std::numeric_limits<float>::min();
        int m_timeLag = 1;
        bool m_forwardBackward = false;
        Vector2i m_offsetROI = Vector2i::Zero();
        Vector2i m_sizeROI = Vector2i::Zero();
        std::deque<std::shared_ptr<Image>> m_frameBuffer;
 
};
 
}