#ifndef MODEL_H
#define MODEL_H
#include "dataType.h"
#include <algorithm>
#include "../config.h"

// * Each rect's data structure.
// * tlwh: topleft point & (w,h).
// * confidence: detection confidence.
// * feature: the rect's 128d feature.
// */
class DETECTION_ROW
{
public:
    int obj_id;
    DETECTBOX tlwh;
    float confidence;
    FEATURE feature;
    DETECTBOX to_xyah() const;
    DETECTBOX to_tlbr() const;

    bool isFall;
    int fallScore;

    bool isRun;
    int runScore;

    bool is_hat;
    int hatScore;

    int helmetScore;

    int headScore;

    bool is_mask;
    int maskScore;

    bool is_smoke;
    int smokeScore;

};

typedef std::vector<DETECTION_ROW> DETECTIONS;


#endif // MODEL_H