|
|
|
@ -30,7 +30,7 @@ |
|
|
|
|
* rgbtestsrc is ported from MPlayer libmpcodecs/vf_rgbtest.c by |
|
|
|
|
* Michael Niedermayer. |
|
|
|
|
* |
|
|
|
|
* smptebars is by Paul B Mahol. |
|
|
|
|
* smptebars and smptehdbars is by Paul B Mahol. |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
#include <float.h> |
|
|
|
@ -654,10 +654,7 @@ AVFilter avfilter_vsrc_rgbtestsrc = { |
|
|
|
|
|
|
|
|
|
#endif /* CONFIG_RGBTESTSRC_FILTER */ |
|
|
|
|
|
|
|
|
|
#if CONFIG_SMPTEBARS_FILTER |
|
|
|
|
|
|
|
|
|
#define smptebars_options options |
|
|
|
|
AVFILTER_DEFINE_CLASS(smptebars); |
|
|
|
|
#if CONFIG_SMPTEBARS_FILTER || CONFIG_SMPTEHDBARS_FILTER |
|
|
|
|
|
|
|
|
|
static const uint8_t rainbow[7][4] = { |
|
|
|
|
{ 191, 191, 191, 255 }, /* gray */ |
|
|
|
@ -690,6 +687,17 @@ static const uint8_t pos4ire[4] = { 29, 29, 29, 255 }; /* 11.5% intensity bla |
|
|
|
|
static const uint8_t i_pixel[4] = { 0, 68, 130, 255 }; |
|
|
|
|
static const uint8_t q_pixel[4] = { 67, 0, 130, 255 }; |
|
|
|
|
|
|
|
|
|
static const uint8_t gray40[4] = { 102, 102, 102, 255 }; |
|
|
|
|
static const uint8_t gray15[4] = { 38, 38, 38, 255 }; |
|
|
|
|
static const uint8_t cyan[4] = { 0, 255, 255, 255 }; |
|
|
|
|
static const uint8_t yellow[4] = { 255, 255, 0, 255 }; |
|
|
|
|
static const uint8_t blue[4] = { 0, 0, 255, 255 }; |
|
|
|
|
static const uint8_t red[4] = { 255, 0, 0, 255 }; |
|
|
|
|
static const uint8_t black0[4] = { 5, 5, 5, 255 }; |
|
|
|
|
static const uint8_t black2[4] = { 10, 10, 10, 255 }; |
|
|
|
|
static const uint8_t black4[4] = { 15, 15, 15, 255 }; |
|
|
|
|
static const uint8_t neg2[4] = { 0, 0, 0, 255 }; |
|
|
|
|
|
|
|
|
|
static void inline draw_bar(TestSourceContext *test, const uint8_t *color, |
|
|
|
|
unsigned x, unsigned y, unsigned w, unsigned h, |
|
|
|
|
AVFrame *frame) |
|
|
|
@ -709,6 +717,37 @@ static void inline draw_bar(TestSourceContext *test, const uint8_t *color, |
|
|
|
|
frame->data, frame->linesize, x, y, w, h); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static int smptebars_query_formats(AVFilterContext *ctx) |
|
|
|
|
{ |
|
|
|
|
ff_set_common_formats(ctx, ff_draw_supported_pixel_formats(0)); |
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static int smptebars_config_props(AVFilterLink *outlink) |
|
|
|
|
{ |
|
|
|
|
AVFilterContext *ctx = outlink->src; |
|
|
|
|
TestSourceContext *test = ctx->priv; |
|
|
|
|
|
|
|
|
|
ff_draw_init(&test->draw, outlink->format, 0); |
|
|
|
|
|
|
|
|
|
return config_props(outlink); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static const AVFilterPad smptebars_outputs[] = { |
|
|
|
|
{ |
|
|
|
|
.name = "default", |
|
|
|
|
.type = AVMEDIA_TYPE_VIDEO, |
|
|
|
|
.request_frame = request_frame, |
|
|
|
|
.config_props = smptebars_config_props, |
|
|
|
|
}, |
|
|
|
|
{ NULL } |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
#if CONFIG_SMPTEHDBARS_FILTER |
|
|
|
|
|
|
|
|
|
#define smptebars_options options |
|
|
|
|
AVFILTER_DEFINE_CLASS(smptebars); |
|
|
|
|
|
|
|
|
|
static void smptebars_fill_picture(AVFilterContext *ctx, AVFrame *picref) |
|
|
|
|
{ |
|
|
|
|
TestSourceContext *test = ctx->priv; |
|
|
|
@ -755,37 +794,115 @@ static av_cold int smptebars_init(AVFilterContext *ctx) |
|
|
|
|
return init(ctx); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static int smptebars_query_formats(AVFilterContext *ctx) |
|
|
|
|
{ |
|
|
|
|
ff_set_common_formats(ctx, ff_draw_supported_pixel_formats(0)); |
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|
AVFilter avfilter_vsrc_smptebars = { |
|
|
|
|
.name = "smptebars", |
|
|
|
|
.description = NULL_IF_CONFIG_SMALL("Generate SMPTE color bars."), |
|
|
|
|
.priv_size = sizeof(TestSourceContext), |
|
|
|
|
.init = smptebars_init, |
|
|
|
|
.uninit = uninit, |
|
|
|
|
|
|
|
|
|
static int smptebars_config_props(AVFilterLink *outlink) |
|
|
|
|
.query_formats = smptebars_query_formats, |
|
|
|
|
.inputs = NULL, |
|
|
|
|
.outputs = smptebars_outputs, |
|
|
|
|
.priv_class = &smptebars_class, |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
#endif /* CONFIG_SMPTEBARS_FILTER */ |
|
|
|
|
|
|
|
|
|
#if CONFIG_SMPTEHDBARS_FILTER |
|
|
|
|
|
|
|
|
|
#define smptehdbars_options options |
|
|
|
|
AVFILTER_DEFINE_CLASS(smptehdbars); |
|
|
|
|
|
|
|
|
|
static void smptehdbars_fill_picture(AVFilterContext *ctx, AVFrame *picref) |
|
|
|
|
{ |
|
|
|
|
AVFilterContext *ctx = outlink->src; |
|
|
|
|
TestSourceContext *test = ctx->priv; |
|
|
|
|
int d_w, r_w, r_h, l_w, i, tmp, x = 0, y = 0; |
|
|
|
|
const AVPixFmtDescriptor *pixdesc = av_pix_fmt_desc_get(picref->format); |
|
|
|
|
|
|
|
|
|
ff_draw_init(&test->draw, outlink->format, 0); |
|
|
|
|
d_w = FFALIGN(test->w / 8, 1 << pixdesc->log2_chroma_w); |
|
|
|
|
r_h = FFALIGN(test->h * 7 / 12, 1 << pixdesc->log2_chroma_h); |
|
|
|
|
draw_bar(test, gray40, x, 0, d_w, r_h, picref); |
|
|
|
|
x += d_w; |
|
|
|
|
|
|
|
|
|
return config_props(outlink); |
|
|
|
|
r_w = FFALIGN((((test->w + 3) / 4) * 3) / 7, 1 << pixdesc->log2_chroma_w); |
|
|
|
|
for (i = 0; i < 7; i++) { |
|
|
|
|
draw_bar(test, rainbow[i], x, 0, r_w, r_h, picref); |
|
|
|
|
x += r_w; |
|
|
|
|
} |
|
|
|
|
draw_bar(test, gray40, x, 0, test->w - x, r_h, picref); |
|
|
|
|
y = r_h; |
|
|
|
|
r_h = FFALIGN(test->h / 12, 1 << pixdesc->log2_chroma_h); |
|
|
|
|
draw_bar(test, cyan, 0, y, d_w, r_h, picref); |
|
|
|
|
x = d_w; |
|
|
|
|
draw_bar(test, i_pixel, x, y, r_w, r_h, picref); |
|
|
|
|
x += r_w; |
|
|
|
|
tmp = r_w * 6; |
|
|
|
|
draw_bar(test, rainbow[0], x, y, tmp, r_h, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
l_w = x; |
|
|
|
|
draw_bar(test, blue, x, y, test->w - x, r_h, picref); |
|
|
|
|
y += r_h; |
|
|
|
|
draw_bar(test, yellow, 0, y, d_w, r_h, picref); |
|
|
|
|
x = d_w; |
|
|
|
|
draw_bar(test, q_pixel, x, y, r_w, r_h, picref); |
|
|
|
|
x += r_w; |
|
|
|
|
|
|
|
|
|
static const AVFilterPad smptebars_outputs[] = { |
|
|
|
|
for (i = 0; i < tmp; i += 1 << pixdesc->log2_chroma_w) { |
|
|
|
|
uint8_t yramp[4] = {0}; |
|
|
|
|
|
|
|
|
|
yramp[0] = |
|
|
|
|
yramp[1] = |
|
|
|
|
yramp[2] = i * 255 / tmp; |
|
|
|
|
yramp[3] = 255; |
|
|
|
|
|
|
|
|
|
draw_bar(test, yramp, x, y, 1 << pixdesc->log2_chroma_w, r_h, picref); |
|
|
|
|
x += 1 << pixdesc->log2_chroma_w; |
|
|
|
|
} |
|
|
|
|
draw_bar(test, red, x, y, test->w - x, r_h, picref); |
|
|
|
|
y += r_h; |
|
|
|
|
draw_bar(test, gray15, 0, y, d_w, test->h - y, picref); |
|
|
|
|
x = d_w; |
|
|
|
|
tmp = FFALIGN(r_w * 3 / 2, 1 << pixdesc->log2_chroma_w); |
|
|
|
|
draw_bar(test, black0, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
tmp = FFALIGN(r_w * 2, 1 << pixdesc->log2_chroma_w); |
|
|
|
|
draw_bar(test, white, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
tmp = FFALIGN(r_w * 5 / 6, 1 << pixdesc->log2_chroma_w); |
|
|
|
|
draw_bar(test, black0, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
tmp = FFALIGN(r_w / 3, 1 << pixdesc->log2_chroma_w); |
|
|
|
|
draw_bar(test, neg2, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
draw_bar(test, black0, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
draw_bar(test, black2, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
draw_bar(test, black0, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
draw_bar(test, black4, x, y, tmp, test->h - y, picref); |
|
|
|
|
x += tmp; |
|
|
|
|
r_w = l_w - x; |
|
|
|
|
draw_bar(test, black0, x, y, r_w, test->h - y, picref); |
|
|
|
|
x += r_w; |
|
|
|
|
draw_bar(test, gray15, x, y, test->w - x, test->h - y, picref); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static av_cold int smptehdbars_init(AVFilterContext *ctx) |
|
|
|
|
{ |
|
|
|
|
.name = "default", |
|
|
|
|
.type = AVMEDIA_TYPE_VIDEO, |
|
|
|
|
.request_frame = request_frame, |
|
|
|
|
.config_props = smptebars_config_props, |
|
|
|
|
}, |
|
|
|
|
{ NULL } |
|
|
|
|
}; |
|
|
|
|
TestSourceContext *test = ctx->priv; |
|
|
|
|
|
|
|
|
|
AVFilter avfilter_vsrc_smptebars = { |
|
|
|
|
.name = "smptebars", |
|
|
|
|
.description = NULL_IF_CONFIG_SMALL("Generate SMPTE color bars."), |
|
|
|
|
test->fill_picture_fn = smptehdbars_fill_picture; |
|
|
|
|
test->draw_once = 1; |
|
|
|
|
return init(ctx); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
AVFilter avfilter_vsrc_smptehdbars = { |
|
|
|
|
.name = "smptehdbars", |
|
|
|
|
.description = NULL_IF_CONFIG_SMALL("Generate SMPTE HD color bars."), |
|
|
|
|
.priv_size = sizeof(TestSourceContext), |
|
|
|
|
.init = smptebars_init, |
|
|
|
|
.init = smptehdbars_init, |
|
|
|
|
.uninit = uninit, |
|
|
|
|
|
|
|
|
|
.query_formats = smptebars_query_formats, |
|
|
|
@ -794,4 +911,5 @@ AVFilter avfilter_vsrc_smptebars = { |
|
|
|
|
.priv_class = &smptebars_class, |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
#endif /* CONFIG_SMPTEBARS_FILTER */ |
|
|
|
|
#endif /* CONFIG_SMPTEHDBARS_FILTER */ |
|
|
|
|
#endif /* CONFIG_SMPTEHDBARS_FILTER || CONFIG_SMPTEHDBARS_FILTER */ |
|
|
|
|