Hu Chunming / village_ascend_arm

  #include "./ai_engine_module/include.h"
  #include "./ai_engine_module/VehicleAnalysis.h"
  #include "./ai_engine_module/VehicleHeadTail.h"

  #include "./ai_engine_module/VidClothes.h"

  #include "./ai_engine_module/HumanAnalysis.h"
  #include "./ai_engine_module/HumanCarAnalysis.h"

  #include "./ai_engine_module/MotorRainshedAnalysis.h"
  #include "./ai_engine_module/MotorPhoneAnalysis.h"

  #include "./ai_engine_module/RoadSegAnalysis.h"

  #include "./utils/CropUtil.h"
  

  #include "village_inc.h"
  

  using namespace std;
  
  
  class PicAnalysis
  {
  public:
      PicAnalysis(/* args */);
      ~PicAnalysis();
  

      int init(VillageParam param);

  
      // todo 接口定义是二进制流的，这个地方有待修改

      vector<AnalysisResult> analysis_file(vector<string> file_path);

      vector<AnalysisResult> analysis_img(vector<sy_img> vec_img);

  
  private:

  
      int release();
  

      int human_analysis(vector<sy_img> vec_img);
  
      int check_motor_retrograde_motion(vector<sy_img> vec_img);
  

      vector<AnalysisResult> va_result2AnalysisResult(va_result* result, int batchsize);
  

  private:

      aclrtContext m_ctx{nullptr};
      aclrtStream stream{nullptr};
      DvppProcess* m_dvpp{nullptr};
  
      VehicleAnalysis m_vehicle_analysis;
      VehicleHeadTail m_head_tail_algorithm;

      VidClothes m_clothes_algorithm;

      HumanAnalysis m_human_algorithm;
      HumanCarAnalysis m_human_car_algorithm;

      MotorRainshedAnalysis m_motor_rainshed_algorithm;
      MotorPhoneAnalysis m_motor_phone_algorithm;

      RoadSegAnalysis m_road_seg_algorithm;

  
      CropUtil m_crop_util;

  };