PlayerData.h

Go to the documentation of this file.

#ifndef PLAYERDATA_H
#define PLAYERDATA_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 "TwoDStatsGeneration.h"
#include "ThreeDStatsGeneration.h"

#include "opencv2/opencv.hpp"


typedef enum
{
    TOP_LEFT,
    BOTTOM_LEFT,
    BOTTOM_RIGHT,
    TOP_RIGHT,
    ALL
}WHICH_QUADRANT;

class PlayerData
{
    public:

        PlayerData();
        ~PlayerData();

        void setPlayerBallPts(std::vector<std::vector<cv::Point3f> > &_inputPointSetsInWorld);
        void setPlayerImpactPtIndices(std::vector<std::vector<int> > &_inputDepthIndices);

        void setPlayerDataProcessedFlag();

        std::vector<cv::Point>& getPlayerImpactPtsAllRallies();

        std::vector<std::vector<cv::Point3f> >& getPlayerBallPtsAll();

        std::vector<std::vector<int> >& getPlayerImpactPtIndices();

        float getPercentageDistributionData(int _width, int _height,WHICH_QUADRANT _whichQuadrant);

        std::vector<cv::Point>& getQuadSpecificPoints(WHICH_QUADRANT _whichQuad);

        std::vector<cv::Point3f>& getBarGraphVertices(cv::Point _inputTopXY, int _width, int _height, int _gridResolution);

        std::vector<std::vector<cv::Point3f> >& getPlayerInterpolatedBallPtsAll(float _inputDistanceFrmKinectToPlayArea);

        std::vector<cv::Point3f>& getPlayerImpactPtsIn3DAllRallies();

        std::vector<float>& getPlayerSpeedData(float _inputDistanceFrmKinectToPlayArea);

        std::vector<float>& getRevolutionsPerMinute(float _inputDiameterInMeters,
                                                    float _inputDistanceFrmKinectToPlayArea);


        bool checkDataProcessed();


    private:

        TwoDStatsGeneration *m_twoDStatsGenerator;
        ThreeDStatsGeneration *m_threeDStatsGenerator;



        bool m_dataProcessed;

        std::vector<std::vector<cv::Point3f> > m_playerBallPts;

        std::vector<std::vector<int> > m_playerDeepestPtIndices;

        std::vector<cv::Point> m_playerImpactPtsAllRallies;
        std::vector<cv::Point3f> m_playerImpactPtsAllRalliesIn3D;

        float m_bottomLeftPercentage,m_topLeftPercentage,m_topRightPercentage,m_bottomRightPercentage;

        std::vector<cv::Point> m_playerImpactPtsBottomLeft,m_playerImpactPtsTopLeft;
        std::vector<cv::Point> m_playerImpactPtsTopRight,m_playerImpactPtsBottomRight;

        std::vector<cv::Point3f> m_barGraphVertices;

        std::vector<std::vector<cv::Point3f> > m_playerInterpolatedBallPts;

        std::vector<float> m_speedData;
        std::vector<float> m_rpmData;

        void calculatePlayerImpactPtsFromIndices();

};

#endif // PLAYERDATA_H