97 lines
3.9 KiB
C++
97 lines
3.9 KiB
C++
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
|
|
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
|
|
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
|
|
// PARTICULAR PURPOSE.
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved
|
|
//
|
|
#pragma once
|
|
#include <MMDeviceAPI.h>
|
|
#include <AudioClient.h>
|
|
#include <AudioPolicy.h>
|
|
|
|
//
|
|
// WASAPI Capture class.
|
|
class CWASAPICapture : public IAudioSessionEvents, IMMNotificationClient
|
|
{
|
|
public:
|
|
// Public interface to CWASAPICapture.
|
|
CWASAPICapture(IMMDevice *Endpoint, bool EnableStreamSwitch, ERole EndpointRole);
|
|
bool Initialize(UINT32 EngineLatency);
|
|
void Shutdown();
|
|
bool Start(BYTE *CaptureBuffer, size_t BufferSize);
|
|
void Stop();
|
|
WORD ChannelCount() { return _MixFormat->nChannels; }
|
|
UINT32 SamplesPerSecond() { return _MixFormat->nSamplesPerSec; }
|
|
UINT32 BytesPerSample() { return _MixFormat->wBitsPerSample / 8; }
|
|
size_t FrameSize() { return _FrameSize; }
|
|
WAVEFORMATEX *MixFormat() { return _MixFormat; }
|
|
size_t BytesCaptured() { return _CurrentCaptureIndex; }
|
|
STDMETHOD_(ULONG, AddRef)();
|
|
STDMETHOD_(ULONG, Release)();
|
|
|
|
private:
|
|
~CWASAPICapture(void); // Destructor is private to prevent accidental deletion.
|
|
LONG _RefCount;
|
|
//
|
|
// Core Audio Capture member variables.
|
|
//
|
|
IMMDevice * _Endpoint;
|
|
IAudioClient * _AudioClient;
|
|
IAudioCaptureClient *_CaptureClient;
|
|
|
|
HANDLE _CaptureThread;
|
|
HANDLE _ShutdownEvent;
|
|
WAVEFORMATEX * _MixFormat;
|
|
size_t _FrameSize;
|
|
UINT32 _BufferSize;
|
|
|
|
//
|
|
// Capture buffer management.
|
|
//
|
|
BYTE *_CaptureBuffer;
|
|
size_t _CaptureBufferSize;
|
|
size_t _CurrentCaptureIndex;
|
|
|
|
static DWORD __stdcall WASAPICaptureThread(LPVOID Context);
|
|
DWORD CWASAPICapture::DoCaptureThread();
|
|
//
|
|
// Stream switch related members and methods.
|
|
//
|
|
bool _EnableStreamSwitch;
|
|
ERole _EndpointRole;
|
|
HANDLE _StreamSwitchEvent; // Set when the current session is disconnected.
|
|
HANDLE _StreamSwitchCompleteEvent; // Set when the default device changed.
|
|
IAudioSessionControl * _AudioSessionControl;
|
|
IMMDeviceEnumerator * _DeviceEnumerator;
|
|
LONG _EngineLatencyInMS;
|
|
bool _InStreamSwitch;
|
|
|
|
bool InitializeStreamSwitch();
|
|
void TerminateStreamSwitch();
|
|
bool HandleStreamSwitchEvent();
|
|
|
|
STDMETHOD(OnDisplayNameChanged) (LPCWSTR /*NewDisplayName*/, LPCGUID /*EventContext*/) { return S_OK; };
|
|
STDMETHOD(OnIconPathChanged) (LPCWSTR /*NewIconPath*/, LPCGUID /*EventContext*/) { return S_OK; };
|
|
STDMETHOD(OnSimpleVolumeChanged) (float /*NewSimpleVolume*/, BOOL /*NewMute*/, LPCGUID /*EventContext*/) { return S_OK; }
|
|
STDMETHOD(OnChannelVolumeChanged) (DWORD /*ChannelCount*/, float /*NewChannelVolumes*/[], DWORD /*ChangedChannel*/, LPCGUID /*EventContext*/) { return S_OK; };
|
|
STDMETHOD(OnGroupingParamChanged) (LPCGUID /*NewGroupingParam*/, LPCGUID /*EventContext*/) {return S_OK; };
|
|
STDMETHOD(OnStateChanged) (AudioSessionState /*NewState*/) { return S_OK; };
|
|
STDMETHOD(OnSessionDisconnected) (AudioSessionDisconnectReason DisconnectReason);
|
|
STDMETHOD(OnDeviceStateChanged) (LPCWSTR /*DeviceId*/, DWORD /*NewState*/) { return S_OK; }
|
|
STDMETHOD(OnDeviceAdded) (LPCWSTR /*DeviceId*/) { return S_OK; };
|
|
STDMETHOD(OnDeviceRemoved) (LPCWSTR /*DeviceId(*/) { return S_OK; };
|
|
STDMETHOD(OnDefaultDeviceChanged) (EDataFlow Flow, ERole Role, LPCWSTR NewDefaultDeviceId);
|
|
STDMETHOD(OnPropertyValueChanged) (LPCWSTR /*DeviceId*/, const PROPERTYKEY /*Key*/){return S_OK; };
|
|
//
|
|
// IUnknown
|
|
//
|
|
STDMETHOD(QueryInterface)(REFIID iid, void **pvObject);
|
|
|
|
//
|
|
// Utility functions.
|
|
//
|
|
bool InitializeAudioEngine();
|
|
bool LoadFormat();
|
|
};
|