EdgeModelPotts.h

// Potts Edge Model class interface
// Written by Sergey G. Kosov in 2019 for Project X 
#pragma once

#include "IEdgeModel.h"
#include "permutohedral/permutohedral.h"

namespace DirectGraphicalModels {
    // ================================ Potts Edge Model ================================
    class CEdgeModelPotts : public IEdgeModel {
    public:
        CEdgeModelPotts(const Mat& features, float weight = 1.0f, const std::function<void(const Mat& src, Mat& dst)>& semiMetricFunction = {}, bool perPixelNormalization = true);
        virtual ~CEdgeModelPotts(void) = default;
    
        void apply(const Mat &src, Mat &dst) const override;
    

    private:
        std::unique_ptr<CPermutohedral>                 m_pLattice;     
        float                                           m_weight;       
        Mat                                             m_norm;         
        std::function<void(const Mat &src, Mat &dst)>   m_function;     
    };
}