Go to the documentation of this file.
30 #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);
180 VK_BUFFER_USAGE_SHADER_DEVICE_ADDRESS_BIT |
181 VK_BUFFER_USAGE_STORAGE_BUFFER_BIT,
182 VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT));
191 VK_FORMAT_UNDEFINED));
211 spv = ff_vk_spirv_init();
221 VK_SHADER_STAGE_COMPUTE_BIT, 0));
223 VK_SHADER_STAGE_COMPUTE_BIT, 0));
227 .name =
"input_images",
228 .type = VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER,
231 .stages = VK_SHADER_STAGE_COMPUTE_BIT,
235 .name =
"output_images",
236 .type = VK_DESCRIPTOR_TYPE_STORAGE_IMAGE,
238 .mem_quali =
"writeonly",
241 .stages = 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)
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
#define FF_FILTER_FLAG_HWFRAME_AWARE
The filter is aware of hardware frames, and any hardware frame context should not be automatically pr...
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.
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.
int ff_vk_pipeline_descriptor_set_add(FFVulkanContext *s, FFVulkanPipeline *pl, FFVkSPIRVShader *shd, FFVulkanDescriptorSetBinding *desc, int nb, int read_only, int print_to_shader_only)
Add descriptor to a pipeline.
void ff_vk_shader_set_compute_sizes(FFVkSPIRVShader *shd, int x, int y, int z)
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.
int(* init)(AVBSFContext *ctx)
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_INPUTS(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
int ff_vk_shader_init(FFVulkanPipeline *pl, FFVkSPIRVShader *shd, const char *name, VkShaderStageFlags stage, uint32_t required_subgroup_size)
Shader management.
int main(int argc, char **argv)
#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.
#define FILTER_SINGLE_PIXFMT(pix_fmt_)
static void init_kernel_size(GBlurVulkanContext *s, int *out_size)
#define i(width, name, range_min, range_max)
int w
agreed upon image width
static const struct @363 planes[]
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)
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
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.
#define FILTER_OUTPUTS(array)
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.
static av_cold int uninit(AVCodecContext *avctx)
void ff_vk_shader_free(FFVulkanContext *s, FFVkSPIRVShader *shd)
static float gaussian(float sigma, float x)