From patchwork Tue May 22 02:46:14 2018 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Rostislav Pehlivanov X-Patchwork-Id: 9050 Delivered-To: ffmpegpatchwork@gmail.com Received: by 2002:a02:155:0:0:0:0:0 with SMTP id c82-v6csp988124jad; Mon, 21 May 2018 19:55:01 -0700 (PDT) X-Google-Smtp-Source: AB8JxZoBRE5ECmb4p5FWoS8ihWxpWM8KV6Ibt4b4UgNVlcizMepPz46VtHCRFPJ+MBN6FkwoXO9b X-Received: by 2002:adf:8143:: with SMTP id 61-v6mr3368819wrm.192.1526957700966; Mon, 21 May 2018 19:55:00 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1526957700; cv=none; d=google.com; s=arc-20160816; b=qlD9Nlb8IgdT2X35zctoT/n/7V+6mQTH4EaqeXWC7Rf6rG6MFm2kIrCazZmVndYxo8 OSIHQ5OBTdgMe+467K0FIM/essl12oTjOIJQH0S28tOewOUcSGY961uNtVgLczS1M5I9 nAHmOEM4FSSB8rJRsGPP96EUs8SCWTR3jV8W8iQqbl24vt+XWLTnnonmZU7x4Y7hE/dZ /ZmVC3o8xYy52nS9VMtm0wP+WDu0n2wgzewiQnLSyzPDZlediJcP3K7BqPNLcbGzixSw dJ84dWsS7TcsnjgkhQ4TjfmTbKGhPVl9aCxIQASqgpmuDxZGrUBRd1FB9H3/AqFdYcwh 6byA== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:mime-version:cc:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:references:in-reply-to:message-id:date :to:from:dkim-signature:delivered-to:arc-authentication-results; bh=EGy5YatCf7dt7r5aiQH/bUUibH2hEj1oPdjyBAxQ64Y=; b=RJHFvVq/KGh6k2YOTPgZ16r7eNFozNvx2+464ymSx0M/G4IxlouaOOAYJKYxc4NQ1Q Gc5xYEqJ6NhNmJDUiPs9x15Wjzrn/Rq5RZiMAMDnTM03cnpKsejLpsur0sqtL0qtZU4a SKgIq3Wv+93zhhPNsqkk9czF3fkZ/5Lts9OBfzEsKiUec4nOeg6h2wWJq8tpcn6rdgI+ wTQWmHI3sF75CoA+i9WpAieWCUPjgIu97CKZQ/KIkHlfWltXt+HwBY9I+9j0CYKKxO1Y CxtUtRX1hKwdQ9UEGSOZr79wz+CGbBjR6TCis4IWO9UXhgmog+VDGYlpJ5V0p/XP9aSA x24A== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=qFUj4ez6; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=gmail.com Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id 195-v6si11092342wmx.64.2018.05.21.19.55.00; Mon, 21 May 2018 19:55:00 -0700 (PDT) Received-SPF: pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) client-ip=79.124.17.100; Authentication-Results: mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=qFUj4ez6; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=gmail.com Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id BC17D689E84; Tue, 22 May 2018 05:54:18 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-wm0-f65.google.com (mail-wm0-f65.google.com [74.125.82.65]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id B2092689CE5 for ; Tue, 22 May 2018 05:54:12 +0300 (EEST) Received: by mail-wm0-f65.google.com with SMTP id t11-v6so28479030wmt.0 for ; Mon, 21 May 2018 19:54:53 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=from:to:cc:subject:date:message-id:in-reply-to:references; bh=ycmJrvQj9scgEGMqnAKysUavaXiJvvG0Zwim7KAxMzQ=; b=qFUj4ez6Bnpvy4KCXPTM8WXhL9dqy3jl9EmoTGdKZUnKoXLbgZSieFlIqsgn7mEvA0 2/ntFHwtMShx73VH4mTt73etj7xG0ufp6cDojDAxByfCej3Z9/6X2hF+bmohhl4b4JRq GdI5s/s+StW262f2YdN7H9W/3uOKcukKDJR4rS07ZQl2dcg/+xZGySRQlHKcAnpPPASC PVUdQJob5S+OCVMHYgwgJetSPnWxJkPAujgbYfSHeZ0/ExEK67W1tYawnNWAGVom8kqP O2++4Qmcbu3rUPiIf5R+eiJiRdsGBA0HYvABt+DrwHgA0POsVckKQ89NE7mR9nkhz0+F yNYg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:from:to:cc:subject:date:message-id:in-reply-to :references; bh=ycmJrvQj9scgEGMqnAKysUavaXiJvvG0Zwim7KAxMzQ=; b=JczPnhm89ytfIyVcPQOUZ2UMTNfKztpKcqVMiuzvOffmBZdnWjN0pIkp74a3RSCsSz d+e2mYDvjf6x8dlN0AOG8uo9fe43MOOnh9sn3BV/pufA3kjnPKNW4HPAXG8LLOBasTYU a4FZltayYckYQMERuwiQZTUg7No4yuqzH2IaLoGImb6zK4o3nzjzFH9Sy4J5L2fDvwE8 LwU+LBcqNuef8X9D8+9vhhfwAor/hhADZrU8NGi03ZBbJCBfdM42kgxTGGd7PtoBpzsX zueW/+jQL5u+pCbYDkMWC0h7W9MS6HyF3yUpOA12zMplHwkCHgU/MwWXaFtc0BQaBpsb vbdg== X-Gm-Message-State: ALKqPwcbsGq9FkUp+BsjSqmH/5FMuyia9fkFmbSdClD5fzozBtIoHNDR ieWYq7FYUXZodJ1neTU2ZxzCC782 X-Received: by 2002:a1c:e182:: with SMTP id y124-v6mr712979wmg.57.1526957305971; Mon, 21 May 2018 19:48:25 -0700 (PDT) Received: from moonbase.pars.ee ([2a00:23c4:7c8c:8e00:150:af35:e548:96e6]) by smtp.gmail.com with ESMTPSA id k5-v6sm7736085wrd.13.2018.05.21.19.48.24 (version=TLS1_2 cipher=ECDHE-RSA-AES128-GCM-SHA256 bits=128/128); Mon, 21 May 2018 19:48:25 -0700 (PDT) From: Rostislav Pehlivanov To: ffmpeg-devel@ffmpeg.org Date: Tue, 22 May 2018 03:46:14 +0100 Message-Id: <20180522024616.3091-7-atomnuker@gmail.com> X-Mailer: git-send-email 2.17.0 In-Reply-To: <20180522024616.3091-1-atomnuker@gmail.com> References: <20180522024616.3091-1-atomnuker@gmail.com> Subject: [FFmpeg-devel] [PATCH v3 6/8] lavfi: add a Vulkan chromatic aberration filter X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Rostislav Pehlivanov MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" It tries to do something similar to it with YUV images, but RGB images are done properly. Signed-off-by: Rostislav Pehlivanov --- configure | 1 + libavfilter/Makefile | 1 + libavfilter/allfilters.c | 1 + libavfilter/vf_chromaticaberration_vulkan.c | 342 ++++++++++++++++++++ 4 files changed, 345 insertions(+) create mode 100644 libavfilter/vf_chromaticaberration_vulkan.c diff --git a/configure b/configure index c34a48b2bc..eb81cc1ed5 100755 --- a/configure +++ b/configure @@ -3313,6 +3313,7 @@ azmq_filter_deps="libzmq" blackframe_filter_deps="gpl" boxblur_filter_deps="gpl" bs2b_filter_deps="libbs2b" +chromaticaberration_vulkan_filter_deps="vulkan libshaderc" colormatrix_filter_deps="gpl" convolution_opencl_filter_deps="opencl" convolve_filter_deps="avcodec" diff --git a/libavfilter/Makefile b/libavfilter/Makefile index fd8cf8c13c..976955959c 100644 --- a/libavfilter/Makefile +++ b/libavfilter/Makefile @@ -156,6 +156,7 @@ OBJS-$(CONFIG_BLEND_FILTER) += vf_blend.o framesync.o OBJS-$(CONFIG_BOXBLUR_FILTER) += vf_boxblur.o OBJS-$(CONFIG_BWDIF_FILTER) += vf_bwdif.o OBJS-$(CONFIG_CHROMAKEY_FILTER) += vf_chromakey.o +OBJS-$(CONFIG_CHROMATICABERRATION_VULKAN_FILTER) += vf_chromaticaberration_vulkan.o vulkan.o OBJS-$(CONFIG_CIESCOPE_FILTER) += vf_ciescope.o OBJS-$(CONFIG_CODECVIEW_FILTER) += vf_codecview.o OBJS-$(CONFIG_COLORBALANCE_FILTER) += vf_colorbalance.o diff --git a/libavfilter/allfilters.c b/libavfilter/allfilters.c index c53cb2154e..7be81e4706 100644 --- a/libavfilter/allfilters.c +++ b/libavfilter/allfilters.c @@ -149,6 +149,7 @@ extern AVFilter ff_vf_blend; extern AVFilter ff_vf_boxblur; extern AVFilter ff_vf_bwdif; extern AVFilter ff_vf_chromakey; +extern AVFilter ff_vf_chromaticaberration_vulkan; extern AVFilter ff_vf_ciescope; extern AVFilter ff_vf_codecview; extern AVFilter ff_vf_colorbalance; diff --git a/libavfilter/vf_chromaticaberration_vulkan.c b/libavfilter/vf_chromaticaberration_vulkan.c new file mode 100644 index 0000000000..1d1aeb95a0 --- /dev/null +++ b/libavfilter/vf_chromaticaberration_vulkan.c @@ -0,0 +1,342 @@ +/* + * 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 + */ + +#include "libavutil/opt.h" +#include "vulkan.h" +#include "internal.h" + +typedef struct ChromaticAberrationVulkanContext { + VulkanFilterContext vkctx; + + int initialized; + FFVkExecContext exec; + + /* Shader updators, must be in the main filter struct */ + VkDescriptorImageInfo input_images[3]; + VkDescriptorImageInfo output_images[3]; + + float dist_x; + float dist_y; +} ChromaticAberrationVulkanContext; + +static const char distort_chroma_kernel[] = { + C(0, void distort_rgb(ivec2 size, ivec2 pos) ) + C(0, { ) + C(1, const vec2 p = ((vec2(pos)/vec2(size)) - 0.5f)*2.0f; ) + C(1, const vec2 o = p * (FILTER_DIST - 1.0f); ) + C(0, ) + C(1, vec4 res; ) + C(1, res.r = texture(input_img[0], ((p - o)/2.0f) + 0.5f).r; ) + C(1, res.g = texture(input_img[0], (( p)/2.0f) + 0.5f).g; ) + C(1, res.b = texture(input_img[0], ((p + o)/2.0f) + 0.5f).b; ) + C(1, res.a = texture(input_img[0], (( p)/2.0f) + 0.5f).a; ) + C(1, imageStore(output_img[0], pos, res); ) + C(0, } ) + C(0, ) + C(0, void distort_chroma(int idx, ivec2 size, ivec2 pos) ) + C(0, { ) + C(1, vec2 p = ((vec2(pos)/vec2(size)) - 0.5f)*2.0f; ) + C(1, float d = sqrt(p.x*p.x + p.y*p.y); ) + C(1, p *= d / (d*FILTER_DIST); ) + C(1, vec4 res = texture(input_img[idx], (p/2.0f) + 0.5f); ) + C(1, imageStore(output_img[idx], pos, res); ) + C(0, } ) +}; + +static av_cold int init_filter(AVFilterContext *ctx, AVFrame *in) +{ + int err; + ChromaticAberrationVulkanContext *s = ctx->priv; + + /* Create a sampler */ + const VulkanSampler *sampler = ff_vk_init_sampler(ctx, NULL, 0, + VK_FILTER_LINEAR); + if (!sampler) + return AVERROR_EXTERNAL; + + { /* Create the shader */ + const float dist_x = (s->dist_x / 100.0f) + 1.0f; + const float dist_y = (s->dist_y / 100.0f) + 1.0f; + const int planes = av_pix_fmt_count_planes(s->vkctx.output_format); + + SPIRVShader *shd = ff_vk_init_shader(ctx, "chromaticaberration_compute", + VK_SHADER_STAGE_COMPUTE_BIT); + ff_vk_set_compute_shader_sizes(ctx, shd, (int [3]){ 16, 16, 1 }); + + VulkanDescriptorSetBinding desc_i[2] = { + { + .name = "input_img", + .type = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, + .dimensions = 2, + .elems = planes, + .stages = VK_SHADER_STAGE_COMPUTE_BIT, + .updater = s->input_images, + .samplers = DUP_SAMPLER_ARRAY4(sampler->sampler), + }, + { + .name = "output_img", + .type = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE, + .mem_layout = ff_vk_shader_rep_fmt(s->vkctx.output_format), + .mem_quali = "writeonly", + .dimensions = 2, + .elems = planes, + .stages = VK_SHADER_STAGE_COMPUTE_BIT, + .updater = s->output_images, + }, + }; + + RET(ff_vk_add_descriptor_set(ctx, shd, desc_i, 2, 0)); /* set 0 */ + + GLSLF(0, #define FILTER_DIST vec2(%f, %f) ,dist_x, dist_y); + GLSLD( distort_chroma_kernel ); + GLSLC(0, void main() ); + GLSLC(0, { ); + GLSLC(1, ivec2 pos = ivec2(gl_GlobalInvocationID.xy); ); + GLSLF(1, int planes = %i; ,planes); + GLSLC(1, for (int i = 0; i < planes; i++) { ); + GLSLC(2, ivec2 size = imageSize(output_img[i]); ); + GLSLC(2, if (!IS_WITHIN(pos, size)) ); + GLSLC(3, continue; ); + GLSLC(2, if (planes == 1) { ); + GLSLC(3, distort_rgb(size, pos); ); + GLSLC(2, } else if (i > 0) { ); + GLSLC(3, distort_chroma(i, size, pos); ); + GLSLC(2, } else { ); + GLSLC(3, vec2 npos = vec2(pos)/vec2(size); ); + GLSLC(3, vec4 res = texture(input_img[i], npos); ); + GLSLC(3, imageStore(output_img[i], pos, res); ); + GLSLC(2, } ); + GLSLC(1, } ); + GLSLC(0, } ); + + RET(ff_vk_compile_shader(ctx, shd, "main")); + } + + RET(ff_vk_init_pipeline_layout(ctx)); + + /* Execution context */ + RET(ff_vk_create_exec_ctx(ctx, &s->exec, + s->vkctx.hwctx->queue_family_comp_index)); + + /* The pipeline */ + RET(ff_vk_init_compute_pipeline(ctx)); + + s->initialized = 1; + + return 0; + +fail: + return err; +} + +static int process_frames(AVFilterContext *avctx, AVVkFrame *out, AVVkFrame *in) +{ + int err; + ChromaticAberrationVulkanContext *s = avctx->priv; + int planes = av_pix_fmt_count_planes(s->vkctx.output_format); + + VkCommandBufferBeginInfo cmd_start = { + .sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO, + .flags = VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT, + }; + + VkComponentMapping null_map = { + .r = VK_COMPONENT_SWIZZLE_IDENTITY, + .g = VK_COMPONENT_SWIZZLE_IDENTITY, + .b = VK_COMPONENT_SWIZZLE_IDENTITY, + .a = VK_COMPONENT_SWIZZLE_IDENTITY, + }; + + for (int i = 0; i < planes; i++) { + RET(ff_vk_create_imageview(avctx, &s->input_images[i].imageView, in, + ff_vk_plane_rep_fmt(s->vkctx.input_format, i), + ff_vk_aspect_flags(s->vkctx.input_format, i), + null_map, NULL)); + + RET(ff_vk_create_imageview(avctx, &s->output_images[i].imageView, out, + ff_vk_plane_rep_fmt(s->vkctx.output_format, i), + ff_vk_aspect_flags(s->vkctx.output_format, i), + null_map, NULL)); + + s->input_images[i].imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL; + s->output_images[i].imageLayout = VK_IMAGE_LAYOUT_GENERAL; + } + + ff_vk_update_descriptor_set(avctx, 0); + + vkBeginCommandBuffer(s->exec.buf, &cmd_start); + + { + VkImageMemoryBarrier bar[2] = { + { + .sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER, + .srcAccessMask = 0, + .dstAccessMask = VK_ACCESS_SHADER_READ_BIT, + .oldLayout = in->layout, + .newLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL, + .srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED, + .dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED, + .image = in->img, + .subresourceRange.aspectMask = ff_vk_aspect_flags(s->vkctx.input_format, -1), + .subresourceRange.levelCount = 1, + .subresourceRange.layerCount = 1, + }, + { + .sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER, + .srcAccessMask = 0, + .dstAccessMask = VK_ACCESS_SHADER_WRITE_BIT, + .oldLayout = out->layout, + .newLayout = VK_IMAGE_LAYOUT_GENERAL, + .srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED, + .dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED, + .image = out->img, + .subresourceRange.aspectMask = ff_vk_aspect_flags(s->vkctx.output_format, -1), + .subresourceRange.levelCount = 1, + .subresourceRange.layerCount = 1, + }, + }; + + vkCmdPipelineBarrier(s->exec.buf, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT, + VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT, 0, + 0, NULL, 0, NULL, 2, bar); + + in->layout = bar[0].newLayout; + in->access = bar[0].dstAccessMask; + + out->layout = bar[1].newLayout; + out->access = bar[1].dstAccessMask; + } + + vkCmdBindPipeline(s->exec.buf, VK_PIPELINE_BIND_POINT_COMPUTE, s->vkctx.pipeline); + vkCmdBindDescriptorSets(s->exec.buf, VK_PIPELINE_BIND_POINT_COMPUTE, s->vkctx.pipeline_layout, 0, s->vkctx.descriptor_sets_num, s->vkctx.desc_set, 0, 0); + vkCmdDispatch(s->exec.buf, + FFALIGN(s->vkctx.output_width, s->vkctx.shaders[0].local_size[0])/s->vkctx.shaders[0].local_size[0], + FFALIGN(s->vkctx.output_height, s->vkctx.shaders[0].local_size[1])/s->vkctx.shaders[0].local_size[1], 1); + + vkEndCommandBuffer(s->exec.buf); + + VkSubmitInfo s_info = { + .sType = VK_STRUCTURE_TYPE_SUBMIT_INFO, + .commandBufferCount = 1, + .pCommandBuffers = &s->exec.buf, + }; + + VkResult ret = vkQueueSubmit(s->exec.queue, 1, &s_info, s->exec.fence); + if (ret != VK_SUCCESS) { + av_log(avctx, AV_LOG_ERROR, "Unable to submit command buffer: %s\n", + ff_vk_ret2str(ret)); + return AVERROR_EXTERNAL; + } else { + vkWaitForFences(s->vkctx.hwctx->act_dev, 1, &s->exec.fence, VK_TRUE, UINT64_MAX); + vkResetFences(s->vkctx.hwctx->act_dev, 1, &s->exec.fence); + } + +fail: + + for (int i = 0; i < planes; i++) { + ff_vk_destroy_imageview(avctx, s->input_images[i].imageView); + ff_vk_destroy_imageview(avctx, s->output_images[i].imageView); + } + + return err; +} + +static int chromaticaberration_vulkan_filter_frame(AVFilterLink *link, AVFrame *in) +{ + int err; + AVFilterContext *ctx = link->dst; + ChromaticAberrationVulkanContext *s = ctx->priv; + AVFilterLink *outlink = ctx->outputs[0]; + + AVFrame *out = ff_get_video_buffer(outlink, outlink->w, outlink->h); + if (!out) { + err = AVERROR(ENOMEM); + goto fail; + } + + if (!s->initialized) + RET(init_filter(ctx, in)); + + RET(process_frames(ctx, (AVVkFrame *)out->data[0], + (AVVkFrame *) in->data[0])); + + err = av_frame_copy_props(out, in); + if (err < 0) + goto fail; + + av_frame_free(&in); + + return ff_filter_frame(outlink, out); + +fail: + av_frame_free(&in); + av_frame_free(&out); + return err; +} + +static void chromaticaberration_vulkan_uninit(AVFilterContext *avctx) +{ + ChromaticAberrationVulkanContext *s = avctx->priv; + + ff_vk_free_exec_ctx(avctx, &s->exec); + ff_vk_filter_uninit(avctx); + + s->initialized = 0; +} + +#define OFFSET(x) offsetof(ChromaticAberrationVulkanContext, x) +#define FLAGS (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM) +static const AVOption chromaticaberration_vulkan_options[] = { + { "dist_x", "Set horizontal distortion amount", OFFSET(dist_x), AV_OPT_TYPE_FLOAT, {.dbl = 0.0f}, 0.0f, 10.0f, .flags = FLAGS }, + { "dist_y", "Set vertical distortion amount", OFFSET(dist_y), AV_OPT_TYPE_FLOAT, {.dbl = 0.0f}, 0.0f, 10.0f, .flags = FLAGS }, + { NULL }, +}; + +AVFILTER_DEFINE_CLASS(chromaticaberration_vulkan); + +static const AVFilterPad chromaticaberration_vulkan_inputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_VIDEO, + .filter_frame = &chromaticaberration_vulkan_filter_frame, + .config_props = &ff_vk_filter_config_input, + }, + { NULL } +}; + +static const AVFilterPad chromaticaberration_vulkan_outputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_VIDEO, + .config_props = &ff_vk_filter_config_output, + }, + { NULL } +}; + +AVFilter ff_vf_chromaticaberration_vulkan = { + .name = "chromaticaberration_vulkan", + .description = NULL_IF_CONFIG_SMALL("Offset chroma of input video"), + .priv_size = sizeof(ChromaticAberrationVulkanContext), + .init = &ff_vk_filter_init, + .uninit = &chromaticaberration_vulkan_uninit, + .query_formats = &ff_vk_filter_query_formats, + .inputs = chromaticaberration_vulkan_inputs, + .outputs = chromaticaberration_vulkan_outputs, + .priv_class = &chromaticaberration_vulkan_class, + .flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE, +};