mirror of
https://github.com/obsproject/obs-studio.git
synced 2024-07-18 17:19:14 +00:00
100 lines
2.3 KiB
C++
100 lines
2.3 KiB
C++
#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);
|
|
PropVariantClear(&nameVar);
|
|
}
|
|
}
|
|
|
|
return device_name;
|
|
}
|
|
|
|
static 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);
|
|
}
|
|
}
|