sampleshadowsoccer.h

#ifndef SAMPLESHADOWSOCCER_H
#define SAMPLESHADOWSOCCER_H

#include <map>
#include <string>
#include <QImage>
#include "Datapool.h"
#include "ModuleInterface.h"
#include <QDir>
#include "SegmentationTool.h"

#include <opencv2/video/background_segm.hpp>
#include "opencv2/opencv.hpp"

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

    //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();

    //Spatial Filter
    void spatialFilter();

    //Labeling Nodes Skeletons
    void labelingNodes(cv::Mat &skeletons);
    void updateConnections(int new_v, int old_v, std::map<int, int> &con);

    void filterLeafs(int *labelmatrix);

    void PrintShadowSamplePoints();
    void PrintShadowFeatures();

private:
    void setLabel();

    void SampleShadow();
    void SampleForeground();
    cv::Mat asSamplesVectors(std::vector<std::vector<int> > Features);

    SegmentationTool st;

    cv::Mat frame; //current frame
    cv::Mat resizeF;
    cv::Mat fgMask;//fg mask generated by MOG method
    cv::Mat skel;//Skeleton Image
    cv::Mat thinnedSkel;//Skeleton Image
    cv::Mat connectivityMap; //Map whit connectivity of nodes
    cv::Mat foregroundSam; //Image for foreground samples

    cv::Ptr<cv::BackgroundSubtractor> mog;

    QImage *current;

    int threshold; //Classification Shadow Threshold
    int maxShadowFeatures; //Cantidad Máxima de Features de Sombra detectar
    int maxForegroundFeatures; //Cantidad Máxima de Features de Foreground a detectard
    bool endTrain; //Flag que indica si finalizo el entrenamiento
    bool setExtofFeatures; //Flag que indica si se quiere realizar extracción de caracteristicas

    cv::Mat _means_S ;
    cv::Mat _weights_S ;
    std::vector<cv::Mat> _covs_S;

    cv::Mat _means_F;
    cv::Mat _weights_F ;
    std::vector<cv::Mat> _covs_F ;

    std::vector<bool> excluded_zone; //Define zones of intersections
    std::vector<CvPoint> intersections;
    std::vector<CvPoint> SsamplePoints; //Shadow Sample Points
    std::vector<CvPoint> FsamplePoints; //Foreground Sample Points

    //Classify pixel whit tri-variate mixture
    //    true = foreground    false = shadow
    std::vector<bool> classifyPixel;

    std::vector<std::vector<int> > sFeatures; //Shadow Feature Vectors
    std::vector<std::vector<int> > fFeatures; //Foreground Feature Vectors
};

#endif // SAMPLESHADOWSOCCER_H