avfilter/vsrc_testsrc: smpte(hd)bars: use yuv directly
Also set color space. Signed-off-by: Paul B Mahol <onemda@gmail.com>
This commit is contained in:
Paul B Mahol
parent
1d8ce109e9
commit
cfc9a4c732
|
|
@ -793,52 +793,54 @@ AVFilter avfilter_vsrc_rgbtestsrc = {
|
|||
#if CONFIG_SMPTEBARS_FILTER || CONFIG_SMPTEHDBARS_FILTER
|
||||
|
||||
static const uint8_t rainbow[7][4] = {
|
||||
{ 191, 191, 191, 255 }, /* gray */
|
||||
{ 191, 191, 0, 255 }, /* yellow */
|
||||
{ 0, 191, 191, 255 }, /* cyan */
|
||||
{ 0, 191, 0, 255 }, /* green */
|
||||
{ 191, 0, 191, 255 }, /* magenta */
|
||||
{ 191, 0, 0, 255 }, /* red */
|
||||
{ 0, 0, 191, 255 }, /* blue */
|
||||
{ 180, 128, 128, 255 }, /* gray */
|
||||
{ 168, 44, 136, 255 }, /* yellow */
|
||||
{ 145, 147, 44, 255 }, /* cyan */
|
||||
{ 133, 63, 52, 255 }, /* green */
|
||||
{ 63, 193, 204, 255 }, /* magenta */
|
||||
{ 51, 109, 212, 255 }, /* red */
|
||||
{ 28, 212, 120, 255 }, /* blue */
|
||||
};
|
||||
|
||||
static const uint8_t wobnair[7][4] = {
|
||||
{ 0, 0, 191, 255 }, /* blue */
|
||||
{ 19, 19, 19, 255 }, /* 7.5% intensity black */
|
||||
{ 191, 0, 191, 255 }, /* magenta */
|
||||
{ 19, 19, 19, 255 }, /* 7.5% intensity black */
|
||||
{ 0, 191, 191, 255 }, /* cyan */
|
||||
{ 19, 19, 19, 255 }, /* 7.5% intensity black */
|
||||
{ 191, 191, 191, 255 }, /* gray */
|
||||
{ 32, 240, 118, 255 }, /* blue */
|
||||
{ 19, 128, 128, 255 }, /* 7.5% intensity black */
|
||||
{ 54, 184, 198, 255 }, /* magenta */
|
||||
{ 19, 128, 128, 255 }, /* 7.5% intensity black */
|
||||
{ 188, 154, 16, 255 }, /* cyan */
|
||||
{ 19, 128, 128, 255 }, /* 7.5% intensity black */
|
||||
{ 191, 128, 128, 255 }, /* gray */
|
||||
};
|
||||
|
||||
static const uint8_t white[4] = { 255, 255, 255, 255 };
|
||||
static const uint8_t black[4] = { 19, 19, 19, 255 }; /* 7.5% intensity black */
|
||||
static const uint8_t white[4] = { 235, 128, 128, 255 };
|
||||
static const uint8_t black[4] = { 19, 128, 128, 255 }; /* 7.5% intensity black */
|
||||
|
||||
/* pluge pulses */
|
||||
static const uint8_t neg4ire[4] = { 9, 9, 9, 255 }; /* 3.5% intensity black */
|
||||
static const uint8_t pos4ire[4] = { 29, 29, 29, 255 }; /* 11.5% intensity black */
|
||||
static const uint8_t neg4ire[4] = { 9, 128, 128, 255 }; /* 3.5% intensity black */
|
||||
static const uint8_t pos4ire[4] = { 29, 128, 128, 255 }; /* 11.5% intensity black */
|
||||
|
||||
/* fudged Q/-I */
|
||||
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 i_pixel[4] = { 61, 153, 99, 255 };
|
||||
static const uint8_t q_pixel[4] = { 35, 174, 152, 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 const uint8_t gray40[4] = { 104, 128, 128, 255 };
|
||||
static const uint8_t gray15[4] = { 49, 128, 128, 255 };
|
||||
static const uint8_t cyan[4] = { 188, 154, 16, 255 };
|
||||
static const uint8_t yellow[4] = { 219, 16, 138, 255 };
|
||||
static const uint8_t blue[4] = { 32, 240, 118, 255 };
|
||||
static const uint8_t red[4] = { 63, 102, 240, 255 };
|
||||
static const uint8_t black0[4] = { 16, 128, 128, 255 };
|
||||
static const uint8_t black2[4] = { 20, 128, 128, 255 };
|
||||
static const uint8_t black4[4] = { 25, 128, 128, 255 };
|
||||
static const uint8_t neg2[4] = { 12, 128, 128, 255 };
|
||||
|
||||
static void inline draw_bar(TestSourceContext *test, const uint8_t *color,
|
||||
unsigned x, unsigned y, unsigned w, unsigned h,
|
||||
AVFrame *frame)
|
||||
static void draw_bar(TestSourceContext *test, const uint8_t color[4],
|
||||
unsigned x, unsigned y, unsigned w, unsigned h,
|
||||
AVFrame *frame)
|
||||
{
|
||||
FFDrawColor draw_color;
|
||||
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame->format);
|
||||
uint8_t *p, *p0;
|
||||
int plane;
|
||||
|
||||
x = FFMIN(x, test->w - 1);
|
||||
y = FFMIN(y, test->h - 1);
|
||||
|
|
@ -848,33 +850,49 @@ static void inline draw_bar(TestSourceContext *test, const uint8_t *color,
|
|||
av_assert0(x + w <= test->w);
|
||||
av_assert0(y + h <= test->h);
|
||||
|
||||
ff_draw_color(&test->draw, &draw_color, color);
|
||||
ff_fill_rectangle(&test->draw, &draw_color,
|
||||
frame->data, frame->linesize, x, y, w, h);
|
||||
for (plane = 0; frame->data[plane]; plane++) {
|
||||
const int c = color[plane];
|
||||
const int linesize = frame->linesize[plane];
|
||||
int i, px, py, pw, ph;
|
||||
|
||||
if (plane == 1 || plane == 2) {
|
||||
px = x >> desc->log2_chroma_w;
|
||||
pw = w >> desc->log2_chroma_w;
|
||||
py = y >> desc->log2_chroma_h;
|
||||
ph = h >> desc->log2_chroma_h;
|
||||
} else {
|
||||
px = x;
|
||||
pw = w;
|
||||
py = y;
|
||||
ph = h;
|
||||
}
|
||||
|
||||
p0 = p = frame->data[plane] + py * linesize + px;
|
||||
memset(p, c, pw);
|
||||
p += linesize;
|
||||
for (i = 1; i < ph; i++, p += linesize)
|
||||
memcpy(p, p0, pw);
|
||||
}
|
||||
}
|
||||
|
||||
static int smptebars_query_formats(AVFilterContext *ctx)
|
||||
{
|
||||
ff_set_common_formats(ctx, ff_draw_supported_pixel_formats(0));
|
||||
static const enum AVPixelFormat pix_fmts[] = {
|
||||
AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P,
|
||||
AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV444P,
|
||||
AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV411P,
|
||||
AV_PIX_FMT_NONE,
|
||||
};
|
||||
ff_set_common_formats(ctx, ff_make_format_list(pix_fmts));
|
||||
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,
|
||||
.config_props = config_props,
|
||||
},
|
||||
{ NULL }
|
||||
};
|
||||
|
|
@ -890,6 +908,8 @@ static void smptebars_fill_picture(AVFilterContext *ctx, AVFrame *picref)
|
|||
int r_w, r_h, w_h, p_w, p_h, i, tmp, x = 0;
|
||||
const AVPixFmtDescriptor *pixdesc = av_pix_fmt_desc_get(picref->format);
|
||||
|
||||
av_frame_set_colorspace(picref, AVCOL_SPC_BT470BG);
|
||||
|
||||
r_w = FFALIGN((test->w + 6) / 7, 1 << pixdesc->log2_chroma_w);
|
||||
r_h = FFALIGN(test->h * 2 / 3, 1 << pixdesc->log2_chroma_h);
|
||||
w_h = FFALIGN(test->h * 3 / 4 - r_h, 1 << pixdesc->log2_chroma_h);
|
||||
|
|
@ -955,6 +975,8 @@ static void smptehdbars_fill_picture(AVFilterContext *ctx, AVFrame *picref)
|
|||
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);
|
||||
|
||||
av_frame_set_colorspace(picref, AVCOL_SPC_BT709);
|
||||
|
||||
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);
|
||||
|
|
@ -986,9 +1008,9 @@ static void smptehdbars_fill_picture(AVFilterContext *ctx, AVFrame *picref)
|
|||
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[0] = i * 255 / tmp;
|
||||
yramp[1] = 128;
|
||||
yramp[2] = 128;
|
||||
yramp[3] = 255;
|
||||
|
||||
draw_bar(test, yramp, x, y, 1 << pixdesc->log2_chroma_w, r_h, picref);
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user