NiagaraDataInterfaceSTAudio.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "NiagaraDataInterface.h"
#include "Main/ST_AudioVisSettings.h"
#include "RenderGraphFwd.h"
#include "RenderGraphResources.h"
#include "Data/ST_AudioIOLib.h"
#include "NiagaraDataInterfaceSTAudio.generated.h"
 
struct FNiagaraDataInterfaceGeneratedFunction;
struct FNiagaraFunctionSignature;

UCLASS(EditInlineNew, Category = "ShaderTech|AudioVisualisation", meta = (DisplayName = "ST Audio Query"))
class ST_AUDIOVISUALISATION_API UNiagaraDataInterfaceSTAudio : public UNiagaraDataInterface
{
    GENERATED_UCLASS_BODY()
 
    
    BEGIN_SHADER_PARAMETER_STRUCT(FShaderParameters, )
        SHADER_PARAMETER(uint32, IODataSize)
 
        
        SHADER_PARAMETER_RDG_BUFFER_SRV(StructuredBuffer<uint>, IOData)
 
        
        SHADER_PARAMETER(uint32, frequencyBandsLength)
 
        
        SHADER_PARAMETER_RDG_BUFFER_SRV(StructuredBuffer<float>, frequencyBands)
 
        
        SHADER_PARAMETER(uint32, amplitudesLength)
 
        
        SHADER_PARAMETER_RDG_BUFFER_SRV(StructuredBuffer<float>, amplitudes)
 
        
        SHADER_PARAMETER(uint32, pitchesLength)
 
        
        SHADER_PARAMETER_RDG_BUFFER_SRV(StructuredBuffer<float>, pitches)
 
        
        SHADER_PARAMETER(uint32, beatsLength)
 
        
        SHADER_PARAMETER_RDG_BUFFER_SRV(StructuredBuffer<float>, beats)
    END_SHADER_PARAMETER_STRUCT()
 
private:
    FST_AudioIOData_GameThread ST_AudioIOData_GT;
 
public:
    void SetSTAudioBuffers(FST_AudioIOData_GameThread InST_AudioIOData_GT)
    {
        ST_AudioIOData_GT = InST_AudioIOData_GT;
    }
 
    //~ Begin UObject Interface
    virtual void PostInitProperties() override;
    //~ End UObject Interface
 
    //~ Begin UNiagaraDataInterface Interface
    virtual void GetFunctions(TArray<FNiagaraFunctionSignature>& OutFunctions) override;

    virtual bool CanExecuteOnTarget(ENiagaraSimTarget Target) const override { return Target == ENiagaraSimTarget::GPUComputeSim; }

    virtual void BuildShaderParameters(FNiagaraShaderParametersBuilder& ShaderParametersBuilder) const override;

    virtual void SetShaderParameters(const FNiagaraDataInterfaceSetShaderParametersContext& Context) const override;
 
#if WITH_EDITORONLY_DATA
    virtual bool AppendCompileHash(FNiagaraCompileHashVisitor* InVisitor) const override;

    virtual bool GetFunctionHLSL(const FNiagaraDataInterfaceGPUParamInfo& ParamInfo, const FNiagaraDataInterfaceGeneratedFunction& FunctionInfo, int FunctionInstanceIndex, FString& OutHLSL) override;

    virtual void GetParameterDefinitionHLSL(const FNiagaraDataInterfaceGPUParamInfo& ParamInfo, FString& OutHLSL) override;
#endif

    virtual bool HasPreSimulateTick() const override { return true; }

    virtual bool PerInstanceTick(void* PerInstanceData, FNiagaraSystemInstance* SystemInstance, float DeltaSeconds) override;

    virtual int32 PerInstanceDataSize() const override;

    virtual bool InitPerInstanceData(void* PerInstanceData, FNiagaraSystemInstance* SystemInstance) override;

    virtual void DestroyPerInstanceData(void* PerInstanceData, FNiagaraSystemInstance* SystemInstance) override;
    //~ End UNiagaraDataInterface Interface
};

struct FNDISTAudioProxy : public FNiagaraDataInterfaceProxy
{
    virtual void ConsumePerInstanceDataFromGameThread(void* PerInstanceData, const FNiagaraSystemInstanceID& Instance) override {}

    virtual int32 PerInstanceDataPassedToRenderThreadSize() const override { return 0; }

    TMap<FNiagaraSystemInstanceID, FST_AudioIOData_RenderThread> SystemInstancesToProxyData_RT;
};