Go to the documentation of this file.
32 #define GBLUR_MAX_KERNEL_SIZE 127
54 C(0,
void gblur(
const ivec2
pos,
const int index) )
56 C(1, vec4 sum = texture(input_images[
index],
pos) * kernel[0]; )
58 C(1,
for(
int i = 1;
i < kernel.length();
i++) { )
63 C(1, imageStore(output_images[
index],
pos, sum); )
67 static inline float gaussian(
float sigma,
float x)
69 return 1.0 / (sqrt(2.0 *
M_PI) * sigma) *
70 exp(-(x * x) / (2.0 * sigma * sigma));
75 return (
b -
a) * (1.0 / 6.0) * ((
gaussian(sigma,
a) +
85 for (x = 0; x < kernel_size; x++) {
90 sum += kernel[x] * 2.0;
94 for (x = 0; x < kernel_size; x++) {
114 s->sigmaV =
s->sigma;
129 uint8_t *kernel_mapped;
132 void *spv_opaque =
NULL;
138 .type = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
139 .mem_quali =
"readonly",
140 .mem_layout =
"std430",
141 .stages = VK_SHADER_STAGE_COMPUTE_BIT,
145 char *kernel_def =
av_asprintf(
"float kernel[%i];", ksize);
157 GLSLC(1,
const ivec2
pos = ivec2(gl_GlobalInvocationID.xy); );
163 if (
s->planes & (1 <<
i)) {
166 GLSLF(1, vec4 res = texture(input_images[%
i],
pos); ,
i);
167 GLSLF(1, imageStore(output_images[%
i],
pos, res); ,
i);
179 VK_BUFFER_USAGE_SHADER_DEVICE_ADDRESS_BIT |
180 VK_BUFFER_USAGE_STORAGE_BUFFER_BIT,
181 VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT));
188 params_buf, 0, params_buf->
size,
189 VK_FORMAT_UNDEFINED));
209 spv = ff_vk_spirv_init();
221 .name =
"input_images",
222 .type = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
225 .stages = VK_SHADER_STAGE_COMPUTE_BIT,
229 .name =
"output_images",
230 .type = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
232 .mem_quali =
"writeonly",
235 .stages = VK_SHADER_STAGE_COMPUTE_BIT,
244 VK_SHADER_STAGE_COMPUTE_BIT,
258 VK_SHADER_STAGE_COMPUTE_BIT,
342 #define OFFSET(x) offsetof(GBlurVulkanContext, x)
343 #define FLAGS (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)
373 .
name =
"gblur_vulkan",
381 .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)
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
void ff_vk_shader_free(FFVulkanContext *s, FFVulkanShader *shd)
Free a shader.
int ff_vk_shader_init(FFVulkanContext *s, FFVulkanShader *shd, const char *name, VkPipelineStageFlags stage, const char *extensions[], int nb_extensions, int lg_x, int lg_y, int lg_z, uint32_t required_subgroup_size)
Initialize a shader object, with a specific set of extensions, type+bind, local group size,...
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.
static int ff_vk_map_buffer(FFVulkanContext *s, FFVkBuffer *buf, uint8_t **mem, int invalidate)
static const struct @467 planes[]
void ff_vk_uninit(FFVulkanContext *s)
Frees main context.
void(* uninit)(struct FFVkSPIRVCompiler **ctx)
const char * name
Filter name.
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
int ff_vk_shader_register_exec(FFVulkanContext *s, FFVkExecPool *pool, FFVulkanShader *shd)
Register a shader with an exec pool.
int ff_vk_shader_add_descriptor_set(FFVulkanContext *s, FFVulkanShader *shd, FFVulkanDescriptorSetBinding *desc, int nb, int singular, int print_to_shader_only)
Add descriptor to a shader.
static const AVFilterPad gblur_vulkan_outputs[]
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[]
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)
const char * ff_vk_shader_rep_fmt(enum AVPixelFormat pix_fmt, enum FFVkShaderRepFormat rep_fmt)
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
static int init_gblur_pipeline(GBlurVulkanContext *s, FFVulkanShader *shd, FFVkBuffer *params_buf, int ksize, float sigma, FFVkSPIRVCompiler *spv)
int ff_vk_filter_config_output(AVFilterLink *outlink)
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(* 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.
int(* compile_shader)(FFVulkanContext *s, struct FFVkSPIRVCompiler *ctx, FFVulkanShader *shd, uint8_t **data, size_t *size, const char *entrypoint, void **opaque)
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.
int ff_vk_shader_update_desc_buffer(FFVulkanContext *s, FFVkExecContext *e, FFVulkanShader *shd, int set, int bind, int elem, FFVkBuffer *buf, VkDeviceSize offset, VkDeviceSize len, VkFormat fmt)
Update a descriptor in a buffer with a buffer.
@ AV_OPT_TYPE_FLOAT
Underlying C type is float.
static void uninit(AVBSFContext *ctx)
static void init_kernel_size(GBlurVulkanContext *s, int *out_size)
int ff_vk_filter_process_2pass(FFVulkanContext *vkctx, FFVkExecPool *e, FFVulkanShader *shd_list[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.
#define i(width, name, range_min, range_max)
int w
agreed upon image width
int ff_vk_shader_link(FFVulkanContext *s, FFVulkanShader *shd, uint8_t *spirv, size_t spirv_len, const char *entrypoint)
Link a shader into an executable.
static int ff_vk_unmap_buffer(FFVulkanContext *s, FFVkBuffer *buf, int flush)
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)
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
VkDevice act_dev
Active device.
int ff_vk_init_sampler(FFVulkanContext *s, VkSampler *sampler, int unnorm_coords, VkFilter filt)
Create a sampler.
#define FILTER_SINGLE_PIXFMT(pix_fmt_)
static float gaussian(float sigma, float x)