src/PicAnalysis.cpp

#include "PicAnalysis.h"
#include "./utils/logger.hpp"
PicAnalysis::PicAnalysis(/* args */)
{
    aclInit(nullptr);
}
PicAnalysis::~PicAnalysis()
{
    release();
    aclFinalize();
}
int PicAnalysis::init(int dev_id) {
    int ret = SY_FAILED;
    ret = m_vehicle_analysis.init(dev_id, 16);
    if(0 != ret){
        return -1;
    }
    head_tail_param ht_param;
    ht_param.devId = dev_id;
    ht_param.max_batch = 16;
    ret = m_head_tail_algorithm.init(ht_param);
    if(0 != ret){
        return -1;
    }
    ret = m_clothes_algorithm.init(dev_id);
    if(0 != ret){
        return -1;
    }
    ret = m_human_algorithm.init(dev_id);
    if(0 != ret){
        return -1;
    }
    ret = m_human_car_algorithm.init(dev_id);
    if(0 != ret){
        return -1;
    }
    ret = m_motor_rainshed_algorithm.init(dev_id);
    if(0 != ret){
        return -1;
    }
    ret = m_motor_phone_algorithm.init(dev_id);
    if(0 != ret){
        return -1;
    }
    ret = m_road_seg_algorithm.init(dev_id);
    if(0 != ret){
        return -1;
    }
    ACL_CALL(aclrtCreateContext(&m_ctx, 0), ACL_SUCCESS, SY_FAILED);
	ACL_CALL(aclrtSetCurrentContext(m_ctx), ACL_SUCCESS, SY_FAILED);
	ACL_CALL(aclrtCreateStream(&stream), ACL_SUCCESS, SY_FAILED);
    m_dvpp = new DvppProcess();
	m_dvpp->InitResource(stream);
    return 0;
}
int PicAnalysis::analysis_sync(vector<string> vec_file_path){
    ACL_CALL(aclrtSetCurrentContext(m_ctx), ACL_SUCCESS, SY_FAILED);
    const int batch_size = vec_file_path.size();
    vector<sy_img> vec_img;
    int ret = SY_FAILED;
    // ImageData 内部是智能指针，分析未处理完成前不得释放
    ImageData dvpp_data[batch_size];
    for (size_t i = 0; i < vec_file_path.size(); i++)
    {
        string file_path = vec_file_path[i];
        ImageData src;
        ret = Utils::ReadImageFile(src, file_path); //将二进制图像读入内存，并读取宽高信息
        if(ret != SY_SUCCESS){
            LOG_ERROR("ReadImageFile failed!");
            return -1;
        }
        ret = m_dvpp->CvtJpegToYuv420sp(dvpp_data[i], src); //解码
        if(ret != SY_SUCCESS){
            LOG_ERROR("CvtJpegToYuv420sp failed!");
            return -1;
        }
        
        sy_img img;
        img.w_ = dvpp_data[i].width;
        img.h_ = dvpp_data[i].height;
        img.data_ = dvpp_data[i].data.get();
        vec_img.push_back(img);
    }
    
    m_vehicle_analysis.detect(vec_img);
    vector<HeadTailResult> head_tail_result;
    ret = m_head_tail_algorithm.detect(vec_img, head_tail_result);
    if (0 != ret) {
        LOG_ERROR("m_head_tail_algorithm failed!");
        head_tail_result.clear();
    }
    m_clothes_algorithm.detect(vec_img);
    m_human_algorithm.detect(vec_img);
    m_human_car_algorithm.detect(vec_img);
    m_motor_rainshed_algorithm.detect(vec_img);
    m_motor_phone_algorithm.detect(vec_img);
    m_road_seg_algorithm.detect(vec_img);
    
    LOG_INFO("analysis_sync finished!");
    return 0;
}
int PicAnalysis::release() {
    ACL_CALL(aclrtSetCurrentContext(m_ctx), ACL_SUCCESS, SY_FAILED);
    delete m_dvpp;
    m_dvpp = nullptr;
    if (stream != nullptr) {
        int ret = aclrtDestroyStream(stream);
        if (ret != ACL_SUCCESS) {
            LOG_ERROR("destroy stream failed");
        }
        stream = nullptr;
    }
    aclrtDestroyContext(m_ctx);
    return 0;
}