NNEModelData.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 "Memory/SharedBuffer.h"
#include "Misc/Guid.h"
#include "Templates/SharedPointer.h"
 
#include "NNEModelData.generated.h"
 
namespace UE::NNE
{
    class FSharedModelData
    {
 
    public:
        NNE_API FSharedModelData(FSharedBuffer InData, uint32 InMemoryAlignment);
 
        NNE_API FSharedModelData();
 
        NNE_API TConstArrayView64<uint8> GetView() const;
 
        NNE_API uint32 GetMemoryAlignment() const;
 
    private:
 
        FSharedBuffer Data;
        
        uint32 MemoryAlignment;
    };
}
 
UCLASS(BlueprintType, Category = "NNE", MinimalAPI)
class UNNEModelData : public UObject
{
    GENERATED_BODY()
 
public:
 
    // UObject interface
    NNE_API virtual void GetAssetRegistryTags(FAssetRegistryTagsContext Context) const override;
    NNE_API virtual void Serialize(FArchive& Ar) override;
    // End of UObject interface
 
    NNE_API void Init(const FString& Type, TConstArrayView64<uint8> Buffer, const TMap<FString, TConstArrayView64<uint8>>& AdditionalBuffers = TMap<FString, TConstArrayView64<uint8>>());
 
    NNE_API TArrayView<const FString> GetTargetRuntimes() const;
 
    NNE_API void SetTargetRuntimes(TArrayView<const FString> RuntimeNames);
 
    NNE_API FString GetFileType() const;
 
    NNE_API TConstArrayView64<uint8> GetFileData() const;
 
    NNE_API TConstArrayView64<uint8> GetAdditionalFileData(const FString& Key) const;
 
    NNE_API void ClearFileDataAndFileType();
 
    NNE_API FGuid GetFileId() const;
 
    NNE_API TSharedPtr<UE::NNE::FSharedModelData> GetModelData(const FString& RuntimeName);
 
    NNE_API void ClearModelData();
 
private:
    TArray<FString> TargetRuntimes;
 
    FString FileType;
 
    TArray64<uint8> FileData;
 
    TMap<FString, TArray64<uint8>> AdditionalFileData;
 
    FGuid FileId;
 
    TMap<FString, TSharedPtr<UE::NNE::FSharedModelData>> ModelData;
};