DataflowContextAssetStore.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "UObject/ObjectPtr.h"
#include "Containers/StringFwd.h"
#include "UObject/GCObject.h"
 
class UObject;
 
namespace UE::Dataflow
{
    struct IContextAssetStoreInterface
    {
    public:
        template <typename T>
        inline T* AddAssetTyped(const FString& AssetPath)
        {
            return Cast<T>(AddAsset(AssetPath, T::StaticClass()));
        }
 
        virtual UObject* AddAsset(const FString& PersistentAssetPath, const UClass* AssetClass) = 0;
 
        virtual UObject* CommitAsset(const FString& TransientAssetPath) = 0;
 
        virtual void ClearAssets() = 0;
 
    };
 
    struct FContextAssetStore: public FGCObject
    {
    public:
        DATAFLOWCORE_API UObject* AddAsset(const FString& PersistentAssetPath, const UClass* AssetClass);
 
        DATAFLOWCORE_API UObject* CommitAsset(const FString& TransientAssetPath);
 
        DATAFLOWCORE_API UObject* CommitAsset(const UObject* TransientAsset);
        
        DATAFLOWCORE_API void ClearAssets();
 
        //~ Begin FGCObject interface
        DATAFLOWCORE_API virtual void AddReferencedObjects(FReferenceCollector& Collector) override;
        virtual FString GetReferencerName() const override { return TEXT("FContextAssetStore"); }
        //~ End FGCObject interface
 
    private:
        struct FAssetData
        {
            FString             PersistentPath;
            FString             TransientPath;
            TObjectPtr<UObject> TransientAsset = nullptr;
        };
 
        const FAssetData* FindByTransientPath(const FString& TransientAssetPath) const;
        const FAssetData* FindByTransientAsset(const UObject* TransientAsset) const;
 
        UObject* CommitAsset(const FAssetData& AssetData);
 
        TArray<FAssetData> AssetData;
    };
 
}