From 6ba01e8d2ec9570699d59e67b068f5abc8783ed8 Mon Sep 17 00:00:00 2001 From: Wu Jianhua Date: Wed, 23 Feb 2022 01:03:19 +0800 Subject: [PATCH 03/10] avfilter/vf_blend_vulkan: add subtract blend mode Use the commands below to test: (href: https://trac.ffmpeg.org/wiki/Blend) I. make an image for test ffmpeg -f lavfi -i color=s=256x256,geq=r='H-1-Y':g='H-1-Y':b='H-1-Y' -frames 1 \ -y -pix_fmt yuv420p test.jpg II. blend in sw ffmpeg -i test.jpg -vf "split[a][b];[b]transpose[b];[a][b]blend=all_mode=subtract,\ pseudocolor=preset=turbo" -y subtract_sw.jpg III. blend in vulkan ffmpeg -init_hw_device vulkan -i test.jpg -vf "split[a][b];[b]transpose[b];\ [a]hwupload[a];[b]hwupload[b];[a][b]blend_vulkan=all_mode=subtract,hwdownload,\ format=yuv420p,pseudocolor=preset=turbo" -y subtract_vulkan.jpg Signed-off-by: Wu Jianhua --- libavfilter/vf_blend_vulkan.c | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/libavfilter/vf_blend_vulkan.c b/libavfilter/vf_blend_vulkan.c index 2fc6e9f0c5..670df47499 100644 --- a/libavfilter/vf_blend_vulkan.c +++ b/libavfilter/vf_blend_vulkan.c @@ -70,15 +70,21 @@ static const char blend_##MODE##_func[] = { \ #define MAX vec4(1.0) #define HALF vec4(0.5) +#define MIN vec4(0.0) + #ifdef min #undef min #endif +#ifdef max +#undef max +#endif #define FN(EXPR) A + ((EXPR) - A) * opacity DEFINE_BLEND_MODE(NORMAL, A * opacity + B * (1.0f - opacity)) DEFINE_BLEND_MODE(ADDITION, FN(min(MAX, A + B))) DEFINE_BLEND_MODE(AVERAGE, FN((A + B) * HALF)) +DEFINE_BLEND_MODE(SUBTRACT, FN(max(MIN, A - B))) DEFINE_BLEND_MODE(MULTIPLY, FN(1.0f * A * B / 1.0f)) static inline void init_blend_func(FilterParamsVulkan *param) @@ -91,6 +97,7 @@ static inline void init_blend_func(FilterParamsVulkan *param) switch (param->mode) { CASE(ADDITION) CASE(AVERAGE) + CASE(SUBTRACT) CASE(NORMAL) CASE(MULTIPLY) default: param->blend = NULL; break; @@ -464,6 +471,7 @@ static const AVOption blend_vulkan_options[] = { { "all_mode", "set blend mode for all components", OFFSET(all_mode), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, BLEND_NB - 1, FLAGS, "mode" }, { "addition", "", 0, AV_OPT_TYPE_CONST, { .i64 = BLEND_ADDITION }, 0, 0, FLAGS, "mode" }, { "average", "", 0, AV_OPT_TYPE_CONST, { .i64 = BLEND_AVERAGE }, 0, 0, FLAGS, "mode" }, + { "subtract", "", 0, AV_OPT_TYPE_CONST, { .i64 = BLEND_SUBTRACT }, 0, 0, FLAGS, "mode" }, { "normal", "", 0, AV_OPT_TYPE_CONST, { .i64 = BLEND_NORMAL }, 0, 0, FLAGS, "mode" }, { "multiply", "", 0, AV_OPT_TYPE_CONST, { .i64 = BLEND_MULTIPLY }, 0, 0, FLAGS, "mode" }, -- 2.25.1