Go to the documentation of this file.
32 #define GBLUR_MAX_KERNEL_SIZE 127
56 C(0,
void gblur(
const ivec2
pos,
const int index) )
58 C(1, vec4 sum = texture(input_images[
index],
pos) * kernel[0]; )
60 C(1,
for(
int i = 1;
i < kernel.length();
i++) { )
65 C(1, imageStore(output_images[
index],
pos, sum); )
69 static inline float gaussian(
float sigma,
float x)
71 return 1.0 / (sqrt(2.0 *
M_PI) * sigma) *
72 exp(-(x * x) / (2.0 * sigma * sigma));
77 return (
b -
a) * (1.0 / 6.0) * ((
gaussian(sigma,
a) +
87 for (x = 0; x < kernel_size; x++) {
92 sum += kernel[x] * 2.0;
96 for (x = 0; x < kernel_size; x++) {
116 s->sigmaV =
s->sigma;
131 uint8_t *kernel_mapped;
134 void *spv_opaque =
NULL;
140 .type = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
141 .mem_quali =
"readonly",
142 .mem_layout =
"std430",
143 .stages = VK_SHADER_STAGE_COMPUTE_BIT,
147 char *kernel_def =
av_asprintf(
"float kernel[%i];", ksize);
159 GLSLC(1,
const ivec2
pos = ivec2(gl_GlobalInvocationID.xy); );
165 if (
s->planes & (1 <<
i)) {
168 GLSLF(1, vec4 res = texture(input_images[%
i],
pos); ,
i);
169 GLSLF(1, imageStore(output_images[%
i],
pos, res); ,
i);
182 VK_BUFFER_USAGE_SHADER_DEVICE_ADDRESS_BIT |
183 VK_BUFFER_USAGE_STORAGE_BUFFER_BIT,
184 VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT));
193 VK_FORMAT_UNDEFINED));
213 spv = ff_vk_spirv_init();
223 VK_SHADER_STAGE_COMPUTE_BIT, 0));
225 VK_SHADER_STAGE_COMPUTE_BIT, 0));
229 .name =
"input_images",
230 .type = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
233 .stages = VK_SHADER_STAGE_COMPUTE_BIT,
237 .name =
"output_images",
238 .type = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
240 .mem_quali =
"writeonly",
243 .stages = VK_SHADER_STAGE_COMPUTE_BIT,
344 #define OFFSET(x) offsetof(GBlurVulkanContext, x)
345 #define FLAGS (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)
375 .
name =
"gblur_vulkan",
383 .priv_class = &gblur_vulkan_class,
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
int ff_vk_create_buf(FFVulkanContext *s, FFVkBuffer *buf, size_t size, void *pNext, void *alloc_pNext, VkBufferUsageFlags usage, VkMemoryPropertyFlagBits flags)
#define AV_LOG_WARNING
Something somehow does not look correct.
AVFILTER_DEFINE_CLASS(gblur_vulkan)
void ff_vk_pipeline_free(FFVulkanContext *s, FFVulkanPipeline *pl)
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
static float gaussian_simpson_integration(float sigma, float a, float b)
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
int ff_vk_qf_init(FFVulkanContext *s, FFVkQueueFamilyCtx *qf, VkQueueFlagBits dev_family)
Chooses a QF and loads it into a context.
char * av_asprintf(const char *fmt,...)
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
#define FILTER_INPUTS(array)
This structure describes decoded (raw) audio or video data.
int ff_vk_filter_init(AVFilterContext *avctx)
General lavfi IO functions.
int ff_vk_shader_create(FFVulkanContext *s, FFVkSPIRVShader *shd, uint8_t *spirv, size_t spirv_size, const char *entrypoint)
void ff_vk_uninit(FFVulkanContext *s)
Frees main context.
void(* uninit)(struct FFVkSPIRVCompiler **ctx)
const char * name
Filter name.
void ff_vk_shader_set_compute_sizes(FFVkSPIRVShader *shd, int x, int y, int z)
static const struct @446 planes[]
A link between two filters.
@ AV_PIX_FMT_VULKAN
Vulkan hardware images.
static const AVFilterPad gblur_vulkan_inputs[]
static void init_gaussian_kernel(float *kernel, float sigma, float kernel_size)
int av_pix_fmt_count_planes(enum AVPixelFormat pix_fmt)
const VkAllocationCallbacks * alloc
Custom memory allocator, else NULL.
void * priv
private data for use by the filter
static const AVFilterPad gblur_vulkan_outputs[]
int ff_vk_filter_process_2pass(FFVulkanContext *vkctx, FFVkExecPool *e, FFVulkanPipeline *pls[2], AVFrame *out, AVFrame *tmp, AVFrame *in, VkSampler sampler, void *push_src, size_t push_size)
Submit a compute shader with a single in and single out with 2 stages.
A filter pad used for either input or output.
s EdgeDetect Foobar g libavfilter vf_edgedetect c libavfilter vf_foobar c edit libavfilter and add an entry for foobar following the pattern of the other filters edit libavfilter allfilters and add an entry for foobar following the pattern of the other filters configure make j< whatever > ffmpeg ffmpeg i you should get a foobar png with Lena edge detected That s your new playground is ready Some little details about what s going which in turn will define variables for the build system and the C
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
static const AVOption gblur_vulkan_options[]
static int ff_vk_unmap_buffer(FFVulkanContext *s, FFVkBuffer *buf, int flush)
int(* compile_shader)(struct FFVkSPIRVCompiler *ctx, void *avctx, struct FFVkSPIRVShader *shd, uint8_t **data, size_t *size, const char *entrypoint, void **opaque)
void ff_vk_exec_pool_free(FFVulkanContext *s, FFVkExecPool *pool)
#define FILTER_OUTPUTS(array)
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a link
static av_cold int init_filter(AVFilterContext *ctx, AVFrame *in)
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
int ff_vk_filter_config_output(AVFilterLink *outlink)
int ff_vk_init_compute_pipeline(FFVulkanContext *s, FFVulkanPipeline *pl, FFVkSPIRVShader *shd)
int ff_vk_exec_pool_init(FFVulkanContext *s, FFVkQueueFamilyCtx *qf, FFVkExecPool *pool, int nb_contexts, int nb_queries, VkQueryType query_type, int query_64bit, const void *query_create_pnext)
Allocates/frees an execution pool.
#define GBLUR_MAX_KERNEL_SIZE
#define FF_FILTER_FLAG_HWFRAME_AWARE
The filter is aware of hardware frames, and any hardware frame context should not be automatically pr...
int ff_vk_shader_init(FFVulkanPipeline *pl, FFVkSPIRVShader *shd, const char *name, VkShaderStageFlags stage, uint32_t required_subgroup_size)
Shader management.
int(* init)(AVBSFContext *ctx)
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
#define AVFILTER_FLAG_HWDEVICE
The filter can create hardware frames using AVFilterContext.hw_device_ctx.
The reader does not expect b to be semantically here and if the code is changed by maybe adding a a division or other the signedness will almost certainly be mistaken To avoid this confusion a new type was SUINT is the C unsigned type but it holds a signed int to use the same example SUINT a
static av_cold void gblur_vulkan_uninit(AVFilterContext *avctx)
#define AVERROR_EXTERNAL
Generic error in an external library.
@ AV_OPT_TYPE_FLOAT
Underlying C type is float.
static void uninit(AVBSFContext *ctx)
static void init_kernel_size(GBlurVulkanContext *s, int *out_size)
#define i(width, name, range_min, range_max)
int w
agreed upon image width
const char * ff_vk_shader_rep_fmt(enum AVPixelFormat pixfmt)
Returns the format to use for images in shaders.
const char * name
Pad name.
void ff_vk_free_buf(FFVulkanContext *s, FFVkBuffer *buf)
void(* free_shader)(struct FFVkSPIRVCompiler *ctx, void **opaque)
static av_cold void init_gaussian_params(GBlurVulkanContext *s)
int ff_vk_pipeline_descriptor_set_add(FFVulkanContext *s, FFVulkanPipeline *pl, FFVkSPIRVShader *shd, FFVulkanDescriptorSetBinding *desc, int nb, int singular, int print_to_shader_only)
Add descriptor to a pipeline.
static int ff_vk_map_buffer(FFVulkanContext *s, FFVkBuffer *buf, uint8_t **mem, int invalidate)
static int gblur_vulkan_filter_frame(AVFilterLink *link, AVFrame *in)
static const char gblur_func[]
int h
agreed upon image height
const AVFilter ff_vf_gblur_vulkan
@ AV_OPT_TYPE_INT
Underlying C type is int.
int ff_vk_filter_config_input(AVFilterLink *inlink)
AVVulkanDeviceContext * hwctx
int ff_vk_set_descriptor_buffer(FFVulkanContext *s, FFVulkanPipeline *pl, FFVkExecContext *e, int set, int bind, int offs, VkDeviceAddress addr, VkDeviceSize len, VkFormat fmt)
VkDevice act_dev
Active device.
int ff_vk_init_sampler(FFVulkanContext *s, VkSampler *sampler, int unnorm_coords, VkFilter filt)
Create a sampler.
static int init_gblur_pipeline(GBlurVulkanContext *s, FFVulkanPipeline *pl, FFVkSPIRVShader *shd, FFVkBuffer *params_buf, int ksize, float sigma, FFVkSPIRVCompiler *spv)
int ff_vk_exec_pipeline_register(FFVulkanContext *s, FFVkExecPool *pool, FFVulkanPipeline *pl)
Register a pipeline with an exec pool.
void ff_vk_shader_free(FFVulkanContext *s, FFVkSPIRVShader *shd)
#define FILTER_SINGLE_PIXFMT(pix_fmt_)
static float gaussian(float sigma, float x)