From patchwork Fri Apr 20 04:30:08 2018 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Rostislav Pehlivanov X-Patchwork-Id: 8523 Delivered-To: ffmpegpatchwork@gmail.com Received: by 2002:a02:155:0:0:0:0:0 with SMTP id c82-v6csp228536jad; Thu, 19 Apr 2018 21:31:23 -0700 (PDT) X-Google-Smtp-Source: AIpwx4+RbZkke6sWssJorShfQRHQwFD5nUSLOWCon1in1Rpwm0mYJVLayoXjeJOMI6YRb8WuRmnm X-Received: by 10.28.32.132 with SMTP id g126mr841590wmg.52.1524198683685; Thu, 19 Apr 2018 21:31:23 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1524198683; cv=none; d=google.com; s=arc-20160816; b=QDV096IcCPnehMEm4kLS9zV7WsveWPTDOUNSA9xmjRTXDYXxhkrOE81ukR6kqaqS3+ YCvzndPqYXRHXZFGXqVZ2oOduqvPnCFs9pf15OeWiGsFxhSEjVT0y7sDjv/rln231KN2 mYc/oX90t8SzjBcnqeKTpjMBq6ZWsj7yuOZ0RA9YeKYfVsJCvMUfAxmgC7np7hHk447R h8pvFmQ6Bm7cjssQNAe6rTm75szCPkrcMZXxjIkfXLO2LcY0e2uERWzFEeVtpTXK+JFj 9mAYA8KgBnNtfi8myVtHXKwj+n8o/eERW0szLKu/kVobtQujgt6v6NJHzKoC0KIGn/qp sPFg== 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=hJfyWLkx29Niq4VPZMkOMZ1EtTe9tHhdVMoVz4eNOvc=; b=yHo0eWiVKMJ6gUoREwOTYU+0DFACt74v6v8RN6QISRA8gosBDRKs4z8UMUXrrIBTNi p5DbqmgHdm7Md7HZk77no6W6Yd+G0Qsp9WUARnfDFV2VMQR5LVa4g3eRdJvoeV5dUQDS 2CpKQC839cZmGZvWnQXYIg6eEie49Pkb2qqn+E+2DSjalSqmuoySyTJpwWItLNhOjAgk vBiFSM0ktgVbr5FOEEW5f8rnkJPpKzYYIKSvVw8d0aV+cfoIjvK9VWdfYcKG52FAESbY 5AYSkUMcVxiZPbDoSmoBjJOeFBNa+3H5Gah05Y00DJt6oBUzCovvMtkjivv5vQbOQE5x JIjg== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=ufQ5IjxR; 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 g7-v6si3784920wrb.33.2018.04.19.21.31.23; Thu, 19 Apr 2018 21:31:23 -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=ufQ5IjxR; 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 9708368A1A3; Fri, 20 Apr 2018 07:30:31 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-wr0-f194.google.com (mail-wr0-f194.google.com [209.85.128.194]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 3BD5C68A16E for ; Fri, 20 Apr 2018 07:30:23 +0300 (EEST) Received: by mail-wr0-f194.google.com with SMTP id w3-v6so19297759wrg.2 for ; Thu, 19 Apr 2018 21:30:51 -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=yFn+EAs6en2K+p3BPwP2NxuVfVmEvOLGFleYf6PyZAc=; b=ufQ5IjxRpmjWDdXy2z/mYvdsSXh8odQx6TWoifAzGRbJM0v9JdzdH6aUqrSF0nFhTx C+pMgnowKMieYdsUsLI7YcK/TeoJiX1bh5cwkBLd8T4RWzwI+kq4i+sPl0sY1FY0hvwB Owi0Z6S14aUWa/ZhkqGNxFdwBEOLNYg2RtqSk3wvhuBIJiWEfL3mTrQyki2ekD0A4i1s tzl81Z0KvvsV7BiqTYqcbMFlygplxCflsYlF6ueVsp/oM+DBQEu8Vyy3tB6vnFQ24jXx ixbNONqgEISRqPPa7pmmh7el86A8ZKSPrM8yE/9icc1mEEFn/d4L2NTAwSEZy6s46Wjd uPAQ== 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=yFn+EAs6en2K+p3BPwP2NxuVfVmEvOLGFleYf6PyZAc=; b=UiOcAzmSgx2qx3YiSOiQN0SAbEcUH3Hyh7R5Am/G9Dy4NsPfDbBQ7bA9nCUMIZTX2m 8JQlVuPYzLeCI7tANcp0UAdk2qsaL6Oift/NmYbQHp1np5FTFRTqSrs7mUTQXiwDhEst 6lVUKRSFoh2ZZDOsGvYTEKA3GiK4U1pIHCQLOhhpfJ+T055auFrq/z+SZ33pVlENCecM oBbt/PR6MgWHBSBtPf1uhReCcw9U0Im73R9c2j/+pyg2dwTQbuCXL8tovHJVV6Rc8Sax Qkd+HANPjyd0FHdNLwHEwt6C2R+aoZGUwx/m6lnsGG2b6qFF1eAmvITui3XSt0p3KfWY 7KiA== X-Gm-Message-State: ALQs6tDMuPNkGvCznAv3Kd2hmrVyqjp3rM/8IwSYQkq4dzsF1hLwYLw+ S4rblIIwObrSzx4bQavpMJM24F6L X-Received: by 2002:adf:8822:: with SMTP id d31-v6mr6633939wrd.18.1524198650219; Thu, 19 Apr 2018 21:30:50 -0700 (PDT) Received: from moonbase.pars.ee ([2a00:23c4:7c88:af00:4c9e:9728:c86:66cf]) by smtp.gmail.com with ESMTPSA id 80sm935304wml.44.2018.04.19.21.30.49 (version=TLS1_2 cipher=ECDHE-RSA-AES128-GCM-SHA256 bits=128/128); Thu, 19 Apr 2018 21:30:49 -0700 (PDT) From: Rostislav Pehlivanov To: ffmpeg-devel@ffmpeg.org Date: Fri, 20 Apr 2018 05:30:08 +0100 Message-Id: <20180420043010.3709-7-atomnuker@gmail.com> X-Mailer: git-send-email 2.17.0 In-Reply-To: <20180420043010.3709-1-atomnuker@gmail.com> References: <20180420043010.3709-1-atomnuker@gmail.com> Subject: [FFmpeg-devel] [PATCH 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 99128d74dc..b93bf0b00f 100755 --- a/configure +++ b/configure @@ -3304,6 +3304,7 @@ azmq_filter_deps="libzmq" blackframe_filter_deps="gpl" boxblur_filter_deps="gpl" bs2b_filter_deps="libbs2b" +chromaticabberation_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 915c4009af..b4c10daed6 100644 --- a/libavfilter/Makefile +++ b/libavfilter/Makefile @@ -153,6 +153,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 65f1628249..a4b59f43ac 100644 --- a/libavfilter/allfilters.c +++ b/libavfilter/allfilters.c @@ -146,6 +146,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..e814d2442b --- /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 = "rgba8", + .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.input_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, +};