#ifndef __DETECTOR_YOLOX_H__
#define __DETECTOR_YOLOX_H__

#include <migraphx/program.hpp>
#include <CommonDefinition.h>

namespace migraphxSamples
{

typedef struct _YOLOXParameter
{
    int numberOfClasses;
    float confidenceThreshold;
    float nmsThreshold;
    float objectThreshold;

}YOLOXParameter;

struct Object
{
    cv::Rect_<float> rect;
    int label;
    float prob;
};

struct GridAndStride
{
    int grid0;
    int grid1;
    int stride;
};

class DetectorYOLOX
{
public:
    DetectorYOLOX();
    
    ~DetectorYOLOX();

    ErrorCode Initialize(InitializationParameterOfDetector initializationParameterOfDetector, bool dynamic);
    
    void decode_outputs(float* prob, std::vector<Object>& objects, float scalew, float scaleh, const int img_w, const int img_h);
    void generate_grids_and_stride(std::vector<int>& strides, std::vector<GridAndStride>& grid_strides);
    void generate_yolox_proposals(std::vector<GridAndStride> grid_strides, float* feat_blob, float prob_threshold, std::vector<Object>& objects);
    void qsort_descent_inplace(std::vector<Object>& faceobjects, int left, int right);
    void qsort_descent_inplace(std::vector<Object>& objects);
    void nms_sorted_bboxes(const std::vector<Object>& faceobjects, std::vector<int>& picked, float nms_threshold);
    inline float intersection_area(const Object& a, const Object& b);

    ErrorCode Detect(const cv::Mat &srcImage, std::vector<std::size_t> &relInputShape, std::vector<ResultOfDetection> &resultsOfDetection, bool dynamic);

private:
    cv::FileStorage configurationFile;

    migraphx::program net;
    cv::Size inputSize;
    std::string inputName;
    std::string modelPath;
    migraphx::shape inputShape;
    
    bool useFP16;
    std::vector<std::string> classNames;

    YOLOXParameter yoloxParameter;
};

}


#endif