MaterialCacheTagProvider.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "MaterialCache/IMaterialCacheTagProvider.h"
#include "RenderGraphBuilder.h"
 
struct FMaterialCacheTagLayout;
namespace UE::HLSL { struct FMaterialCacheTagEntry; }
 
class FMaterialCacheTagProvider : public IMaterialCacheTagProvider
{
public:
    uint32 AllocatePrimitiveTagOffset();
    
    void FreePrimitiveTagOffset(uint32 TagOffset);
    
    void SetTagEntry(uint32 TagOffset, const FGuid& TagGuid, const UE::HLSL::FMaterialCacheTagEntry& Entry);
    
    void Update(FRDGBuilder& GraphBuilder);
 
    void CallPendingCallbacks();
    
public: 
    virtual FMaterialCacheTagBindingData GetBindingData(const FGuid& Guid) override;
    virtual FMaterialCacheTagUniformData GetUniformData(const FGuid& Guid) override;
    virtual void Register(FSceneInterface* Scene, FPrimitiveComponentId PrimitiveComponentId, const FMaterialCacheTagLayout& TagLayout, IAllocatedVirtualTexture* VirtualTexture) override;
    virtual void Unregister(FSceneInterface* Scene, FPrimitiveComponentId PrimitiveComponentId, const FGuid& TagGuid, IAllocatedVirtualTexture* VirtualTexture) override;
    virtual void Flush(FSceneInterface* Scene, FPrimitiveComponentId PrimitiveComponentId, const FGuid& TagGuid) override;
    virtual IVirtualTexture* CreateProducer(FSceneInterface* Scene, FPrimitiveComponentId PrimitiveComponentId, const FMaterialCacheTagLayout& TagLayout, const FVTProducerDescription& ProducerDesc) override;
    virtual void AddTagSceneInvalidationCallback(const FGuid& TagGuid, FMaterialCacheTagProviderSceneInvalidationDelegate Delegate, void* Baton) override;
    virtual void RemoveTagSceneInvalidationCallbacks(void* Baton) override;
 
public: 
    static void Initialize();
    
    static void Shutdown();
    
    static FMaterialCacheTagProvider& Get();
 
private:
    void CreateDeviceBuffersOrResize(struct FMaterialCacheTagBucket& Bucket);
 
    void NotifyTagSceneInvalidation(const FGuid& TagGuid);
 
private:
    FCriticalSection Mutex;
    
    TMap<FGuid, struct FMaterialCacheTagBucket*> TagBuckets;
 
    TSet<FGuid> PendingTagSceneInvalidations;
 
    TMap<TTuple<FPrimitiveComponentId, FGuid>, IAllocatedVirtualTexture*> VirtualTextures;
 
    TArray<uint32> FreeTagIndices;
 
    uint32 TagOffsetAllocator = 0;
 
private:
    struct FTagSceneListener
    {
        FMaterialCacheTagProviderSceneInvalidationDelegate Delegate = nullptr;
        void* Baton = nullptr;
    };
    
    struct FTagBatonEntry
    {
        int32 CallbackIndex = UINT32_MAX;
        FGuid TagGuid;
    };
 
    TArray<FTagSceneListener>  TagSceneInvalidationListeners;
 
    TArray<int32> FreeTagSceneInvalidationCallbackIndices;
 
    TMultiMap<FGuid, int32>  TagCallbackIndices;
    TMultiMap<void*, FTagBatonEntry> BatonCallbackIndices;
};