vf_nlmeans_vulkan.c File Reference

#include "libavutil/mem.h"
#include "libavutil/random_seed.h"
#include "libavutil/vulkan_spirv.h"
#include "libavutil/opt.h"
#include "vulkan_filter.h"
#include "filters.h"
#include "video.h"

Go to the source code of this file.

Data Structures
struct	NLMeansVulkanContext
struct	NLMeansVulkanContext::nlmeans_opts
struct	IntegralPushData
struct	WeightsPushData
struct	DenoisePushData

Macros
#define	TYPE_NAME   "vec4"
#define	TYPE_ELEMS   4
#define	TYPE_SIZE   (TYPE_ELEMS*4)
#define	TYPE_BLOCK_ELEMS   16
#define	TYPE_BLOCK_SIZE   (TYPE_SIZE * TYPE_BLOCK_ELEMS)
#define	WG_SIZE   32
#define	OFFSET(x)   offsetof(NLMeansVulkanContext, x)
#define	FLAGS   (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)

Functions
static void	shared_shd_def (FFVulkanShader *shd)
static av_cold int	init_integral_pipeline (FFVulkanContext *vkctx, FFVkExecPool *exec, FFVulkanShader *shd_horizontal, FFVulkanShader *shd_vertical, FFVkSPIRVCompiler *spv, const AVPixFmtDescriptor *desc, int planes)
static av_cold int	init_weights_pipeline (FFVulkanContext *vkctx, FFVkExecPool *exec, FFVulkanShader *shd, FFVkSPIRVCompiler *spv, const AVPixFmtDescriptor *desc, int planes)
static av_cold int	init_denoise_pipeline (FFVulkanContext *vkctx, FFVkExecPool *exec, FFVulkanShader *shd, FFVkSPIRVCompiler *spv, const AVPixFmtDescriptor *desc, int planes)
static av_cold int	init_filter (AVFilterContext *ctx)
static int	denoise_pass (NLMeansVulkanContext *s, FFVkExecContext *exec, FFVkBuffer *ws_vk, uint32_t comp_offs[4], uint32_t comp_planes[4], uint32_t ws_offset[4], uint32_t ws_stride[4], uint32_t ws_count, uint32_t t, uint32_t nb_components)
static int	nlmeans_vulkan_filter_frame (AVFilterLink *link, AVFrame *in)
static void	nlmeans_vulkan_uninit (AVFilterContext *avctx)
	AVFILTER_DEFINE_CLASS (nlmeans_vulkan)

Variables
static const AVOption	nlmeans_vulkan_options []
static const AVFilterPad	nlmeans_vulkan_inputs []
static const AVFilterPad	nlmeans_vulkan_outputs []
const FFFilter	ff_vf_nlmeans_vulkan

Macro Definition Documentation

TYPE_NAME

#define TYPE_NAME   "vec4"

Definition at line 30 of file vf_nlmeans_vulkan.c.

TYPE_ELEMS

#define TYPE_ELEMS   4

Definition at line 31 of file vf_nlmeans_vulkan.c.

TYPE_SIZE

#define TYPE_SIZE   (TYPE_ELEMS*4)

Definition at line 32 of file vf_nlmeans_vulkan.c.

TYPE_BLOCK_ELEMS

#define TYPE_BLOCK_ELEMS   16

Definition at line 33 of file vf_nlmeans_vulkan.c.

TYPE_BLOCK_SIZE

#define TYPE_BLOCK_SIZE   (TYPE_SIZE * TYPE_BLOCK_ELEMS)

Definition at line 34 of file vf_nlmeans_vulkan.c.

WG_SIZE

#define WG_SIZE   32

Definition at line 35 of file vf_nlmeans_vulkan.c.

OFFSET

#define OFFSET

(

x

)

offsetof(NLMeansVulkanContext, x)

Definition at line 1174 of file vf_nlmeans_vulkan.c.

FLAGS

#define FLAGS   (AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM)

Definition at line 1175 of file vf_nlmeans_vulkan.c.

Function Documentation

shared_shd_def()

static void shared_shd_def ( FFVulkanShader *  shd )

static

Definition at line 83 of file vf_nlmeans_vulkan.c.

Referenced by init_integral_pipeline().

init_integral_pipeline()

static av_cold int init_integral_pipeline ( FFVulkanContext *  vkctx, FFVkExecPool *  exec, FFVulkanShader *  shd_horizontal, FFVulkanShader *  shd_vertical, FFVkSPIRVCompiler *  spv, const AVPixFmtDescriptor *  desc, int  planes  )

static

Definition at line 118 of file vf_nlmeans_vulkan.c.

Referenced by init_filter().

init_weights_pipeline()

static av_cold int init_weights_pipeline ( FFVulkanContext *  vkctx, FFVkExecPool *  exec, FFVulkanShader *  shd, FFVkSPIRVCompiler *  spv, const AVPixFmtDescriptor *  desc, int  planes  )

static

Definition at line 307 of file vf_nlmeans_vulkan.c.

Referenced by init_filter().

init_denoise_pipeline()

static av_cold int init_denoise_pipeline ( FFVulkanContext *  vkctx, FFVkExecPool *  exec, FFVulkanShader *  shd, FFVkSPIRVCompiler *  spv, const AVPixFmtDescriptor *  desc, int  planes  )

static

Definition at line 487 of file vf_nlmeans_vulkan.c.

Referenced by init_filter().

init_filter()

static av_cold int init_filter ( AVFilterContext *  ctx )

static

Definition at line 606 of file vf_nlmeans_vulkan.c.

Referenced by nlmeans_vulkan_filter_frame().

denoise_pass()

static int denoise_pass ( NLMeansVulkanContext *  s, FFVkExecContext *  exec, FFVkBuffer *  ws_vk, uint32_t  comp_offs[4], uint32_t  comp_planes[4], uint32_t  ws_offset[4], uint32_t  ws_stride[4], uint32_t  ws_count, uint32_t  t, uint32_t  nb_components  )

static

Definition at line 736 of file vf_nlmeans_vulkan.c.

Referenced by nlmeans_vulkan_filter_frame().

nlmeans_vulkan_filter_frame()

static int nlmeans_vulkan_filter_frame ( AVFilterLink *  link, AVFrame *  in  )

static

Definition at line 794 of file vf_nlmeans_vulkan.c.

nlmeans_vulkan_uninit()

static void nlmeans_vulkan_uninit ( AVFilterContext *  avctx )

static

Definition at line 1152 of file vf_nlmeans_vulkan.c.

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

nlmeans_vulkan

)

Variable Documentation

nlmeans_vulkan_options

const AVOption nlmeans_vulkan_options[]

static

Initial value:

= {
    { "s",  "denoising strength for all components", OFFSET(opts.s), AV_OPT_TYPE_DOUBLE, { .dbl = 1.0 }, 0.0, 100.0, FLAGS },
    { "p",  "patch size for all components", OFFSET(opts.p), AV_OPT_TYPE_INT, { .i64 = 3*2+1 }, 0, 99, FLAGS },
    { "r",  "research window size", OFFSET(opts.r), AV_OPT_TYPE_INT, { .i64 = 7*2+1 }, 0, 99, FLAGS },
    { "t",  "parallelism", OFFSET(opts.t), AV_OPT_TYPE_INT, { .i64 = 8 }, 1, 64, FLAGS },
 
    { "s1", "denoising strength for component 1", OFFSET(opts.sc[0]), AV_OPT_TYPE_DOUBLE, { .dbl = NAN }, 0.0, 100.0, FLAGS },
    { "s2", "denoising strength for component 2", OFFSET(opts.sc[1]), AV_OPT_TYPE_DOUBLE, { .dbl = NAN }, 0.0, 100.0, FLAGS },
    { "s3", "denoising strength for component 3", OFFSET(opts.sc[2]), AV_OPT_TYPE_DOUBLE, { .dbl = NAN }, 0.0, 100.0, FLAGS },
    { "s4", "denoising strength for component 4", OFFSET(opts.sc[3]), AV_OPT_TYPE_DOUBLE, { .dbl = NAN }, 0.0, 100.0, FLAGS },
 
    { "p1", "patch size for component 1", OFFSET(opts.pc[0]), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 99, FLAGS },
    { "p2", "patch size for component 2", OFFSET(opts.pc[1]), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 99, FLAGS },
    { "p3", "patch size for component 3", OFFSET(opts.pc[2]), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 99, FLAGS },
    { "p4", "patch size for component 4", OFFSET(opts.pc[3]), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 99, FLAGS },
 
    { NULL }
}

Definition at line 1176 of file vf_nlmeans_vulkan.c.

nlmeans_vulkan_inputs

const AVFilterPad nlmeans_vulkan_inputs[]

static

Initial value:

= {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_VIDEO,
        .filter_frame = &nlmeans_vulkan_filter_frame,
        .config_props = &ff_vk_filter_config_input,
    },
}

Definition at line 1197 of file vf_nlmeans_vulkan.c.

nlmeans_vulkan_outputs

const AVFilterPad nlmeans_vulkan_outputs[]

static

Initial value:

= {
    {
        .name = "default",
        .type = AVMEDIA_TYPE_VIDEO,
        .config_props = &ff_vk_filter_config_output,
    },
}

Definition at line 1206 of file vf_nlmeans_vulkan.c.

ff_vf_nlmeans_vulkan

const FFFilter ff_vf_nlmeans_vulkan

Initial value:

= {
    .p.name         = "nlmeans_vulkan",
    .p.description  = NULL_IF_CONFIG_SMALL("Non-local means denoiser (Vulkan)"),
    .p.priv_class   = &nlmeans_vulkan_class,
    .p.flags        = AVFILTER_FLAG_HWDEVICE,
    .priv_size      = sizeof(NLMeansVulkanContext),
    .init           = &ff_vk_filter_init,
    .uninit         = &nlmeans_vulkan_uninit,
    FILTER_INPUTS(nlmeans_vulkan_inputs),
    FILTER_OUTPUTS(nlmeans_vulkan_outputs),
    FILTER_SINGLE_PIXFMT(AV_PIX_FMT_VULKAN),
    .flags_internal = FF_FILTER_FLAG_HWFRAME_AWARE,
}

Definition at line 1214 of file vf_nlmeans_vulkan.c.