Updated to fit with half life patch
On Sat, Nov 1, 2014 at 7:12 AM, Kevin Mitchell <kevmi...@gmail.com> wrote:
>
From d31cbea4430043ad71f8ad3f099ee8cbbc997875 Mon Sep 17 00:00:00 2001
From: Kevin Mitchell <kevmi...@gmail.com>
Date: Sat, 1 Nov 2014 05:38:35 -0700
Subject: [PATCH 2/2] avfilter/vf_idet: add a repeated field detection
This can be useful for determining telecine.
---
doc/filters.texi | 18 ++++++++++++++++--
libavfilter/version.h | 2 +-
libavfilter/vf_idet.c | 40 +++++++++++++++++++++++++++++++++++++++-
libavfilter/vf_idet.h | 8 ++++++++
4 files changed, 64 insertions(+), 4 deletions(-)
diff --git a/doc/filters.texi b/doc/filters.texi
index b12beae..5a5ac87 100644
--- a/doc/filters.texi
+++ b/doc/filters.texi
@@ -5574,8 +5574,9 @@ value.
Detect video interlacing type.
-This filter tries to detect the input frames as interlaced, progressive,
-top or bottom field first.
+This filter tries to detect if the input frames as interlaced, progressive,
+top or bottom field first. It will also try and detect fields that are
+repeated between adjacent frames (a sign of telecine).
There are two detection modes. Single frame detection considers only
immediately adjacent frames when classifying each frame. Multiple frame
@@ -5621,6 +5622,17 @@ Cumulative number of frames that could not be classified using single-frame dete
@item multiple.undetermined
Cumulative number of frames that could not be classified using multiple-frame detection.
+@item repeated.current_frame
+Which field in the current frame is repeated from the last. One of ``neither'', ``top'', or ``bottom''.
+
+@item repeated.neither
+Cumulative number of frames with no repeated field.
+
+@item repeated.top
+Cumulative number of frames with the top field repeated from the previous frame's top field.
+
+@item repeated.bottom
+Cumulative number of frames with the bottom field repeated from the previous frame's bottom field.
@end table
The filter accepts the following options:
@@ -5630,6 +5642,8 @@ The filter accepts the following options:
Set interlacing threshold.
@item prog_thres
Set progressive threshold.
+@item repeat_thres
+Threshold for repeated field detection.
@item half_life
Number of frames after which a given frame's contribution to the
statistics is halved (i.e., it contributes only 0.5 to it's
diff --git a/libavfilter/version.h b/libavfilter/version.h
index dab9b45..6f61aee 100644
--- a/libavfilter/version.h
+++ b/libavfilter/version.h
@@ -31,7 +31,7 @@
#define LIBAVFILTER_VERSION_MAJOR 5
#define LIBAVFILTER_VERSION_MINOR 2
-#define LIBAVFILTER_VERSION_MICRO 102
+#define LIBAVFILTER_VERSION_MICRO 103
#define LIBAVFILTER_VERSION_INT AV_VERSION_INT(LIBAVFILTER_VERSION_MAJOR, \
LIBAVFILTER_VERSION_MINOR, \
diff --git a/libavfilter/vf_idet.c b/libavfilter/vf_idet.c
index b6a42ce..ad1fa54 100644
--- a/libavfilter/vf_idet.c
+++ b/libavfilter/vf_idet.c
@@ -32,6 +32,7 @@
static const AVOption idet_options[] = {
{ "intl_thres", "set interlacing threshold", OFFSET(interlace_threshold), AV_OPT_TYPE_FLOAT, {.dbl = 1.04}, -1, FLT_MAX, FLAGS },
{ "prog_thres", "set progressive threshold", OFFSET(progressive_threshold), AV_OPT_TYPE_FLOAT, {.dbl = 1.5}, -1, FLT_MAX, FLAGS },
+ { "rep_thres", "set repeat threshold", OFFSET(repeat_threshold), AV_OPT_TYPE_FLOAT, {.dbl = 3.0}, -1, FLT_MAX, FLAGS },
{ "half_life", "half life of cumulative statistics", OFFSET(half_life), AV_OPT_TYPE_FLOAT, { .dbl = 0.0 }, -1, INT_MAX, FLAGS },
{ NULL }
};
@@ -57,6 +58,16 @@ static int av_dict_set_float(AVDictionary **pm, const char *key, float value,
return av_dict_set(pm, key, valuestr, flags);
}
+static const char *rep2str(RepeatedField repeated_field)
+{
+ switch(repeated_field) {
+ case REPEAT_NONE : return "neither";
+ case REPEAT_TOP : return "top";
+ case REPEAT_BOTTOM : return "bottom";
+ }
+ return NULL;
+}
+
int ff_idet_filter_line_c(const uint8_t *a, const uint8_t *b, const uint8_t *c, int w)
{
int x;
@@ -87,6 +98,12 @@ static void log_cumulative_stats(AVFilterContext *ctx)
{
IDETContext *idet = ctx->priv;
+ av_log(ctx, AV_LOG_INFO, "Repeated Fields: Neither:%6.1f Top:%6.1f Bottom:%6.1f Total: %6.1f\n",
+ idet->repeats[REPEAT_NONE],
+ idet->repeats[REPEAT_TOP],
+ idet->repeats[REPEAT_BOTTOM],
+ idet->frames_total
+ );
av_log(ctx, AV_LOG_INFO, "Single frame detection: TFF:%6.1f BFF:%6.1f Progressive:%6.1f Undetermined:%6.1f Total: %6.1f\n",
idet->prestat[TFF],
idet->prestat[BFF],
@@ -109,7 +126,9 @@ static void filter(AVFilterContext *ctx)
int y, i;
int64_t alpha[2]={0};
int64_t delta=0;
+ int64_t gamma[2]={0};
Type type, best_type;
+ RepeatedField repeat;
int match = 0;
AVDictionary **metadata = avpriv_frame_get_metadatap(idet->cur);
@@ -130,6 +149,7 @@ static void filter(AVFilterContext *ctx)
alpha[ y &1] += idet->filter_line(cur-refs, prev, cur+refs, w);
alpha[(y^1)&1] += idet->filter_line(cur-refs, next, cur+refs, w);
delta += idet->filter_line(cur-refs, cur, cur+refs, w);
+ gamma[(y^1)&1] += idet->filter_line(cur , prev, cur , w);
}
}
@@ -143,6 +163,14 @@ static void filter(AVFilterContext *ctx)
type = UNDETERMINED;
}
+ if ( gamma[0] > idet->repeat_threshold * gamma[1] ){
+ repeat = REPEAT_TOP;
+ } else if ( gamma[1] > idet->repeat_threshold * gamma[0] ){
+ repeat = REPEAT_BOTTOM;
+ } else {
+ repeat = REPEAT_NONE;
+ }
+
memmove(idet->history+1, idet->history, HIST_SIZE-1);
idet->history[0] = type;
best_type = UNDETERMINED;
@@ -176,6 +204,10 @@ static void filter(AVFilterContext *ctx)
}
idet->frames_total *= idet->decay_coefficient;
+
+ for(i=0; i<3; i++)
+ idet->repeats[i] *= idet->decay_coefficient;
+
for(i=0; i<4; i++){
idet->prestat [i] *= idet->decay_coefficient;
idet->poststat[i] *= idet->decay_coefficient;
@@ -184,11 +216,17 @@ static void filter(AVFilterContext *ctx)
idet->frames_total += 1.0;
idet->prestat [ type] += 1.0;
idet->poststat[idet->last_type] += 1.0;
+ idet->repeats [ repeat] += 1.0;
- av_log(ctx, AV_LOG_DEBUG, "Single frame:%12s, Multi frame:%12s\n", type2str(type), type2str(idet->last_type));
+ av_log(ctx, AV_LOG_DEBUG, "Repeated Field:%12s, Single frame:%12s, Multi frame:%12s\n", rep2str(repeat), type2str(type), type2str(idet->last_type));
av_dict_set_float(metadata, "lavfi.idet.frames_total", idet->frames_total, 0);
+ av_dict_set (metadata, "lavfi.idet.repeated.current_frame", rep2str(repeat), 0);
+ av_dict_set_float(metadata, "lavfi.idet.repeated.neither", idet->repeats[REPEAT_NONE], 0);
+ av_dict_set_float(metadata, "lavfi.idet.repeated.top", idet->repeats[REPEAT_TOP], 0);
+ av_dict_set_float(metadata, "lavfi.idet.repeated.bottom", idet->repeats[REPEAT_BOTTOM], 0);
+
av_dict_set (metadata, "lavfi.idet.single.current_frame", type2str(type), 0);
av_dict_set_float(metadata, "lavfi.idet.single.tff", idet->prestat[TFF], 0);
av_dict_set_float(metadata, "lavfi.idet.single.bff", idet->prestat[BFF], 0);
diff --git a/libavfilter/vf_idet.h b/libavfilter/vf_idet.h
index 005ac67..2acd0af 100644
--- a/libavfilter/vf_idet.h
+++ b/libavfilter/vf_idet.h
@@ -33,15 +33,23 @@ typedef enum {
UNDETERMINED,
} Type;
+typedef enum {
+ REPEAT_NONE,
+ REPEAT_TOP,
+ REPEAT_BOTTOM,
+} RepeatedField;
+
typedef struct {
const AVClass *class;
float interlace_threshold;
float progressive_threshold;
+ float repeat_threshold;
float half_life;
float decay_coefficient;
Type last_type;
float frames_total;
+ float repeats[3];
float prestat[4];
float poststat[4];
--
2.1.1
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
http://ffmpeg.org/mailman/listinfo/ffmpeg-devel
