From patchwork Thu Jun 21 16:55:41 2018 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Rostislav Pehlivanov X-Patchwork-Id: 9462 Delivered-To: ffmpegpatchwork@gmail.com Received: by 2002:a02:115:0:0:0:0:0 with SMTP id c21-v6csp60269jad; Thu, 21 Jun 2018 09:56:22 -0700 (PDT) X-Google-Smtp-Source: ADUXVKI9IMvME5NudgZMcZRjnKSmM42D7zlcE6TP5Wd88DZaGLc6ZZG3yCdDqMPqz7H2uPg2AsNY X-Received: by 2002:adf:b445:: with SMTP id v5-v6mr21833542wrd.67.1529600182072; Thu, 21 Jun 2018 09:56:22 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1529600182; cv=none; d=google.com; s=arc-20160816; b=lEjAxjL+9giuEhpP2dMPJ1lQl1H4bnSYpvm9K8wVTNqZyPc/gCONlB8y9sOYyZj6Ya wdow8fn9eqSEW5gE67LyadhUXhiZVBO9pF7AagpZmFlbX93mUVCknj8qFphv8XcxsKs3 +8g7vY1r8S2O50G5IyIofR2EATCpn3iJBj6ZQT/a2ZLzIN8Mga28MvGWPTAxjxX4Bqwd GlWD6n3srcR+xp62fRE7/JBMhQQhbwjY/jN6u1tQecAbcLA6KWGpFL+a9JY0FGue/WV3 zDilAM9ac5fgHg9l1ARZFkHwlCcmhkwOp5Eu0olCQQlDf9sBsRF6x8aUoiMa0hiQEL/Y Pa7w== 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=tbDVQQmXmbgZkYfZTHYwO6praA3N5M2JfMl+ju/dwbc=; b=v33wZVNK+fsteLgXo3mRGdwHDpCuvfz4W9DlGUqmMZJjT+79r9dYmFK/WxaZR7ea+m 1dtGS44UFasDlpIx3PzZpIykQ5c/mKBgcs6XSM3c8pbavsdsZ5NXjgl6YrhbehlmELDa saviBZpBSuzG1KYvlembduwuSqiYll8opVvkSz2YrTrd2zZ3yHjo5WXoUNAraG1RRIeu pyKq4Un+PzXRSwgnl8RRIBTuxXRMao/1VJ+MKdAOCR3U66PSknsaJk7OFv1VIcB5xt6O ez6Ohq4pNv9ZhogW8X2xNbN62HlsXSe+GPsyUot1dIbZ3IR1mpPsLt0gmUvlUpm6NeNp 64Fw== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20161025 header.b=pVvuoPug; 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 a4-v6si5184420wrq.395.2018.06.21.09.56.13; Thu, 21 Jun 2018 09:56:22 -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=pVvuoPug; 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 56D9B68A5BB; Thu, 21 Jun 2018 19:55:08 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-wr0-f170.google.com (mail-wr0-f170.google.com [209.85.128.170]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 0989C68A59F for ; Thu, 21 Jun 2018 19:55:02 +0300 (EEST) Received: by mail-wr0-f170.google.com with SMTP id w10-v6so3918551wrk.9 for ; Thu, 21 Jun 2018 09:55:55 -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=YPgm1GgD3m5INicpNWfIDEL1vBgMXqE8cpsE7ZcYz+M=; b=pVvuoPugoKIcv4+OpJmOm5W7cQwA5ZPh6kogVHsd98ExMhvxOVwyT595KIsbINIBZ8 5NAPAttD+eB4ql2oBoDiAEYv2cmahbbPBVrzUdmiyMDi36DGO274JM9q7UPtxC8wnQEe 8O0QVlHxfQl3tvWoaepQT9Yv5ZYDnhIC2pAkV9urnskR+o8eg13/Sb0lJRyjlifYSZo5 8qK1bo28TZSet53DZRedSOhjg9eW/2oP49II0IVAWHpjvho6/QM5cdNuzSXPjbeGOS/s hM3Cqm/bz4h4+FMnd3P7f+g4Juex2Cvf2mMxc4ZqQDuDrA90bEdYx3DlIXrTkNAiOegl vNaA== 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=YPgm1GgD3m5INicpNWfIDEL1vBgMXqE8cpsE7ZcYz+M=; b=jgxnM/+Q5lU/26YQOrfK+nP4aTh07IqApdgmwn83lT+xfdtpMQyDwf/RkINMHOggRL k6m27c7vqyCMkHvVtsH08eSIzU33dB3pmTCAoMGWBlSzso58sALdtzTI3V4/9mGNcUNC /2tT4oExxy29czo1xlivHtvcGYT1cogHn9k7vnzcBoU1tnplcHMxe/F+2ipAqCkCnJRf BgPFZkXlh43cvvwUztFg8c2s17wK7GMxu2oJ6o+dz4t+Ey5LJslCfTSSFfWqIrJcBFJ/ 7a/cLaUuo7RLFE+ZFCoXnqy7rqmyifSPQ1pr2m1UaWHomL1SPaHCck47j+t9Z5yVMaLg rPrA== X-Gm-Message-State: APt69E3GJH5j0yxV3cQ70tTQJqqPftPbkMtSok8JQjqSPh4W+Fi7IkPa CUqr90sHoRkLRTKg9mlGO0A6PNuF X-Received: by 2002:a5d:49d1:: with SMTP id t17-v6mr20981049wrs.242.1529600155013; Thu, 21 Jun 2018 09:55:55 -0700 (PDT) Received: from skyhide.pars.ee ([2a00:23c4:7c8c:8e00:2934:b4d2:9f8c:e74f]) by smtp.gmail.com with ESMTPSA id m16-v6sm7222357wro.91.2018.06.21.09.55.53 (version=TLS1_2 cipher=ECDHE-RSA-AES128-GCM-SHA256 bits=128/128); Thu, 21 Jun 2018 09:55:53 -0700 (PDT) From: Rostislav Pehlivanov To: ffmpeg-devel@ffmpeg.org Date: Thu, 21 Jun 2018 17:55:41 +0100 Message-Id: <20180621165543.5198-5-atomnuker@gmail.com> X-Mailer: git-send-email 2.18.0.rc2.435.gd175cea416 In-Reply-To: <20180621165543.5198-1-atomnuker@gmail.com> References: <20180621165543.5198-1-atomnuker@gmail.com> Subject: [FFmpeg-devel] [PATCH 4/6] 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 eab7c5b630..cd5229ef2d 100755 --- a/configure +++ b/configure @@ -3318,6 +3318,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 d1cab35e2e..dbb7a859dd 100644 --- a/libavfilter/Makefile +++ b/libavfilter/Makefile @@ -160,6 +160,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 a05616f728..5848ad9b4f 100644 --- a/libavfilter/allfilters.c +++ b/libavfilter/allfilters.c @@ -151,6 +151,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, +};