StatsVisualization.h

Go to the documentation of this file.

#ifndef VISUALIZATIONWINDOW_H
#define VISUALIZATIONWINDOW_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
  */

#include "ScreenQuad.h"
#include "PlayerData.h"

#include <ngl/Camera.h>
#include <ngl/Colour.h>
#include <ngl/TransformStack.h>
#include <ngl/ShaderLib.h>
#include <ngl/VertexArrayObject.h>
#include <ngl/Text.h>


#include "opencv2/opencv.hpp"
#include <opencv2/core/core.hpp>
#include <iostream>
#include <string>

// must be included after our stuff becuase GLEW needs to be first
#include <QtOpenGL>



typedef enum
{
    PLAYERA,
    PLAYERB,
    BOTH
}WHOSE_STATS;

typedef enum
{
    IMPACT_POINTS_VIEW,
    IMPACT_PERCENTAGE_DISTRIBUTION_VIEW,
    IMPACT_HEIGHT_MAP_VIEW,
    TRAJECTORY_VIEW,
    VELOCITY_VIEW,
    RPM_VIEW

}WHICH_STAT;

struct VertData
{
  //GLfloat u; // tex cords
  //GLfloat v; // tex cords
  GLfloat nx; // normal
  GLfloat ny;
  GLfloat nz;
  GLfloat x; // position
  GLfloat y;
  GLfloat z;
};

class StatsVisualization : public QGLWidget
{
Q_OBJECT        // must include this if you use Qt signals/slots
public :

    //def constructor
    StatsVisualization();

  StatsVisualization(
           QWidget *_parent, PlayerData *_playerA, PlayerData *_playerB
          );


  StatsVisualization(
           QWidget *_parent, bool _importFlag
          );

  ~StatsVisualization();
  void processKey(
                   QKeyEvent *_event
                 );

  void updateDisplay();

private :

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  int m_spinXFace;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  int m_spinYFace;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  bool m_rotate;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  bool m_translate;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  int m_origX;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  int m_origY;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  int m_origXPos;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  int m_origYPos;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  ngl::Vec3 m_modelPos;
  //----------------------------------------------------------------------------------------------------------------------

  ngl::Camera *m_cam;
  ngl::TransformStack m_transformStack;

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  ngl::VertexArrayObject *m_vao;

  ScreenQuad *m_2DScreen;

  int m_updateTimer;

  // stats array

  std::vector<cv::Point> m_playerAImpactData;
  std::vector<cv::Point> m_playerBImpactData;
  float m_percentageABL,m_percentageATL,m_percentageATR,m_percentageABR;
  float m_percentageBBL,m_percentageBTL,m_percentageBTR,m_percentageBBR;

  std::vector<cv::Point> m_quadSpecABL,m_quadSpecATL,m_quadSpecATR,m_quadSpecABR;
  std::vector<cv::Point> m_quadSpecBBL,m_quadSpecBTL,m_quadSpecBTR,m_quadSpecBBR;
  std::vector<std::vector<cv::Point3f> > m_playerAInterpPtsMappedToRallies;
  std::vector<std::vector<cv::Point3f> > m_playerBInterpPtsMappedToRallies;

  std::vector<float> m_playerATrajectoryFromFile;
  std::vector<float> m_playerBTrajectoryFromFile;

  std::vector<VertData> m_dataVertsA, m_dataVertsB;

  std::vector<float> m_speedAData,m_speedBData;
  std::vector<cv::Point3f> m_playerAImpactPtsAllRalliesIn3D,m_playerBImpactPtsAllRalliesIn3D;
  std::vector<float> m_playerARPM,m_playerBRPM;

  cv::Rect m_fieldOfPlay;
  float m_finalCourtWidthWorld,m_finalCourtDepthWorld;
  int m_width,m_height,m_midTopX,m_midTopY,m_midBottomX,m_midBottomY,m_midLeftX;
  int m_midLeftY,m_midRightX,m_midRightY;

  PlayerData *m_playerA,*m_playerB;

  bool m_impactPtsFilled;
  bool m_speedMapBuilt,m_rpmDataBuilt,m_quadMapBuilt,m_trajectoryBuilt;

  WHOSE_STATS m_whichPlayer;
  WHICH_STAT m_whichStat;

  WHICH_QUADRANT m_whichQuad;

  bool m_3dFlag;

  float m_kinectToPlayAreaInMeters;
  float m_diameterOfBAllInMeters;

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  ngl::Vec3 m_lightPos;

  bool m_importFlag;

  ngl::Text *m_text;

  GLuint m_crowdTextureName,m_tableTextureName,m_speedLegend,m_rpmLegend,m_trajectoryLegend,m_gridXYHtMap;
  GLuint m_heightMapLegend;

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  ngl::Mat4 m_view;
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  ngl::Mat4 m_projection;

  int m_legendWidth,m_legendHeight;

protected:

  void initializeGL();

  void resizeGL(
                const int _w,
                const int _h
               );
  void paintGL();

private :

  void drawEntireScene();

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  void mouseMoveEvent (
                        QMouseEvent * _event
                      );
  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  void mousePressEvent (
                          QMouseEvent *_event
                       );

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  void mouseReleaseEvent (
                          QMouseEvent *_event
                          );

  //----------------------------------------------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------------------------------------------
  void wheelEvent(
                    QWheelEvent *_event
                 );
  void timerEvent(
                  QTimerEvent *_event
                 );

  void loadMatricesToADSShader();

  void loadMatricesToTextureShader();

  void loadMatricesToLegendShader();

  void fillAllImpactPointsBothPlayers();
  void buildQuadSpecificData();
  void buildPointGraphBars();
  void buildTrajectory();
  void buildSpeedMap();
  void buildRPMData();

  void calculateFieldOfPlayInWorldCoords();
  void setColorBasedOnSpeedRange(float _inputVelocity);
  void setColorBasedOnRPMRange(float _inputRPM);

  void fillDataVertsWithNormals(std::vector<cv::Point3f>& _inputVertList,
                                std::vector<VertData>& o_normalFilledDataVerts);

  void fillDataFromFileForBothPlayers(cv::FileStorage& _fs,WHOSE_STATS _whichPlayer);

  void drawCrowd(const std::string& _inputPlaneName);
  void drawTable(const std::string& _inputTableName);

  void renderPercentages();

  void legendWindow(GLuint _inputLegendName);

  void drawContainingGridScaleXY();

public slots:

  void setWhichStatToView(int _inputWhichStat);

  void setWhoseStatToView(int _inputWhoseStat);

  void setWhichQuadToView(int _inputWhichQuad);

  void setKinectToPlayAreaDistance(double _inputDistanceInMeters);

  void setDiameterOfBallInMeters(double _inputDiaInMeters);

  void readFile(QString _inputFileName);

  void writeFile();

};

#endif // VISUALIZATIONWINDOW_H