Chiebot-Mirror
/
FFmpeg
8
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

avfilter/vf_vectorscope: add graticule

Browse Source 
Signed-off-by: Paul B Mahol <onemda@gmail.com>
pull/189/head
Paul B Mahol 9 years ago
parent 85dd497baa
commit a7897bd3a6
2 changed files with 288 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. 22
      doc/filters.texi
  2. 266
      libavfilter/vf_vectorscope.c

22
doc/filters.texi
Unescape View File

@ -12559,6 +12559,28 @@ can still spot out of range values without constantly looking at vectorscope.
@item peak+instant
Peak and instant envelope combined together.
@end table
@item graticule, g
Set what kind of graticule to draw.
@table @samp
@item none
@item green
@item color
@end table
@item opacity, o
Set graticule opacity.
@item flags, f
Set graticule flags.
@table @samp
@item white
Draw graticule for white point.
@item black
Draw graticule for black point.
@end table
@end table
@anchor{vidstabdetect}

266
libavfilter/vf_vectorscope.c
Unescape View File

@ -50,12 +50,19 @@ typedef struct VectorscopeContext {
int x, y, pd;
int is_yuv;
int size;
int depth;
int mult;
int envelope;
int graticule;
float opacity;
int flags;
int cs;
uint8_t peak[1024][1024];
void (*vectorscope)(struct VectorscopeContext *s,
AVFrame *in, AVFrame *out, int pd);
void (*graticulef)(struct VectorscopeContext *s, AVFrame *out,
int X, int Y, int D, int P);
} VectorscopeContext;
#define OFFSET(x) offsetof(VectorscopeContext, x)
@ -80,6 +87,17 @@ static const AVOption vectorscope_options[] = {
{ "instant", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "envelope" },
{ "peak", 0, 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, FLAGS, "envelope" },
{ "peak+instant", 0, 0, AV_OPT_TYPE_CONST, {.i64=3}, 0, 0, FLAGS, "envelope" },
{ "graticule", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=0}, 0, 2, FLAGS, "graticule"},
{ "g", "set graticule", OFFSET(graticule), AV_OPT_TYPE_INT, {.i64=0}, 0, 2, FLAGS, "graticule"},
{ "none", 0, 0, AV_OPT_TYPE_CONST, {.i64=0}, 0, 0, FLAGS, "graticule" },
{ "green", 0, 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "graticule" },
{ "color", 0, 0, AV_OPT_TYPE_CONST, {.i64=2}, 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},
{ "flags", "set graticule flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64=0}, 0, 3, FLAGS, "flags"},
{ "f", "set graticule flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64=0}, 0, 3, FLAGS, "flags"},
{ "white", "draw white point", 0, AV_OPT_TYPE_CONST, {.i64=1}, 0, 0, FLAGS, "flags" },
{ "black", "draw black point", 0, AV_OPT_TYPE_CONST, {.i64=2}, 0, 0, FLAGS, "flags" },
{ NULL }
};
@ -662,6 +680,228 @@ static void vectorscope8(VectorscopeContext *s, AVFrame *in, AVFrame *out, int p
}
}
const static uint16_t positions[][14][3] = {
{ { 210, 16, 146 }, { 170, 166, 16 }, { 145, 54, 34 },
{ 106, 202, 222 }, { 81, 90, 240 }, { 41, 240, 110 },
{ 162, 44, 142 }, { 131, 156, 44 }, { 112, 72, 58 },
{ 84, 184, 198 }, { 65, 100, 212 }, { 35, 212, 114 },
{ 235, 128, 128 }, { 16, 128, 128 } },
{ { 63, 102, 240 }, { 32, 240, 118 }, { 188, 154, 16 },
{ 219, 16, 138 }, { 173, 42, 26 }, { 78, 214, 230 },
{ 28, 212, 120 }, { 51, 109, 212 }, { 63, 193, 204 },
{ 133, 63, 52 }, { 145, 147, 44 }, { 168, 44, 136 },
{ 235, 128, 128 }, { 16, 128, 128 } },
{ { 210*2, 16*2, 146*2 }, { 170*2, 166*2, 16*2 }, { 145*2, 54*2, 34*2 },
{ 106*2, 202*2, 222*2 }, { 81*2, 90*2, 240*2 }, { 41*2, 240*2, 110*2 },
{ 162*2, 44*2, 142*2 }, { 131*2, 156*2, 44*2 }, { 112*2, 72*2, 58*2 },
{ 84*2, 184*2, 198*2 }, { 65*2, 100*2, 212*2 }, { 35*2, 212*2, 114*2 },
{ 470, 256, 256 }, { 32, 256, 256 } },
{ { 63*2, 102*2, 240*2 }, { 32*2, 240*2, 118*2 }, { 188*2, 154*2, 16*2 },
{ 219*2, 16*2, 138*2 }, { 173*2, 42*2, 26*2 }, { 78*2, 214*2, 230*2 },
{ 28*2, 212*2, 120*2 }, { 51*2, 109*2, 212*2 }, { 63*2, 193*2, 204*2 },
{ 133*2, 63*2, 52*2 }, { 145*2, 147*2, 44*2 }, { 168*2, 44*2, 136*2 },
{ 470, 256, 256 }, { 32, 256, 256 } },
{ { 210*4, 16*4, 146*4 }, { 170*4, 166*4, 16*4 }, { 145*4, 54*4, 34*4 },
{ 106*4, 202*4, 222*4 }, { 81*4, 90*4, 240*4 }, { 41*4, 240*4, 110*4 },
{ 162*4, 44*4, 142*4 }, { 131*4, 156*4, 44*4 }, { 112*4, 72*4, 58*4 },
{ 84*4, 184*4, 198*4 }, { 65*4, 100*4, 212*4 }, { 35*4, 212*4, 114*4 },
{ 940, 512, 512 }, { 64, 512, 512 } },
{ { 63*4, 102*4, 240*4 }, { 32*4, 240*4, 118*4 }, { 188*4, 154*4, 16*4 },
{ 219*4, 16*4, 138*4 }, { 173*4, 42*4, 26*4 }, { 78*4, 214*4, 230*4 },
{ 28*4, 212*4, 120*4 }, { 51*4, 109*4, 212*4 }, { 63*4, 193*4, 204*4 },
{ 133*4, 63*4, 52*4 }, { 145*4, 147*4, 44*4 }, { 168*4, 44*4, 136*4 },
{ 940, 512, 512 }, { 64, 512, 512 } },
};
static void draw_dots(uint8_t *dst, int L, int v, float o)
{
const float f = 1. - o;
const float V = o * v;
int l = L * 2;
dst[ l - 3] = dst[ l - 3] * f + V;
dst[ l + 3] = dst[ l + 3] * f + V;
dst[-l - 3] = dst[-l - 3] * f + V;
dst[-l + 3] = dst[-l + 3] * f + V;
l += L;
dst[ l - 3] = dst[ l - 3] * f + V;
dst[ l + 3] = dst[ l + 3] * f + V;
dst[ l - 2] = dst[ l - 2] * f + V;
dst[ l + 2] = dst[ l + 2] * f + V;
dst[-l - 3] = dst[-l - 3] * f + V;
dst[-l + 3] = dst[-l + 3] * f + V;
dst[-l - 2] = dst[-l - 2] * f + V;
dst[-l + 2] = dst[-l + 2] * f + V;
}
static void draw_dots16(uint16_t *dst, int L, int v, float o)
{
const float f = 1. - o;
const float V = o * v;
int l = L * 2;
dst[ l - 3] = dst[ l - 3] * f + V;
dst[ l + 3] = dst[ l + 3] * f + V;
dst[-l - 3] = dst[-l - 3] * f + V;
dst[-l + 3] = dst[-l + 3] * f + V;
l += L;
dst[ l - 3] = dst[ l - 3] * f + V;
dst[ l + 3] = dst[ l + 3] * f + V;
dst[ l - 2] = dst[ l - 2] * f + V;
dst[ l + 2] = dst[ l + 2] * f + V;
dst[-l - 3] = dst[-l - 3] * f + V;
dst[-l + 3] = dst[-l + 3] * f + V;
dst[-l - 2] = dst[-l - 2] * f + V;
dst[-l + 2] = dst[-l + 2] * f + V;
}
static void none_graticule(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
{
}
static void color_graticule16(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
{
const float o = s->opacity;
int i;
for (i = 0; i < 12; i++) {
int x = positions[P][i][X];
int y = positions[P][i][Y];
int d = positions[P][i][D];
draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, d, o);
draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, x, o);
draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, y, o);
}
if (s->flags & 1) {
int x = positions[P][12][X];
int y = positions[P][12][Y];
int d = positions[P][12][D];
draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, d, o);
draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, x, o);
draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, y, o);
}
if (s->flags & 2) {
int x = positions[P][13][X];
int y = positions[P][13][Y];
int d = positions[P][13][D];
draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, d, o);
draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, x, o);
draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, y, o);
}
}
static void color_graticule(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
{
const float o = s->opacity;
int i;
for (i = 0; i < 12; i++) {
int x = positions[P][i][X];
int y = positions[P][i][Y];
int d = positions[P][i][D];
draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], d, o);
draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], x, o);
draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], y, o);
}
if (s->flags & 1) {
int x = positions[P][12][X];
int y = positions[P][12][Y];
int d = positions[P][12][D];
draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], d, o);
draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], x, o);
draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], y, o);
}
if (s->flags & 2) {
int x = positions[P][13][X];
int y = positions[P][13][Y];
int d = positions[P][12][D];
draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], d, o);
draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], x, o);
draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], y, o);
}
}
static void green_graticule16(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
{
const float o = s->opacity;
const int m = s->mult;
int i;
for (i = 0; i < 12; i++) {
int x = positions[P][i][X];
int y = positions[P][i][Y];
draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, 128 * m, o);
draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, 0, o);
draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, 0, o);
}
if (s->flags & 1) {
int x = positions[P][12][X];
int y = positions[P][12][Y];
draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, 128 * m, o);
draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, 0, o);
draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, 0, o);
}
if (s->flags & 2) {
int x = positions[P][13][X];
int y = positions[P][13][Y];
draw_dots16((uint16_t *)(out->data[D] + y * out->linesize[D] + x * 2), out->linesize[D] / 2, 128 * m, o);
draw_dots16((uint16_t *)(out->data[X] + y * out->linesize[X] + x * 2), out->linesize[X] / 2, 0, o);
draw_dots16((uint16_t *)(out->data[Y] + y * out->linesize[Y] + x * 2), out->linesize[Y] / 2, 0, o);
}
}
static void green_graticule(VectorscopeContext *s, AVFrame *out, int X, int Y, int D, int P)
{
const float o = s->opacity;
int i;
for (i = 0; i < 12; i++) {
int x = positions[P][i][X];
int y = positions[P][i][Y];
draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], 128, o);
draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], 0, o);
draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], 0, o);
}
if (s->flags & 1) {
int x = positions[P][12][X];
int y = positions[P][12][Y];
draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], 128, o);
draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], 0, o);
draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], 0, o);
}
if (s->flags & 2) {
int x = positions[P][13][X];
int y = positions[P][13][Y];
draw_dots(out->data[D] + y * out->linesize[D] + x, out->linesize[D], 128, o);
draw_dots(out->data[X] + y * out->linesize[X] + x, out->linesize[X], 0, o);
draw_dots(out->data[Y] + y * out->linesize[Y] + x, out->linesize[Y], 0, o);
}
}
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
{
AVFilterContext *ctx = inlink->dst;
@ -669,6 +909,16 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
AVFilterLink *outlink = ctx->outputs[0];
AVFrame *out;
switch (av_frame_get_colorspace(in)) {
case AVCOL_SPC_SMPTE170M:
case AVCOL_SPC_BT470BG:
s->cs = (s->depth - 8) * 2 + 0;
break;
case AVCOL_SPC_BT709:
default:
s->cs = (s->depth - 8) * 2 + 1;
}
out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
if (!out) {
av_frame_free(&in);
@ -677,6 +927,7 @@ static int filter_frame(AVFilterLink *inlink, AVFrame *in)
out->pts = in->pts;
s->vectorscope(s, in, out, s->pd);
s->graticulef(s, out, s->x, s->y, s->pd, s->cs);
av_frame_free(&in);
return ff_filter_frame(outlink, out);
@ -690,6 +941,7 @@ static int config_input(AVFilterLink *inlink)
s->is_yuv = !(desc->flags & AV_PIX_FMT_FLAG_RGB);
s->size = 1 << desc->comp[0].depth;
s->mult = s->size / 256;
s->depth = desc->comp[0].depth;
if (s->mode == GRAY && s->is_yuv)
s->pd = 0;
@ -707,6 +959,20 @@ static int config_input(AVFilterLink *inlink)
else
s->vectorscope = vectorscope16;
s->graticulef = none_graticule;
if (s->is_yuv && s->size == 256) {
if (s->graticule == 1)
s->graticulef = green_graticule;
else if (s->graticule == 2)
s->graticulef = color_graticule;
} else if (s->is_yuv) {
if (s->graticule == 1)
s->graticulef = green_graticule16;
else if (s->graticule == 2)
s->graticulef = color_graticule16;
}
switch (inlink->format) {
case AV_PIX_FMT_GBRP10:
case AV_PIX_FMT_GBRP9:

Write Preview
Loading…
Cancel
Save