Hu Chunming / mvpt_dq_ascend_arm

  #include "../ai_platform/stl_aiplatform.h"
  #include <chrono>
  #include <thread>
  #include <stdio.h>
  #include <string.h>
  #include <stdlib.h>
  #include <unistd.h>

  #include <iostream>

  #include "opencv2/opencv.hpp"
  #include "opencv2/imgcodecs/legacy/constants_c.h"
  #include "opencv2/imgproc/types_c.h"
  
  #include "demo_util.hpp"
  

  using namespace std;

  using namespace cv;

  
  static long long get_cur_time(){
      chrono::time_point<chrono::system_clock, chrono::milliseconds> tpMicro
          = chrono::time_point_cast<chrono::milliseconds>(chrono::system_clock::now());
      return tpMicro.time_since_epoch().count();
  }
  

  string get_ipc_url(int gi){
    string ipc_url = "/data/share/data/Street.uvf";

    switch(gi){
      case 0:

        ipc_url = "rtsp://admin:admin@123456@192.168.60.176:554/cam/realmonitor?channel=1&subtype=0";

        break;
      case 1:

        ipc_url = "rtsp://122.97.218.170:8604/openUrl/V5nXRHa?params=eyJwcm90b2NhbCI6InJ0c3AiLCJjbGllbnRUeXBlIjoib3Blbl9hcGkiLCJleHByaWVUaW1lIjotMSwicHJvdG9jb2wiOiJydHNwIiwiZXhwaXJlVGltZSI6MzAwLCJlbmFibGVNR0MiOnRydWUsImV4cGFuZCI6InN0YW5kYXJkPXJ0c3Amc3RyZWFtZm9ybT1ydHAiLCJhIjoiMTBjZjM4N2JjY2Y5NDg3YzhjNWYzNjE2M2ViMWUyNTJ8MXwwfDEiLCJ0IjoxfQ==";

        break;
      case 2:

        ipc_url = "rtsp://admin:ad123456@192.168.10.166:554/cam/realmonitor?channel=1&subtype=0";

        break;
      case 3:

        ipc_url = "rtsp://122.97.218.170:8604/openUrl/LBBYTra?params=eyJwcm90b2NhbCI6InJ0c3AiLCJjbGllbnRUeXBlIjoib3Blbl9hcGkiLCJleHByaWVUaW1lIjotMSwicHJvdG9jb2wiOiJydHNwIiwiZXhwaXJlVGltZSI6MzAwLCJlbmFibGVNR0MiOnRydWUsImV4cGFuZCI6InN0YW5kYXJkPXJ0c3Amc3RyZWFtZm9ybT1ydHAiLCJhIjoiOTgzYjRjMmUxMThlNGU1OTlkYThmMTI3NTkyMGViODV8MXwwfDEiLCJ0IjoxfQ==";
        // ipc_url = "rtsp://122.97.218.170:8604/openUrl/V5nXRHa?params=eyJwcm90b2NhbCI6InJ0c3AiLCJjbGllbnRUeXBlIjoib3Blbl9hcGkiLCJleHByaWVUaW1lIjotMSwicHJvdG9jb2wiOiJydHNwIiwiZXhwaXJlVGltZSI6MzAwLCJlbmFibGVNR0MiOnRydWUsImV4cGFuZCI6InN0YW5kYXJkPXJ0c3Amc3RyZWFtZm9ybT1ydHAiLCJhIjoiMTBjZjM4N2JjY2Y5NDg3YzhjNWYzNjE2M2ViMWUyNTJ8MXwwfDEiLCJ0IjoxfQ==";

        break;
      case 4:

        ipc_url = "/data/share/data/Street.uvf";

        break;
      case 5:

        ipc_url = "/data/share/data/公安局老桥头_CVR15F89410_1465819864_1B.mp4";

        break;
      case 6:

        ipc_url = "/data/share/data/不带头盔2.mp4";

        break;
      case 7:

        ipc_url = "/data/share/data/hczr1.mp4";

        break;
      case 8:

        ipc_url = "/data/share/data/燕高路口高点_CVR_2015-12-30_09-00-00_2015-12-30.mp4";

        break;
      case 9:

        // ipc_url = "/opt/share/data/1-00000002d55_h265.mp4";
        ipc_url = "/data/share/data/duan1.mp4";

        break;
      case 10:

        ipc_url = "/data/share/data/gs1-00000001.mp4";

        break;
      case 11:

        ipc_url = "/data/share/data/98.mp4";

        break;
      case 12:

        ipc_url = "/data/share/data/AV_12345_1_20230108110849847_D_1673147329847729.mp4";

        break;
      case 13:

        ipc_url = "/data/share/data/nx_1-00000002.mp4";

        break;
      case 14:

        ipc_url = "/data/share/data/nx1-0000001.mp4";

        break;
      case 15:

        ipc_url = "/data/share/data/nx1-00000002.mp4";

        break;
      case 16:

        ipc_url = "/data/share/data/Street.uvf";

        break;
      case 17:

        ipc_url = "/data/share/data/xiancun.mp4";

        break;
      case 18:

        ipc_url = "rtsp://192.168.60.146:8554/w_1_out";

        break;
      case 19:

        ipc_url = "/data/share/data/42_44.avi";

        break;
      case 20:

        ipc_url = "/data/share/data/37_70.avi";

        break;
      case 21:

        ipc_url = "/data/share/data/192_25.avi";

        break;
      case 22:

        ipc_url = "/data/share/data/31_4.avi";

        break;
      case 23:

        ipc_url = "/data/share/data/146_192.avi";

        break;
      case 24:

        ipc_url = "/data/share/data/146_187.mp4";

        break;
      case 25:
        // 公安内网

        ipc_url = "rtsp://192.168.51.200:8554/320202202308231001001002";

        break;
      case 26:
        // FFmpeg生成的解码错误数据流

        ipc_url = "rtsp://192.168.10.4:8554/street";

        break;
      case 27:

        ipc_url = "/data/share/data/Street_4k_265.mp4";

        break;
      case 28:

        ipc_url = "http://192.168.60.179:10016/公安局老桥头_CVR15F89410_1465819864_1B.mp4";

        break;
      case 29:

        ipc_url = "http://192.168.60.179:10016/不带头盔2.mp4";

        break;
      case 30:

        ipc_url = "/data/share/data/149_45.avi";

        break;
      case 31:

        ipc_url = "/data/share/data/146_88.avi";

        break;
      case 32:

        ipc_url = "/data/share/data/141_74.avi";

        break;
      default:

        ipc_url = "/data/share/data/Street.uvf";

        break;
    }

    return ipc_url;

  }
  

  static int nTaskId = 0;

  string createTask(void *handle, int gi, bool bFlag = true){

    task_param tparam;
  

    tparam.ipc_url = get_ipc_url(gi);

  
    if (bFlag){
      nTaskId = gi;
    }

    std::string task_id_str = "test_task_id_" + std::to_string(nTaskId);
    tparam.task_id = task_id_str.c_str();

    nTaskId++;
  

    tparam.result_folder_little = "./res";         //目标快照抠图保存地址
  	tparam.result_folder = "./res";                //目标快照大图保存地址

  
    const int result_code = add_task(handle, tparam);
    if (result_code != 0)
      printf("[Error]: ");
    printf("--- task_id: %s result code: %d\n", tparam.task_id, result_code);
  

    return task_id_str;
  }
  

  void show_result(ObjectData obj)
  {

    printf("%s  %d \n", obj.task_id.c_str(), obj.task_frame_count);

    printf("%d %d (%d,%d,%d,%d) %lf \n", obj.index, obj.object_id, obj.left, obj.top, obj.right, obj.bottom, obj.confidence);
  

    cv::Mat image = cv::imread(obj.ori_pic_path);
  
    cv::Rect rc(obj.left, obj.top,obj.right - obj.left, obj.bottom - obj.top);
    const cv::Scalar color(0, 0, 255);
    cv::rectangle(image, rc, color);
  

    if (obj.index == 0)
    {// person

      for (size_t i = 0; i < HP_ATTRI_SIZE; i++)
      {
        auto hp_cls = obj.hp_cls[i];
        printf("(%d, %f)  ", hp_cls.res_index, hp_cls.res_prob);
      }
      printf("\n");
      for (size_t i = 0; i < HUMANREID_FEATURE_SIZE; i++)
      {
        auto hp_cls = obj.hp_feature[i];
        printf("%f ", obj.hp_feature[i]);
      }
      printf("\n");

    } 
    else if (obj.index >= 4 && obj.index <= 8)
    {// vehicle

      printf("vehicle color:%d  prob:%f \n", obj.vehicle_color_index, obj.vehicle_color_prob);
      for (size_t i = 0; i < obj.vec_vpd_cls.size(); i++)
      {
        auto info = obj.vec_vpd_cls[i];
        printf("%d %f (%d, %d, %d, %d) \n", info.index, info.confidence, info.left_, info.top_, info.width_, info.height_);
  
        cv::Rect rc(info.left_, info.top_, info.width_, info.height_);
        const cv::Scalar color(0, 255, 255);
        cv::rectangle(image, rc, color);
      }
  
      auto& plate_info = obj.vehicle_plate;
      printf("plate : (%d, %d, %d, %d)  %f \n", plate_info.left_, plate_info.top_, plate_info.width_, plate_info.height_, plate_info.detect_score );
      cv::Rect rc(plate_info.left_, plate_info.top_, plate_info.width_, plate_info.height_);
      const cv::Scalar color(0, 255, 0);
      cv::rectangle(image, rc, color);
      for (size_t i = 0; i < 8; i++)
      {
        printf("%s:%f  ", plate_info.recg[i].character, plate_info.recg[i].maxprob);
      }
      printf("\n");
      printf("num_score: %f \n", plate_info.num_score);
      printf("type: %d \n", plate_info.type);
      printf("state:%d  %f\n", plate_info.state, plate_info.state_score);

    }
    

    string result_path = "./res/result/" +  obj.task_id + "_" + to_string(obj.object_id) + ".jpg";
    cv::imwrite(result_path, image);
    image.release();

  }

  
  void test_gpu(int gpuID){
      tsl_aiplatform_param vptParam;
      vptParam.gpuid = gpuID;

      vptParam.models_dir = ".";
  
      vptParam.log_days = 1;
      vptParam.log_level = AI_LOG_LEVEL_TRACE;
      // vptParam.log_level = AI_LOG_LEVEL_DEBUG;
      vptParam.log_mem = 64 * 1024 * 1024; // 64MB.
      vptParam.log_path = "logs/main.log";
  

      vptParam.result_cbk = show_result;
  

      void *handle;
      int flag = tsl_aiplatform_init(&handle, vptParam);
      if (0 != flag) {
          printf("Init Failed! Error Code: %d\n", flag);
          return;
      } else {
          printf("Init Success\n");
      }
  

      CreateResultFolder("./res/result", "");
  

    // createTask(handle, algor_vec2, 19, false);
    // createTask(handle, algor_vec2, 20, false);
    // createTask(handle, algor_vec2, 21, false);
    // createTask(handle, algor_vec2, 22, false);
    // createTask(handle, algor_vec2, 23, false);
    // createTask(handle, algor_vec2, 24, false);
    // createTask(handle, algor_vec2, 30, false);
    // createTask(handle, algor_vec2, 31, false);

    createTask(handle, 4, false);

  
      while (getchar() != 'q');
  
      tsl_aiplatform_release(&handle);
  }
  

  int main(int argc, char *argv[]) {
      printf("new test\n");
  
      // if (argc < 4) {
      //   fprintf(stderr, "./xxx 0 2 10 1 ## [start_ai_id, end_ai_id) repeat_num gpu_id\n");
      //   return -1;
      // }
  
      // //! load params.
      // int start_id = atoi(argv[1]);
      // int end_id = atoi(argv[2]);
      // int repeat_num = atoi(argv[3]);
      // int gpuID = atoi(argv[4]);
  
      test_gpu(0);
      // test_gpu(1);
      // test_gpu(2);
      // test_gpu(3);
  

  
      printf("Done.\n"); 
  
      return 0;
  }