segmentationmodule.h

#ifndef SEGMENTATIONMODULE_H
#define SEGMENTATIONMODULE_H

#include <map>
#include <string>
#include <QImage>
#include "Datapool.h"
#include "ModuleInterface.h"
#include <QDir>
#include "src/json.h"
#include "ReliabilityBg.h"

#include "opencv2/opencv.hpp"


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

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

  //Global updating settings functions
  bool setGlobalZones();
  bool setGlobalMask();
  bool insideInZone(int i, int j);
  bool outsideOutZone(int i, int j);

  void updateBGGlobally(QImage *current);

  bool readImageOnDisk(QString dir,QImage **o_rpOutputImage);

  void tuneParameters();

  //Processes segmentation with provided data, and returns the number of foreground pixels in the region:
  static int getSegmentation(cv::Mat& input, cv::Mat& output, cv::Mat& bg, cv::Mat& mask, int threshold);
  static int getSegmentation(Datapool *dp, BoundingBox &roi, cv::Mat& output, int threshold);

  void getDiff(QImage *, QImage *);

  uchar T;

private:
  int currentFrame;
  int line;
  int cont;
  bool BGup;
  double alpha;    //tasa de aprendizaje
  bool m_global;
  int m_N;
  int m_K;
  QString m_InZone;
  QString m_OutZone;
  bool m_inDefined;
  bool m_outDefined;
  bool m_aperture;
  bool m_tuningActivated;
  int m_currentTuneValue;
  int m_tuneStep;
  int m_maxTuneValue;

  bool globalMaskReady;
  QImage *globalMask;
  int globalValidPoints;
  bool global_validated;
  bool validate_once;
  bool different_bg_current;
  std::map<int, std::deque<RLESegment> > RLEglobal;

  zoneType zInType, zOutType;
  QSharedPointer<world::Zone> zin3D, zout3D;
  QSharedPointer<world::Zone2D> zin2D, zout2D;
  QSharedPointer<world::ZoneH> zinH, zoutH;

  QString bgImName;
  bool first;
  bool globalNotSet;
  bool m_mask;
  QString m_maskImName;
  void applyAperture(QImage *, QImage *);
  void updateBG(QImage *);
  std::map<int, std::deque<int> > globalPoints;
  cv::Mat element;
  cv::Mat featureMap, reliabilityMap;
  float minScaleThreshold, maxScaleThreshold, minScaleThresholdRatio, maxScaleThresholdRatio;

  int width, height;

  bool chromaticSegmentationIsActivated;
  int minGrade;
  int maxGrade;
  int minSaturation;
  int maxLight;
  // Added by Barrientos Nettle - CV 2014-1
  bool chromaticSegmentationFromHSVdataFiles;
  std::vector<std::vector<int> > hsvFGparams;
  std::vector<std::vector<int> > hsvBGparams;
  std::vector<std::vector<int> > readHSVparamsFile(QString fileName);
  int BgED;
  int FgED;
  int BgSS;
  int FgSS;

  enum BG_FG_ErodeDilate
  {
    erode,
    dilate,
    none
  };

};

#endif // SEGMENTATIONMODULE_H