/*
 * @Author: xiewenji 527774126@qq.com
 * @Date: 2025-09-03 10:33:48
 * @LastEditors: xiewenji 527774126@qq.com
 * @LastEditTime: 2025-09-14 20:28:10
 * @FilePath: /AI_SO_Test/AIEngineModule/src/AI_Factory.cpp
 * @Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 */
#include "AI_Factory.h"

#include "AIModel_Impl.h"

std::shared_ptr<AIModel_Base> AIModel_Base::GetInstance()
{
    std::shared_ptr<AIModel_Base> instance = std::make_shared<AIModel_Impl>();
    return instance;
}

AIFactory::AIFactory()
{
}

AIFactory::~AIFactory()
{
}
std::shared_ptr<AIFactory> AIFactory::GetInstance()
{
    static std::shared_ptr<AIFactory> instance = std::make_shared<AIFactory>();
    return instance;
}

int AIFactory::InitALLAIModle(GPU_Config gupconfig)
{
    if (gupconfig.GetNum() == 0)
    {
        printf("GPU Num Error \n");
        return 1;
    }

    if (!Defect)
    {
        Defect = AIModel_Base::GetInstance();
        AIModel_Base::AIModelRun_Config edge_config;
        edge_config.gpuconfig.copy(gupconfig);
        edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/Defect.engine";
        edge_config.strName = "CA_Det";
        edge_config.inputType = AIModel_Base::Input_HWC;
        edge_config.Stream_num = 2;
        Defect->Init(edge_config);
    }
    if (!Class)
    {
        Class = AIModel_Base::GetInstance();
        AIModel_Base::AIModelRun_Config jbl_config;
        jbl_config.gpuconfig.copy(gupconfig);
        jbl_config.strPath = "/home/aidlux/BOE/UseModel_ZB/Class_10.engine";
        jbl_config.inputType = AIModel_Base::Input_HWC;
        jbl_config.strName = "CA_Class";
        jbl_config.IsClass = true;
        Class->Init(jbl_config);
    }
    if (!Pin_Loc)
    {
        Pin_Loc = AIModel_Base::GetInstance();
        AIModel_Base::AIModelRun_Config edge_config;
        edge_config.gpuconfig.copy(gupconfig);
        edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/PinLoc.engine";
        edge_config.strName = "TA_Det";
        edge_config.inputType = AIModel_Base::Input_HWC;
        edge_config.Stream_num = 2;
        Pin_Loc->Init(edge_config);
    }
    if (!Align_Outer)
    {
        Align_Outer = AIModel_Base::GetInstance();
        AIModel_Base::AIModelRun_Config edge_config;
        edge_config.gpuconfig.copy(gupconfig);
        edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/AlignOuter.engine";
        edge_config.strName = "Align";
        edge_config.inputType = AIModel_Base::Input_HWC;
        Align_Outer->Init(edge_config);
    }
    if (!Align_Inner)
    {
        Align_Inner = AIModel_Base::GetInstance();
        AIModel_Base::AIModelRun_Config edge_config;
        edge_config.gpuconfig.copy(gupconfig);
        edge_config.strPath = "/home/aidlux/BOE/UseModel_ZB/AlignInner.engine";
        edge_config.strName = "Mark";
        edge_config.inputType = AIModel_Base::Input_HWC;
        Align_Inner->Init(edge_config);
    }
    m_bInitSucc = true;
    return 0;
}