codebookmodel.h

#ifndef CODEBOOKMODEL_H
#define CODEBOOKMODEL_H

#include <opencv2/core/core.hpp>
#include <opencv2/legacy/legacy.hpp>
#include <vector>
#include "commonimage.h"

class CodebookModel
{
public:
    CodebookModel();
    CodebookModel(CommonImage* cimg);
    ~CodebookModel();

    void                setCBParameters(std::vector<uchar> params);

    // Learning step
    bool                learnFrame();

    // Foreground retrieving step
    bool                updateFGMask();
    void                getFG(cv::Mat& output);
    void                getFG(QImage& output);

    double              getFGPercentage() const;
    static double       getFGPercentageExtern(const cv::Mat mask);

    void                associateCommonImage(CommonImage* asoc);

    int                 getLearnedFrames() const;

    // Clear and resetting
    void                setRelativeStaleThreshold(double rel);
    double              getRelativeStaleThreshold() const;
    void                clearStales();
    void                resetCBModel();

    // Some misc function I didn't know where to put it.
    static void         getConnectedComponents(const cv::Mat& fginput, std::vector<cv::Mat>& masks, std::vector<cv::Rect>& boundingBoxes);

protected:
    void                cleanse();          // To change the CB Model, look this function first.
    void                allocateCBData();
    void                setCBDefaults();
    bool                verifyNotNullity();

private:
    // Associated image, which we will learn and extract FG/BG.
    CommonImage*        asocimg;

    // Relative stale threshold
    double              relstalethreshold;

    // Stored and retrieved masks
    cv::Mat             fgmask;

    // Functionality with OpenCV
    cv::Mat             yuvimg;
    IplImage            yuvimgshadow;

    // The model.
    CvBGCodeBookModel*  cbmodel;
};

#endif // CODEBOOKMODEL_H