ST_AudioIOLib.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "RenderGraphUtils.h"
#include "RenderGraphBuilder.h"
#include "GlobalShader.h"
#include "RenderGraphResources.h"
#include "RenderGraphFwd.h"
#include "RenderGraphDefinitions.h"
 
#define AudioIO_Items 12
#define AudioIO_Items_Active 0
#define AudioIO_Items_Order 1
#define AudioIO_Items_NumChannels 2
#define AudioIO_Items_NumFrequencyBands 3
#define AudioIO_Items_FrequencyBandHistorySize 4
#define AudioIO_Items_FrequencyBandsStartIndex 5
#define AudioIO_Items_AmpltitudeHistorySize 6
#define AudioIO_Items_AmplitudesStartIndex 7
#define AudioIO_Items_PitchHistorySize 8
#define AudioIO_Items_PitchesStartIndex 9
#define AudioIO_Items_BeatHistorySize 10
#define AudioIO_Items_BeatsStartIndex 11

struct FST_AudioIOData_GameThread
{
    uint32 IODataSize = 0;

    uint32 frequencyBandsLength = 0;

    uint32 amplitudesLength = 0;

    uint32 pitchesLength = 0;

    uint32 beatsLength = 0;

    TArray<uint32> IOData;

    TArray<float> frequencyBands;

    TArray<float> amplitudes;

    TArray<float> pitches;

    TArray<float> beats;

    ~FST_AudioIOData_GameThread()
    {
        ReleaseData();
    }

    void ReleaseData()
    {
        IODataSize = 0;
        frequencyBandsLength = 0;
        amplitudesLength = 0;
        pitchesLength = 0;
        beatsLength = 0;
        IOData.Empty();
        frequencyBands.Empty();
        amplitudes.Empty();
        pitches.Empty();
        beats.Empty();
    }
};

struct FST_AudioIOData_RenderThread
{
    uint32 IODataSize = 0;

    uint32 frequencyBandsLength = 0;

    uint32 amplitudesLength = 0;

    uint32 pitchesLength = 0;

    uint32 beatsLength = 0;

    TArray<uint32> IOData;

    TArray<float> frequencyBands;

    TArray<float> amplitudes;

    TArray<float> pitches;

    TArray<float> beats;

    ~FST_AudioIOData_RenderThread()
    {
        ReleaseData();
    }

    void UpdateData(FST_AudioIOData_GameThread& InstanceData_GT)
    {
        ReleaseData();
        IOData = InstanceData_GT.IOData;
        frequencyBands = InstanceData_GT.frequencyBands;
        amplitudes = InstanceData_GT.amplitudes;
        pitches = InstanceData_GT.pitches;
        beats = InstanceData_GT.beats;
        IODataSize = InstanceData_GT.IODataSize;
        frequencyBandsLength = InstanceData_GT.frequencyBandsLength;
        amplitudesLength = InstanceData_GT.amplitudesLength;
        pitchesLength = InstanceData_GT.pitchesLength;
        beatsLength = InstanceData_GT.beatsLength;
    }

    void ReleaseData()
    {
        IODataSize = 0;
        frequencyBandsLength = 0;
        amplitudesLength = 0;
        pitchesLength = 0;
        beatsLength = 0;
        IOData.Empty();
        frequencyBands.Empty();
        amplitudes.Empty();
        pitches.Empty();
        beats.Empty();
    }
};

struct FST_AudioIOBuffers_RenderThread
{
    uint32 IODataSize = 0;

    uint32 frequencyBandsLength = 0;

    uint32 amplitudesLength = 0;

    uint32 pitchesLength = 0;

    uint32 beatsLength = 0;

    TRefCountPtr<FRDGPooledBuffer> ReadIODataPooled = nullptr;

    FRDGBufferRef ReadIODataScopedRef = nullptr;

    FRDGBufferSRVRef ReadIODataScopedSRV = nullptr;

    TRefCountPtr<FRDGPooledBuffer> ReadFrequencyBandsPooled = nullptr;

    FRDGBufferRef ReadFrequencyBandsScopedRef = nullptr;

    FRDGBufferSRVRef ReadFrequencyBandsScopedSRV = nullptr;

    TRefCountPtr<FRDGPooledBuffer> ReadAmplitudesPooled = nullptr;

    FRDGBufferRef ReadAmplitudesScopedRef = nullptr;

    FRDGBufferSRVRef ReadAmplitudesScopedSRV = nullptr;

    TRefCountPtr<FRDGPooledBuffer> ReadPitchesPooled = nullptr;

    FRDGBufferRef ReadPitchesScopedRef = nullptr;

    FRDGBufferSRVRef ReadPitchesScopedSRV = nullptr;

    TRefCountPtr<FRDGPooledBuffer> ReadBeatsPooled = nullptr;

    FRDGBufferRef ReadBeatsScopedRef = nullptr;

    FRDGBufferSRVRef ReadBeatsScopedSRV = nullptr;

    ~FST_AudioIOBuffers_RenderThread()
    {
        ReleaseData();
    }

    void ReleaseData()
    {
        IODataSize = 0;
        frequencyBandsLength = 0;
        amplitudesLength = 0;
        pitchesLength = 0;
        beatsLength = 0;
        ReadIODataPooled.SafeRelease();
        ReadFrequencyBandsPooled.SafeRelease();
        ReadAmplitudesPooled.SafeRelease();
        ReadPitchesPooled.SafeRelease();
        ReadBeatsPooled.SafeRelease();
    }
};