labeler.hpp

#ifndef _QDISPLAY_LABELER_H_
#define _QDISPLAY_LABELER_H_

#include <QObject>
#include <QApplication>


#include "qdisplay/Viewer.hpp"
#include "qdisplay/ImageView.hpp"
#include "qdisplay/Shape.hpp"
#include "kernel/threads.hpp"


namespace jafar {
namespace qdisplay {

/******************************************************************************/

enum LabelMode { lmBoundingBox = 0, lmSizeOri, lmBlobs, lmCOUNT };
static const std::string LabelModeNames[lmCOUNT] = {"BoundingBox", "SizeOri", "Blobs" };

class LabelBoundingBox
{
public:
  enum Quality { qlGood=0, qlOk, qlBad };
public:
  LabelBoundingBox(): rect(), quality(qlGood) {}
  cv::Rect_<double> rect;
  Quality quality;
};


class LabelSizeOri
{
public:
  double size;
  double ori;
  double center_x, center_y;
};


class LabelBlobs
{
public:
  std::vector<std::pair<double,double> > centers;
};


class LabeledImage
{
public:
  enum Status { stUninit=0, stManual, stInterpol, stNothing };
public:
  LabeledImage(): status(stUninit) {}
  std::string name;
  Status status;
  LabelBoundingBox lbbox;
  LabelSizeOri lsori;
  LabelBlobs lblob;
};



class Labeler: public QObject
{
  Q_OBJECT
private:
  qdisplay::Viewer *viewer;
  qdisplay::ImageView *view;
  image::Image img;
  std::vector<LabeledImage> images;
  int nimages;
  int pos, lastpos;
  double w,h;
  LabelBoundingBox::Quality qual;
  qdisplay::Shape *cursor;
  qdisplay::Shape *mark;
  LabelMode mode;
  unsigned state;
private:
  void display();
  void displayRects();
  void interpol(int start, LabeledImage::Status statusFind1, LabeledImage::Status statusFill1,
                LabeledImage::Status statusFind2, LabeledImage::Status statusFill2);
  void interpol(int before, int after, LabeledImage::Status status);
  void save();
  void load();
  void increaseSize(double inc);
  void decreaseSize(double inc);
  void remove();
  void next();
  void prev();
  void find(LabeledImage::Status status);
  void quality(LabelBoundingBox::Quality quality);
  void switch_mode();
public:
  Labeler(int nimages, char** images);
  ~Labeler();
public slots:
  void onKeyPress(QKeyEvent *event);
  void onMouseClick(QGraphicsSceneMouseEvent *mouseEvent, bool isClick);
  void onMouseMove(QGraphicsSceneMouseEvent *mouseEvent);
};



}}

namespace cv {
template<typename T> Rect_<T> operator+(Rect_<T> const& A, Rect_<T> const& B)
  { return Rect(A.x+B.x, A.y+B.y, A.width+B.width, A.height+B.height); }
template<typename T> Rect_<T> operator-(Rect_<T> const& A, Rect_<T> const& B)
  { return Rect(A.x-B.x, A.y-B.y, A.width-B.width, A.height-B.height); }
template<typename T> Rect_<T> operator*(Rect_<T> const& A, T b)
  { return Rect(A.x*b, A.y*b, A.width*b, A.height*b); }
template<typename T> Rect_<T> operator/(Rect_<T> const& A, T b)
  { return Rect(A.x/b, A.y/b, A.width/b, A.height/b); }
}

#endif