WasapiStreamManager.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "AudioCaptureCore.h"
#include "WasapiDeviceEnumeration.h"
#include "WasapiInputStream.h"
#include "WasapiCaptureThread.h"
 
 
namespace Audio
{
    class FWasapiStreamManager
    {
    public:
 
        using FDeviceInfo = FWasapiDeviceEnumeration::FDeviceInfo;
 
        FWasapiStreamManager();
        FWasapiStreamManager(FWasapiStreamManager&& InOther) = delete;
        FWasapiStreamManager(const FWasapiStreamManager& InOther) = delete;
 
        virtual ~FWasapiStreamManager() {}
 
        FWasapiStreamManager& operator=(FWasapiStreamManager&& InOther) = delete;
        FWasapiStreamManager& operator=(const FWasapiStreamManager& InOther) = delete;
 
        FString GetDefaultInputDeviceId();
        FString GetDefaultOutputDeviceId();
 
        bool GetDeviceInfo(const FString& InDeviceId, FDeviceInfo& OutDeviceInfo);
 
        bool GetDeviceIndexFromId(const FString& InDeviceId, int32& OutDeviceIndex);
 
        bool GetDeviceIdFromIndex(int32 InDeviceIndex, EDataFlow InDataFlow, FString& OutDeviceId);
 
        bool GetInputDevicesAvailable(TArray<FDeviceInfo>& OutDevices);
 
        bool OpenStream(
            const FString& InDeviceId, 
            const FWasapiAudioFormat& InFormat,
            uint32 InNumFramesDesired,
            FWasapiOnAudioCaptureFunction InCallback);
 
        bool IsStreamOpen() const;
 
        void StartStream();
        bool IsCapturing() const;
        void StopStream();
 
        void AbortStream();
        void CloseStream();
 
        uint32 GetStreamBufferSizeBytes() const;
        double GetStreamPosition() const;
 
    private:
        enum class EStreamState {
            INVALID_STATE,
            STREAM_CLOSED,
            STREAM_STOPPED,
            STREAM_STOPPING,
            STREAM_CAPTURING
        };
 
        EStreamState State = EStreamState::STREAM_CLOSED;
 
        FWasapiDeviceEnumeration DeviceEnumerator;
 
        TSharedPtr<FWasapiInputStream> InputStream;
 
        TUniquePtr<FWasapiCaptureThread> CaptureThread;
    };
}