NNE.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Algo/RemoveIf.h"
#include "Containers/ArrayView.h"
#include "Containers/UnrealString.h"
#include "NNERuntime.h"
#include "NNEStatus.h"
#include "UObject/WeakInterfacePtr.h"
 
NNE_API DECLARE_LOG_CATEGORY_EXTERN(LogNNE, Log, All);
 
namespace UE::NNE
{
    using ERegisterRuntimeStatus = EResultStatus;
    using EUnregisterRuntimeStatus = EResultStatus;
 
    NNE_API ERegisterRuntimeStatus RegisterRuntime(TWeakInterfacePtr<INNERuntime> Runtime);
 
    NNE_API EUnregisterRuntimeStatus UnregisterRuntime(TWeakInterfacePtr<INNERuntime> Runtime);
    
    NNE_API TArray<FString> GetAllRuntimeNames();
 
    NNE_API TWeakInterfacePtr<INNERuntime> GetRuntime(const FString& Name);
 
    template<class T>
    TWeakInterfacePtr<T> GetRuntime(const FString& Name)
    {
        TWeakInterfacePtr<INNERuntime> Runtime = GetRuntime(Name);
 
        T* RuntimePtr = Cast<T>(Runtime.Get());
 
        return TWeakInterfacePtr<T>(RuntimePtr);
    }
 
    template<class T>
    TArray<FString> GetAllRuntimeNames()
    {
        TArray<FString> Result = GetAllRuntimeNames();
        Result.SetNum(Algo::RemoveIf(Result, [] (const FString &RuntimeName)
        {
            return !GetRuntime<T>(RuntimeName).IsValid();
        }));
 
        return Result;
    }
}