virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-08-15 14:13:16 +02:00
Code Issues Releases Activity

avfilter/vf_readeia608: use special struct to hold line items

Browse Source
This commit is contained in:
Paul B Mahol
2019-12-23 12:28:30 +01:00
parent 0115dbd043
commit 16968b619d
1 changed files with 70 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

150
libavfilter/vf_readeia608.c
View File

@@ -40,6 +40,16 @@
#define SYNC_MIN 12.f #define SYNC_MIN 12.f
#define SYNC_MAX 15.f #define SYNC_MAX 15.f
typedef struct LineItem {
int input;
int output;
float unfiltered;
float filtered;
float average;
float deviation;
} LineItem;
typedef struct CodeItem { typedef struct CodeItem {
uint8_t bit; uint8_t bit;
int size; int size;
@@ -57,13 +67,8 @@ typedef struct ReadEIA608Context {
uint64_t histogram[256]; uint64_t histogram[256];
uint8_t *temp;
uint8_t *signal;
CodeItem *code; CodeItem *code;
float *unfiltered; LineItem *line;
float *filtered;
float *avg_filter;
float *std_filter;
} ReadEIA608Context; } ReadEIA608Context;
#define OFFSET(x) offsetof(ReadEIA608Context, x) #define OFFSET(x) offsetof(ReadEIA608Context, x)
@@ -114,26 +119,20 @@ static int config_input(AVFilterLink *inlink)
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
s->unfiltered = av_calloc(size, sizeof(*s->unfiltered)); s->line = av_calloc(size, sizeof(*s->line));
s->filtered = av_calloc(size, sizeof(*s->filtered));
s->avg_filter = av_calloc(size, sizeof(*s->avg_filter));
s->std_filter = av_calloc(size, sizeof(*s->std_filter));
s->signal = av_calloc(size, sizeof(*s->signal));
s->code = av_calloc(size, sizeof(*s->code)); s->code = av_calloc(size, sizeof(*s->code));
s->temp = av_calloc(size, sizeof(*s->temp)); if (!s->line || !s->code)
if (!s->unfiltered || !s->filtered || !s->avg_filter ||
!s->std_filter || !s->signal || !s->code || !s->temp)
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
return 0; return 0;
} }
static void build_histogram(ReadEIA608Context *s, const uint8_t *src, int len) static void build_histogram(ReadEIA608Context *s, const LineItem *line, int len)
{ {
memset(s->histogram, 0, sizeof(s->histogram)); memset(s->histogram, 0, sizeof(s->histogram));
for (int i = 0; i < len; i++) for (int i = LAG; i < len + LAG; i++)
s->histogram[src[i]]++; s->histogram[line[i].input]++;
} }
static void find_black_and_white(ReadEIA608Context *s) static void find_black_and_white(ReadEIA608Context *s)
@@ -178,87 +177,86 @@ static void find_black_and_white(ReadEIA608Context *s)
s->white = white; s->white = white;
} }
static float meanf(float *data, int len) static float meanf(const LineItem *line, int len)
{ {
float sum = 0.0, mean = 0.0; float sum = 0.0, mean = 0.0;
for (int i = 0; i < len; i++) for (int i = 0; i < len; i++)
sum += data[i]; sum += line[i].filtered;
mean = sum / len; mean = sum / len;
return mean; return mean;
} }
static float stddevf(float *data, int len) static float stddevf(const LineItem *line, int len)
{ {
float m = meanf(data, len); float m = meanf(line, len);
float standard_deviation = 0.f; float standard_deviation = 0.f;
for (int i = 0; i < len; i++) for (int i = 0; i < len; i++)
standard_deviation += (data[i] - m) * (data[i] - m); standard_deviation += (line[i].filtered - m) * (line[i].filtered - m);
return sqrtf(standard_deviation / (len - 1)); return sqrtf(standard_deviation / (len - 1));
} }
static void thresholding(ReadEIA608Context *s, const uint8_t *y, uint8_t *signal, static void thresholding(ReadEIA608Context *s, LineItem *line,
float *unfiltered, float *filtered, float *avg_filter, float *std_filter,
int lag, float threshold, float influence, int len) int lag, float threshold, float influence, int len)
{ {
for (int i = lag; i < len + lag; i++) { for (int i = lag; i < len + lag; i++) {
unfiltered[i] = y[i - lag] / 255.f; line[i].unfiltered = line[i].input / 255.f;
filtered[i] = unfiltered[i]; line[i].filtered = line[i].unfiltered;
} }
for (int i = 0; i < lag; i++) { for (int i = 0; i < lag; i++) {
unfiltered[i] = meanf(unfiltered, len * s->spw); line[i].unfiltered = meanf(line, len * s->spw);
filtered[i] = unfiltered[i]; line[i].filtered = line[i].unfiltered;
} }
memset(signal, 0, len); line[lag - 1].average = meanf(line, lag);
line[lag - 1].deviation = stddevf(line, lag);
avg_filter[lag - 1] = meanf(unfiltered, lag);
std_filter[lag - 1] = stddevf(unfiltered, lag);
for (int i = lag; i < len + lag; i++) { for (int i = lag; i < len + lag; i++) {
if (fabsf(unfiltered[i] - avg_filter[i-1]) > threshold * std_filter[i-1]) { if (fabsf(line[i].unfiltered - line[i-1].average) > threshold * line[i-1].deviation) {
if (unfiltered[i] > avg_filter[i-1]) { if (line[i].unfiltered > line[i-1].average) {
signal[i - lag] = 255; line[i].output = 255;
} else { } else {
signal[i - lag] = 0; line[i].output = 0;
} }
filtered[i] = influence * unfiltered[i] + (1.f - influence) * filtered[i-1];
line[i].filtered = influence * line[i].unfiltered + (1.f - influence) * line[i-1].filtered;
} else { } else {
int distance_from_black, distance_from_white; int distance_from_black, distance_from_white;
distance_from_black = FFABS(y[i - lag] - s->black); distance_from_black = FFABS(line[i].input - s->black);
distance_from_white = FFABS(y[i - lag] - s->white); distance_from_white = FFABS(line[i].input - s->white);
signal[i - lag] = distance_from_black <= distance_from_white ? 0 : 255; line[i].output = distance_from_black <= distance_from_white ? 0 : 255;
} }
avg_filter[i] = meanf(filtered + i - lag, lag);
std_filter[i] = stddevf(filtered + i - lag, lag); line[i].average = meanf(line + i - lag, lag);
line[i].deviation = stddevf(line + i - lag, lag);
} }
} }
static int periods(const uint8_t *signal, CodeItem *code, int len) static int periods(const LineItem *line, CodeItem *code, int len)
{ {
int hold = signal[0], cnt = 0; int hold = line[LAG].output, cnt = 0;
int last = 0; int last = LAG;
memset(code, 0, len * sizeof(*code)); memset(code, 0, len * sizeof(*code));
for (int i = 1; i < len; i++) { for (int i = LAG + 1; i < len + LAG; i++) {
if (signal[i] != hold) { if (line[i].output != hold) {
code[cnt].size = i - last; code[cnt].size = i - last;
code[cnt].bit = hold; code[cnt].bit = hold;
hold = signal[i]; hold = line[i].output;
last = i; last = i;
cnt++; cnt++;
} }
} }
code[cnt].size = len - last; code[cnt].size = LAG + len - last;
code[cnt].bit = hold; code[cnt].bit = hold;
return cnt + 1; return cnt + 1;
@@ -275,9 +273,10 @@ static void dump_code(AVFilterContext *ctx, int len, int item)
av_log(ctx, AV_LOG_DEBUG, "\n"); av_log(ctx, AV_LOG_DEBUG, "\n");
} }
static void extract_line(AVFilterContext *ctx, AVFilterLink *inlink, AVFrame *in, int line) static void extract_line(AVFilterContext *ctx, AVFrame *in, int w, int nb_line)
{ {
ReadEIA608Context *s = ctx->priv; ReadEIA608Context *s = ctx->priv;
LineItem *line = s->line;
int i, j, ch, len; int i, j, ch, len;
const uint8_t *src; const uint8_t *src;
uint8_t byte[2] = { 0 }; uint8_t byte[2] = { 0 };
@@ -285,40 +284,38 @@ static void extract_line(AVFilterContext *ctx, AVFilterLink *inlink, AVFrame *in
float bit_size = 0.f; float bit_size = 0.f;
int parity; int parity;
src = &in->data[0][line * in->linesize[0]]; memset(line, 0, (w + LAG) * sizeof(*line));
if (s->lp) {
uint8_t *dst = s->temp;
int w = inlink->w - 1;
for (i = 0; i < inlink->w; i++) { src = &in->data[0][nb_line * in->linesize[0]];
if (s->lp) {
for (i = 0; i < w; i++) {
int a = FFMAX(i - 3, 0); int a = FFMAX(i - 3, 0);
int b = FFMAX(i - 2, 0); int b = FFMAX(i - 2, 0);
int c = FFMAX(i - 1, 0); int c = FFMAX(i - 1, 0);
int d = FFMIN(i + 3, w); int d = FFMIN(i + 3, w-1);
int e = FFMIN(i + 2, w); int e = FFMIN(i + 2, w-1);
int f = FFMIN(i + 1, w); int f = FFMIN(i + 1, w-1);
dst[i] = (src[a] + src[b] + src[c] + src[i] + src[d] + src[e] + src[f] + 6) / 7; line[LAG + i].input = (src[a] + src[b] + src[c] + src[i] + src[d] + src[e] + src[f] + 6) / 7;
}
} else {
for (i = 0; i < w; i++) {
line[LAG + i].input = src[i];
}
} }
src = s->temp; build_histogram(s, line, w);
}
build_histogram(s, src, inlink->w);
find_black_and_white(s); find_black_and_white(s);
if (s->white - s->black < 5) if (s->white - s->black < 5)
return; return;
thresholding(s, src, s->signal, s->unfiltered, s->filtered, thresholding(s, line, LAG, 1, 0, w);
s->avg_filter, s->std_filter, len = periods(line, s->code, w);
LAG, 1, 0, inlink->w); dump_code(ctx, len, nb_line);
//memcpy(&in->data[0][line * in->linesize[0]], s->signal, inlink->w);
len = periods(s->signal, s->code, inlink->w);
dump_code(ctx, len, line);
if (len < 15 || if (len < 15 ||
s->code[14].bit != 0 || s->code[14].bit != 0 ||
inlink->w / (float)s->code[14].size < SYNC_MIN || w / (float)s->code[14].size < SYNC_MIN ||
inlink->w / (float)s->code[14].size > SYNC_MAX) { w / (float)s->code[14].size > SYNC_MAX) {
return; return;
} }
@@ -383,7 +380,7 @@ static void extract_line(AVFilterContext *ctx, AVFilterLink *inlink, AVFrame *in
av_dict_set(&in->metadata, key, value, 0); av_dict_set(&in->metadata, key, value, 0);
snprintf(key, sizeof(key), "lavfi.readeia608.%d.line", s->nb_found); snprintf(key, sizeof(key), "lavfi.readeia608.%d.line", s->nb_found);
snprintf(value, sizeof(value), "%d", line); snprintf(value, sizeof(value), "%d", nb_line);
av_dict_set(&in->metadata, key, value, 0); av_dict_set(&in->metadata, key, value, 0);
} }
@@ -399,7 +396,7 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
s->nb_found = 0; s->nb_found = 0;
for (i = s->start; i <= s->end; i++) for (i = s->start; i <= s->end; i++)
extract_line(ctx, inlink, in, i); extract_line(ctx, in, inlink->w, i);
return ff_filter_frame(outlink, in); return ff_filter_frame(outlink, in);
} }
@@ -408,13 +405,8 @@ static av_cold void uninit(AVFilterContext *ctx)
{ {
ReadEIA608Context *s = ctx->priv; ReadEIA608Context *s = ctx->priv;
av_freep(&s->temp);
av_freep(&s->code); av_freep(&s->code);
av_freep(&s->signal); av_freep(&s->line);
av_freep(&s->unfiltered);
av_freep(&s->filtered);
av_freep(&s->avg_filter);
av_freep(&s->std_filter);
} }
static const AVFilterPad readeia608_inputs[] = { static const AVFilterPad readeia608_inputs[] = {