quantum/winscap
1
0
Fork 0
mirror of https://github.com/quantum5/winscap.git synced 2025-04-24 05:31:58 -04:00
Code Issues Packages Projects Releases Wiki Activity

Handle default output device change by capturing from new device

Browse source
This commit is contained in:
Quantum 2020-01-08 23:12:06 -08:00
parent df1299f5de
commit e8be8a4a14
1 changed files with 91 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

130
winscap.cpp
View file

@ -32,6 +32,44 @@ _COM_SMARTPTR_TYPEDEF(IMMDevice, __uuidof(IMMDevice));
_COM_SMARTPTR_TYPEDEF(IAudioClient, __uuidof(IAudioClient)); _COM_SMARTPTR_TYPEDEF(IAudioClient, __uuidof(IAudioClient));
_COM_SMARTPTR_TYPEDEF(IAudioCaptureClient, __uuidof(IAudioCaptureClient)); _COM_SMARTPTR_TYPEDEF(IAudioCaptureClient, __uuidof(IAudioCaptureClient));
class DeviceChangeNotification : public IMMNotificationClient {
volatile ULONG ref;
volatile bool &changed;
public:
DeviceChangeNotification(volatile bool &changed) : changed(changed) {}
// This is meant to be allocated on stack, so we don't actually free.
STDMETHODIMP_(ULONG) AddRef() { return InterlockedIncrement(&ref); }
STDMETHODIMP_(ULONG) Release() { return InterlockedDecrement(&ref); }
STDMETHODIMP QueryInterface(REFIID iid, void **ppv) {
if (iid == IID_IUnknown) {
AddRef();
*ppv = (IUnknown *)this;
} else if (iid == __uuidof(IMMNotificationClient)) {
AddRef();
*ppv = (IMMNotificationClient *)this;
} else {
*ppv = nullptr;
return E_NOINTERFACE;
}
return S_OK;
}
STDMETHODIMP OnDefaultDeviceChanged(EDataFlow flow, ERole role, LPCWSTR) {
if (flow == eRender && role == eConsole) {
changed = true;
}
return S_OK;
}
STDMETHODIMP OnDeviceAdded(LPCWSTR) { return S_OK; }
STDMETHODIMP OnDeviceRemoved(LPCWSTR) { return S_OK; }
STDMETHODIMP OnDeviceStateChanged(LPCWSTR, DWORD) { return S_OK; }
STDMETHODIMP OnPropertyValueChanged(LPCWSTR, const PROPERTYKEY) { return S_OK; }
};
class EnsureCaptureStop { class EnsureCaptureStop {
IAudioClientPtr m_client; IAudioClientPtr m_client;
@ -99,15 +137,6 @@ int main(int argc, char *argv[]) {
CCoInitialize comInit; CCoInitialize comInit;
ensure(comInit); ensure(comInit);
IMMDeviceEnumeratorPtr pEnumerator;
ensure(pEnumerator.CreateInstance(__uuidof(MMDeviceEnumerator)));
IMMDevicePtr pDevice;
ensure(pEnumerator->GetDefaultAudioEndpoint(eRender, eConsole, &pDevice));
IAudioClientPtr pClient;
ensure(pDevice->Activate(__uuidof(IAudioClient), CLSCTX_ALL, nullptr, (void **)&pClient));
WAVEFORMATEX wfx; WAVEFORMATEX wfx;
wfx.wFormatTag = WAVE_FORMAT_PCM; wfx.wFormatTag = WAVE_FORMAT_PCM;
wfx.nChannels = (WORD)channels; wfx.nChannels = (WORD)channels;
@ -116,44 +145,67 @@ int main(int argc, char *argv[]) {
wfx.nBlockAlign = wfx.nChannels * wfx.wBitsPerSample / 8; wfx.nBlockAlign = wfx.nChannels * wfx.wBitsPerSample / 8;
wfx.nAvgBytesPerSec = wfx.nSamplesPerSec * wfx.nBlockAlign; wfx.nAvgBytesPerSec = wfx.nSamplesPerSec * wfx.nBlockAlign;
ensure(pClient->Initialize(AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_LOOPBACK, IMMDeviceEnumeratorPtr pEnumerator;
16 * REFTIMES_PER_MILLISEC, 0, &wfx, nullptr)); ensure(pEnumerator.CreateInstance(__uuidof(MMDeviceEnumerator)));
UINT32 bufferFrameCount; volatile bool deviceChanged = false;
ensure(pClient->GetBufferSize(&bufferFrameCount)); DeviceChangeNotification deviceChangeNotification(deviceChanged);
pEnumerator->RegisterEndpointNotificationCallback(&deviceChangeNotification);
IAudioCaptureClientPtr pCapture;
ensure(pClient->GetService(__uuidof(IAudioCaptureClient), (void **)&pCapture));
DWORD dwDelay = (DWORD)(((double)REFTIMES_PER_SEC * bufferFrameCount / wfx.nSamplesPerSec) /
REFTIMES_PER_MILLISEC / 2);
LPBYTE pSilence = (LPBYTE)malloc(bufferFrameCount * wfx.nBlockAlign);
EnsureFree freeSilence(pSilence);
ZeroMemory(pSilence, bufferFrameCount * wfx.nBlockAlign);
ensure(pClient->Start());
EnsureCaptureStop autoStop(pClient);
for (;;) { for (;;) {
Sleep(dwDelay); IMMDevicePtr pDevice;
ensure(pEnumerator->GetDefaultAudioEndpoint(eRender, eConsole, &pDevice));
UINT32 packetLength; IAudioClientPtr pClient;
ensure(pCapture->GetNextPacketSize(&packetLength)); ensure(pDevice->Activate(__uuidof(IAudioClient), CLSCTX_ALL, nullptr, (void **)&pClient));
while (packetLength) { ensure(pClient->Initialize(AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_LOOPBACK,
LPBYTE pData; 16 * REFTIMES_PER_MILLISEC, 0, &wfx, nullptr));
UINT32 numFramesAvailable;
DWORD flags;
ensure(pCapture->GetBuffer(&pData, &numFramesAvailable, &flags, nullptr, nullptr)); UINT32 bufferFrameCount;
ensure(pClient->GetBufferSize(&bufferFrameCount));
if (flags & AUDCLNT_BUFFERFLAGS_SILENT) IAudioCaptureClientPtr pCapture;
pData = pSilence; ensure(pClient->GetService(__uuidof(IAudioCaptureClient), (void **)&pCapture));
_write(_fileno(stdout), pData, wfx.nBlockAlign * numFramesAvailable); DWORD dwDelay = (DWORD)(((double)REFTIMES_PER_SEC * bufferFrameCount / wfx.nSamplesPerSec) /
ensure(pCapture->ReleaseBuffer(numFramesAvailable)); REFTIMES_PER_MILLISEC / 2);
ensure(pCapture->GetNextPacketSize(&packetLength));
LPBYTE pSilence = (LPBYTE)malloc(bufferFrameCount * wfx.nBlockAlign);
EnsureFree freeSilence(pSilence);
ZeroMemory(pSilence, bufferFrameCount * wfx.nBlockAlign);
ensure(pClient->Start());
EnsureCaptureStop autoStop(pClient);
while (!deviceChanged) {
Sleep(dwDelay);
UINT32 packetLength;
HRESULT hrNext = pCapture->GetNextPacketSize(&packetLength);
if (hrNext == AUDCLNT_E_DEVICE_INVALIDATED) {
while (!deviceChanged)
;
break;
} else {
ensure(hrNext);
}
while (packetLength) {
LPBYTE pData;
UINT32 numFramesAvailable;
DWORD flags;
ensure(pCapture->GetBuffer(&pData, &numFramesAvailable, &flags, nullptr, nullptr));
if (flags & AUDCLNT_BUFFERFLAGS_SILENT)
pData = pSilence;
_write(_fileno(stdout), pData, wfx.nBlockAlign * numFramesAvailable);
ensure(pCapture->ReleaseBuffer(numFramesAvailable));
ensure(pCapture->GetNextPacketSize(&packetLength));
}
} }
deviceChanged = false;
} }
} }