f53df7da64
Code submissions have continually suffered from formatting inconsistencies that constantly have to be addressed. Using clang-format simplifies this by making code formatting more consistent, and allows automation of the code formatting so that maintainers can focus more on the code itself instead of code formatting.
99 lines
2.3 KiB
C++
99 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);
|
|
}
|
|
}
|
|
|
|
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);
|
|
}
|
|
}
|