playerlabelingfrombagoffeatures.h

#ifndef PLAYERLABELINGFROMBAGOFFEATURES_H
#define PLAYERLABELINGFROMBAGOFFEATURES_H

#include <QDomNode>
#include <QImage>

#include "reliabilitysinglemodelinterface.h"
#include "src/blob.h"
#include "src/footballConstants.h"

class PlayerLabelingFromBagOfFeatures : public ReliabilitySingleModelInterface
{
public:
    PlayerLabelingFromBagOfFeatures(int buffer_size);

    virtual ~PlayerLabelingFromBagOfFeatures();

    //IMPLEMENT!! Initializes the model attributes for the interface.
    //The function is called by the interface constructor
    virtual void initInstanceAttributes();

    //IMPLEMENT!! Initializes the dynamics model attributes for the interface.
    //The function is called by the interface constructor
    virtual void initDynamicsAttributes();

    //IMPLEMENT!! Copy function for specific data from the instance (general_copy calls it after copying base info)
    virtual void copy(SpReliabilitySingleModelInterface);

    //IMPLEMENT!! Copy general structures and parameters of specific model, without considering instance specific data
    virtual void copy_structure(SpReliabilitySingleModelInterface);

    //IMPLEMENT!! sets parameters from xml file
    virtual void setParameters(QDomNode &i_parameters);

    //IMPLEMENT!! sets activation criteria for each model: reliability on input (distance, bad data),
    //   needs (occlusion, priority),
    //sets priority of models (hierarchy).
    //Checked before update phase
    virtual void activate(BoundingBox &bb);

    //IMPLEMENT!! initializes every activated model, according to their own input (region-blob-segments schema)
    //Executed only once at the begginning of single model creation (after 'copy_structure' and 'copy' methods
    //are used to copy the information).
    virtual void init();

    //IMPLEMENT!! updates activated model instance attributes for current frame, and returns the set of detected/enriched objects.
    //            It notifies if roi (bounding box, or segmentation) changes compared to the input, for reprocessing.
    virtual bool updateInstance(BoundingBox &, std::set<SpRMMHypothesis, hypothesesOrderedByBestProbabilityOperator> &);

    //IMPLEMENT!! updates activated model instance attributes for current frame and object, and returns the detected/enriched object.
    //            It notifies if roi (bounding box, or segmentation) changes compared to the input, for reprocessing.
    //            This function does NOT generate mobiles or hypotheses. It processes attributes for the current object.
    //            This function could be used by updateInstance to generate mobile information.
    virtual void updateSingleInstance(BoundingBox &bb);


    //IMPLEMENT!! updates model dynamics, based on instances information
    virtual void updateModelDynamics();

    //IMPLEMENT!! sets global probability for the model (how good is the model)
    virtual void setProbabilityAndReliability();

    //IMPLEMENT!! sets blob enclosing the model in forward process (update)
    virtual void setForwardInterface();

    //OPTIONAL
    virtual bool draw(QPainter &painter);

    virtual bool draw(QPainter &painter, int r, int g, int b);

    virtual BoundingBox getEstimator(double &R);

    ObjectType attributeNameToObjectType(QString s);

private:

    appHistType::iterator itClassHist;

    double threshold_histogram;
    double weightCorr;
    double weightBatha;
    double weightLuma;
    double weightChroma;
    double weightLuv;

    bool useForegroundMask;
    bool countAll;
    bool useClassified;
    int distType;
    bool drawFlag;
    unsigned long showHistogram;

    std::map< typeCluster, QString > nameFeature;


    double *countPixelClusterBlob_Chroma(const std::vector<cv::Point> &c,BoundingBox rect, bool useForgroundMask = true, bool countAll = true, int distType = 1, bool useClasdified = true, bool showH = false);
    double *countPixelClusterBlob_Luv(const std::vector<cv::Point> &c,BoundingBox rect, bool useForgroundMask = true, bool countAll = true, int distType = 1, bool useClasdified = true, bool showH = false);
    double *countPixelClusterBlob_Luma(const std::vector<cv::Point> &c,BoundingBox rect, bool useForgroundMask = true, bool countAll = true, int distType = 1, bool useClasdified = true, bool showH = false);
    void calcDistanceAttributes(std::map< typeCluster,  double *> blobHistograms, std::map< typeCluster,  int> bins, std::map< typeCluster, double > weights, ModelInstanceDataPack &m);

};

#endif // PLAYERLABELINGFROMBAGOFFEATURES_H