NNERuntime.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Containers/ContainersFwd.h"
#include "Containers/UnrealString.h"
#include "NNEStatus.h"
#include "UObject/Interface.h"
 
#include "NNERuntime.generated.h"
 
class ITargetPlatform;
 
namespace UE::NNE
{
    class FSharedModelData;
}
 
UINTERFACE(MinimalAPI)
class UNNERuntime : public UInterface
{
    GENERATED_BODY()
};
 
class INNERuntime
{
    GENERATED_BODY()
 
public:
 
    enum class ECanCreateModelDataStatus : uint8
    {
        Ok = 0,
        Fail = 1,
        FailFileIdNotSupported = 2
    };
 
    virtual FString GetRuntimeName() const = 0;
 
    virtual ECanCreateModelDataStatus CanCreateModelData(const FString& FileType, TConstArrayView64<uint8> FileData, const TMap<FString, TConstArrayView64<uint8>>& AdditionalFileData, const FGuid& FileId, const ITargetPlatform* TargetPlatform) const = 0;
 
    virtual TSharedPtr<UE::NNE::FSharedModelData> CreateModelData(const FString& FileType, TConstArrayView64<uint8> FileData, const TMap<FString, TConstArrayView64<uint8>>& AdditionalFileData, const FGuid& FileId, const ITargetPlatform* TargetPlatform) = 0;
 
    virtual FString GetModelDataIdentifier(const FString& FileType, TConstArrayView64<uint8> FileData, const TMap<FString, TConstArrayView64<uint8>>& AdditionalFileData, const FGuid& FileId, const ITargetPlatform* TargetPlatform) const = 0;
};