VTrain/engine/SaveMoveImageEngine/SaveMoveImageEngine.cpp

#include "SaveMoveImageEngine.h"


using namespace ai_matrix;

SaveMoveImageEngine::SaveMoveImageEngine() {}

SaveMoveImageEngine::~SaveMoveImageEngine() {}

APP_ERROR SaveMoveImageEngine::Init()
{

    strPort0_ = engineName_ + "_" + std::to_string(engineId_) + "_0";

    this->baseConfig_ = Config::getins()->getBaseConfig();
    vecCompressionParams_.push_back(cv::IMWRITE_JPEG_QUALITY);  //选择jpeg
    vecCompressionParams_.push_back(this->iPicQuality_);        //图片质量

    LogInfo << "SaveMoveImageEngine Init ok";
    return APP_ERR_OK;
}

APP_ERROR SaveMoveImageEngine::DeInit()
{
    LogInfo << "SaveMoveImageEngine DeInit ok";
    return APP_ERR_OK;
}

APP_ERROR SaveMoveImageEngine::Process()
{


    int iRet = APP_ERR_OK;
    while (!isStop_)
    {
        std::shared_ptr<void> pvoidd = nullptr;
        inputQueMap_[strPort0_]->pop(pvoidd);
        if (nullptr == pvoidd)
        {
            usleep(1000);
            continue;
        }
        
        std::shared_ptr<SaveImgData> pSaveImgData = std::static_pointer_cast<SaveImgData>(pvoidd);

        //图片绝对路径
        if (pSaveImgData->strFilePath.back() != '/')
        {
            pSaveImgData->strFilePath += "/";
        }
        std::string strImgFilePath = pSaveImgData->strFilePath + pSaveImgData->strFileName;

        if (!FileUtil::getins()->createDirPath(pSaveImgData->strFilePath))
        {
            LogError << " 创建图片存储目录失败:" << strImgFilePath;
            continue;
        }

        if (pSaveImgData->frameData.pData != nullptr && pSaveImgData->frameData.iSize != 0)
        {
            cv::Mat image(pSaveImgData->frameData.iHeight, pSaveImgData->frameData.iWidth, CV_8UC3, static_cast<uint8_t *>(pSaveImgData->frameData.pData.get()));   //RGB

//            cv::cvtColor(image, image, cv::COLOR_RGB2BGR);

//            if (this->baseConfig_.bTestModel)
//            {
//                std::stringstream ss;
//                std::string strTitle;
//                ss << pSaveImgData->strTrainStage << "   "
//                << pSaveImgData->strDirection << "   "
//                << pSaveImgData->strTrainStatus;
//                ss >> strTitle;
//
//                cv::putText(image,
//                            strTitle,
//                            cv::Point(15, 45),
//                            cv::FONT_HERSHEY_SIMPLEX, 1, cv::Scalar(0, 255, 0, 255), 2);
//                for (int i = 0; i < pSaveImgData->vecCoordinate.size(); ++i)
//                {
//                    float centerX = pSaveImgData->vecCoordinate.at(i).fLTX + (pSaveImgData->vecCoordinate.at(i).fRBX - pSaveImgData->vecCoordinate.at(i).fLTX)/2;
//                    float centerY = pSaveImgData->vecCoordinate.at(i).fLTY + (pSaveImgData->vecCoordinate.at(i).fRBY - pSaveImgData->vecCoordinate.at(i).fLTY)/2;
//
//                    cv::rectangle(image,
//                                  cv::Point(pSaveImgData->vecCoordinate.at(i).fLTX, pSaveImgData->vecCoordinate.at(i).fLTY),
//                                  cv::Point(pSaveImgData->vecCoordinate.at(i).fRBX, pSaveImgData->vecCoordinate.at(i).fRBY),
//                                  cv::Scalar(0, 255, 0, 200), 2);
//                    cv::line(image,
//                             cv::Point(centerX, centerY-10), cv::Point(centerX, centerY+10),
//                             cv::Scalar(0, 255, 0, 100), 3);
//                    cv::putText(image,
//                                pSaveImgData->vecCoordinate.at(i).strClassName,
//                                cv::Point(pSaveImgData->vecCoordinate.at(i).fLTX+10, pSaveImgData->vecCoordinate.at(i).fLTY+10),
//                                cv::FONT_HERSHEY_SIMPLEX, 1, cv::Scalar(0, 255, 0, 180), 2);
//                }
//            }

            if (cv::imwrite(strImgFilePath, image, this->vecCompressionParams_))
            {
                std::shared_ptr<VToStep1Data> pToStep1Data = std::make_shared<VToStep1Data>();
                pToStep1Data->bIsEnd = pSaveImgData->bIsEnd;
                pToStep1Data->strImagePath = pSaveImgData->strFilePath;
                pToStep1Data->strImageName = pSaveImgData->strFileName;

                outputQueMap_[strPort0_]->push(std::static_pointer_cast<void>(pToStep1Data), true);
            }
            else
            {
                LogError << "图片存储失败：" << strImgFilePath;
            }
        }
    }
    return APP_ERR_OK;
}