ST_C_VisAudioReactiveBoids.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Main/Base/ST_VisComputeBase.h"
#include "Data/ST_AudioReactiveBoidsLib.h"
#include "Compute/Utils/ST_VJ_PingPongBuffer.h"
#include "GlobalShader.h"
#include "NiagaraComponent.h"
#include "DataDrivenShaderPlatformInfo.h"
#include "Data/ST_AudioIOLib.h"
#include "Materials/MaterialInstanceDynamic.h"
#include "ST_C_VisAudioReactiveBoids.generated.h"

UCLASS()
class ST_AUDIOVISUALISATION_API AST_C_VisAudioReactiveBoids : public AST_VisComputeBase
{
    GENERATED_BODY()
 
public:
    AST_C_VisAudioReactiveBoids();
 
protected:
    virtual void BeginPlay() override;

    virtual void BeginDestroy() override;
 
    // --- Start Managed Compute Shader Interface ---

    virtual void InitComputeShader_GameThread() override;

    virtual void InitComputeShader_RenderThread(FRHICommandListImmediate& RHICmdList) override;

    virtual void ExecuteComputeShader_GameThread(float DeltaTime) override;

    virtual void ExecuteComputeShader_RenderThread(FRHICommandListImmediate& RHICmdList) override;

    virtual void DisposeComputeShader_GameThread() override;

    virtual void DisposeComputeShader_RenderThread(FRHICommandListImmediate& RHICmdList) override;
 
    // --- End Managed Compute Shader Interface ---
 
protected:
    virtual bool SetConstantParameters() override;

    virtual bool SetDynamicParameters() override;

    virtual FString GetOwnerName() const override;
 
protected:
    virtual void OnAudioIOStatusChanged_Implementation(bool bIsActive, FIntPoint IOId) override;

    virtual void OnFrequencyBandsReady_Implementation(const TArray<float>& FrequencyBands, int32 NumFrequencyBands, FIntPoint IOId) override;

    virtual void OnAmplitudesReady_Implementation(const TArray<float>& Amplitudes, FIntPoint IOId) override;

    virtual void OnPitchesReady_Implementation(const TArray<float>& Pitches, FIntPoint IOId) override;

    virtual void OnBeatDetected_Implementation(const TArray<float>& Beats, const TArray<bool>& BeatDetectedForChannel, FIntPoint IOId) override;
 
private:
    TArray<class UST_AudioIOBase*> RegisteredIOs;
 
protected:
    UPROPERTY(Transient, VisibleAnywhere, BlueprintReadOnly, Category = "ShaderTech|Boids")
    FST_VJ_BoidCurrentParameters BoidCurrentParameters;

    UPROPERTY(Transient, VisibleAnywhere, BlueprintReadOnly, Category = "ShaderTech|Boids")
    FST_VJ_ReactiveBoidDynamicParameters ReactiveBoidDynamicParameters;

    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "ShaderTech|Boids")
    TArray<FST_VJ_BoidItem> BoidsArray;

    UPROPERTY(Transient, VisibleAnywhere, BlueprintReadOnly, Category = "Niagara")
    UNiagaraComponent* Niagara = nullptr;

    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Transient, Category = "ShaderTech|Boids")
    UMaterialInstanceDynamic* MatiD = nullptr;

    FST_VJ_PingPongBuffer BoidsPingPongBuffer;

    FST_VJ_BoidsRDGStateData BoidsRDGStateData;

    FST_AudioIOData_GameThread ST_AudioIOData_GT;

    FST_AudioIOBuffers_RenderThread ST_AudioIOBuffers_RT;

    virtual TSoftObjectPtr<UNiagaraSystem> GetNiagaraSystem() const;
 
private:
    FMatrix BoundsMatrix;
};
 
inline FString AST_C_VisAudioReactiveBoids::GetOwnerName() const
{
    return "VisAudioReactiveBoids";
}