ST_AudioSoundWaveProcedural.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Sound/SoundWaveProcedural.h"
#include "Main/ST_AudioProcessingInterface.h"
#include "Main/ST_AudioProcessingSettings.h"
#include "ST_AudioSoundWaveProcedural.generated.h"

UCLASS(BlueprintType)
class ST_AUDIOPROCESSING_API UST_AudioSoundWaveProcedural : public USoundWaveProcedural, public IST_AudioProcessingInterface
{
    GENERATED_BODY()
 
public:

    UST_AudioSoundWaveProcedural(const FObjectInitializer& ObjectInitializer);

    ~UST_AudioSoundWaveProcedural();

    virtual TUniquePtr<Audio::IProxyData> CreateNewProxyData(const Audio::FProxyDataInitParams& InitParams) override;

    virtual int32 OnGeneratePCMAudio(TArray<uint8>& OutAudio, int32 NumSamples);

    void QueueAudioBuffer(const uint8* AudioData, const int32 BufferSize);

    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioProcessing|AudioSoundWaveProcedural")
    void Initialise(FFFTIOData FFTIOData, bool bUsedByMixerOnly, EST_SampleFormat NewSampleFormat);

    virtual Audio::EAudioMixerStreamDataFormat::Type GetGeneratedPCMDataFormat() const override;

    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioProcessing|AudioSoundWaveProcedural")
    virtual bool RegisterToIO(EAudioIOType NewAudioIOType);

    UFUNCTION(BlueprintCallable, Category = "ShaderTech|AudioProcessing|AudioSoundWaveProcedural")
    virtual void DeregisterFromIO(EAudioIOType CurrAudioIOType);

    virtual void OnAudioIOStatusChanged_Implementation(bool bIsActive, FIntPoint IOId) override;

    UPROPERTY(Transient, VisibleAnywhere, BlueprintReadOnly, Category = "ShaderTech|AudioProcessing|AudioSoundWaveProcedural")
    EAudioIOType Registered_AudioIOType = EAudioIOType::None;
 
    const FThreadSafeBool& GetInitialisedFlag() const;
 
private:

    bool bMixerOnly = false;

    EST_SampleFormat SampleFormat = EST_SampleFormat::S16;

    FThreadSafeBool bInitialised = false;
};
 
inline const FThreadSafeBool& UST_AudioSoundWaveProcedural::GetInitialisedFlag() const
{ 
    return bInitialised;
}