VTrain/ai_matrix/framework/EngineBase.cpp

#include "EngineBase.h"

namespace ai_matrix
{
    //初始化engine
    void EngineBase::AssignInitArgs(const EngineInitArguments &initArgs)
    {
        deviceId_ = initArgs.deviceId;

        engineName_ = initArgs.engineName;
        engineId_ = initArgs.engineId;
    }

    // get the data from input queue then call Process function in the new thread
    void EngineBase::ProcessThread()
    {
        Process();
    }

    //设置engine输入队列
    void EngineBase::SetInputMap(std::string engineAddress, std::shared_ptr<MyQueue<std::shared_ptr<void>>> inputQueue)
    {
        inputQueMap_[engineAddress] = inputQueue;
    }
    //得到输入队列
    std::shared_ptr<MyQueue<std::shared_ptr<void>>> EngineBase::GetInputMap(std::string engineAddress)
    {
        if (inputQueMap_.find(engineAddress) == inputQueMap_.end())
        {
            return nullptr;
        }
        return inputQueMap_.at(engineAddress);
    }

    //设置engine输出队列
    void EngineBase::SetOutputMap(std::string engineAddress, std::shared_ptr<MyQueue<std::shared_ptr<void>>> outputQue)
    {
        outputQueMap_[engineAddress] = outputQue;
    }

    //启动engine
    APP_ERROR EngineBase::Run()
    {
        LogInfo << engineName_ << "[" << engineId_ << "] Run";
        isStop_ = false;
        processThr_ = std::thread(&EngineBase::ProcessThread, this);
        return APP_ERR_OK;
    }

    // 停止engine
    APP_ERROR EngineBase::Stop()
    {
        //停止线程
        isStop_ = true;

        //停止所有输入队列
        for (auto it = inputQueMap_.begin(); it != inputQueMap_.end(); it++)
        {
            it->second->stop();
        }

        //等待线程结束
        if (processThr_.joinable())
        {
            processThr_.join();
        }

        //其他清理
        return DeInit();
    }
}