mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-07 11:13:41 +02:00
790f793844
There are lots of files that don't need it: The number of object files that actually need it went down from 2011 to 884 here. Keep it for external users in order to not cause breakages. Also improve the other headers a bit while just at it. Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
837 lines
25 KiB
C
837 lines
25 KiB
C
/*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#define VK_NO_PROTOTYPES
|
|
#define VK_ENABLE_BETA_EXTENSIONS
|
|
|
|
#include "config.h"
|
|
#include "ffplay_renderer.h"
|
|
|
|
#if (SDL_VERSION_ATLEAST(2, 0, 6) && CONFIG_LIBPLACEBO)
|
|
/* Get PL_API_VER */
|
|
#include <libplacebo/config.h>
|
|
#define HAVE_VULKAN_RENDERER (PL_API_VER >= 278)
|
|
#else
|
|
#define HAVE_VULKAN_RENDERER 0
|
|
#endif
|
|
|
|
#if HAVE_VULKAN_RENDERER
|
|
|
|
#if defined(_WIN32) && !defined(VK_USE_PLATFORM_WIN32_KHR)
|
|
#define VK_USE_PLATFORM_WIN32_KHR
|
|
#endif
|
|
|
|
#include <libplacebo/vulkan.h>
|
|
#include <libplacebo/utils/frame_queue.h>
|
|
#include <libplacebo/utils/libav.h>
|
|
#include <SDL_vulkan.h>
|
|
|
|
#include "libavutil/bprint.h"
|
|
#include "libavutil/mem.h"
|
|
|
|
#endif
|
|
|
|
struct VkRenderer {
|
|
const AVClass *class;
|
|
|
|
int (*create)(VkRenderer *renderer, SDL_Window *window, AVDictionary *dict);
|
|
|
|
int (*get_hw_dev)(VkRenderer *renderer, AVBufferRef **dev);
|
|
|
|
int (*display)(VkRenderer *renderer, AVFrame *frame);
|
|
|
|
int (*resize)(VkRenderer *renderer, int width, int height);
|
|
|
|
void (*destroy)(VkRenderer *renderer);
|
|
};
|
|
|
|
#if HAVE_VULKAN_RENDERER
|
|
|
|
typedef struct RendererContext {
|
|
VkRenderer api;
|
|
|
|
// Can be NULL when vulkan instance is created by avutil
|
|
pl_vk_inst placebo_instance;
|
|
pl_vulkan placebo_vulkan;
|
|
pl_swapchain swapchain;
|
|
VkSurfaceKHR vk_surface;
|
|
pl_renderer renderer;
|
|
pl_tex tex[4];
|
|
|
|
pl_log vk_log;
|
|
|
|
AVBufferRef *hw_device_ref;
|
|
AVBufferRef *hw_frame_ref;
|
|
enum AVPixelFormat *transfer_formats;
|
|
AVHWFramesConstraints *constraints;
|
|
|
|
PFN_vkGetInstanceProcAddr get_proc_addr;
|
|
// This field is a copy from pl_vk_inst->instance or hw_device_ref instance.
|
|
VkInstance inst;
|
|
|
|
AVFrame *vk_frame;
|
|
} RendererContext;
|
|
|
|
static void vk_log_cb(void *log_priv, enum pl_log_level level,
|
|
const char *msg)
|
|
{
|
|
static const int level_map[] = {
|
|
AV_LOG_QUIET,
|
|
AV_LOG_FATAL,
|
|
AV_LOG_ERROR,
|
|
AV_LOG_WARNING,
|
|
AV_LOG_INFO,
|
|
AV_LOG_DEBUG,
|
|
AV_LOG_TRACE,
|
|
};
|
|
|
|
if (level > 0 && level < FF_ARRAY_ELEMS(level_map))
|
|
av_log(log_priv, level_map[level], "%s\n", msg);
|
|
}
|
|
|
|
// Should keep sync with optional_device_exts inside hwcontext_vulkan.c
|
|
static const char *optional_device_exts[] = {
|
|
/* Misc or required by other extensions */
|
|
VK_KHR_PORTABILITY_SUBSET_EXTENSION_NAME,
|
|
VK_KHR_PUSH_DESCRIPTOR_EXTENSION_NAME,
|
|
VK_KHR_SAMPLER_YCBCR_CONVERSION_EXTENSION_NAME,
|
|
VK_EXT_DESCRIPTOR_BUFFER_EXTENSION_NAME,
|
|
VK_EXT_PHYSICAL_DEVICE_DRM_EXTENSION_NAME,
|
|
VK_EXT_SHADER_ATOMIC_FLOAT_EXTENSION_NAME,
|
|
VK_KHR_COOPERATIVE_MATRIX_EXTENSION_NAME,
|
|
|
|
/* Imports/exports */
|
|
VK_KHR_EXTERNAL_MEMORY_FD_EXTENSION_NAME,
|
|
VK_EXT_EXTERNAL_MEMORY_DMA_BUF_EXTENSION_NAME,
|
|
VK_EXT_IMAGE_DRM_FORMAT_MODIFIER_EXTENSION_NAME,
|
|
VK_KHR_EXTERNAL_SEMAPHORE_FD_EXTENSION_NAME,
|
|
VK_EXT_EXTERNAL_MEMORY_HOST_EXTENSION_NAME,
|
|
#ifdef _WIN32
|
|
VK_KHR_EXTERNAL_MEMORY_WIN32_EXTENSION_NAME,
|
|
VK_KHR_EXTERNAL_SEMAPHORE_WIN32_EXTENSION_NAME,
|
|
#endif
|
|
|
|
/* Video encoding/decoding */
|
|
VK_KHR_VIDEO_QUEUE_EXTENSION_NAME,
|
|
VK_KHR_VIDEO_DECODE_QUEUE_EXTENSION_NAME,
|
|
VK_KHR_VIDEO_DECODE_H264_EXTENSION_NAME,
|
|
VK_KHR_VIDEO_DECODE_H265_EXTENSION_NAME,
|
|
"VK_MESA_video_decode_av1",
|
|
};
|
|
|
|
static inline int enable_debug(const AVDictionary *opt)
|
|
{
|
|
AVDictionaryEntry *entry = av_dict_get(opt, "debug", NULL, 0);
|
|
int debug = entry && strtol(entry->value, NULL, 10);
|
|
return debug;
|
|
}
|
|
|
|
static void hwctx_lock_queue(void *priv, uint32_t qf, uint32_t qidx)
|
|
{
|
|
AVHWDeviceContext *avhwctx = priv;
|
|
const AVVulkanDeviceContext *hwctx = avhwctx->hwctx;
|
|
hwctx->lock_queue(avhwctx, qf, qidx);
|
|
}
|
|
|
|
static void hwctx_unlock_queue(void *priv, uint32_t qf, uint32_t qidx)
|
|
{
|
|
AVHWDeviceContext *avhwctx = priv;
|
|
const AVVulkanDeviceContext *hwctx = avhwctx->hwctx;
|
|
hwctx->unlock_queue(avhwctx, qf, qidx);
|
|
}
|
|
|
|
static int add_instance_extension(const char **ext, unsigned num_ext,
|
|
const AVDictionary *opt,
|
|
AVDictionary **dict)
|
|
{
|
|
const char *inst_ext_key = "instance_extensions";
|
|
AVDictionaryEntry *entry;
|
|
AVBPrint buf;
|
|
char *ext_list = NULL;
|
|
int ret;
|
|
|
|
av_bprint_init(&buf, 0, AV_BPRINT_SIZE_AUTOMATIC);
|
|
for (int i = 0; i < num_ext; i++) {
|
|
if (i)
|
|
av_bprintf(&buf, "+%s", ext[i]);
|
|
else
|
|
av_bprintf(&buf, "%s", ext[i]);
|
|
}
|
|
|
|
entry = av_dict_get(opt, inst_ext_key, NULL, 0);
|
|
if (entry && entry->value && entry->value[0]) {
|
|
if (num_ext)
|
|
av_bprintf(&buf, "+");
|
|
av_bprintf(&buf, "%s", entry->value);
|
|
}
|
|
|
|
ret = av_bprint_finalize(&buf, &ext_list);
|
|
if (ret < 0)
|
|
return ret;
|
|
return av_dict_set(dict, inst_ext_key, ext_list, AV_DICT_DONT_STRDUP_VAL);
|
|
}
|
|
|
|
static int add_device_extension(const AVDictionary *opt,
|
|
AVDictionary **dict)
|
|
{
|
|
const char *dev_ext_key = "device_extensions";
|
|
AVDictionaryEntry *entry;
|
|
AVBPrint buf;
|
|
char *ext_list = NULL;
|
|
int ret;
|
|
|
|
av_bprint_init(&buf, 0, AV_BPRINT_SIZE_AUTOMATIC);
|
|
av_bprintf(&buf, "%s", VK_KHR_SWAPCHAIN_EXTENSION_NAME);
|
|
for (int i = 0; i < pl_vulkan_num_recommended_extensions; i++)
|
|
av_bprintf(&buf, "+%s", pl_vulkan_recommended_extensions[i]);
|
|
|
|
entry = av_dict_get(opt, dev_ext_key, NULL, 0);
|
|
if (entry && entry->value && entry->value[0])
|
|
av_bprintf(&buf, "+%s", entry->value);
|
|
|
|
ret = av_bprint_finalize(&buf, &ext_list);
|
|
if (ret < 0)
|
|
return ret;
|
|
return av_dict_set(dict, dev_ext_key, ext_list, AV_DICT_DONT_STRDUP_VAL);
|
|
}
|
|
|
|
static const char *select_device(const AVDictionary *opt)
|
|
{
|
|
const AVDictionaryEntry *entry;
|
|
|
|
entry = av_dict_get(opt, "device", NULL, 0);
|
|
if (entry)
|
|
return entry->value;
|
|
return NULL;
|
|
}
|
|
|
|
static int create_vk_by_hwcontext(VkRenderer *renderer,
|
|
const char **ext, unsigned num_ext,
|
|
const AVDictionary *opt)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
AVHWDeviceContext *dev;
|
|
AVVulkanDeviceContext *hwctx;
|
|
AVDictionary *dict = NULL;
|
|
int ret;
|
|
|
|
ret = add_instance_extension(ext, num_ext, opt, &dict);
|
|
if (ret < 0)
|
|
return ret;
|
|
ret = add_device_extension(opt, &dict);
|
|
if (ret) {
|
|
av_dict_free(&dict);
|
|
return ret;
|
|
}
|
|
|
|
ret = av_hwdevice_ctx_create(&ctx->hw_device_ref, AV_HWDEVICE_TYPE_VULKAN,
|
|
select_device(opt), dict, 0);
|
|
av_dict_free(&dict);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
dev = (AVHWDeviceContext *) ctx->hw_device_ref->data;
|
|
hwctx = dev->hwctx;
|
|
|
|
// There is no way to pass SDL GetInstanceProcAddr to hwdevice.
|
|
// Check the result and return error if they don't match.
|
|
if (hwctx->get_proc_addr != SDL_Vulkan_GetVkGetInstanceProcAddr()) {
|
|
av_log(renderer, AV_LOG_ERROR,
|
|
"hwdevice and SDL use different get_proc_addr. "
|
|
"Try -vulkan_params create_by_placebo=1\n");
|
|
return AVERROR_PATCHWELCOME;
|
|
}
|
|
|
|
ctx->get_proc_addr = hwctx->get_proc_addr;
|
|
ctx->inst = hwctx->inst;
|
|
ctx->placebo_vulkan = pl_vulkan_import(ctx->vk_log,
|
|
pl_vulkan_import_params(
|
|
.instance = hwctx->inst,
|
|
.get_proc_addr = hwctx->get_proc_addr,
|
|
.phys_device = hwctx->phys_dev,
|
|
.device = hwctx->act_dev,
|
|
.extensions = hwctx->enabled_dev_extensions,
|
|
.num_extensions = hwctx->nb_enabled_dev_extensions,
|
|
.features = &hwctx->device_features,
|
|
.lock_queue = hwctx_lock_queue,
|
|
.unlock_queue = hwctx_unlock_queue,
|
|
.queue_ctx = dev,
|
|
.queue_graphics = {
|
|
.index = hwctx->queue_family_index,
|
|
.count = hwctx->nb_graphics_queues,
|
|
},
|
|
.queue_compute = {
|
|
.index = hwctx->queue_family_comp_index,
|
|
.count = hwctx->nb_comp_queues,
|
|
},
|
|
.queue_transfer = {
|
|
.index = hwctx->queue_family_tx_index,
|
|
.count = hwctx->nb_tx_queues,
|
|
},
|
|
));
|
|
if (!ctx->placebo_vulkan)
|
|
return AVERROR_EXTERNAL;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void placebo_lock_queue(struct AVHWDeviceContext *dev_ctx,
|
|
uint32_t queue_family, uint32_t index)
|
|
{
|
|
RendererContext *ctx = dev_ctx->user_opaque;
|
|
pl_vulkan vk = ctx->placebo_vulkan;
|
|
vk->lock_queue(vk, queue_family, index);
|
|
}
|
|
|
|
static void placebo_unlock_queue(struct AVHWDeviceContext *dev_ctx,
|
|
uint32_t queue_family,
|
|
uint32_t index)
|
|
{
|
|
RendererContext *ctx = dev_ctx->user_opaque;
|
|
pl_vulkan vk = ctx->placebo_vulkan;
|
|
vk->unlock_queue(vk, queue_family, index);
|
|
}
|
|
|
|
static int get_decode_queue(VkRenderer *renderer, int *index, int *count)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
VkQueueFamilyProperties *queue_family_prop = NULL;
|
|
uint32_t num_queue_family_prop = 0;
|
|
PFN_vkGetPhysicalDeviceQueueFamilyProperties get_queue_family_prop;
|
|
PFN_vkGetInstanceProcAddr get_proc_addr = ctx->get_proc_addr;
|
|
|
|
*index = -1;
|
|
*count = 0;
|
|
get_queue_family_prop = (PFN_vkGetPhysicalDeviceQueueFamilyProperties)
|
|
get_proc_addr(ctx->placebo_instance->instance,
|
|
"vkGetPhysicalDeviceQueueFamilyProperties");
|
|
get_queue_family_prop(ctx->placebo_vulkan->phys_device,
|
|
&num_queue_family_prop, NULL);
|
|
if (!num_queue_family_prop)
|
|
return AVERROR_EXTERNAL;
|
|
|
|
queue_family_prop = av_calloc(num_queue_family_prop,
|
|
sizeof(*queue_family_prop));
|
|
if (!queue_family_prop)
|
|
return AVERROR(ENOMEM);
|
|
|
|
get_queue_family_prop(ctx->placebo_vulkan->phys_device,
|
|
&num_queue_family_prop,
|
|
queue_family_prop);
|
|
|
|
for (int i = 0; i < num_queue_family_prop; i++) {
|
|
if (queue_family_prop[i].queueFlags & VK_QUEUE_VIDEO_DECODE_BIT_KHR) {
|
|
*index = i;
|
|
*count = queue_family_prop[i].queueCount;
|
|
break;
|
|
}
|
|
}
|
|
av_free(queue_family_prop);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int create_vk_by_placebo(VkRenderer *renderer,
|
|
const char **ext, unsigned num_ext,
|
|
const AVDictionary *opt)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
AVHWDeviceContext *device_ctx;
|
|
AVVulkanDeviceContext *vk_dev_ctx;
|
|
int decode_index;
|
|
int decode_count;
|
|
int ret;
|
|
|
|
ctx->get_proc_addr = SDL_Vulkan_GetVkGetInstanceProcAddr();
|
|
|
|
ctx->placebo_instance = pl_vk_inst_create(ctx->vk_log, pl_vk_inst_params(
|
|
.get_proc_addr = ctx->get_proc_addr,
|
|
.debug = enable_debug(opt),
|
|
.extensions = ext,
|
|
.num_extensions = num_ext
|
|
));
|
|
if (!ctx->placebo_instance) {
|
|
return AVERROR_EXTERNAL;
|
|
}
|
|
ctx->inst = ctx->placebo_instance->instance;
|
|
|
|
ctx->placebo_vulkan = pl_vulkan_create(ctx->vk_log, pl_vulkan_params(
|
|
.instance = ctx->placebo_instance->instance,
|
|
.get_proc_addr = ctx->placebo_instance->get_proc_addr,
|
|
.surface = ctx->vk_surface,
|
|
.allow_software = false,
|
|
.opt_extensions = optional_device_exts,
|
|
.num_opt_extensions = FF_ARRAY_ELEMS(optional_device_exts),
|
|
.extra_queues = VK_QUEUE_VIDEO_DECODE_BIT_KHR,
|
|
.device_name = select_device(opt),
|
|
));
|
|
if (!ctx->placebo_vulkan)
|
|
return AVERROR_EXTERNAL;
|
|
ctx->hw_device_ref = av_hwdevice_ctx_alloc(AV_HWDEVICE_TYPE_VULKAN);
|
|
if (!ctx->hw_device_ref) {
|
|
return AVERROR(ENOMEM);
|
|
}
|
|
|
|
device_ctx = (AVHWDeviceContext *) ctx->hw_device_ref->data;
|
|
device_ctx->user_opaque = ctx;
|
|
|
|
vk_dev_ctx = device_ctx->hwctx;
|
|
vk_dev_ctx->lock_queue = placebo_lock_queue,
|
|
vk_dev_ctx->unlock_queue = placebo_unlock_queue;
|
|
|
|
vk_dev_ctx->get_proc_addr = ctx->placebo_instance->get_proc_addr;
|
|
|
|
vk_dev_ctx->inst = ctx->placebo_instance->instance;
|
|
vk_dev_ctx->phys_dev = ctx->placebo_vulkan->phys_device;
|
|
vk_dev_ctx->act_dev = ctx->placebo_vulkan->device;
|
|
|
|
vk_dev_ctx->device_features = *ctx->placebo_vulkan->features;
|
|
|
|
vk_dev_ctx->enabled_inst_extensions = ctx->placebo_instance->extensions;
|
|
vk_dev_ctx->nb_enabled_inst_extensions = ctx->placebo_instance->num_extensions;
|
|
|
|
vk_dev_ctx->enabled_dev_extensions = ctx->placebo_vulkan->extensions;
|
|
vk_dev_ctx->nb_enabled_dev_extensions = ctx->placebo_vulkan->num_extensions;
|
|
|
|
vk_dev_ctx->queue_family_index = ctx->placebo_vulkan->queue_graphics.index;
|
|
vk_dev_ctx->nb_graphics_queues = ctx->placebo_vulkan->queue_graphics.count;
|
|
|
|
vk_dev_ctx->queue_family_tx_index = ctx->placebo_vulkan->queue_transfer.index;
|
|
vk_dev_ctx->nb_tx_queues = ctx->placebo_vulkan->queue_transfer.count;
|
|
|
|
vk_dev_ctx->queue_family_comp_index = ctx->placebo_vulkan->queue_compute.index;
|
|
vk_dev_ctx->nb_comp_queues = ctx->placebo_vulkan->queue_compute.count;
|
|
|
|
ret = get_decode_queue(renderer, &decode_index, &decode_count);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
vk_dev_ctx->queue_family_decode_index = decode_index;
|
|
vk_dev_ctx->nb_decode_queues = decode_count;
|
|
|
|
ret = av_hwdevice_ctx_init(ctx->hw_device_ref);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int create(VkRenderer *renderer, SDL_Window *window, AVDictionary *opt)
|
|
{
|
|
int ret = 0;
|
|
unsigned num_ext = 0;
|
|
const char **ext = NULL;
|
|
int w, h;
|
|
struct pl_log_params vk_log_params = {
|
|
.log_cb = vk_log_cb,
|
|
.log_level = PL_LOG_DEBUG,
|
|
.log_priv = renderer,
|
|
};
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
AVDictionaryEntry *entry;
|
|
|
|
ctx->vk_log = pl_log_create(PL_API_VER, &vk_log_params);
|
|
|
|
if (!SDL_Vulkan_GetInstanceExtensions(window, &num_ext, NULL)) {
|
|
av_log(NULL, AV_LOG_FATAL, "Failed to get vulkan extensions: %s\n",
|
|
SDL_GetError());
|
|
return AVERROR_EXTERNAL;
|
|
}
|
|
|
|
ext = av_calloc(num_ext, sizeof(*ext));
|
|
if (!ext) {
|
|
ret = AVERROR(ENOMEM);
|
|
goto out;
|
|
}
|
|
|
|
SDL_Vulkan_GetInstanceExtensions(window, &num_ext, ext);
|
|
|
|
entry = av_dict_get(opt, "create_by_placebo", NULL, 0);
|
|
if (entry && strtol(entry->value, NULL, 10))
|
|
ret = create_vk_by_placebo(renderer, ext, num_ext, opt);
|
|
else
|
|
ret = create_vk_by_hwcontext(renderer, ext, num_ext, opt);
|
|
if (ret < 0)
|
|
goto out;
|
|
|
|
if (!SDL_Vulkan_CreateSurface(window, ctx->inst, &ctx->vk_surface)) {
|
|
ret = AVERROR_EXTERNAL;
|
|
goto out;
|
|
}
|
|
|
|
ctx->swapchain = pl_vulkan_create_swapchain(
|
|
ctx->placebo_vulkan,
|
|
pl_vulkan_swapchain_params(
|
|
.surface = ctx->vk_surface,
|
|
.present_mode = VK_PRESENT_MODE_FIFO_KHR));
|
|
if (!ctx->swapchain) {
|
|
ret = AVERROR_EXTERNAL;
|
|
goto out;
|
|
}
|
|
|
|
SDL_Vulkan_GetDrawableSize(window, &w, &h);
|
|
pl_swapchain_resize(ctx->swapchain, &w, &h);
|
|
|
|
ctx->renderer = pl_renderer_create(ctx->vk_log, ctx->placebo_vulkan->gpu);
|
|
if (!ctx->renderer) {
|
|
ret = AVERROR_EXTERNAL;
|
|
goto out;
|
|
}
|
|
|
|
ctx->vk_frame = av_frame_alloc();
|
|
if (!ctx->vk_frame) {
|
|
ret = AVERROR(ENOMEM);
|
|
goto out;
|
|
}
|
|
|
|
ret = 0;
|
|
|
|
out:
|
|
av_free(ext);
|
|
return ret;
|
|
}
|
|
|
|
static int get_hw_dev(VkRenderer *renderer, AVBufferRef **dev)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
|
|
*dev = ctx->hw_device_ref;
|
|
return 0;
|
|
}
|
|
|
|
static int create_hw_frame(VkRenderer *renderer, AVFrame *frame)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
AVHWFramesContext *src_hw_frame = (AVHWFramesContext *)
|
|
frame->hw_frames_ctx->data;
|
|
AVHWFramesContext *hw_frame;
|
|
AVVulkanFramesContext *vk_frame_ctx;
|
|
int ret;
|
|
|
|
if (ctx->hw_frame_ref) {
|
|
hw_frame = (AVHWFramesContext *) ctx->hw_frame_ref->data;
|
|
|
|
if (hw_frame->width == frame->width &&
|
|
hw_frame->height == frame->height &&
|
|
hw_frame->sw_format == src_hw_frame->sw_format)
|
|
return 0;
|
|
|
|
av_buffer_unref(&ctx->hw_frame_ref);
|
|
}
|
|
|
|
if (!ctx->constraints) {
|
|
ctx->constraints = av_hwdevice_get_hwframe_constraints(
|
|
ctx->hw_device_ref, NULL);
|
|
if (!ctx->constraints)
|
|
return AVERROR(ENOMEM);
|
|
}
|
|
|
|
// Check constraints and skip create hwframe. Don't take it as error since
|
|
// we can fallback to memory copy from GPU to CPU.
|
|
if ((ctx->constraints->max_width &&
|
|
ctx->constraints->max_width < frame->width) ||
|
|
(ctx->constraints->max_height &&
|
|
ctx->constraints->max_height < frame->height) ||
|
|
(ctx->constraints->min_width &&
|
|
ctx->constraints->min_width > frame->width) ||
|
|
(ctx->constraints->min_height &&
|
|
ctx->constraints->min_height > frame->height))
|
|
return 0;
|
|
|
|
if (ctx->constraints->valid_sw_formats) {
|
|
enum AVPixelFormat *sw_formats = ctx->constraints->valid_sw_formats;
|
|
while (*sw_formats != AV_PIX_FMT_NONE) {
|
|
if (*sw_formats == src_hw_frame->sw_format)
|
|
break;
|
|
sw_formats++;
|
|
}
|
|
if (*sw_formats == AV_PIX_FMT_NONE)
|
|
return 0;
|
|
}
|
|
|
|
ctx->hw_frame_ref = av_hwframe_ctx_alloc(ctx->hw_device_ref);
|
|
if (!ctx->hw_frame_ref)
|
|
return AVERROR(ENOMEM);
|
|
|
|
hw_frame = (AVHWFramesContext *) ctx->hw_frame_ref->data;
|
|
hw_frame->format = AV_PIX_FMT_VULKAN;
|
|
hw_frame->sw_format = src_hw_frame->sw_format;
|
|
hw_frame->width = frame->width;
|
|
hw_frame->height = frame->height;
|
|
|
|
if (frame->format == AV_PIX_FMT_CUDA) {
|
|
vk_frame_ctx = hw_frame->hwctx;
|
|
vk_frame_ctx->flags = AV_VK_FRAME_FLAG_DISABLE_MULTIPLANE;
|
|
}
|
|
|
|
ret = av_hwframe_ctx_init(ctx->hw_frame_ref);
|
|
if (ret < 0) {
|
|
av_log(renderer, AV_LOG_ERROR, "Create hwframe context failed, %s\n",
|
|
av_err2str(ret));
|
|
return ret;
|
|
}
|
|
|
|
av_hwframe_transfer_get_formats(ctx->hw_frame_ref,
|
|
AV_HWFRAME_TRANSFER_DIRECTION_TO,
|
|
&ctx->transfer_formats, 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline int check_hw_transfer(RendererContext *ctx, AVFrame *frame)
|
|
{
|
|
if (!ctx->hw_frame_ref || !ctx->transfer_formats)
|
|
return 0;
|
|
|
|
for (int i = 0; ctx->transfer_formats[i] != AV_PIX_FMT_NONE; i++)
|
|
if (ctx->transfer_formats[i] == frame->format)
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline int move_to_output_frame(RendererContext *ctx, AVFrame *frame)
|
|
{
|
|
int ret = av_frame_copy_props(ctx->vk_frame, frame);
|
|
if (ret < 0)
|
|
return ret;
|
|
av_frame_unref(frame);
|
|
av_frame_move_ref(frame, ctx->vk_frame);
|
|
return 0;
|
|
}
|
|
|
|
static int map_frame(VkRenderer *renderer, AVFrame *frame, int use_hw_frame)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
int ret;
|
|
|
|
if (use_hw_frame && !ctx->hw_frame_ref)
|
|
return AVERROR(ENOSYS);
|
|
|
|
// Try map data first
|
|
av_frame_unref(ctx->vk_frame);
|
|
if (use_hw_frame) {
|
|
ctx->vk_frame->hw_frames_ctx = av_buffer_ref(ctx->hw_frame_ref);
|
|
ctx->vk_frame->format = AV_PIX_FMT_VULKAN;
|
|
}
|
|
ret = av_hwframe_map(ctx->vk_frame, frame, 0);
|
|
if (!ret)
|
|
return move_to_output_frame(ctx, frame);
|
|
|
|
if (ret != AVERROR(ENOSYS))
|
|
av_log(NULL, AV_LOG_FATAL, "Map frame failed: %s\n", av_err2str(ret));
|
|
return ret;
|
|
}
|
|
|
|
static int transfer_frame(VkRenderer *renderer, AVFrame *frame, int use_hw_frame)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
int ret;
|
|
|
|
if (use_hw_frame && !check_hw_transfer(ctx, frame))
|
|
return AVERROR(ENOSYS);
|
|
|
|
av_frame_unref(ctx->vk_frame);
|
|
if (use_hw_frame)
|
|
av_hwframe_get_buffer(ctx->hw_frame_ref, ctx->vk_frame, 0);
|
|
ret = av_hwframe_transfer_data(ctx->vk_frame, frame, 1);
|
|
if (!ret)
|
|
return move_to_output_frame(ctx, frame);
|
|
|
|
if (ret != AVERROR(ENOSYS))
|
|
av_log(NULL, AV_LOG_FATAL, "Transfer frame failed: %s\n",
|
|
av_err2str(ret));
|
|
return ret;
|
|
}
|
|
|
|
static int convert_frame(VkRenderer *renderer, AVFrame *frame)
|
|
{
|
|
int ret;
|
|
|
|
if (!frame->hw_frames_ctx)
|
|
return 0;
|
|
|
|
if (frame->format == AV_PIX_FMT_VULKAN)
|
|
return 0;
|
|
|
|
ret = create_hw_frame(renderer, frame);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
for (int use_hw = 1; use_hw >=0; use_hw--) {
|
|
ret = map_frame(renderer, frame, use_hw);
|
|
if (!ret)
|
|
return 0;
|
|
if (ret != AVERROR(ENOSYS))
|
|
return ret;
|
|
|
|
ret = transfer_frame(renderer, frame, use_hw);
|
|
if (!ret)
|
|
return 0;
|
|
if (ret != AVERROR(ENOSYS))
|
|
return ret;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static int display(VkRenderer *renderer, AVFrame *frame)
|
|
{
|
|
struct pl_swapchain_frame swap_frame = {0};
|
|
struct pl_frame pl_frame = {0};
|
|
struct pl_frame target = {0};
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
int ret = 0;
|
|
|
|
ret = convert_frame(renderer, frame);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
if (!pl_map_avframe_ex(ctx->placebo_vulkan->gpu, &pl_frame, pl_avframe_params(
|
|
.frame = frame,
|
|
.tex = ctx->tex))) {
|
|
av_log(NULL, AV_LOG_ERROR, "pl_map_avframe_ex failed\n");
|
|
return AVERROR_EXTERNAL;
|
|
}
|
|
|
|
if (!pl_swapchain_start_frame(ctx->swapchain, &swap_frame)) {
|
|
av_log(NULL, AV_LOG_ERROR, "start frame failed\n");
|
|
ret = AVERROR_EXTERNAL;
|
|
goto out;
|
|
}
|
|
|
|
pl_frame_from_swapchain(&target, &swap_frame);
|
|
if (!pl_render_image(ctx->renderer, &pl_frame, &target,
|
|
&pl_render_default_params)) {
|
|
av_log(NULL, AV_LOG_ERROR, "pl_render_image failed\n");
|
|
ret = AVERROR_EXTERNAL;
|
|
goto out;
|
|
}
|
|
|
|
if (!pl_swapchain_submit_frame(ctx->swapchain)) {
|
|
av_log(NULL, AV_LOG_ERROR, "pl_swapchain_submit_frame failed\n");
|
|
ret = AVERROR_EXTERNAL;
|
|
goto out;
|
|
}
|
|
pl_swapchain_swap_buffers(ctx->swapchain);
|
|
|
|
out:
|
|
pl_unmap_avframe(ctx->placebo_vulkan->gpu, &pl_frame);
|
|
return ret;
|
|
}
|
|
|
|
static int resize(VkRenderer *renderer, int width, int height)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
|
|
if (!pl_swapchain_resize(ctx->swapchain, &width, &height))
|
|
return AVERROR_EXTERNAL;
|
|
return 0;
|
|
}
|
|
|
|
static void destroy(VkRenderer *renderer)
|
|
{
|
|
RendererContext *ctx = (RendererContext *) renderer;
|
|
PFN_vkDestroySurfaceKHR vkDestroySurfaceKHR;
|
|
|
|
av_frame_free(&ctx->vk_frame);
|
|
av_freep(&ctx->transfer_formats);
|
|
av_hwframe_constraints_free(&ctx->constraints);
|
|
av_buffer_unref(&ctx->hw_frame_ref);
|
|
|
|
if (ctx->placebo_vulkan) {
|
|
for (int i = 0; i < FF_ARRAY_ELEMS(ctx->tex); i++)
|
|
pl_tex_destroy(ctx->placebo_vulkan->gpu, &ctx->tex[i]);
|
|
pl_renderer_destroy(&ctx->renderer);
|
|
pl_swapchain_destroy(&ctx->swapchain);
|
|
pl_vulkan_destroy(&ctx->placebo_vulkan);
|
|
}
|
|
|
|
if (ctx->vk_surface) {
|
|
vkDestroySurfaceKHR = (PFN_vkDestroySurfaceKHR)
|
|
ctx->get_proc_addr(ctx->inst, "vkDestroySurfaceKHR");
|
|
vkDestroySurfaceKHR(ctx->inst, ctx->vk_surface, NULL);
|
|
ctx->vk_surface = NULL;
|
|
}
|
|
|
|
av_buffer_unref(&ctx->hw_device_ref);
|
|
pl_vk_inst_destroy(&ctx->placebo_instance);
|
|
|
|
pl_log_destroy(&ctx->vk_log);
|
|
}
|
|
|
|
static const AVClass vulkan_renderer_class = {
|
|
.class_name = "Vulkan Renderer",
|
|
.item_name = av_default_item_name,
|
|
.version = LIBAVUTIL_VERSION_INT,
|
|
};
|
|
|
|
VkRenderer *vk_get_renderer(void)
|
|
{
|
|
RendererContext *ctx = av_mallocz(sizeof(*ctx));
|
|
VkRenderer *renderer;
|
|
|
|
if (!ctx)
|
|
return NULL;
|
|
|
|
renderer = &ctx->api;
|
|
renderer->class = &vulkan_renderer_class;
|
|
renderer->get_hw_dev = get_hw_dev;
|
|
renderer->create = create;
|
|
renderer->display = display;
|
|
renderer->resize = resize;
|
|
renderer->destroy = destroy;
|
|
|
|
return renderer;
|
|
}
|
|
|
|
#else
|
|
|
|
VkRenderer *vk_get_renderer(void)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
#endif
|
|
|
|
int vk_renderer_create(VkRenderer *renderer, SDL_Window *window,
|
|
AVDictionary *opt)
|
|
{
|
|
return renderer->create(renderer, window, opt);
|
|
}
|
|
|
|
int vk_renderer_get_hw_dev(VkRenderer *renderer, AVBufferRef **dev)
|
|
{
|
|
return renderer->get_hw_dev(renderer, dev);
|
|
}
|
|
|
|
int vk_renderer_display(VkRenderer *renderer, AVFrame *frame)
|
|
{
|
|
return renderer->display(renderer, frame);
|
|
}
|
|
|
|
int vk_renderer_resize(VkRenderer *renderer, int width, int height)
|
|
{
|
|
return renderer->resize(renderer, width, height);
|
|
}
|
|
|
|
void vk_renderer_destroy(VkRenderer *renderer)
|
|
{
|
|
renderer->destroy(renderer);
|
|
}
|