ST_AudioIOCaptureBase.h

// Copyright (c) 2025 Aaron Trotter (ShaderTech). All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "IO/Base/ST_AudioIOBase.h"
#include "Main/ST_AudioProcessingSettings.h"
#include "ST_AudioIOCaptureBase.generated.h"
 
struct ma_device;
struct ma_encoder;
using ma_uint32 = unsigned int;

UCLASS(BlueprintType, Abstract)
class ST_AUDIOPROCESSING_API UST_AudioIOCaptureBase : public UST_AudioIOBase
{
    GENERATED_BODY()
 
public:

    bool IsCapturing() const;

    static void MA_CaptureCallBack(ma_device* pDevice, void* pOutput, const void* pInput, ma_uint32 frameCount);

    FEncoderConfig GetEncoderConfig() const;

    virtual EST_SampleFormat GetAudioFormat() const override;
 
protected:

    virtual void DeinitializeAudioIO() override;

    virtual ma_encoder* GetCaptureEncoder() const override;

    void SetIsCapturing(bool bValue);

    ma_device* CaptureDevice = nullptr;

    ma_encoder* CaptureEncoder = nullptr;
 
private:

    bool bIsCapturing = false;
};
 
inline bool UST_AudioIOCaptureBase::IsCapturing() const
{
    return bIsCapturing;
}
 
inline ma_encoder* UST_AudioIOCaptureBase::GetCaptureEncoder() const
{
    return CaptureEncoder;
}
 
inline void UST_AudioIOCaptureBase::SetIsCapturing(bool value)
{
    bIsCapturing = value;
}
 
inline FEncoderConfig UST_AudioIOCaptureBase::GetEncoderConfig() const
{
    return FEncoderConfig(CachedAudioIOConfig);
}
 
inline EST_SampleFormat UST_AudioIOCaptureBase::GetAudioFormat() const
{
    if (CaptureDevice == nullptr)
    {
        return Super::GetAudioFormat();
    }
    return (EST_SampleFormat)CaptureDevice->capture.format;
}