2021-11-16 07:35:24 +00:00
|
|
|
/*
|
|
|
|
* copyright (c) 2021 Wu Jianhua <jianhua.wu@intel.com>
|
|
|
|
* 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/random_seed.h"
|
|
|
|
#include "libavutil/opt.h"
|
lavfi/vulkan: split off lavfi-specific code into vulkan_filter.c
The issue is that libavfilter depends on libavcodec, and when doing a
static build, if libavcodec also includes "libavfilter/vulkan.c", then
during link-time, compiling programs will fail as there would be multiple
definitions of the same symbols in both libavfilter and libavcodec's
object files.
Linkers are, however, more permitting if both files that include
a common file that's used as a template are one-to-one identical.
Hence, to make both files the same in the future, export all avfilter
specific functions to a separate file.
There is some work in progress to make templated files like this be
compiled only once, so this is not a long-term solution.
This also removes a macro that could be used to toggle SPIRV compilation
capability on #include-time, as this could cause the files to be different.
2021-11-19 14:13:35 +00:00
|
|
|
#include "vulkan_filter.h"
|
2021-11-16 07:35:24 +00:00
|
|
|
#include "internal.h"
|
|
|
|
|
|
|
|
#define CGS 32
|
|
|
|
#define GBLUR_MAX_KERNEL_SIZE 127
|
|
|
|
|
|
|
|
typedef struct GBlurVulkanContext {
|
|
|
|
FFVulkanContext vkctx;
|
|
|
|
FFVkQueueFamilyCtx qf;
|
|
|
|
FFVkExecContext *exec;
|
|
|
|
FFVulkanPipeline *pl_hor;
|
|
|
|
FFVulkanPipeline *pl_ver;
|
|
|
|
FFVkBuffer params_buf_hor;
|
|
|
|
FFVkBuffer params_buf_ver;
|
|
|
|
|
|
|
|
VkDescriptorImageInfo input_images[3];
|
|
|
|
VkDescriptorImageInfo tmp_images[3];
|
|
|
|
VkDescriptorImageInfo output_images[3];
|
|
|
|
VkDescriptorBufferInfo params_desc_hor;
|
|
|
|
VkDescriptorBufferInfo params_desc_ver;
|
|
|
|
|
|
|
|
int initialized;
|
|
|
|
int size;
|
|
|
|
int planes;
|
|
|
|
int kernel_size;
|
|
|
|
float sigma;
|
|
|
|
float sigmaV;
|
|
|
|
AVFrame *tmpframe;
|
|
|
|
} GBlurVulkanContext;
|
|
|
|
|
|
|
|
static const char gblur_horizontal[] = {
|
|
|
|
C(0, void gblur(const ivec2 pos, const int index) )
|
|
|
|
C(0, { )
|
|
|
|
C(1, vec4 sum = texture(input_image[index], pos) * kernel[0]; )
|
|
|
|
C(0, )
|
|
|
|
C(1, for(int i = 1; i < kernel.length(); i++) { )
|
|
|
|
C(2, sum += texture(input_image[index], pos + vec2(i, 0.0)) * kernel[i]; )
|
|
|
|
C(2, sum += texture(input_image[index], pos - vec2(i, 0.0)) * kernel[i]; )
|
|
|
|
C(1, } )
|
|
|
|
C(0, )
|
|
|
|
C(1, imageStore(output_image[index], pos, sum); )
|
|
|
|
C(0, } )
|
|
|
|
};
|
|
|
|
|
|
|
|
static const char gblur_vertical[] = {
|
|
|
|
C(0, void gblur(const ivec2 pos, const int index) )
|
|
|
|
C(0, { )
|
|
|
|
C(1, vec4 sum = texture(input_image[index], pos) * kernel[0]; )
|
|
|
|
C(0, )
|
|
|
|
C(1, for(int i = 1; i < kernel.length(); i++) { )
|
|
|
|
C(2, sum += texture(input_image[index], pos + vec2(0.0, i)) * kernel[i]; )
|
|
|
|
C(2, sum += texture(input_image[index], pos - vec2(0.0, i)) * kernel[i]; )
|
|
|
|
C(1, } )
|
|
|
|
C(0, )
|
|
|
|
C(1, imageStore(output_image[index], pos, sum); )
|
|
|
|
C(0, } )
|
|
|
|
};
|
|
|
|
|
|
|
|
static inline float gaussian(float sigma, float x)
|
|
|
|
{
|
|
|
|
return 1.0 / (sqrt(2.0 * M_PI) * sigma) *
|
|
|
|
exp(-(x * x) / (2.0 * sigma * sigma));
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline float gaussian_simpson_integration(float sigma, float a, float b)
|
|
|
|
{
|
|
|
|
return (b - a) * (1.0 / 6.0) * ((gaussian(sigma, a) +
|
|
|
|
4.0 * gaussian(sigma, (a + b) * 0.5) + gaussian(sigma, b)));
|
|
|
|
}
|
|
|
|
|
|
|
|
static void init_gaussian_kernel(float *kernel, float sigma, float kernel_size)
|
|
|
|
{
|
|
|
|
int x;
|
|
|
|
float sum;
|
|
|
|
|
|
|
|
sum = 0;
|
|
|
|
for (x = 0; x < kernel_size; x++) {
|
|
|
|
kernel[x] = gaussian_simpson_integration(sigma, x - 0.5f, x + 0.5f);
|
|
|
|
if (!x)
|
|
|
|
sum += kernel[x];
|
|
|
|
else
|
|
|
|
sum += kernel[x] * 2.0;
|
|
|
|
}
|
|
|
|
/* Normalized */
|
|
|
|
sum = 1.0 / sum;
|
|
|
|
for (x = 0; x < kernel_size; x++) {
|
|
|
|
kernel[x] *= sum;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static av_cold void init_gaussian_params(GBlurVulkanContext *s)
|
|
|
|
{
|
|
|
|
if (!(s->size & 1)) {
|
|
|
|
av_log(s, AV_LOG_WARNING, "kernel size should be odd\n");
|
|
|
|
s->size++;
|
|
|
|
}
|
|
|
|
if (s->sigmaV <= 0)
|
|
|
|
s->sigmaV = s->sigma;
|
|
|
|
|
|
|
|
s->kernel_size = (s->size >> 1) + 1;
|
|
|
|
s->tmpframe = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static av_cold int init_filter(AVFilterContext *ctx, AVFrame *in)
|
|
|
|
{
|
2021-11-17 05:04:54 +00:00
|
|
|
int err = 0;
|
2021-11-16 07:35:24 +00:00
|
|
|
char *kernel_def;
|
|
|
|
uint8_t *kernel_mapped;
|
2021-11-19 06:46:15 +00:00
|
|
|
FFVkSPIRVShader *shd;
|
2021-11-16 07:35:24 +00:00
|
|
|
GBlurVulkanContext *s = ctx->priv;
|
|
|
|
const int planes = av_pix_fmt_count_planes(s->vkctx.output_format);
|
|
|
|
|
|
|
|
FFVulkanDescriptorSetBinding image_descs[] = {
|
|
|
|
{
|
|
|
|
.name = "input_image",
|
|
|
|
.type = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
|
|
|
|
.dimensions = 2,
|
|
|
|
.elems = planes,
|
|
|
|
.stages = VK_SHADER_STAGE_COMPUTE_BIT,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
.name = "output_image",
|
|
|
|
.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,
|
|
|
|
},
|
|
|
|
};
|
|
|
|
|
|
|
|
FFVulkanDescriptorSetBinding buf_desc = {
|
|
|
|
.name = "data",
|
|
|
|
.type = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
|
|
|
|
.mem_quali = "readonly",
|
|
|
|
.mem_layout = "std430",
|
|
|
|
.stages = VK_SHADER_STAGE_COMPUTE_BIT,
|
|
|
|
.updater = NULL,
|
|
|
|
.buf_content = NULL,
|
|
|
|
};
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
image_descs[0].sampler = ff_vk_init_sampler(&s->vkctx, 1, VK_FILTER_LINEAR);
|
2021-11-16 07:35:24 +00:00
|
|
|
if (!image_descs[0].sampler)
|
|
|
|
return AVERROR_EXTERNAL;
|
|
|
|
|
|
|
|
init_gaussian_params(s);
|
|
|
|
|
|
|
|
kernel_def = av_asprintf("float kernel[%i];", s->kernel_size);
|
|
|
|
if (!kernel_def)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
|
|
|
buf_desc.buf_content = kernel_def;
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_qf_init(&s->vkctx, &s->qf, VK_QUEUE_COMPUTE_BIT, 0);
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
{ /* Create shader for the horizontal pass */
|
|
|
|
image_descs[0].updater = s->input_images;
|
|
|
|
image_descs[1].updater = s->tmp_images;
|
|
|
|
buf_desc.updater = &s->params_desc_hor;
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
s->pl_hor = ff_vk_create_pipeline(&s->vkctx, &s->qf);
|
2021-11-16 07:35:24 +00:00
|
|
|
if (!s->pl_hor) {
|
|
|
|
err = AVERROR(ENOMEM);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
shd = ff_vk_init_shader(s->pl_hor, "gblur_compute_hor", image_descs[0].stages);
|
2021-11-16 07:35:24 +00:00
|
|
|
if (!shd) {
|
|
|
|
err = AVERROR(ENOMEM);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_set_compute_shader_sizes(shd, (int [3]){ CGS, CGS, 1 });
|
|
|
|
RET(ff_vk_add_descriptor_set(&s->vkctx, s->pl_hor, shd, image_descs, FF_ARRAY_ELEMS(image_descs), 0));
|
|
|
|
RET(ff_vk_add_descriptor_set(&s->vkctx, s->pl_hor, shd, &buf_desc, 1, 0));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
GLSLD( gblur_horizontal );
|
|
|
|
GLSLC(0, void main() );
|
|
|
|
GLSLC(0, { );
|
|
|
|
GLSLC(1, ivec2 size; );
|
|
|
|
GLSLC(1, const ivec2 pos = ivec2(gl_GlobalInvocationID.xy); );
|
|
|
|
for (int i = 0; i < planes; i++) {
|
|
|
|
GLSLC(0, );
|
|
|
|
GLSLF(1, size = imageSize(output_image[%i]); ,i);
|
|
|
|
GLSLC(1, if (IS_WITHIN(pos, size)) { );
|
|
|
|
if (s->planes & (1 << i)) {
|
|
|
|
GLSLF(2, gblur(pos, %i); ,i);
|
|
|
|
} else {
|
|
|
|
GLSLF(2, vec4 res = texture(input_image[%i], pos); ,i);
|
|
|
|
GLSLF(2, imageStore(output_image[%i], pos, res); ,i);
|
|
|
|
}
|
|
|
|
GLSLC(1, } );
|
|
|
|
}
|
|
|
|
GLSLC(0, } );
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_compile_shader(&s->vkctx, shd, "main"));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_init_pipeline_layout(&s->vkctx, s->pl_hor));
|
|
|
|
RET(ff_vk_init_compute_pipeline(&s->vkctx, s->pl_hor));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_create_buf(&s->vkctx, &s->params_buf_hor, sizeof(float) * s->kernel_size,
|
2021-11-16 07:35:24 +00:00
|
|
|
VK_BUFFER_USAGE_STORAGE_BUFFER_BIT, VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT));
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_map_buffers(&s->vkctx, &s->params_buf_hor, &kernel_mapped, 1, 0));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
init_gaussian_kernel((float *)kernel_mapped, s->sigma, s->kernel_size);
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_unmap_buffers(&s->vkctx, &s->params_buf_hor, 1, 1));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
s->params_desc_hor.buffer = s->params_buf_hor.buf;
|
|
|
|
s->params_desc_hor.range = VK_WHOLE_SIZE;
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_update_descriptor_set(&s->vkctx, s->pl_hor, 1);
|
2021-11-16 07:35:24 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
{ /* Create shader for the vertical pass */
|
|
|
|
image_descs[0].updater = s->tmp_images;
|
|
|
|
image_descs[1].updater = s->output_images;
|
|
|
|
buf_desc.updater = &s->params_desc_ver;
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
s->pl_ver = ff_vk_create_pipeline(&s->vkctx, &s->qf);
|
2021-11-16 07:35:24 +00:00
|
|
|
if (!s->pl_ver) {
|
|
|
|
err = AVERROR(ENOMEM);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
shd = ff_vk_init_shader(s->pl_ver, "gblur_compute_ver", image_descs[0].stages);
|
2021-11-16 07:35:24 +00:00
|
|
|
if (!shd) {
|
|
|
|
err = AVERROR(ENOMEM);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_set_compute_shader_sizes(shd, (int [3]){ CGS, CGS, 1 });
|
|
|
|
RET(ff_vk_add_descriptor_set(&s->vkctx, s->pl_ver, shd, image_descs, FF_ARRAY_ELEMS(image_descs), 0));
|
|
|
|
RET(ff_vk_add_descriptor_set(&s->vkctx, s->pl_ver, shd, &buf_desc, 1, 0));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
GLSLD( gblur_vertical );
|
|
|
|
GLSLC(0, void main() );
|
|
|
|
GLSLC(0, { );
|
|
|
|
GLSLC(1, ivec2 size; );
|
|
|
|
GLSLC(1, const ivec2 pos = ivec2(gl_GlobalInvocationID.xy); );
|
|
|
|
for (int i = 0; i < planes; i++) {
|
|
|
|
GLSLC(0, );
|
|
|
|
GLSLF(1, size = imageSize(output_image[%i]); ,i);
|
|
|
|
GLSLC(1, if (IS_WITHIN(pos, size)) { );
|
|
|
|
if (s->planes & (1 << i)) {
|
|
|
|
GLSLF(2, gblur(pos, %i); ,i);
|
|
|
|
} else {
|
|
|
|
GLSLF(2, vec4 res = texture(input_image[%i], pos); ,i);
|
|
|
|
GLSLF(2, imageStore(output_image[%i], pos, res); ,i);
|
|
|
|
}
|
|
|
|
GLSLC(1, } );
|
|
|
|
}
|
|
|
|
GLSLC(0, } );
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_compile_shader(&s->vkctx, shd, "main"));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_init_pipeline_layout(&s->vkctx, s->pl_ver));
|
|
|
|
RET(ff_vk_init_compute_pipeline(&s->vkctx, s->pl_ver));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_create_buf(&s->vkctx, &s->params_buf_ver, sizeof(float) * s->kernel_size,
|
2021-11-16 07:35:24 +00:00
|
|
|
VK_BUFFER_USAGE_STORAGE_BUFFER_BIT, VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT));
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_map_buffers(&s->vkctx, &s->params_buf_ver, &kernel_mapped, 1, 0));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
init_gaussian_kernel((float *)kernel_mapped, s->sigmaV, s->kernel_size);
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_unmap_buffers(&s->vkctx, &s->params_buf_ver, 1, 1));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
s->params_desc_ver.buffer = s->params_buf_ver.buf;
|
|
|
|
s->params_desc_ver.range = VK_WHOLE_SIZE;
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_update_descriptor_set(&s->vkctx, s->pl_ver, 1);
|
2021-11-16 07:35:24 +00:00
|
|
|
}
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_create_exec_ctx(&s->vkctx, &s->exec, &s->qf));
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
s->initialized = 1;
|
|
|
|
|
|
|
|
fail:
|
|
|
|
av_free(kernel_def);
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
|
|
|
|
static av_cold void gblur_vulkan_uninit(AVFilterContext *avctx)
|
|
|
|
{
|
|
|
|
GBlurVulkanContext *s = avctx->priv;
|
|
|
|
|
|
|
|
av_frame_free(&s->tmpframe);
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_free_buf(&s->vkctx, &s->params_buf_hor);
|
|
|
|
ff_vk_free_buf(&s->vkctx, &s->params_buf_ver);
|
|
|
|
ff_vk_uninit(&s->vkctx);
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
s->initialized = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int process_frames(AVFilterContext *avctx, AVFrame *outframe, AVFrame *inframe)
|
|
|
|
{
|
|
|
|
int err;
|
|
|
|
VkCommandBuffer cmd_buf;
|
|
|
|
|
|
|
|
const VkFormat *input_formats = NULL;
|
|
|
|
const VkFormat *output_formats = NULL;
|
|
|
|
GBlurVulkanContext *s = avctx->priv;
|
|
|
|
FFVulkanFunctions *vk = &s->vkctx.vkfn;
|
|
|
|
AVVkFrame *in = (AVVkFrame *)inframe->data[0];
|
|
|
|
AVVkFrame *tmp = (AVVkFrame *)s->tmpframe->data[0];
|
|
|
|
AVVkFrame *out = (AVVkFrame *)outframe->data[0];
|
|
|
|
|
|
|
|
int planes = av_pix_fmt_count_planes(s->vkctx.output_format);
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_start_exec_recording(&s->vkctx, s->exec);
|
|
|
|
cmd_buf = ff_vk_get_exec_buf(s->exec);
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
input_formats = av_vkfmt_from_pixfmt(s->vkctx.input_format);
|
|
|
|
output_formats = av_vkfmt_from_pixfmt(s->vkctx.output_format);
|
|
|
|
for (int i = 0; i < planes; i++) {
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_create_imageview(&s->vkctx, s->exec, &s->input_images[i].imageView,
|
2021-11-16 07:35:24 +00:00
|
|
|
in->img[i],
|
|
|
|
input_formats[i],
|
|
|
|
ff_comp_identity_map));
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_create_imageview(&s->vkctx, s->exec, &s->tmp_images[i].imageView,
|
2021-11-16 07:35:24 +00:00
|
|
|
tmp->img[i],
|
|
|
|
output_formats[i],
|
|
|
|
ff_comp_identity_map));
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
RET(ff_vk_create_imageview(&s->vkctx, s->exec, &s->output_images[i].imageView,
|
2021-11-16 07:35:24 +00:00
|
|
|
out->img[i],
|
|
|
|
output_formats[i],
|
|
|
|
ff_comp_identity_map));
|
|
|
|
|
|
|
|
s->input_images[i].imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
|
|
|
|
s->tmp_images[i].imageLayout = VK_IMAGE_LAYOUT_GENERAL;
|
|
|
|
s->output_images[i].imageLayout = VK_IMAGE_LAYOUT_GENERAL;
|
|
|
|
}
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_update_descriptor_set(&s->vkctx, s->pl_hor, 0);
|
|
|
|
ff_vk_update_descriptor_set(&s->vkctx, s->pl_ver, 0);
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
for (int i = 0; i < planes; i++) {
|
|
|
|
VkImageMemoryBarrier barriers[] = {
|
|
|
|
{
|
|
|
|
.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER,
|
|
|
|
.srcAccessMask = 0,
|
|
|
|
.dstAccessMask = VK_ACCESS_SHADER_READ_BIT,
|
|
|
|
.oldLayout = in->layout[i],
|
|
|
|
.newLayout = s->input_images[i].imageLayout,
|
|
|
|
.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
|
.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
|
.image = in->img[i],
|
|
|
|
.subresourceRange.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT,
|
|
|
|
.subresourceRange.levelCount = 1,
|
|
|
|
.subresourceRange.layerCount = 1,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER,
|
|
|
|
.srcAccessMask = 0,
|
|
|
|
.dstAccessMask = VK_ACCESS_SHADER_WRITE_BIT | VK_ACCESS_SHADER_READ_BIT,
|
|
|
|
.oldLayout = tmp->layout[i],
|
|
|
|
.newLayout = s->tmp_images[i].imageLayout,
|
|
|
|
.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
|
.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
|
.image = tmp->img[i],
|
|
|
|
.subresourceRange.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT,
|
|
|
|
.subresourceRange.levelCount = 1,
|
|
|
|
.subresourceRange.layerCount = 1,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
.sType = VK_STRUCTURE_TYPE_IMAGE_MEMORY_BARRIER,
|
|
|
|
.srcAccessMask = 0,
|
|
|
|
.dstAccessMask = VK_ACCESS_SHADER_WRITE_BIT,
|
|
|
|
.oldLayout = out->layout[i],
|
|
|
|
.newLayout = s->output_images[i].imageLayout,
|
|
|
|
.srcQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
|
.dstQueueFamilyIndex = VK_QUEUE_FAMILY_IGNORED,
|
|
|
|
.image = out->img[i],
|
|
|
|
.subresourceRange.aspectMask = VK_IMAGE_ASPECT_COLOR_BIT,
|
|
|
|
.subresourceRange.levelCount = 1,
|
|
|
|
.subresourceRange.layerCount = 1,
|
|
|
|
},
|
|
|
|
};
|
|
|
|
|
|
|
|
vk->CmdPipelineBarrier(cmd_buf, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT,
|
|
|
|
VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT, 0,
|
|
|
|
0, NULL, 0, NULL, FF_ARRAY_ELEMS(barriers), barriers);
|
|
|
|
|
|
|
|
in->layout[i] = barriers[0].newLayout;
|
|
|
|
in->access[i] = barriers[0].dstAccessMask;
|
|
|
|
|
|
|
|
tmp->layout[i] = barriers[1].newLayout;
|
|
|
|
tmp->access[i] = barriers[1].dstAccessMask;
|
|
|
|
|
|
|
|
out->layout[i] = barriers[2].newLayout;
|
|
|
|
out->access[i] = barriers[2].dstAccessMask;
|
|
|
|
}
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_bind_pipeline_exec(&s->vkctx, s->exec, s->pl_hor);
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
vk->CmdDispatch(cmd_buf, FFALIGN(s->vkctx.output_width, CGS)/CGS,
|
|
|
|
FFALIGN(s->vkctx.output_height, CGS)/CGS, 1);
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_bind_pipeline_exec(&s->vkctx, s->exec, s->pl_ver);
|
2021-11-16 07:35:24 +00:00
|
|
|
|
|
|
|
vk->CmdDispatch(cmd_buf, FFALIGN(s->vkctx.output_width, CGS)/CGS,
|
|
|
|
FFALIGN(s->vkctx.output_height, CGS)/CGS, 1);
|
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_add_exec_dep(&s->vkctx, s->exec, inframe, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT);
|
|
|
|
ff_vk_add_exec_dep(&s->vkctx, s->exec, outframe, VK_PIPELINE_STAGE_TOP_OF_PIPE_BIT);
|
2021-11-16 07:35:24 +00:00
|
|
|
|
2021-11-19 06:46:15 +00:00
|
|
|
err = ff_vk_submit_exec_queue(&s->vkctx, s->exec);
|
2021-11-16 07:35:24 +00:00
|
|
|
if (err)
|
|
|
|
return err;
|
|
|
|
|
|
|
|
ff_vk_qf_rotate(&s->qf);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
fail:
|
2021-11-19 06:46:15 +00:00
|
|
|
ff_vk_discard_exec_deps(s->exec);
|
2021-11-16 07:35:24 +00:00
|
|
|
return err;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int gblur_vulkan_filter_frame(AVFilterLink *link, AVFrame *in)
|
|
|
|
{
|
|
|
|
int err;
|
|
|
|
AVFrame *out = NULL;
|
|
|
|
AVFilterContext *ctx = link->dst;
|
|
|
|
GBlurVulkanContext *s = ctx->priv;
|
|
|
|
AVFilterLink *outlink = ctx->outputs[0];
|
|
|
|
|
|
|
|
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));
|
|
|
|
s->tmpframe = ff_get_video_buffer(outlink, outlink->w, outlink->h);
|
|
|
|
if (!s->tmpframe) {
|
|
|
|
err = AVERROR(ENOMEM);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
RET(process_frames(ctx, out, in));
|
|
|
|
|
|
|
|
RET(av_frame_copy_props(out, in));
|
|
|
|
|
|
|
|
av_frame_free(&in);
|
|
|
|
|
|
|
|
return ff_filter_frame(outlink, out);
|
|
|
|
|
|
|
|
fail:
|
|
|
|
av_frame_free(&in);
|
|
|
|
av_frame_free(&out);
|
|
|
|
av_frame_free(&s->tmpframe);
|
|
|
|
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
|
|
|
|
#define OFFSET(x) offsetof(GBlurVulkanContext, x)
|
|
|
|
#define FLAGS (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)
|
|
|
|
static const AVOption gblur_vulkan_options[] = {
|
|
|
|
{ "sigma", "Set sigma", OFFSET(sigma), AV_OPT_TYPE_FLOAT, {.dbl = 0.5}, 0.01, 1024.0, FLAGS },
|
|
|
|
{ "sigmaV", "Set vertical sigma", OFFSET(sigmaV), AV_OPT_TYPE_FLOAT, {.dbl = 0}, 0.0, 1024.0, FLAGS },
|
|
|
|
{ "planes", "Set planes to filter", OFFSET(planes), AV_OPT_TYPE_INT, {.i64 = 0xF}, 0, 0xF, FLAGS },
|
|
|
|
{ "size", "Set kernel size", OFFSET(size), AV_OPT_TYPE_INT, {.i64 = 19}, 1, GBLUR_MAX_KERNEL_SIZE, FLAGS },
|
|
|
|
{ NULL },
|
|
|
|
};
|
|
|
|
|
|
|
|
AVFILTER_DEFINE_CLASS(gblur_vulkan);
|
|
|
|
|
|
|
|
static const AVFilterPad gblur_vulkan_inputs[] = {
|
|
|
|
{
|
|
|
|
.name = "default",
|
|
|
|
.type = AVMEDIA_TYPE_VIDEO,
|
|
|
|
.filter_frame = &gblur_vulkan_filter_frame,
|
|
|
|
.config_props = &ff_vk_filter_config_input,
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
static const AVFilterPad gblur_vulkan_outputs[] = {
|
|
|
|
{
|
|
|
|
.name = "default",
|
|
|
|
.type = AVMEDIA_TYPE_VIDEO,
|
|
|
|
.config_props = &ff_vk_filter_config_output,
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
const AVFilter ff_vf_gblur_vulkan = {
|
|
|
|
.name = "gblur_vulkan",
|
|
|
|
.description = NULL_IF_CONFIG_SMALL("Gaussian Blur in Vulkan"),
|
|
|
|
.priv_size = sizeof(GBlurVulkanContext),
|
|
|
|
.init = &ff_vk_filter_init,
|
|
|
|
.uninit = &gblur_vulkan_uninit,
|
|
|
|
FILTER_INPUTS(gblur_vulkan_inputs),
|
|
|
|
FILTER_OUTPUTS(gblur_vulkan_outputs),
|
|
|
|
FILTER_SINGLE_PIXFMT(AV_PIX_FMT_VULKAN),
|
|
|
|
.priv_class = &gblur_vulkan_class,
|
|
|
|
.flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
|
|
|
|
};
|