[FFmpeg-devel] [PATCH 1/3] avcodec/vulkan_video: Don't use sparse table

[Andreas Rheinhardt]

ff_vk_codec_map currently is an array indexed by AVCodecID;
it has AV_CODEC_ID_FIRST_AUDIO (= 65536) entries, but uses
only three of them; only 24B of 1MiB were actually used

This commit fixes this by adding an AVCodecID field to the table
and making it non-sparse.

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt at outlook.com>
---
 libavcodec/vulkan_decode.c | 17 ++++++++++++++---
 libavcodec/vulkan_video.c  | 11 +++++++----
 libavcodec/vulkan_video.h  |  6 ++----
 3 files changed, 23 insertions(+), 11 deletions(-)

diff --git a/libavcodec/vulkan_decode.c b/libavcodec/vulkan_decode.c
index fdbcbb450a..5def908a21 100644
--- a/libavcodec/vulkan_decode.c
+++ b/libavcodec/vulkan_decode.c
@@ -20,6 +20,7 @@
 #include "vulkan_video.h"
 #include "vulkan_decode.h"
 #include "config_components.h"
+#include "libavutil/avassert.h"
 
 #if CONFIG_H264_VULKAN_HWACCEL
 extern const VkExtensionProperties ff_vk_dec_h264_ext;
@@ -43,6 +44,15 @@ static const VkExtensionProperties *dec_ext[] = {
 #endif
 };
 
+static const FFVkCodecMap *get_codecmap(enum AVCodecID codec_id)
+{
+    for (size_t i = 0; i < FF_ARRAY_ELEMS(ff_vk_codec_map); i++)
+        if (ff_vk_codec_map[i].codec_id == codec_id)
+            return &ff_vk_codec_map[i];
+    av_assert1(!"unreachable");
+    return NULL;
+}
+
 static const VkVideoProfileInfoKHR *get_video_profile(FFVulkanDecodeShared *ctx, enum AVCodecID codec_id)
 {
     const VkVideoProfileListInfoKHR *profile_list;
@@ -737,7 +747,7 @@ static int vulkan_decode_get_profile(AVCodecContext *avctx, AVBufferRef *frames_
 {
     VkResult ret;
     int max_level, base_profile, cur_profile;
-    const struct FFVkCodecMap *vk_codec = &ff_vk_codec_map[avctx->codec_id];
+    const FFVkCodecMap *vk_codec = get_codecmap(avctx->codec_id);
     AVHWFramesContext *frames = (AVHWFramesContext *)frames_ref->data;
     AVHWDeviceContext *device = (AVHWDeviceContext *)frames->device_ref->data;
     AVVulkanDeviceContext *hwctx = device->hwctx;
@@ -1111,6 +1121,7 @@ int ff_vk_decode_init(AVCodecContext *avctx)
     FFVulkanContext *s;
     FFVulkanFunctions *vk;
     const VkVideoProfileInfoKHR *profile;
+    const FFVkCodecMap *vk_codec;
 
     VkVideoDecodeH264SessionParametersCreateInfoKHR h264_params = {
         .sType = VK_STRUCTURE_TYPE_VIDEO_DECODE_H264_SESSION_PARAMETERS_CREATE_INFO_KHR,
@@ -1167,9 +1178,9 @@ int ff_vk_decode_init(AVCodecContext *avctx)
     /* Create queue context */
     qf = ff_vk_qf_init(s, &ctx->qf, VK_QUEUE_VIDEO_DECODE_BIT_KHR);
 
+    vk_codec = get_codecmap(avctx->codec_id);
     /* Check for support */
-    if (!(s->video_props[qf].videoCodecOperations &
-          ff_vk_codec_map[avctx->codec_id].decode_op)) {
+    if (!(s->video_props[qf].videoCodecOperations & vk_codec->decode_op)) {
         av_log(avctx, AV_LOG_ERROR, "Decoding %s not supported on the given "
                "queue family %i!\n", avcodec_get_name(avctx->codec_id), qf);
         return AVERROR(EINVAL);
diff --git a/libavcodec/vulkan_video.c b/libavcodec/vulkan_video.c
index fb20315db4..a87df52871 100644
--- a/libavcodec/vulkan_video.c
+++ b/libavcodec/vulkan_video.c
@@ -20,20 +20,23 @@
 
 #include "vulkan_video.h"
 
-const FFVkCodecMap ff_vk_codec_map[AV_CODEC_ID_FIRST_AUDIO] = {
-    [AV_CODEC_ID_H264] = {
+const FFVkCodecMap ff_vk_codec_map[3] = {
+    {
+        .codec_id = AV_CODEC_ID_H264,
                            0,
                            0,
                            FF_VK_EXT_VIDEO_DECODE_H264,
                            VK_VIDEO_CODEC_OPERATION_DECODE_H264_BIT_KHR,
     },
-    [AV_CODEC_ID_HEVC] = {
+    {
+        .codec_id = AV_CODEC_ID_HEVC,
                            0,
                            0,
                            FF_VK_EXT_VIDEO_DECODE_H265,
                            VK_VIDEO_CODEC_OPERATION_DECODE_H265_BIT_KHR
     },
-    [AV_CODEC_ID_AV1] = {
+    {
+        .codec_id = AV_CODEC_ID_AV1,
                            0,
                            0,
                            FF_VK_EXT_VIDEO_DECODE_AV1,
diff --git a/libavcodec/vulkan_video.h b/libavcodec/vulkan_video.h
index 51f44dd543..b06e369abd 100644
--- a/libavcodec/vulkan_video.h
+++ b/libavcodec/vulkan_video.h
@@ -32,6 +32,7 @@
 #define CODEC_VER(ver) CODEC_VER_MAJ(ver), CODEC_VER_MIN(ver), CODEC_VER_PAT(ver)
 
 typedef struct FFVkCodecMap {
+    enum AVCodecID                   codec_id;
     FFVulkanExtensions               encode_extension;
     VkVideoCodecOperationFlagBitsKHR encode_op;
     FFVulkanExtensions               decode_extension;
@@ -46,10 +47,7 @@ typedef struct FFVkVideoSession {
     AVBufferPool *buf_pool;
 } FFVkVideoCommon;
 
-/**
- * Index is codec_id.
- */
-extern const FFVkCodecMap ff_vk_codec_map[AV_CODEC_ID_FIRST_AUDIO];
+extern const FFVkCodecMap ff_vk_codec_map[3];
 
 /**
  * Get pixfmt from a Vulkan format.
-- 
2.40.1