gstreamer/subprojects/gst-plugins-bad/sys/qsv/gstqsvutils.cpp
Seungha Yang 865d7a1351 qsvh265dec: Add support for GBR decoding
Use GBR equivalent output formats if RGB colorspace is detected

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/5706>
2023-11-30 21:50:26 +09:00

344 lines
8.7 KiB
C++

/* GStreamer
* Copyright (C) 2021 Seungha Yang <seungha@centricular.com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "gstqsvutils.h"
#ifdef G_OS_WIN32
#include <gst/d3d11/gstd3d11.h>
#include <wrl.h>
/* *INDENT-OFF* */
using namespace Microsoft::WRL;
/* *INDENT-ON* */
#else
#include <gst/va/gstva.h>
#endif
static mfxLoader _loader = nullptr;
mfxLoader
gst_qsv_get_loader (void)
{
GST_QSV_CALL_ONCE_BEGIN {
_loader = MFXLoad ();
} GST_QSV_CALL_ONCE_END;
return _loader;
}
void
gst_qsv_deinit (void)
{
g_clear_pointer (&_loader, MFXUnload);
}
#ifdef G_OS_WIN32
static GList *
gst_qsv_get_d3d11_devices (void)
{
GList *rst = nullptr;
HRESULT hr;
ComPtr < IDXGIFactory1 > factory;
hr = CreateDXGIFactory1 (IID_PPV_ARGS (&factory));
if (FAILED (hr))
return nullptr;
for (guint idx = 0;; idx++) {
ComPtr < IDXGIAdapter1 > adapter;
DXGI_ADAPTER_DESC desc;
gint64 luid;
GstD3D11Device *device;
ComPtr < ID3D10Multithread > multi_thread;
ID3D11Device *device_handle;
hr = factory->EnumAdapters1 (idx, &adapter);
if (FAILED (hr))
return rst;
hr = adapter->GetDesc (&desc);
if (FAILED (hr))
continue;
if (desc.VendorId != 0x8086)
continue;
luid = gst_d3d11_luid_to_int64 (&desc.AdapterLuid);
device = gst_d3d11_device_new_for_adapter_luid (luid,
D3D11_CREATE_DEVICE_BGRA_SUPPORT);
if (!device)
continue;
device_handle = gst_d3d11_device_get_device_handle (device);
hr = device_handle->QueryInterface (IID_PPV_ARGS (&multi_thread));
if (FAILED (hr)) {
gst_object_unref (device);
continue;
}
/* Should enable mutithread protection layer, otherwise QSV will return
* error code when this handle is passed to QSV via
* MFXVideoCORE_SetHandle() */
multi_thread->SetMultithreadProtected (TRUE);
rst = g_list_append (rst, device);
}
return rst;
}
#else /* G_OS_WIN32 */
static GList *
gst_qsv_get_va_displays (void)
{
gchar path[64];
GList *rst = nullptr;
for (guint i = 0; i < 8; i++) {
GstVaDisplay *display;
GstVaImplementation impl;
g_snprintf (path, sizeof (path), "/dev/dri/renderD%d", 128 + i);
if (!g_file_test (path, G_FILE_TEST_EXISTS))
continue;
display = gst_va_display_drm_new_from_path (path);
if (!display)
continue;
impl = gst_va_display_get_implementation (display);
if (impl != GST_VA_IMPLEMENTATION_INTEL_I965 &&
impl != GST_VA_IMPLEMENTATION_INTEL_IHD) {
gst_object_unref (display);
continue;
}
rst = g_list_append (rst, display);
}
return rst;
}
#endif
GList *
gst_qsv_get_platform_devices (void)
{
#ifdef G_OS_WIN32
return gst_qsv_get_d3d11_devices ();
#else
return gst_qsv_get_va_displays ();
#endif
}
const gchar *
gst_qsv_status_to_string (mfxStatus status)
{
#define CASE(err) \
case err: \
return G_STRINGIFY (err);
switch (status) {
CASE (MFX_ERR_NONE);
CASE (MFX_ERR_UNKNOWN);
CASE (MFX_ERR_NULL_PTR);
CASE (MFX_ERR_UNSUPPORTED);
CASE (MFX_ERR_MEMORY_ALLOC);
CASE (MFX_ERR_NOT_ENOUGH_BUFFER);
CASE (MFX_ERR_INVALID_HANDLE);
CASE (MFX_ERR_LOCK_MEMORY);
CASE (MFX_ERR_NOT_INITIALIZED);
CASE (MFX_ERR_NOT_FOUND);
CASE (MFX_ERR_MORE_DATA);
CASE (MFX_ERR_MORE_SURFACE);
CASE (MFX_ERR_ABORTED);
CASE (MFX_ERR_DEVICE_LOST);
CASE (MFX_ERR_INCOMPATIBLE_VIDEO_PARAM);
CASE (MFX_ERR_INVALID_VIDEO_PARAM);
CASE (MFX_ERR_UNDEFINED_BEHAVIOR);
CASE (MFX_ERR_DEVICE_FAILED);
CASE (MFX_ERR_MORE_BITSTREAM);
CASE (MFX_ERR_GPU_HANG);
CASE (MFX_ERR_REALLOC_SURFACE);
CASE (MFX_ERR_RESOURCE_MAPPED);
CASE (MFX_ERR_NOT_IMPLEMENTED);
CASE (MFX_WRN_IN_EXECUTION);
CASE (MFX_WRN_DEVICE_BUSY);
CASE (MFX_WRN_VIDEO_PARAM_CHANGED);
CASE (MFX_WRN_PARTIAL_ACCELERATION);
CASE (MFX_WRN_INCOMPATIBLE_VIDEO_PARAM);
CASE (MFX_WRN_VALUE_NOT_CHANGED);
CASE (MFX_WRN_OUT_OF_RANGE);
CASE (MFX_WRN_FILTER_SKIPPED);
CASE (MFX_ERR_NONE_PARTIAL_OUTPUT);
CASE (MFX_WRN_ALLOC_TIMEOUT_EXPIRED);
default:
break;
}
#undef CASE
return "Unknown";
}
GstVideoFormat
gst_qsv_frame_info_format_to_gst (const mfxFrameInfo * info, gboolean is_gbr)
{
GstVideoFormat format = GST_VIDEO_FORMAT_UNKNOWN;
switch (info->FourCC) {
case MFX_FOURCC_NV12:
format = GST_VIDEO_FORMAT_NV12;
break;
case MFX_FOURCC_P010:
format = GST_VIDEO_FORMAT_P010_10LE;
break;
case MFX_FOURCC_P016:
if (info->BitDepthLuma == 12)
format = GST_VIDEO_FORMAT_P012_LE;
else
format = GST_VIDEO_FORMAT_P016_LE;
break;
case MFX_FOURCC_YUY2:
format = GST_VIDEO_FORMAT_YUY2;
break;
case MFX_FOURCC_Y210:
format = GST_VIDEO_FORMAT_Y210;
break;
case MFX_FOURCC_Y216:
format = GST_VIDEO_FORMAT_Y212_LE;
break;
break;
case MFX_FOURCC_AYUV:
if (is_gbr)
format = GST_VIDEO_FORMAT_RBGA;
else
format = GST_VIDEO_FORMAT_VUYA;
break;
case MFX_FOURCC_Y410:
if (is_gbr)
format = GST_VIDEO_FORMAT_BGR10A2_LE;
else
format = GST_VIDEO_FORMAT_Y410;
break;
case MFX_FOURCC_Y416:
if (is_gbr)
format = GST_VIDEO_FORMAT_BGRA64_LE;
else
format = GST_VIDEO_FORMAT_Y412_LE;
break;
case MFX_FOURCC_RGB4:
format = GST_VIDEO_FORMAT_BGRA;
break;
default:
break;
}
return format;
}
gboolean
gst_qsv_frame_info_set_format (mfxFrameInfo * info, GstVideoFormat format)
{
switch (format) {
case GST_VIDEO_FORMAT_NV12:
info->FourCC = MFX_FOURCC_NV12;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV420;
info->BitDepthLuma = 8;
info->BitDepthChroma = 8;
info->Shift = 0;
break;
case GST_VIDEO_FORMAT_P010_10LE:
info->FourCC = MFX_FOURCC_P010;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV420;
info->BitDepthLuma = 10;
info->BitDepthChroma = 10;
info->Shift = 1;
break;
case GST_VIDEO_FORMAT_P012_LE:
info->FourCC = MFX_FOURCC_P016;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV420;
info->BitDepthLuma = 12;
info->BitDepthChroma = 12;
info->Shift = 1;
break;
case GST_VIDEO_FORMAT_P016_LE:
info->FourCC = MFX_FOURCC_P016;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV420;
info->BitDepthLuma = 16;
info->BitDepthChroma = 16;
info->Shift = 0;
break;
case GST_VIDEO_FORMAT_YUY2:
info->FourCC = MFX_FOURCC_YUY2;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV422;
info->BitDepthLuma = 8;
info->BitDepthChroma = 8;
info->Shift = 0;
break;
case GST_VIDEO_FORMAT_Y210:
info->FourCC = MFX_FOURCC_Y210;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV422;
info->BitDepthLuma = 10;
info->BitDepthChroma = 10;
info->Shift = 1;
break;
case GST_VIDEO_FORMAT_Y212_LE:
info->FourCC = MFX_FOURCC_Y216;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV422;
info->BitDepthLuma = 12;
info->BitDepthChroma = 12;
info->Shift = 1;
break;
case GST_VIDEO_FORMAT_VUYA:
case GST_VIDEO_FORMAT_RBGA:
info->FourCC = MFX_FOURCC_AYUV;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV444;
info->BitDepthLuma = 8;
info->BitDepthChroma = 8;
info->Shift = 0;
break;
case GST_VIDEO_FORMAT_Y410:
case GST_VIDEO_FORMAT_BGR10A2_LE:
info->FourCC = MFX_FOURCC_Y410;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV444;
info->BitDepthLuma = 10;
info->BitDepthChroma = 10;
info->Shift = 0;
break;
case GST_VIDEO_FORMAT_Y412_LE:
case GST_VIDEO_FORMAT_BGRA64_LE:
info->FourCC = MFX_FOURCC_Y416;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV444;
info->BitDepthLuma = 12;
info->BitDepthChroma = 12;
info->Shift = 1;
break;
case GST_VIDEO_FORMAT_BGRA:
info->FourCC = MFX_FOURCC_RGB4;
info->ChromaFormat = MFX_CHROMAFORMAT_YUV444;
break;
default:
return FALSE;
}
return TRUE;
}