BallTrackingVisualization.h

Go to the documentation of this file.

#ifndef DEPTH_WINDOW_H__
#define DEPTH_WINDOW_H__

/*
  LICENSING: You are free to use any part of this project provided I am informed about it via email
  at santoshwins@hotmail.com and referenced appropriately in your works
  */

// must be included after our stuff becuase GLEW needs to be first
#include "KinectInterface.h"
#include "ScreenQuad.h"
#include "QuadCurveFitUtility.h"

#include <QTime>
#include <QEvent>
#include <QResizeEvent>

//#include <opencv2/gpu/gpu.hpp>


class StatsVisualization;

enum VIDEOMODE{RGB,DEPTH};

class BallTrackingVisualization : public QGLWidget
{
Q_OBJECT        // must include this if you use Qt signals/slots
public :
  BallTrackingVisualization(
             QWidget *_parent
             );

  ~BallTrackingVisualization();

private :

    //----------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    cv::Mat m_rgb, m_gray;

    //---------------------------------------------------------------------------------------------------------------------
    //---------------------------------------------------------------------------------------------------------------------
    cv::Mat m_nextBuffer,m_prevBuffer, m_diffBuffer, m_bufferDepthRaw, m_bufferDepthRaw1;


    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    float m_depthinm[640*480];

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    ScreenQuad *m_screen;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    std::vector<cv::Moments> m_momentList;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    cv::Mat canny_output;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    std::vector<std::vector<cv::Point> > contours;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    cv::Mat hierarchy;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    double area,rectArea;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    std::vector<cv::Moments> mu;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    std::vector<cv::Point2f> mc;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    double m_binaryLowerThresh,m_binaryMaxVal,m_cannyLowerThresh,m_cannyHigherThresh;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    double m_contourAreaThresh,m_diffAreaThresh;

    //---------------------------------------------------------------------------------------------------------------------
    //----------------------------------------------------------------------------------------------------------------------
    int m_ellipsesize,m_blurSize,m_noOfTimesToDialate, m_masterRigCheck;


protected:

  void initializeGL();

  void resizeGL(
                const int _w,
                const int _h
               );
  // be re-drawn
  void paintGL();

private :

  void mouseMoveEvent (
                       QMouseEvent * _event
                      );

  void mousePressEvent (
                        QMouseEvent *_event
                       );

  void mouseReleaseEvent (
                          QMouseEvent *_event
                         );

  void timerEvent(
                    QTimerEvent *_event
                   );


public slots:

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  void setBinaryLowerThresh(double _lowerThresh);



  void setStructuringElementSize(int _size);


  void setBlurSize(int _size);


  void setCannyLowerThresh(double _lowerThresh);



  void setCannyHigherThresh(double _higherThresh);



  void setNumerOfDialtions(int _numberOfDialtions);


  void setContourAreaThresh(double _contourAreaThresh);


  void setDifferenceInAreaThresh(double _diffAreaThresh);


  void setMasterRigCheck(int _masterRigCheck);

};

#endif