MOGSegmentationModule.h

#ifndef MOGSEGMENTATIONMODULE_H
#define MOGSEGMENTATIONMODULE_H

#include <map>
#include <string>
#include <QImage>
#include "Datapool.h"
#include "ModuleInterface.h"
#include <QDir>
#include "src/json.h"

#include <opencv2/opencv.hpp>
#include "src/ReliabilityMog2.h"
#include "ReliabilityBg.h"

class MOGSegmentationModule: public ModuleInterface {
public:
    MOGSegmentationModule(Datapool *i_data);
    ~MOGSegmentationModule();

    //Set module configuration parameters
    bool setParameters(QDomNode& config);

    //Initialization after reading parameters
    bool init();

    //Function executed at each frame
    bool run();

    //update parameters at runtime.
    bool updateParameters();

    bool readImageOnDisk(QString dir,QImage **o_rpOutputImage);

    void tuneParameters();

    void applyAperture(cv::Mat &);

    void applyMask(QImage *im);

    private:
        cv::Ptr<cv::BackgroundSubtractorMOG2> mog;
        ReliabilityMog2 mog2;

        int width;
        int height;

        double alpha;    //learning rate (set -1 to autoadjust it)
        int m_history;
        int m_nmixtures;
        int m_noise;
        double m_bgThreshold;
        bool m_aperture;
        bool m_tuningActivated;
        int m_currentTuneValue;
        int m_tuneStep;
        int m_maxTuneValue;

        bool validate_once;
        bool different_bg_current;

        QString bgImName;
        bool first;
        bool m_mask;
        QString m_maskImName;
        bool m_getContrastReliability;
        cv::Mat element;
        cv::Mat featureMap;

};

#endif // MOGSEGMENTATIONMODULE_H