DominantColorSegmentation.h

#ifndef DOMINANTCOLORSEGMENTATION_H
#define DOMINANTCOLORSEGMENTATION_H

/***
 * Author: Daniel Cárdenas, dcardenasn@gmail.com
 * Paper: Automatic Soccer Video Analysis and Summarization
 *
 * Tipo: Segmentacion
 * Methodo: Dominant Color Segmentation
 *
*/

#include <QImage>
#include "Datapool.h"
#include "ModuleInterface.h"
#include <QDir>
#include "SegmentationTool.h"
#include "opencv2/opencv.hpp"
#include <math.h>
#include <stdio.h>

using namespace cv;

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

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

private:

    double HistogramPeakThreshold;
    double Tcolor;
    double upperpenalty;
    double IntensityThreshold;

    bool first_time;
    double Hmean;
    double Smean;
    double Imean;

    bool getHSIColorHistogram(QImage *img);

    double find_histogram_mean(int *histogram, int Hpeak, uchar ipeak);
    double get_cylindrical_dist(int H, int S, int I);

};

#endif // DOMINANTCOLORSEGMENTATION_H