ST_AudioReactiveBoidsLib.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "RHIGPUReadback.h"
#include "RenderGraphDefinitions.h"
#include "Main/ST_AudioProcessingSettings.h"
#include "ST_AudioReactiveBoidsLib.generated.h"

USTRUCT(BlueprintType)
struct ST_AUDIOVISUALISATION_API FST_VJ_BoundsConstantParameters
{
    GENERATED_BODY()
 
public:
    UPROPERTY(EditAnywhere, BlueprintReadWrite, AdvancedDisplay, Category = "ShaderTech|Bounds|Constants")
    bool bSphere = true;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Bounds|Constants", meta = (EditCondition = "bSphere"))
    float Radius = 100;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Bounds|Constants", meta = (EditCondition = "!bSphere"))
    FVector BoundsExtent = FVector(100, 100, 100);
};

USTRUCT(BlueprintType)
struct ST_AUDIOVISUALISATION_API FST_VJ_BoidItem
{
    GENERATED_BODY()
 
public:
    UPROPERTY(BlueprintReadWrite, EditAnywhere, Category = "ShaderTech|Boids")
    FVector3f pos = FVector3f(0.0f, 0.0f, 0.0f);

    UPROPERTY(BlueprintReadWrite, EditAnywhere, Category = "ShaderTech|Boids")
    FVector3f vel = FVector3f(0.0f, 0.0f, 0.0f);

    UPROPERTY(BlueprintReadWrite, EditAnywhere, Category = "ShaderTech|Boids")
    float reactiveNoise0 = 0.0f;

    UPROPERTY(BlueprintReadWrite, EditAnywhere, Category = "ShaderTech|Boids")
    float reactiveNoise1 = 0.0f;

    UPROPERTY(BlueprintReadWrite, EditAnywhere, Category = "ShaderTech|Boids")
    FVector4f col = FVector4f(0.0f, 0.0f, 0.0f, 0.0f);
};

struct FST_VJ_BoidsRDGStateData
{
    TArray<FRDGPassRef> InitPass;

    TArray<FRDGPassRef> ExecutePass;

    FGPUFenceRHIRef InitFence;

    FST_VJ_BoidsRDGStateData()
    {
        InitPass.SetNum(0);
        ExecutePass.SetNum(0);
        InitFence = nullptr;
    }

    FST_VJ_BoidsRDGStateData(int32 InitPassSize, int32 ExecutePassSize)
    {
        InitPass.SetNum(InitPassSize);
        ExecutePass.SetNum(ExecutePassSize);
        InitFence = nullptr;
    };

    void ClearPasses()
    {
        for (int32 Index = 0; Index < InitPass.Num(); ++Index)
        {
            InitPass[Index] = nullptr;
        }
        for (int32 Index = 0; Index < ExecutePass.Num(); ++Index)
        {
            ExecutePass[Index] = nullptr;
        }
    }

    void Dispose()
    {
        if (InitFence->IsValid())
        {
            InitFence->Clear();
            InitFence = nullptr;
        }
    }

    bool WaitingOnInitFence()
    {
        return (!InitFence.IsValid() || !InitFence->Poll());
    }
};

USTRUCT(BlueprintType)
struct ST_AUDIOVISUALISATION_API FST_VJ_BoidConstantParameters
{
    GENERATED_BODY()
 
public:
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Constants")
    int32 numBoids = 32;
};

USTRUCT(BlueprintType)
struct ST_AUDIOVISUALISATION_API FST_VJ_BoidDynamicParameters
{
    GENERATED_BODY()
 
public:
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boid|Variables")
    float simulationSpeed = 1.0f;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    float maxSpeed = 2.0f;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    float minDistance = 0.15f;

    float minDistanceSq() const { return minDistance * minDistance; }

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    FVector2f cohesionFactor = FVector2f::One();

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    FVector2f separationFactor = FVector2f::One();

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    FVector2f alignmentFactor = FVector2f::One();

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    FVector2f perlinNoiseForce = FVector2f::One();

    float turnSpeed() const { return maxSpeed * 3.0f; }

    float minSpeed() const { return maxSpeed * 0.75f; }

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    float meshScale = 0.02f;
};

USTRUCT(BlueprintType)
struct ST_AUDIOVISUALISATION_API FST_VJ_ReactiveBoidDynamicParameters
{
    GENERATED_BODY()
 
public:
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    float audio_ReactiveNoise0_Multiplier = 1.0f;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    int32 audio_ReactiveNoise0_FrequencyIndex = 5;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    EAudioIOType audio_ReactiveNoise0_IO = EAudioIOType::None;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    bool baudio_ReactiveNoise0_Mono = true;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables", meta = (EditCondition = "!baudio_ReactiveNoise0_Mono"))
    int32 audio_ReactiveNoise0_ChannelIndex = 0;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    float audio_ReactiveNoise1_Multiplier = 1.0f;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    int32 audio_ReactiveNoise1_FrequencyIndex = 5;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    EAudioIOType audio_ReactiveNoise1_IO = EAudioIOType::None;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    bool baudio_ReactiveNoise1_Mono = true;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables", meta = (EditCondition = "!baudio_ReactiveNoise1_Mono"))
    int32 audio_ReactiveNoise1_ChannelIndex = 0;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    int32 audio_ReactiveMaterial0_ChannelIndex = 0;

    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "ShaderTech|Boids|Variables")
    int32 audio_ReactiveMaterial1_ChannelIndex = 0;
};

USTRUCT(BlueprintType)
struct ST_AUDIOVISUALISATION_API FST_VJ_BoidCurrentParameters
{
    GENERATED_BODY()
 
public:
    FST_VJ_BoidConstantParameters ConstantParameters;

    FST_VJ_BoidDynamicParameters DynamicParameters;

    float worldScale = 1.0f;

    float boundsRadius;

    FMatrix44f transformMatrix;

    FMatrix44f inverseTransformMatrix;
};