obs-studio/plugins/win-wasapi/enum-wasapi.cpp

97 lines
2.3 KiB
C++
Raw Normal View History

#include "enum-wasapi.hpp"
#include <util/base.h>
#include <util/platform.h>
#include <util/windows/HRError.hpp>
#include <util/windows/ComPtr.hpp>
#include <util/windows/CoTaskMemPtr.hpp>
using namespace std;
string GetDeviceName(IMMDevice *device)
{
string device_name;
ComPtr<IPropertyStore> store;
HRESULT res;
if (SUCCEEDED(device->OpenPropertyStore(STGM_READ, store.Assign()))) {
PROPVARIANT nameVar;
PropVariantInit(&nameVar);
res = store->GetValue(PKEY_Device_FriendlyName, &nameVar);
if (SUCCEEDED(res) && nameVar.pwszVal && *nameVar.pwszVal) {
size_t len = wcslen(nameVar.pwszVal);
size_t size;
size = os_wcs_to_utf8(nameVar.pwszVal, len,
nullptr, 0) + 1;
device_name.resize(size);
os_wcs_to_utf8(nameVar.pwszVal, len,
&device_name[0], size);
}
}
return device_name;
}
void GetWASAPIAudioDevices_(vector<AudioDeviceInfo> &devices, bool input)
{
ComPtr<IMMDeviceEnumerator> enumerator;
ComPtr<IMMDeviceCollection> collection;
UINT count;
HRESULT res;
res = CoCreateInstance(__uuidof(MMDeviceEnumerator), NULL, CLSCTX_ALL,
__uuidof(IMMDeviceEnumerator),
(void**)enumerator.Assign());
if (FAILED(res))
throw HRError("Failed to create enumerator", res);
res = enumerator->EnumAudioEndpoints(input ? eCapture : eRender,
DEVICE_STATE_ACTIVE, collection.Assign());
if (FAILED(res))
throw HRError("Failed to enumerate devices", res);
res = collection->GetCount(&count);
if (FAILED(res))
throw HRError("Failed to get device count", res);
for (UINT i = 0; i < count; i++) {
ComPtr<IMMDevice> device;
CoTaskMemPtr<WCHAR> w_id;
AudioDeviceInfo info;
size_t len, size;
res = collection->Item(i, device.Assign());
if (FAILED(res))
continue;
res = device->GetId(&w_id);
if (FAILED(res) || !w_id || !*w_id)
continue;
info.name = GetDeviceName(device);
len = wcslen(w_id);
size = os_wcs_to_utf8(w_id, len, nullptr, 0) + 1;
info.id.resize(size);
os_wcs_to_utf8(w_id, len, &info.id[0], size);
devices.push_back(info);
}
}
void GetWASAPIAudioDevices(vector<AudioDeviceInfo> &devices, bool input)
{
devices.clear();
try {
GetWASAPIAudioDevices_(devices, input);
} catch (HRError error) {
blog(LOG_WARNING, "[GetWASAPIAudioDevices] %s: %lX",
error.str, error.hr);
}
}