ViBeBGModule.h

#ifndef VIBE_BG_MODULE_H
#define VIBE_BG_MODULE_H

#include "ModuleInterface.h"
#include "ThermalColor.h"
#include "image_display.h"
#include "ReliabilityBg.h"
#include <QImage>

#define FOREGROUND 255
#define BACKGROUND 0

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

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

  void initVibe(cv::Mat firstFrame );
  void segmentation();
  void updateBackground();
  void tuneParameters();

  bool firstTime, displayFeatureMap, displayMeanSample, saveReliabilityMapImage,
    saveFgImage, saveFeatureMapImage, usePreviusFg, saveColouredFgMethod;
  int height, width, numSamples, minSample, minSampleForRealiability, phi, neighbourSize;
  float radius;
  unsigned long randSeed, min_randSeed, max_randSeed, step_randSeed;
  uchar minScale, maxScale, *ptr_colouredFg, *ptr_fgDataImg;
  cv::RNG randGen;
  std::vector <cv::Mat> samples;
  cv::Mat imgARGB32, currImgC3, reliabilityMap, featureMap,
    foreground, meanSample, sample32F, meanSampleU;
  QColor methodColor;

  //for tune parameters
  int min_numSamples, max_numSamples, init_numSamples, step_numSamples, min_minSample, max_minSample,
    init_minSample, step_minSample, min_radius, max_radius, init_radius, step_radius, min_phi, max_phi, init_phi, step_phi, min_neighbourSize, max_neighbourSize, init_neighbourSize, step_neighbourSize;
  bool m_tuningFirst, m_tuningActivated;
};

//V8(3x3) para 640x480, V24(5x5) ( para alfheim 1280 x 960)
//  inicializar vibe primer frame
// clasificar usando R y minSampleForBG
// si phi,= 0,  actualizar dependiendo de la clasificacion
  // fondo:  cambiar una muestra aleatoria por el valor del pixel actual
  // foreground: cambiar una muestra aleatoria por una muestra del vecino V8 o V24.
#endif // VIBE_BG_MODULE_H