#ifndef __DETECTOR_YOLOX_H__
#define __DETECTOR_YOLOX_H__

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

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);
    ErrorCode Detect(const cv::Mat &srcImage,
                     const std::vector<std::size_t> &relInputShape,
                     std::vector<ResultOfDetection> &resultsOfDetection);

  private:
    void decode_outputs(float *prob, std::vector<Object> &objects, float scalew,
                        float scaleh, const int img_w, const int img_h,
                        cv::Size inputSize);
    void generate_grids_and_stride(std::vector<int> &strides,
                                   std::vector<GridAndStride> &grid_strides,
                                   cv::Size inputSize);
    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);

    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;
};

} // namespace migraphxSamples

#endif