aboutsummaryrefslogtreecommitdiffstats
path: root/libavcodec/ffv1enc_vulkan.c
diff options
context:
space:
mode:
authorLynne <dev@lynne.ee>2025-04-29 10:01:21 +0200
committerLynne <dev@lynne.ee>2025-05-01 09:34:44 +0200
commit707c04fe0613ec9daa91361b889d5f9eedf93c5c (patch)
tree53fafda2b48cef9c06a0e2153d782ac515e8e708 /libavcodec/ffv1enc_vulkan.c
parent33a4d3610162758ddb0390a93ea10c4e8fea7e9d (diff)
downloadffmpeg-707c04fe0613ec9daa91361b889d5f9eedf93c5c.tar.gz
ffv1enc_vulkan: support 8 and 16-bit 2-plane YUV formats
This adds support for all 8-bit and 16-bit 2-plane formats. P010 and others require more work as the data's LSB-padded.
Diffstat (limited to 'libavcodec/ffv1enc_vulkan.c')
-rw-r--r--libavcodec/ffv1enc_vulkan.c8
1 files changed, 6 insertions, 2 deletions
diff --git a/libavcodec/ffv1enc_vulkan.c b/libavcodec/ffv1enc_vulkan.c
index 688c14fb81..42a98a5efa 100644
--- a/libavcodec/ffv1enc_vulkan.c
+++ b/libavcodec/ffv1enc_vulkan.c
@@ -141,6 +141,7 @@ typedef struct FFv1VkParameters {
uint8_t micro_version;
uint8_t force_pcm;
uint8_t key_frame;
+ uint8_t components;
uint8_t planes;
uint8_t codec_planes;
uint8_t transparency;
@@ -149,7 +150,7 @@ typedef struct FFv1VkParameters {
uint8_t ec;
uint8_t ppi;
uint8_t chunks;
- uint8_t padding[2];
+ uint8_t padding[1];
} FFv1VkParameters;
static void add_push_data(FFVulkanShader *shd)
@@ -173,6 +174,7 @@ static void add_push_data(FFVulkanShader *shd)
GLSLC(1, uint8_t micro_version; );
GLSLC(1, uint8_t force_pcm; );
GLSLC(1, uint8_t key_frame; );
+ GLSLC(1, uint8_t components; );
GLSLC(1, uint8_t planes; );
GLSLC(1, uint8_t codec_planes; );
GLSLC(1, uint8_t transparency; );
@@ -181,7 +183,7 @@ static void add_push_data(FFVulkanShader *shd)
GLSLC(1, uint8_t ec; );
GLSLC(1, uint8_t ppi; );
GLSLC(1, uint8_t chunks; );
- GLSLC(1, uint8_t padding[2]; );
+ GLSLC(1, uint8_t padding[1]; );
GLSLC(0, }; );
ff_vk_shader_add_push_const(shd, 0, sizeof(FFv1VkParameters),
VK_SHADER_STAGE_COMPUTE_BIT);
@@ -326,6 +328,7 @@ static int vulkan_encode_ffv1_submit_frame(AVCodecContext *avctx,
int has_inter = avctx->gop_size > 1;
uint32_t context_count = f->context_count[f->context_model];
+ const AVPixFmtDescriptor *fmt_desc = av_pix_fmt_desc_get(avctx->sw_pix_fmt);
VkImageView in_views[AV_NUM_DATA_POINTERS];
VkImageView intermediate_views[AV_NUM_DATA_POINTERS];
@@ -498,6 +501,7 @@ static int vulkan_encode_ffv1_submit_frame(AVCodecContext *avctx,
.micro_version = f->micro_version,
.force_pcm = fv->force_pcm,
.key_frame = f->key_frame,
+ .components = fmt_desc->nb_components,
.planes = av_pix_fmt_count_planes(avctx->sw_pix_fmt),
.codec_planes = f->plane_count,
.transparency = f->transparency,