avfilter/vf_estdif: allow to change two more options

This commit is contained in:
Paul B Mahol 2021-11-10 10:19:27 +01:00
parent 946493eb3e
commit aebdffb9c5
2 changed files with 18 additions and 5 deletions

View File

@ -11711,6 +11711,14 @@ Allowed range is from 1 to 15.
Specify the search radius for best edge matching. Default value is 2.
Allowed range is from 0 to 15.
@item ecost
Specify the edge cost for edge matching. Default value is 0.03125.
Allowed range is from 0 to 1.
@item mcost
Specify the middle cost for edge matching. Default value is 0.5.
Allowed range is from 0 to 1.
@item dcost
Specify the distance cost for edge matching. Default value is 0.5.
Allowed range is from 0 to 1.

View File

@ -35,6 +35,8 @@ typedef struct ESTDIFContext {
int deint; ///< which frames to deinterlace
int rslope; ///< best edge slope search radius
int redge; ///< best edge match search radius
float ecost; ///< edge cost for edge matching
float mcost; ///< middle cost for edge matching
float dcost; ///< distance cost for edge matching
int interp; ///< type of interpolation
int linesize[4]; ///< bytes of pixel data per line for each plane
@ -93,6 +95,8 @@ static const AVOption estdif_options[] = {
CONST("interlaced", "only deinterlace frames marked as interlaced", 1, "deint"),
{ "rslope", "specify the search radius for edge slope tracing", OFFSET(rslope), AV_OPT_TYPE_INT, {.i64=1}, 1, MAX_R, FLAGS, },
{ "redge", "specify the search radius for best edge matching", OFFSET(redge), AV_OPT_TYPE_INT, {.i64=2}, 0, MAX_R, FLAGS, },
{ "ecost", "specify the edge cost for edge matching", OFFSET(ecost), AV_OPT_TYPE_FLOAT,{.dbl=0.03125},0,1,FLAGS, },
{ "mcost", "specify the middle cost for edge matching", OFFSET(mcost), AV_OPT_TYPE_FLOAT,{.dbl=0.5}, 0, 1, FLAGS, },
{ "dcost", "specify the distance cost for edge matching", OFFSET(dcost), AV_OPT_TYPE_FLOAT,{.dbl=0.5}, 0, 1, FLAGS, },
{ "interp", "specify the type of interpolation", OFFSET(interp), AV_OPT_TYPE_INT, {.i64=1}, 0, 2, FLAGS, "interp" },
CONST("2p", "two-point interpolation", 0, "interp"),
@ -260,9 +264,10 @@ static void interpolate_##ss(ESTDIFContext *s, uint8_t *ddst, \
const type *const next2_line = (const type *const)nnext2_line; \
const type *const next3_line = (const type *const)nnext3_line; \
const int interp = s->interp; \
const int ecost = s->ecost * 32.f; \
const int dcost = s->dcost * s->max; \
const int end = width - 1; \
const atype f = redge + 2; \
const atype mcost = s->mcost * s->redge * 4.f; \
atype sd[S], sD[S], di = 0; \
atype dmin = amax; \
int k = *K; \
@ -278,8 +283,8 @@ static void interpolate_##ss(ESTDIFContext *s, uint8_t *ddst, \
sum += diff_##ss(next_line, next2_line, xx, yy); \
} \
\
sD[i + rslope] = sum; \
sD[i + rslope] += f * cost_##ss(prev_line, next_line, end, x, i); \
sD[i + rslope] = ecost * sum; \
sD[i + rslope] += mcost * cost_##ss(prev_line, next_line, end, x, i);\
sD[i + rslope] += dcost * abs(i); \
\
dmin = FFMIN(sD[i + rslope], dmin); \
@ -296,8 +301,8 @@ static void interpolate_##ss(ESTDIFContext *s, uint8_t *ddst, \
sum += diff_##ss(next_line, next2_line, xx, yy); \
} \
\
sd[i + rslope] = sum; \
sd[i + rslope] += f * cost_##ss(prev_line, next_line, end, x, k + i); \
sd[i + rslope] = ecost * sum; \
sd[i + rslope] += mcost * cost_##ss(prev_line, next_line, end, x, k+i);\
sd[i + rslope] += dcost * abs(k + i); \
\
dmin = FFMIN(sd[i + rslope], dmin); \