Chiebot-Mirror
/
FFmpeg
mirror of https://github.com/FFmpeg/FFmpeg.git
8
0
Fork 0
Code Issues Projects Releases Wiki Activity

avfilter/vf_waveform: add graticule support

Browse Source 
Signed-off-by: Paul B Mahol <onemda@gmail.com>
pull/184/head
Paul B Mahol 9 years ago
parent f3c00be2a3
commit 361e48f0ed
2 changed files with 187 additions and 0 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 14
      doc/filters.texi
  2. 173
      libavfilter/vf_waveform.c

14
doc/filters.texi
Unescape View File

@ -13161,6 +13161,20 @@ Similar as above, but shows difference between blue and red chroma.
@item color
Displays actual color value on waveform.
@end table
@item graticule, g
Set which graticule to display.
@table @samp
@item none
Do not display graticule.
@item green
Display green graticule showing legal broadcast ranges.
@end table
@item opacity, o
Set graticule opacity.
@end table
@section xbr

173
libavfilter/vf_waveform.c
Unescape View File

@ -49,6 +49,8 @@ typedef struct WaveformContext {
int mirror;
int display;
int envelope;
int graticule;
float opacity;
int estart[4];
int eend[4];
int *emax[4][4];
@ -60,6 +62,7 @@ typedef struct WaveformContext {
int size;
void (*waveform)(struct WaveformContext *s, AVFrame *in, AVFrame *out,
int component, int intensity, int offset, int column);
void (*graticulef)(struct WaveformContext *s, AVFrame *out);
const AVPixFmtDescriptor *desc;
} WaveformContext;
@ -95,6 +98,12 @@ static const AVOption waveform_options[] = {
{ "chroma", NULL, 0, AV_OPT_TYPE_CONST, {.i64=CHROMA}, 0, 0, FLAGS, "filter" },
{ "achroma", NULL, 0, AV_OPT_TYPE_CONST, {.i64=ACHROMA}, 0, 0, FLAGS, "filter" },
{ "color", NULL, 0, AV_OPT_TYPE_CONST, {.i64=COLOR}, 0, 0, FLAGS, "filter" },
{ "graticule", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "graticule" },
{ "g", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=0}, 0, 1, FLAGS, "graticule" },
{ "none", NULL, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "graticule" },
{ "green", NULL, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "graticule" },
{ "opacity", "set graticule opacity", OFFSET(opacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, FLAGS },
{ "o", "set graticule opacity", OFFSET(opacity), AV_OPT_TYPE_FLOAT, {.dbl=0.75}, 0, 1, FLAGS },
{ NULL }
};
@ -1081,7 +1090,163 @@ static void color(WaveformContext *s, AVFrame *in, AVFrame *out,
}
static const uint8_t black_yuva_color[4] = { 0, 127, 127, 255 };
static const uint8_t green_yuva_color[4] = { 255, 0, 0, 255 };
static const uint8_t black_gbrp_color[4] = { 0, 0, 0, 255 };
static const uint8_t lines[4][2] = { { 16, 235 }, { 16, 240 }, { 16, 240 }, { 0, 255 } };
static void blend_vline(uint8_t *dst, int height, int linesize, float o1, float o2, int v)
{
int y;
for (y = 0; y < height; y++) {
dst[0] = v * o1 + dst[0] * o2;
dst += linesize;
}
}
static void blend_vline16(uint16_t *dst, int height, int linesize, float o1, float o2, int v)
{
int y;
for (y = 0; y < height; y++) {
dst[0] = v * o1 + dst[0] * o2;
dst += linesize / 2;
}
}
static void blend_hline(uint8_t *dst, int width, float o1, float o2, int v)
{
int x;
for (x = 0; x < width; x++) {
dst[x] = v * o1 + dst[x] * o2;
}
}
static void blend_hline16(uint16_t *dst, int width, float o1, float o2, int v)
{
int x;
for (x = 0; x < width; x++) {
dst[x] = v * o1 + dst[x] * o2;
}
}
static void graticule_none(WaveformContext *s, AVFrame *out)
{
}
static void graticule_green_row(WaveformContext *s, AVFrame *out)
{
const float o1 = s->opacity;
const float o2 = 1. - o1;
int c, p, l, offset = 0;
for (c = 0; c < s->ncomp; c++) {
if (!((1 << c) & s->pcomp))
continue;
for (p = 0; p < s->ncomp; p++) {
const int is_chroma = (p == 1 || p == 2);
const int shift_w = (is_chroma ? s->desc->log2_chroma_w : 0);
const int shift_h = (is_chroma ? s->desc->log2_chroma_h : 0);
const int v = green_yuva_color[p];
for (l = 0; l < FF_ARRAY_ELEMS(lines[0]); l++) {
int x = offset + (s->mirror ? 255 - lines[c][l] : lines[c][l]);
uint8_t *dst = out->data[p] + (x >> shift_w);
blend_vline(dst, out->height >> shift_h, out->linesize[p], o1, o2, v);
}
}
offset += 256 * s->display;
}
}
static void graticule16_green_row(WaveformContext *s, AVFrame *out)
{
const float o1 = s->opacity;
const float o2 = 1. - o1;
const int mult = s->size / 256;
int c, p, l, offset = 0;
for (c = 0; c < s->ncomp; c++) {
if (!((1 << c) & s->pcomp))
continue;
for (p = 0; p < s->ncomp; p++) {
const int is_chroma = (p == 1 || p == 2);
const int shift_w = (is_chroma ? s->desc->log2_chroma_w : 0);
const int shift_h = (is_chroma ? s->desc->log2_chroma_h : 0);
const int v = green_yuva_color[p] * mult;
for (l = 0; l < FF_ARRAY_ELEMS(lines[0]); l++) {
int x = offset + (s->mirror ? 255 - lines[c][l] : lines[c][l]) * mult;
uint16_t *dst = (uint16_t *)(out->data[p]) + (x >> shift_w);
blend_vline16(dst, out->height >> shift_h, out->linesize[p], o1, o2, v);
}
}
offset += s->size * s->display;
}
}
static void graticule_green_column(WaveformContext *s, AVFrame *out)
{
const float o1 = s->opacity;
const float o2 = 1. - o1;
int c, p, l, offset = 0;
for (c = 0; c < s->ncomp; c++) {
if (!((1 << c) & s->pcomp))
continue;
for (p = 0; p < s->ncomp; p++) {
const int is_chroma = (p == 1 || p == 2);
const int shift_w = (is_chroma ? s->desc->log2_chroma_w : 0);
const int shift_h = (is_chroma ? s->desc->log2_chroma_h : 0);
const int v = green_yuva_color[p];
for (l = 0; l < FF_ARRAY_ELEMS(lines[0]); l++) {
int y = offset + (s->mirror ? 255 - lines[c][l] : lines[c][l]);
uint8_t *dst = out->data[p] + (y >> shift_h) * out->linesize[p];
blend_hline(dst, out->width >> shift_w, o1, o2, v);
}
}
offset += 256 * s->display;
}
}
static void graticule16_green_column(WaveformContext *s, AVFrame *out)
{
const float o1 = s->opacity;
const float o2 = 1. - o1;
const int mult = s->size / 256;
int c, p, l, offset = 0;
for (c = 0; c < s->ncomp; c++) {
if (!((1 << c) & s->pcomp))
continue;
for (p = 0; p < s->ncomp; p++) {
const int is_chroma = (p == 1 || p == 2);
const int shift_w = (is_chroma ? s->desc->log2_chroma_w : 0);
const int shift_h = (is_chroma ? s->desc->log2_chroma_h : 0);
const int v = green_yuva_color[p] * mult;
for (l = 0; l < FF_ARRAY_ELEMS(lines[0]); l++) {
int y = offset + (s->mirror ? 255 - lines[c][l] : lines[c][l]) * mult;
uint16_t *dst = (uint16_t *)(out->data[p] + (y >> shift_h) * out->linesize[p]);
blend_hline16(dst, out->width >> shift_w, o1, o2, v);
}
}
offset += s->size * s->display;
}
}
static int config_input(AVFilterLink *inlink)
{
@ -1094,9 +1259,15 @@ static int config_input(AVFilterLink *inlink)
s->max = 1 << s->bits;
s->intensity = s->fintensity * (s->max - 1);
s->graticulef = graticule_none;
switch (s->filter) {
case LOWPASS:
s->size = 256;
if (s->graticule && s->mode == 1)
s->graticulef = s->bits > 8 ? graticule16_green_column : graticule_green_column;
else if (s->graticule && s->mode == 0)
s->graticulef = s->bits > 8 ? graticule16_green_row : graticule_green_row;
s->waveform = s->bits > 8 ? lowpass16 : lowpass; break;
case FLAT:
s->size = 256 * 3;
@ -1124,6 +1295,7 @@ static int config_input(AVFilterLink *inlink)
case AV_PIX_FMT_GBRP10:
case AV_PIX_FMT_GBRP12:
s->bg_color = black_gbrp_color;
s->graticulef = graticule_none;
break;
default:
s->bg_color = black_yuva_color;
@ -1233,6 +1405,7 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
s->waveform(s, in, out, k, s->intensity, offset, s->mode);
}
}
s->graticulef(s, out);
av_frame_free(&in);
return ff_filter_frame(outlink, out);

Write Preview
Loading…
Cancel
Save