ST_AudioScene.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
#include "Subsystems/EngineSubsystem.h"
#include "Main/ST_AudioProcessingInterface.h"
#include "Engine/World.h"
#include "ST_AudioScene.generated.h"

UCLASS()
class ST_AUDIOVISUALISATION_API UST_AudioScene : public UEngineSubsystem, public IST_AudioProcessingInterface
{
    GENERATED_BODY()
 
public:
     //Begin UEngineSubsystem Interface
    virtual void Initialize(FSubsystemCollectionBase& Collection) override;

    virtual void Deinitialize() override;
    //End UEngineSubsystem Interface

    TSharedPtr<class FST_AudioSceneViewExtension> GetSceneViewExtension() const;
 
protected:
    TSharedPtr<class FST_AudioSceneViewExtension, ESPMode::ThreadSafe> ViewExtension;

    void HandlePostWorldInit(UWorld* World, const UWorld::InitializationValues IVS);

    void HandleWorldEndPlay(UWorld* World, bool bSessionEnded, bool bCleanupResources);
 
private:
    void OnAudioIOStatusChanged_Implementation(const bool bIsActive, const FIntPoint IOId) override;

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

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

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

    virtual void OnBeatDetected_Implementation(const TArray<float>& Beats, const TArray<FChannelBeatDetection>& BeatDetectionForChannel, const FIntPoint IOId) override;

    void Register();

    void Deregister();
};
 
inline TSharedPtr<class FST_AudioSceneViewExtension> UST_AudioScene::GetSceneViewExtension() const
{
    return ViewExtension;
}