footballscenemodel.h

#ifndef FOOTBALLSCENEMODEL_H
#define FOOTBALLSCENEMODEL_H
#include "calibration.h"
#include "footballConstants.h"


/***
 * Author: Cristobal Barrientos Low, cbarrientoslow@gmail.com
 *
 * Tipo: Modelo de escena para campo de juego en el contexto del fútbol
 * Paper:Learning Surveillance Tracking Models for Self-Calibrated Groud Plane
 * Autores: J. Renno, P. Remagnino, G.A Jones
*/

enum fieldZones
{
    teamA_zone,
    teamB_zone,
    outOfField,
    teamA_smallarea,
    teamB_smallarea,
    middleField_zone,
    fieldZone
};

//Estructura que define las variables para representar una muestra de tamaño
typedef struct sizeSample sizeSample;
struct sizeSample
{
        double value;
        QPoint pointImage;
};

class footballSceneModel : public SceneModel
{
public:

    footballSceneModel();
    footballSceneModel(double hmodel, double wmodel, double par1, double par2);
    ~footballSceneModel();

    void setParameters(const cv::Mat &p, const cv::Mat &h);
    bool trainSizesModel();
    std::vector< sizeSample > getSamples();//Obtiene muestras de tamaño desde un archivo CSV
    void setField(); //Definir las lineas que pertenecen al campo de juego dado los parámetros de tamaño de éste
    void setFieldDimensions();//Definir las dimensiones reales en metros del terreno de juego

    double getHeightFromModel(int px, int py);
    double getWidthFromModel(int px, int py, double angle = 0);
    double WmodelfromAngle(double angle);

    bool isOnArea(int px, int py, fieldZones zone);//Entrega Verdadero si el punto en la imagen se encuentra sobre alguna de las areas del campo de juego

    QPointF getRealWorldFieldPos(double px, double py);
    QPoint getScreenFieldPos(double x, double y);

    //Altura del jugador en cierta posición vertical en la imagen
    double _hmodel;
    double _wmodel; //Person
    double _par1;
    double _par2;

    std::map< fieldZones, QRect > zones; //Rectas en la imagen que definen cada una de las zonas del campo de juego

    perspective_matrix pM; //Matriz Perspectiva de la cámara
    homography_matrix hM; //Matriz de Homografía entre el campo de juego y la cámara
};

#endif // FOOTBALLSCENEMODEL_H