//
// Created by Scheaven on 2019/12/27.
//
#include <torch/script.h> // One-stop header.
#include <iostream>
#include <memory>
#include <vector>
#include <string>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/opencv.hpp"
#include "opencv2/videoio.hpp"
#include "reid_feature.h"


using namespace std;
//using namespace cv;

int main(int argc, const char* argv[]) {
//    if (argc != 2) {
//        std::cerr << "usage: reid-app <image path>\n";;
//        return -1;
//    }

//    torch::jit::script::Module module;
//    char cam_id = 'A';
//    ReID_Tracker Tracker;



    /*初始化*/
    int gpu_id = 0;
    ReID_Feature R_Feater;
    bool n_flog = R_Feater.ReID_init(0);
    ReID_Utils r_util;
//    ReID_Feature R_Feater(gpu_id);

    /*opencv加载图片信息*/
    cv::Mat human_img = cv::imread("./03.jpg");
    cv::Mat human_img2 = cv::imread("./01.jpg");
    if (human_img.data == nullptr)
    {
        cerr<<"===图片文件不存在"<<endl;
        return 0;
    }else
    {
        //cv::namedWindow("Display", CV_WINDOW_AUTOSIZE);
        try {
            float pFeature1[2048];
            /*转义图片信息格式*/
            // cv::cvtColor(human_img, human_img, cv::COLOR_RGB2BGR);
            // human_img.convertTo(human_img, CV_32FC3, 1.0f / 255.0f);
            float *my_img_data = r_util.normalize(human_img.data, human_img.cols, human_img.rows, 3);
            bool ex_flag1 = R_Feater.ReID_extractor(my_img_data, pFeature1);
//            for (int k = 0; k < 20; ++k) {
//                cout << "-----11111111111------" <<pFeature1[k+2000]<< endl;
//            }

            float pFeature2[2048];
            // cv::cvtColor(human_img2, human_img2, cv::COLOR_RGB2BGR);
            // human_img2.convertTo(human_img2, CV_32FC3, 1.0f / 255.0f);
            float *my_img_data2 = r_util.normalize(human_img.data, human_img.cols, human_img.rows, 3);
            bool ex_flag2 = R_Feater.ReID_extractor(my_img_data2, pFeature2);
//            for (int k = 0; k < 20; ++k) {
//                cout << "-----2222222222------" <<pFeature2[k+2000]<< endl;
//            }

            /*计算相似度*/
            cout << "--attention_distance human-" << endl;
            float result = R_Feater.ReID_Compare(pFeature1, pFeature2);

//            Tracker.storager(1,human_img);
        }
        catch (const c10::Error& e) {
            std::cerr << "error loading the model\n";
            return -1;
        }

        std::cout << "ok\n";
        //cout<<  human_img  <<endl;
        //cv::imshow("0909",human_img);
        //cv::waitKey(0);
    }

}