1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00

avfilter/vf_hqdn3d: add support for commands

This commit is contained in:
Paul B Mahol 2019-11-21 15:50:35 +01:00
parent 3a61297a67
commit 8c2f81a17a
2 changed files with 38 additions and 9 deletions

View File

@ -11765,6 +11765,13 @@ A floating point number which specifies chroma temporal strength. It defaults to
@var{luma_tmp}*@var{chroma_spatial}/@var{luma_spatial}. @var{luma_tmp}*@var{chroma_spatial}/@var{luma_spatial}.
@end table @end table
@subsection Commands
This filter supports same @ref{commands} as options.
The command accepts the same syntax of the corresponding option.
If the specified expression is not valid, it is kept at its current
value.
@anchor{hwdownload} @anchor{hwdownload}
@section hwdownload @section hwdownload

View File

@ -172,13 +172,10 @@ static int denoise_depth(HQDN3DContext *s,
} \ } \
} while (0) } while (0)
static int16_t *precalc_coefs(double dist25, int depth) static void precalc_coefs(double dist25, int depth, int16_t *ct)
{ {
int i; int i;
double gamma, simil, C; double gamma, simil, C;
int16_t *ct = av_malloc((512<<LUT_BITS)*sizeof(int16_t));
if (!ct)
return NULL;
gamma = log(0.25) / log(1.0 - FFMIN(dist25,252.0)/255.0 - 0.00001); gamma = log(0.25) / log(1.0 - FFMIN(dist25,252.0)/255.0 - 0.00001);
@ -190,7 +187,6 @@ static int16_t *precalc_coefs(double dist25, int depth)
} }
ct[0] = !!dist25; ct[0] = !!dist25;
return ct;
} }
#define PARAM1_DEFAULT 4.0 #define PARAM1_DEFAULT 4.0
@ -254,17 +250,26 @@ static int query_formats(AVFilterContext *ctx)
return ff_set_common_formats(ctx, fmts_list); return ff_set_common_formats(ctx, fmts_list);
} }
static void calc_coefs(AVFilterContext *ctx)
{
HQDN3DContext *s = ctx->priv;
for (int i = 0; i < 4; i++)
precalc_coefs(s->strength[i], s->depth, s->coefs[i]);
}
static int config_input(AVFilterLink *inlink) static int config_input(AVFilterLink *inlink)
{ {
AVFilterContext *ctx = inlink->dst;
HQDN3DContext *s = inlink->dst->priv; HQDN3DContext *s = inlink->dst->priv;
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format); const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
int i; int i, depth;
uninit(inlink->dst); uninit(inlink->dst);
s->hsub = desc->log2_chroma_w; s->hsub = desc->log2_chroma_w;
s->vsub = desc->log2_chroma_h; s->vsub = desc->log2_chroma_h;
s->depth = desc->comp[0].depth; s->depth = depth = desc->comp[0].depth;
for (i = 0; i < 3; i++) { for (i = 0; i < 3; i++) {
s->line[i] = av_malloc_array(inlink->w, sizeof(*s->line[i])); s->line[i] = av_malloc_array(inlink->w, sizeof(*s->line[i]));
@ -273,11 +278,13 @@ static int config_input(AVFilterLink *inlink)
} }
for (i = 0; i < 4; i++) { for (i = 0; i < 4; i++) {
s->coefs[i] = precalc_coefs(s->strength[i], s->depth); s->coefs[i] = av_malloc((512<<LUT_BITS) * sizeof(int16_t));
if (!s->coefs[i]) if (!s->coefs[i])
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
} }
calc_coefs(ctx);
if (ARCH_X86) if (ARCH_X86)
ff_hqdn3d_init_x86(s); ff_hqdn3d_init_x86(s);
@ -346,8 +353,22 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
return ff_filter_frame(outlink, out); return ff_filter_frame(outlink, out);
} }
static int process_command(AVFilterContext *ctx, const char *cmd, const char *args,
char *res, int res_len, int flags)
{
int ret;
ret = ff_filter_process_command(ctx, cmd, args, res, res_len, flags);
if (ret < 0)
return ret;
calc_coefs(ctx);
return 0;
}
#define OFFSET(x) offsetof(HQDN3DContext, x) #define OFFSET(x) offsetof(HQDN3DContext, x)
#define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM #define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_RUNTIME_PARAM
static const AVOption hqdn3d_options[] = { static const AVOption hqdn3d_options[] = {
{ "luma_spatial", "spatial luma strength", OFFSET(strength[LUMA_SPATIAL]), AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS }, { "luma_spatial", "spatial luma strength", OFFSET(strength[LUMA_SPATIAL]), AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS },
{ "chroma_spatial", "spatial chroma strength", OFFSET(strength[CHROMA_SPATIAL]), AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS }, { "chroma_spatial", "spatial chroma strength", OFFSET(strength[CHROMA_SPATIAL]), AV_OPT_TYPE_DOUBLE, { .dbl = 0.0 }, 0, DBL_MAX, FLAGS },
@ -388,4 +409,5 @@ AVFilter ff_vf_hqdn3d = {
.inputs = avfilter_vf_hqdn3d_inputs, .inputs = avfilter_vf_hqdn3d_inputs,
.outputs = avfilter_vf_hqdn3d_outputs, .outputs = avfilter_vf_hqdn3d_outputs,
.flags = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL | AVFILTER_FLAG_SLICE_THREADS, .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_INTERNAL | AVFILTER_FLAG_SLICE_THREADS,
.process_command = process_command,
}; };