ST_VJ_ComputeFunctionLibrary.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Kismet/BlueprintFunctionLibrary.h"
#include "RenderGraphUtils.h"
#include "RenderGraphBuilder.h"
#include "GlobalShader.h"
#include "RenderGraphResources.h"
#include "RenderGraphFwd.h"
#include "RenderGraphDefinitions.h"
#include "ST_VJ_ComputeFunctionLibrary.generated.h"

UCLASS()
class ST_AUDIOVISUALISATION_API UST_VJ_ComputeFunctionLibrary : public UBlueprintFunctionLibrary
{
    GENERATED_BODY()
 
public:
    static FRDGBufferRef CreateStructuredBuffer(
        FRDGBuilder& GraphBuilder,
        const TCHAR* Name,
        uint32 BytesPerElement,
        uint32 NumElements,
        bool bReadOnly = false);

    static void RegisterSRV(
        FRDGBuilder& GraphBuilder,
        TRefCountPtr<FRDGPooledBuffer> buffer,
        FString name,
        FRDGBufferRef& outRDGRef,
        FRDGBufferSRVRef& outSRVRef,
        ERDGBufferFlags flags = ERDGBufferFlags::None);

    static void RegisterUAV(
        FRDGBuilder& GraphBuilder,
        TRefCountPtr<FRDGPooledBuffer> buffer,
        FString name,
        FRDGBufferRef& outRDGRef,
        FRDGBufferUAVRef& outUAVRef,
        ERDGBufferFlags flags = ERDGBufferFlags::None);
};