mirror of https://github.com/FFmpeg/FFmpeg.git
This allows for the loader to be shared with libavcodec and libavfilter.pull/374/head
Lynne
3 years ago
parent
1ffb59c056
commit
d05a18cdc7
3 changed files with 357 additions and 226 deletions
@ -0,0 +1,162 @@ |
||||
/*
|
||||
* 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 |
||||
*/ |
||||
|
||||
#ifndef AVUTIL_VULKAN_FUNCTIONS_H |
||||
#define AVUTIL_VULKAN_FUNCTIONS_H |
||||
|
||||
/* An enum of bitflags for every optional extension we need */ |
||||
typedef enum FFVulkanExtensions { |
||||
FF_VK_EXT_EXTERNAL_DMABUF_MEMORY = 1ULL << 0, /* VK_EXT_external_memory_dma_buf */ |
||||
FF_VK_EXT_DRM_MODIFIER_FLAGS = 1ULL << 1, /* VK_EXT_image_drm_format_modifier */ |
||||
FF_VK_EXT_EXTERNAL_FD_MEMORY = 1ULL << 2, /* VK_KHR_external_memory_fd */ |
||||
FF_VK_EXT_EXTERNAL_FD_SEM = 1ULL << 3, /* VK_KHR_external_semaphore_fd */ |
||||
FF_VK_EXT_EXTERNAL_HOST_MEMORY = 1ULL << 4, /* VK_EXT_external_memory_host */ |
||||
FF_VK_EXT_DEBUG_UTILS = 1ULL << 5, /* VK_EXT_debug_utils */ |
||||
|
||||
FF_VK_EXT_NO_FLAG = 1ULL << 63, |
||||
} FFVulkanExtensions; |
||||
|
||||
/* Macro containing every function that we utilize in our codebase */ |
||||
#define FN_LIST(MACRO) \ |
||||
/* Instance */ \
|
||||
MACRO(0, 0, FF_VK_EXT_NO_FLAG, EnumerateInstanceExtensionProperties) \
|
||||
MACRO(0, 0, FF_VK_EXT_NO_FLAG, CreateInstance) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, DestroyInstance) \
|
||||
\
|
||||
/* Debug */ \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, CreateDebugUtilsMessengerEXT) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, DestroyDebugUtilsMessengerEXT) \
|
||||
\
|
||||
/* Device */ \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetDeviceProcAddr) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, CreateDevice) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetPhysicalDeviceFeatures2) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetPhysicalDeviceProperties) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, DestroyDevice) \
|
||||
\
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, EnumeratePhysicalDevices) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, EnumerateDeviceExtensionProperties) \
|
||||
\
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetPhysicalDeviceProperties2) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetPhysicalDeviceMemoryProperties) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetPhysicalDeviceFormatProperties2) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetPhysicalDeviceImageFormatProperties2) \
|
||||
MACRO(1, 0, FF_VK_EXT_NO_FLAG, GetPhysicalDeviceQueueFamilyProperties) \
|
||||
\
|
||||
/* Command pool */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateCommandPool) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyCommandPool) \
|
||||
\
|
||||
/* Command buffer */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, AllocateCommandBuffers) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, BeginCommandBuffer) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, EndCommandBuffer) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, FreeCommandBuffers) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CmdDispatch) \
|
||||
\
|
||||
/* Queue */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, GetDeviceQueue) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, QueueSubmit) \
|
||||
\
|
||||
/* Fences */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateFence) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, WaitForFences) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, ResetFences) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyFence) \
|
||||
\
|
||||
/* Semaphores */ \
|
||||
MACRO(1, 1, FF_VK_EXT_EXTERNAL_FD_SEM, GetSemaphoreFdKHR) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateSemaphore) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, WaitSemaphores) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroySemaphore) \
|
||||
\
|
||||
/* Memory */ \
|
||||
MACRO(1, 1, FF_VK_EXT_EXTERNAL_FD_MEMORY, GetMemoryFdKHR) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, GetMemoryFdPropertiesKHR) \
|
||||
MACRO(1, 1, FF_VK_EXT_EXTERNAL_HOST_MEMORY, GetMemoryHostPointerPropertiesEXT) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, AllocateMemory) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, MapMemory) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, FlushMappedMemoryRanges) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, InvalidateMappedMemoryRanges) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, UnmapMemory) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, FreeMemory) \
|
||||
\
|
||||
/* Commands */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CmdBindDescriptorSets) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CmdPushConstants) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CmdBindPipeline) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CmdPipelineBarrier) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CmdCopyBufferToImage) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CmdCopyImageToBuffer) \
|
||||
\
|
||||
/* Buffer */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, GetBufferMemoryRequirements2) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateBuffer) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, BindBufferMemory) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyBuffer) \
|
||||
\
|
||||
/* Image */ \
|
||||
MACRO(1, 1, FF_VK_EXT_DRM_MODIFIER_FLAGS, GetImageDrmFormatModifierPropertiesEXT) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, GetImageMemoryRequirements2) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateImage) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, BindImageMemory2) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, GetImageSubresourceLayout) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyImage) \
|
||||
\
|
||||
/* ImageView */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateImageView) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyImageView) \
|
||||
\
|
||||
/* DescriptorSet */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateDescriptorSetLayout) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, AllocateDescriptorSets) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateDescriptorPool) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyDescriptorPool) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyDescriptorSetLayout) \
|
||||
\
|
||||
/* DescriptorUpdateTemplate */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, UpdateDescriptorSetWithTemplate) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateDescriptorUpdateTemplate) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyDescriptorUpdateTemplate) \
|
||||
\
|
||||
/* Pipeline */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreatePipelineLayout) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyPipelineLayout) \
|
||||
\
|
||||
/* PipelineLayout */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateComputePipelines) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyPipeline) \
|
||||
\
|
||||
/* Sampler */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateSampler) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroySampler) \
|
||||
\
|
||||
/* Shaders */ \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, CreateShaderModule) \
|
||||
MACRO(1, 1, FF_VK_EXT_NO_FLAG, DestroyShaderModule) |
||||
|
||||
/* Macro to turn a function name into a definition */ |
||||
#define PFN_DEF(req_inst, req_dev, ext_flag, name) \ |
||||
PFN_vk##name name; |
||||
|
||||
/* Structure with the definition of all listed functions */ |
||||
typedef struct FFVulkanFunctions { |
||||
FN_LIST(PFN_DEF) |
||||
} FFVulkanFunctions; |
||||
|
||||
#endif /* AVUTIL_VULKAN_FUNCTIONS_H */ |
||||
@ -0,0 +1,128 @@ |
||||
/*
|
||||
* 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 |
||||
*/ |
||||
|
||||
#ifndef AVUTIL_VULKAN_LOADER_H |
||||
#define AVUTIL_VULKAN_LOADER_H |
||||
|
||||
#include "vulkan_functions.h" |
||||
|
||||
/* Macro to turn a function name into a loader struct */ |
||||
#define PFN_LOAD_INFO(req_inst, req_dev, ext_flag, name) \ |
||||
{ \
|
||||
req_inst, \
|
||||
req_dev, \
|
||||
offsetof(FFVulkanFunctions, name), \
|
||||
ext_flag, \
|
||||
{ "vk"#name, "vk"#name"EXT", "vk"#name"KHR" } \
|
||||
}, |
||||
|
||||
static inline uint64_t ff_vk_extensions_to_mask(const char * const *extensions, |
||||
int nb_extensions) |
||||
{ |
||||
static const struct ExtensionMap { |
||||
const char *name; |
||||
FFVulkanExtensions flag; |
||||
} extension_map[] = { |
||||
{ VK_EXT_EXTERNAL_MEMORY_DMA_BUF_EXTENSION_NAME, FF_VK_EXT_EXTERNAL_DMABUF_MEMORY }, |
||||
{ VK_EXT_IMAGE_DRM_FORMAT_MODIFIER_EXTENSION_NAME, FF_VK_EXT_DRM_MODIFIER_FLAGS }, |
||||
{ VK_KHR_EXTERNAL_MEMORY_FD_EXTENSION_NAME, FF_VK_EXT_EXTERNAL_FD_MEMORY }, |
||||
{ VK_KHR_EXTERNAL_SEMAPHORE_FD_EXTENSION_NAME, FF_VK_EXT_EXTERNAL_FD_SEM }, |
||||
{ VK_EXT_EXTERNAL_MEMORY_HOST_EXTENSION_NAME, FF_VK_EXT_EXTERNAL_HOST_MEMORY }, |
||||
{ VK_EXT_DEBUG_UTILS_EXTENSION_NAME, FF_VK_EXT_DEBUG_UTILS } |
||||
}; |
||||
|
||||
FFVulkanExtensions mask = 0x0; |
||||
|
||||
for (int i = 0; i < nb_extensions; i++) { |
||||
for (int j = 0; j < FF_ARRAY_ELEMS(extension_map); j++) { |
||||
if (!strcmp(extensions[i], extension_map[j].name)) { |
||||
mask |= extension_map[j].flag; |
||||
continue; |
||||
} |
||||
} |
||||
} |
||||
|
||||
return mask; |
||||
} |
||||
|
||||
/**
|
||||
* Function loader. |
||||
* Vulkan function from scratch loading happens in 3 stages - the first one |
||||
* is before any initialization has happened, and you have neither an instance |
||||
* structure nor a device structure. At this stage, you can only get the bare |
||||
* minimals to initialize an instance. |
||||
* The second stage is when you have an instance. At this stage, you can |
||||
* initialize a VkDevice, and have an idea of what extensions each device |
||||
* supports. |
||||
* Finally, in the third stage, you can proceed and load all core functions, |
||||
* plus you can be sure that any extensions you've enabled during device |
||||
* initialization will be available. |
||||
*/ |
||||
static inline int ff_vk_load_functions(AVHWDeviceContext *ctx, |
||||
FFVulkanFunctions *vk, |
||||
uint64_t extensions_mask, |
||||
int has_inst, int has_dev) |
||||
{ |
||||
AVVulkanDeviceContext *hwctx = ctx->hwctx; |
||||
|
||||
static const struct FunctionLoadInfo { |
||||
int req_inst; |
||||
int req_dev; |
||||
size_t struct_offset; |
||||
FFVulkanExtensions ext_flag; |
||||
const char *names[3]; |
||||
} vk_load_info[] = { |
||||
FN_LIST(PFN_LOAD_INFO) |
||||
}; |
||||
|
||||
for (int i = 0; i < FF_ARRAY_ELEMS(vk_load_info); i++) { |
||||
const struct FunctionLoadInfo *load = &vk_load_info[i]; |
||||
PFN_vkVoidFunction fn; |
||||
|
||||
if (load->req_dev && !has_dev) |
||||
continue; |
||||
if (load->req_inst && !has_inst) |
||||
continue; |
||||
|
||||
for (int j = 0; j < FF_ARRAY_ELEMS(load->names); j++) { |
||||
const char *name = load->names[j]; |
||||
|
||||
if (load->req_dev) |
||||
fn = vk->GetDeviceProcAddr(hwctx->act_dev, name); |
||||
else if (load->req_inst) |
||||
fn = hwctx->get_proc_addr(hwctx->inst, name); |
||||
else |
||||
fn = hwctx->get_proc_addr(NULL, name); |
||||
|
||||
if (fn) |
||||
break; |
||||
} |
||||
|
||||
if (!fn && ((extensions_mask &~ FF_VK_EXT_NO_FLAG) & load->ext_flag)) { |
||||
av_log(ctx, AV_LOG_ERROR, "Loader error, function \"%s\" indicated" |
||||
"as supported, but got NULL function pointer!\n", load->names[0]); |
||||
return AVERROR_EXTERNAL; |
||||
} |
||||
|
||||
*(PFN_vkVoidFunction *)((uint8_t *)vk + load->struct_offset) = fn; |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
#endif /* AVUTIL_VULKAN_LOADER_H */ |
||||
Loading…
Reference in new issue