mirror of
https://github.com/obsproject/obs-studio.git
synced 2026-05-25 00:42:22 -04:00
obs-qsv11: Add adapter checks for Linux
This moves the existing adapter checking into the platform layer and moves the Windows implementation to its platform implementation and adds a Linux implementation based on directly querying VA-API. Unlike Windows, this check takes ~1ms so we have no need to spin out another thread to perform the work. This also fixes up some of the CPP/C mixing going on in common_utils."h" to allow us to call common functions from C files.
This commit is contained in:
committed by
Ryan Foster
parent
c6b470de9a
commit
b8ccaf5649
@@ -1,7 +1,17 @@
|
||||
#include "common_utils.h"
|
||||
|
||||
#include <time.h>
|
||||
#include <cpuid.h>
|
||||
#include <util/c99defs.h>
|
||||
#include <util/dstr.h>
|
||||
#include <va/va_drm.h>
|
||||
#include <va/va_str.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <string.h>
|
||||
#include <dirent.h>
|
||||
|
||||
mfxStatus simple_alloc(mfxHDL pthis, mfxFrameAllocRequest *request,
|
||||
mfxFrameAllocResponse *response)
|
||||
@@ -98,3 +108,172 @@ extern "C" void util_cpuid(int cpuinfo[4], int level)
|
||||
(unsigned int *)&cpuinfo[1], (unsigned int *)&cpuinfo[2],
|
||||
(unsigned int *)&cpuinfo[3]);
|
||||
}
|
||||
|
||||
struct vaapi_device {
|
||||
int fd;
|
||||
VADisplay display;
|
||||
const char *driver;
|
||||
};
|
||||
|
||||
void vaapi_open(char *device_path, struct vaapi_device *device)
|
||||
{
|
||||
int fd = open(device_path, O_RDWR);
|
||||
if (fd < 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
VADisplay display = vaGetDisplayDRM(fd);
|
||||
if (!display) {
|
||||
close(fd);
|
||||
return;
|
||||
}
|
||||
|
||||
// VA-API is noisy by default.
|
||||
vaSetInfoCallback(display, nullptr, nullptr);
|
||||
vaSetErrorCallback(display, nullptr, nullptr);
|
||||
|
||||
int major;
|
||||
int minor;
|
||||
if (vaInitialize(display, &major, &minor) != VA_STATUS_SUCCESS) {
|
||||
vaTerminate(display);
|
||||
close(fd);
|
||||
return;
|
||||
}
|
||||
|
||||
const char *driver = vaQueryVendorString(display);
|
||||
if (strstr(driver, "Intel i965 driver") != nullptr) {
|
||||
blog(LOG_WARNING,
|
||||
"Legacy intel-vaapi-driver detected, incompatible with QSV");
|
||||
vaTerminate(display);
|
||||
close(fd);
|
||||
return;
|
||||
}
|
||||
|
||||
device->fd = fd;
|
||||
device->display = display;
|
||||
device->driver = driver;
|
||||
}
|
||||
|
||||
void vaapi_close(struct vaapi_device *device)
|
||||
{
|
||||
vaTerminate(device->display);
|
||||
close(device->fd);
|
||||
}
|
||||
|
||||
static uint32_t vaapi_check_support(VADisplay display, VAProfile profile,
|
||||
VAEntrypoint entrypoint)
|
||||
{
|
||||
bool ret = false;
|
||||
VAConfigAttrib attrib[1];
|
||||
attrib->type = VAConfigAttribRateControl;
|
||||
|
||||
VAStatus va_status =
|
||||
vaGetConfigAttributes(display, profile, entrypoint, attrib, 1);
|
||||
|
||||
uint32_t rc = 0;
|
||||
switch (va_status) {
|
||||
case VA_STATUS_SUCCESS:
|
||||
rc = attrib->value;
|
||||
break;
|
||||
case VA_STATUS_ERROR_UNSUPPORTED_PROFILE:
|
||||
case VA_STATUS_ERROR_UNSUPPORTED_ENTRYPOINT:
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return (rc & VA_RC_CBR || rc & VA_RC_CQP || rc & VA_RC_VBR);
|
||||
}
|
||||
|
||||
bool vaapi_supports_h264(VADisplay display)
|
||||
{
|
||||
bool ret = false;
|
||||
ret |= vaapi_check_support(display, VAProfileH264ConstrainedBaseline,
|
||||
VAEntrypointEncSlice);
|
||||
ret |= vaapi_check_support(display, VAProfileH264Main,
|
||||
VAEntrypointEncSlice);
|
||||
ret |= vaapi_check_support(display, VAProfileH264High,
|
||||
VAEntrypointEncSlice);
|
||||
|
||||
if (!ret) {
|
||||
ret |= vaapi_check_support(display,
|
||||
VAProfileH264ConstrainedBaseline,
|
||||
VAEntrypointEncSliceLP);
|
||||
ret |= vaapi_check_support(display, VAProfileH264Main,
|
||||
VAEntrypointEncSliceLP);
|
||||
ret |= vaapi_check_support(display, VAProfileH264High,
|
||||
VAEntrypointEncSliceLP);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
bool vaapi_supports_av1(VADisplay display)
|
||||
{
|
||||
bool ret = false;
|
||||
// Are there any devices with non-LowPower entrypoints?
|
||||
ret |= vaapi_check_support(display, VAProfileAV1Profile0,
|
||||
VAEntrypointEncSlice);
|
||||
ret |= vaapi_check_support(display, VAProfileAV1Profile0,
|
||||
VAEntrypointEncSliceLP);
|
||||
return ret;
|
||||
}
|
||||
|
||||
bool vaapi_supports_hevc(VADisplay display)
|
||||
{
|
||||
bool ret = false;
|
||||
ret |= vaapi_check_support(display, VAProfileHEVCMain,
|
||||
VAEntrypointEncSlice);
|
||||
ret |= vaapi_check_support(display, VAProfileHEVCMain,
|
||||
VAEntrypointEncSliceLP);
|
||||
return ret;
|
||||
}
|
||||
|
||||
void check_adapters(struct adapter_info *adapters, size_t *adapter_count)
|
||||
{
|
||||
struct dstr full_path;
|
||||
struct dirent **namelist;
|
||||
int no;
|
||||
int adapter_idx;
|
||||
const char *base_dir = "/dev/dri/";
|
||||
|
||||
dstr_init(&full_path);
|
||||
if ((no = scandir(base_dir, &namelist, 0, alphasort)) > 0) {
|
||||
for (int i = 0; i < no; i++) {
|
||||
struct adapter_info *adapter;
|
||||
struct dirent *dp;
|
||||
struct vaapi_device device = {0};
|
||||
|
||||
dp = namelist[i];
|
||||
if (strstr(dp->d_name, "renderD") == nullptr)
|
||||
goto next_entry;
|
||||
|
||||
adapter_idx = atoi(&dp->d_name[7]) - 128;
|
||||
if (adapter_idx >= (ssize_t)*adapter_count ||
|
||||
adapter_idx < 0)
|
||||
goto next_entry;
|
||||
|
||||
*adapter_count = adapter_idx + 1;
|
||||
dstr_copy(&full_path, base_dir);
|
||||
dstr_cat(&full_path, dp->d_name);
|
||||
vaapi_open(full_path.array, &device);
|
||||
if (!device.display)
|
||||
goto next_entry;
|
||||
|
||||
adapter = &adapters[adapter_idx];
|
||||
adapter->is_intel = strstr(device.driver, "Intel") !=
|
||||
nullptr;
|
||||
// This is currently only used for LowPower coding which is busted on VA-API anyway.
|
||||
adapter->is_dgpu = false;
|
||||
adapter->supports_av1 =
|
||||
vaapi_supports_av1(device.display);
|
||||
adapter->supports_hevc =
|
||||
vaapi_supports_hevc(device.display);
|
||||
vaapi_close(&device);
|
||||
|
||||
next_entry:
|
||||
free(dp);
|
||||
}
|
||||
free(namelist);
|
||||
}
|
||||
dstr_free(&full_path);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user