PoppeFootball.h

#ifndef POPPEFOOTBALLMODULE_H
#define POPPEFOOTBALLMODULE_H

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

#include "opencv2/opencv.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/legacy/legacy.hpp"

// Our own Qt-CV shared image format.
#include "../custom_utils/commonimage.h"
#include "../custom_utils/codebookmodel.h"

#define PFVERBOSEMSG(A) \
    if(verbose){ \
        AppendToLog(QString("[POPPEFOOTBALL] ").append(A)); \
        qDebug() << QString("[POPPEFOOTBALL] ").append(A);  \
    }

#define VALIDXMLPARAMMSG(A,B) \
    if(formatok){ \
        PFVERBOSEMSG(A) \
    }else{ \
        PFVERBOSEMSG(B) \
    }

enum morph_operation
{
    MORPHOP_ERODE,
    MORPHOP_DILATE
};
enum morph_element_type
{
    MORPHELEM_CIRCLE,
    MORPHELEM_SQUARE,
    MORPHELEM_CROSS
};

struct morpho_param
{
    morph_operation     op;
    morph_element_type  type;
    uint                elemsize;
};

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

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

    /*
     *************************************************************************
     *************************************************************************
     */

    // Run operation functions
    void retrieveFGMask();

    void prepareStructuralElem(cv::Mat& output, morpho_param par);

    // Colorpicker
    cv::Scalar colorName(QString colorcode, bool* convsuccess = NULL);
    cv::Scalar colorToHSV(cv::Scalar BGRinput);

private:
    // Config parameters
    bool                verbose;
    int                 param_segm_learningframes;
    double              param_segm_learnagain_percentage_low;
    double              param_segm_learnagain_percentage_high;
    std::vector<morpho_param>   param_segm_morphology_ops;
    int                 param_segm_morphologysize;
    std::vector<uchar>  param_segm_codebook_startingconditions;     // Expected to be of 9 parameters.

    // A shadow for the current image, to be used with our CodebookModel.
    CommonImage     currentimage_shadow;
    cv::Mat         currentimage_mat;

    // Foreground extraction
    cv::Mat         fgmask_raw;
    CommonImage     fgmask;         // The pointer in the datapool points to this.

    // Codebook Segmentation model
    CodebookModel   cbmodel;

    // Frame number in the segmentation
    int             seg_nframe;

    // Some string for verbose reasons.
    QString         logverbose;
};

#endif // POPPEFOOTBALLMODULE_H