fivelink2dbodymodel.h

#ifndef FIVELINK2DBODYMODEL_H
#define FIVELINK2DBODYMODEL_H

#include "opencv2/opencv.hpp"
#include <QString>

#define TORSO 0
#define LIMBSTOPLEFT 1
#define LIMBSTOPRIGHT 2
#define LIMBSBOTTOMLEFT 3
#define LIMBSBOTTOMRIGHT 4
#define HEAD 5

typedef struct posture posture;

struct posture{
    /*Centros del Modelo*/
    cv::Point2f center;//Center of the model
    cv::Point2f centerthighsLeft;
    cv::Point2f centerthighsRight;
    /*Vector de inclinación representando los angulos entre los ejes de las partes del cuerpo y
    el eje vertical y*/
    std::vector<double> angles;
};

typedef struct shape shape;
struct shape{
    //Base-to-height Ratio (alpha = b/l base and length of each rectangle). K= {a1,...,a6}
    std::vector<double> K;
    //Vector de Altura Relativas (ri = li/l1), R = {r1,...,r6}
    std::vector<double> R;
};

typedef struct baseModel baseModel;
struct baseModel{
    //Base-to-height Ratio (alpha = b/l base and length of each rectangle). K= {a1,...,a6}
    posture T;
    //Vector de Altura Relativas (ri = li/l1), R = {r1,...,r6}
    shape S;
};

typedef struct postureDraw postureDraw;
struct postureDraw
{
    float x, y;
    float base;
    float height;
    float angle; //Rad

    float x_prev, y_prev;
};

class FiveLink2DBodyModel
{

public:

    FiveLink2DBodyModel();
    FiveLink2DBodyModel(int id_p);
    ~FiveLink2DBodyModel();

    void drawModel(cv::Mat &img, bool paintExModel = true, int part = TORSO);

    void saveModel(); //SaveModel in .xml File
    bool initModel(cv::Rect roi, bool extPar, int w, int h);
    void setId(unsigned int index);

    bool torsoModelPoints(double height, double width);
    bool thighsModelPoints(double height);
    bool calvesModelPoints(double height);
    bool headModelPoints(double height);

    void setModelRectangle(cv::Point2f center, cv::Size2f size, float angle,int part);

    cv::Mat getDatasetOfModel();

    bool initOKAll();

    int id;
    baseModel Mb; //Modelo Base
    cv::Rect M; //Rectangulo vecino que se le agrega al modelo Base

    /*28 points of the model in a current position of player
    * This points should be in concordance with the position of the player on the image
    */
    std::vector<cv::RotatedRect> modelRectangles;
    void printModelRectangleInfo(int part);

    std::vector<bool> initOK; //Vector que indica si los parámetros de cada una de las partes del cuerpo han sido inicializadas

    double constExtModel; //Constraint for the Extendend body Model
    double torsoPropLF; //Proporción entre la vista lateral y frontal del torso
    double torsoPropLD; //Proporción entre la vista lateral y diagonal del torso

    double beta; //Factor para desplazar la articulación de la parte inferior de la extr.
    double alpha; //Factor para desplazar la articulación de la cabeza
    double gama; //Factor para desplazar la parte inferior de los pies

    //Objetos que actualmente cumplen el Modelo
    std::vector<int> associatedObj;

    //Parametros cálculo función de costo
    double par_Mm_i; //Parámetro de Mediciones faltantes interno del modelo (puede haber uno global en el modulo)
    double par_Ou_i; //Parámetro de Mediciones sin explicación interno del modelo (puede haber uno global en el modulo)
    double costThreshold;//Parámetro para definir el umbral que considera una optimización del modelo adecuada


    double ratio_Torso_height;
    double ratio_Torso_width;
    double ratio_Limbs_height;
    double ratio_Limbs_width;
    double ratio_Head_height;
    double ratio_Head_width;

    double relativePos_LimbsTorso;

    double init_angleHead;
    double init_angleTorso;
    double init_angleLimbsTopLeft;
    double init_angleLimbsTopRight;
    double init_angleLimbsBottomLeft;
    double init_angleLimbsBottomRight;

    cv::Point2f centerROI; //Centro de la region de interes


};

#endif // FIVELINK2DBODYMODEL_H