[FFmpeg-devel] [PATCH] smptebars filter

Thierry Foucu tfoucu at gmail.com
Wed Jun 20 10:19:42 CEST 2012


On Wed, Jun 20, 2012 at 9:54 AM, Paul B Mahol <onemda at gmail.com> wrote:

> Signed-off-by: Paul B Mahol <onemda at gmail.com>
> ---
>  doc/filters.texi             |   20 ++++
>  libavfilter/Makefile         |    1 +
>  libavfilter/allfilters.c     |    1 +
>  libavfilter/vsrc_smptebars.c |  215
> ++++++++++++++++++++++++++++++++++++++++++
>  4 files changed, 237 insertions(+), 0 deletions(-)
>  create mode 100644 libavfilter/vsrc_smptebars.c
>
> diff --git a/doc/filters.texi b/doc/filters.texi
> index 566fbfc..0c0c490 100644
> --- a/doc/filters.texi
> +++ b/doc/filters.texi
> @@ -3799,6 +3799,26 @@ the @code{mp=geq} filter:
>  nullsrc=s=256x256, mp=geq=random(1)*255:128:128
>  @end example
>
> + at section smptebars
> +
> +Generate SMPTE color bars.
>


Am i correct to assume that this SMPTE color bar is the SD version.

For HD, I though to remember the SMPTE color bar is different.
see http://en.wikipedia.org/wiki/File:SMPTE_Color_Bars_16x9.svg

If so, should we not call this one SMPTE_SD? or add both into the same
filter?


> +
> +This source accepts a list of options in the form of
> + at var{key}=@var{value} pairs separated by ":". A description of the
> +accepted options follows.
> +
> + at table @option
> + at item size, s
> +Set the size of the output video.
> +Default is 720x480.
> +
> + at item rate, r
> +Set the video rate, that is the number of frames generated per second.
> +Default is 30.
> +
> + at item sar
> +Set the sample aspect ratio of the sourced video.
> +
>  @c man end VIDEO SOURCES
>
>  @chapter Video Sinks
> diff --git a/libavfilter/Makefile b/libavfilter/Makefile
> index 72bb640..193bfa1 100644
> --- a/libavfilter/Makefile
> +++ b/libavfilter/Makefile
> @@ -136,6 +136,7 @@ OBJS-$(CONFIG_MOVIE_FILTER)                  +=
> src_movie.o
>  OBJS-$(CONFIG_MPTESTSRC_FILTER)              += vsrc_mptestsrc.o
>  OBJS-$(CONFIG_NULLSRC_FILTER)                += vsrc_testsrc.o
>  OBJS-$(CONFIG_RGBTESTSRC_FILTER)             += vsrc_testsrc.o
> +OBJS-$(CONFIG_SMPTEBARS_FILTER)              += vsrc_smptebars.o
>  OBJS-$(CONFIG_TESTSRC_FILTER)                += vsrc_testsrc.o
>
>  OBJS-$(CONFIG_BUFFERSINK_FILTER)             += sink_buffer.o
> diff --git a/libavfilter/allfilters.c b/libavfilter/allfilters.c
> index 35065d5..a4e354b 100644
> --- a/libavfilter/allfilters.c
> +++ b/libavfilter/allfilters.c
> @@ -124,6 +124,7 @@ void avfilter_register_all(void)
>     REGISTER_FILTER (MPTESTSRC,   mptestsrc,   vsrc);
>     REGISTER_FILTER (NULLSRC,     nullsrc,     vsrc);
>     REGISTER_FILTER (RGBTESTSRC,  rgbtestsrc,  vsrc);
> +    REGISTER_FILTER (SMPTEBARS,   smptebars,   vsrc);
>     REGISTER_FILTER (TESTSRC,     testsrc,     vsrc);
>
>     REGISTER_FILTER (BUFFERSINK,  buffersink,  vsink);
> diff --git a/libavfilter/vsrc_smptebars.c b/libavfilter/vsrc_smptebars.c
> new file mode 100644
> index 0000000..70157e8
> --- /dev/null
> +++ b/libavfilter/vsrc_smptebars.c
> @@ -0,0 +1,215 @@
> +/*
> + * Copyright (c) 2012 Paul B Mahol
> + *
> + * This file is part of FFmpeg.
> + *
> + * FFmpeg is free software; you can redistribute it and/or
> + * modify it under the terms of the GNU Lesser General Public
> + * License as published by the Free Software Foundation; either
> + * version 2.1 of the License, or (at your option) any later version.
> + *
> + * FFmpeg is distributed in the hope that it will be useful,
> + * but WITHOUT ANY WARRANTY; without even the implied warranty of
> + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
> + * Lesser General Public License for more details.
> + *
> + * You should have received a copy of the GNU Lesser General Public
> + * License along with FFmpeg; if not, write to the Free Software
> + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
> 02110-1301 USA
> + */
> +
> +#include "libavutil/imgutils.h"
> +#include "libavutil/opt.h"
> +#include "libavutil/parseutils.h"
> +#include "avfilter.h"
> +#include "formats.h"
> +#include "video.h"
> +#include "drawutils.h"
> +
> +typedef struct {
> +    const AVClass *class;
> +    int w, h;
> +    AVRational time_base;
> +    AVRational sar;
> +    char *rate;
> +    int nb_decimals;
> +    FFDrawContext draw;
> +} SMPTEBarsContext;
> +
> +#define OFFSET(x) offsetof(SMPTEBarsContext, x)
> +
> +static const AVOption smptebars_options[]= {
> +    { "size",     "set video size",     OFFSET(w),
>  AV_OPT_TYPE_IMAGE_SIZE, {.str = "720x480"}, 0, 0 },
> +    { "s",        "set video size",     OFFSET(w),
>  AV_OPT_TYPE_IMAGE_SIZE, {.str = "720x480"}, 0, 0 },
> +    { "rate",     "set video rate",     OFFSET(rate),
> AV_OPT_TYPE_STRING, {.str = "30"},      0, 0 },
> +    { "r",        "set video rate",     OFFSET(rate),
> AV_OPT_TYPE_STRING, {.str = "30"},      0, 0 },
> +    { "sar",      "set video sample aspect ratio", OFFSET(sar),
> AV_OPT_TYPE_RATIONAL, {.dbl= 1},  0, INT_MAX },
> +    { NULL },
> +};
> +
> +static const AVClass smptebars_class = {
> +    .class_name = "smptebars",
> +    .item_name  = av_default_item_name,
> +    .option     = smptebars_options,
> +    .version    = LIBAVUTIL_VERSION_INT,
> +    .category   = AV_CLASS_CATEGORY_FILTER,
> +};
> +
> +static av_cold int smptebars_init(AVFilterContext *ctx, const char *args,
> void *opaque)
> +{
> +    SMPTEBarsContext *smpte = ctx->priv;
> +    AVRational frame_rate_q;
> +    int ret = 0;
> +
> +    smpte->class = &smptebars_class;
> +
> +    av_opt_set_defaults(smpte);
> +    if ((ret = (av_set_options_string(smpte, args, "=", ":"))) < 0) {
> +        av_log(ctx, AV_LOG_ERROR, "Error parsing options string: '%s'\n",
> args);
> +        ret = AVERROR(EINVAL);
> +        goto end;
> +    }
> +    if (av_parse_video_rate(&frame_rate_q, smpte->rate) < 0 ||
> +            frame_rate_q.den <= 0 || frame_rate_q.num <= 0) {
> +        av_log(ctx, AV_LOG_ERROR, "Invalid frame rate: %s\n",
> smpte->rate);
> +        ret = AVERROR(EINVAL);
> +        goto end;
> +    }
> +
> +    smpte->time_base.num = frame_rate_q.den;
> +    smpte->time_base.den = frame_rate_q.num;
> +
> +end:
> +    av_opt_free(smpte);
> +    return ret;
> +}
> +
> +static int smptebars_query_formats(AVFilterContext *ctx)
> +{
> +    ff_set_common_formats(ctx, ff_draw_supported_pixel_formats(0));
> +    return 0;
> +}
> +
> +static uint8_t rainbow[7][4] = {
> +    { 191, 191, 191, 255},
> +    { 191, 191,   0, 255},
> +    {   0, 191, 191, 255},
> +    {   0, 191,   0, 255},
> +    { 191,   0, 191, 255},
> +    { 191,   0,   0, 255},
> +    {   0,   0, 191, 255},
> +};
> +
> +static uint8_t wobnair[7][4] = {
> +    {   0,   0, 191, 255},
> +    {   0,   0,   0, 255},
> +    { 191,   0, 191, 255},
> +    {   0,   0,   0, 255},
> +    {   0, 191, 191, 255},
> +    {   0,   0,   0, 255},
> +    { 191, 191, 191, 255},
> +};
> +
> +static uint8_t white[4] = { 255, 255, 255, 255 };
> +static uint8_t black[4] = {   0,   0,   0, 255 };
> +
> +/* made up values */
> +static uint8_t neg4ire[4] = {   7,   7,   7, 255 };
> +static uint8_t pos4ire[4] = {  17,  17,  17, 255 };
> +static uint8_t i_pixel[4] = { 224,   0,   0, 255 };
> +static uint8_t q_pixel[4] = {   0,   0, 176, 255 };
> +
> +static void draw_smptebars(AVFilterContext *ctx, AVFilterBufferRef
> *picref)
> +{
> +    SMPTEBarsContext *smpte = ctx->priv;
> +    FFDrawColor color;
> +    int r_w, r_h, w_h, p_w, p_h, i, x = 0;
> +
> +    r_w = (smpte->w + 6 ) / 7;
> +    r_h = smpte->h * 2 / 3;
> +    w_h = smpte->h * 3 / 4 - r_h;
> +    p_w = r_w * 5 / 4;
> +    p_h = smpte->h - w_h - r_h;
> +    for (i = 0; i < 7; i++) {
> +        ff_draw_color(&smpte->draw, &color, rainbow[i]);
> +        ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x,   0, r_w, r_h);
> +        ff_draw_color(&smpte->draw, &color, wobnair[i]);
> +        ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h, r_w, w_h);
> +        x += r_w;
> +    }
> +    x = 0;
> +    ff_draw_color(&smpte->draw, &color, i_pixel);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, p_w, p_h);
> +    x += p_w;
> +    ff_draw_color(&smpte->draw, &color, white);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, p_w, p_h);
> +    x += p_w;
> +    ff_draw_color(&smpte->draw, &color, q_pixel);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, p_w, p_h);
> +    x += p_w;
> +    ff_draw_color(&smpte->draw, &color, black);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, 5 * r_w - x, p_h);
> +    x += 5 * r_w - x;
> +    ff_draw_color(&smpte->draw, &color, neg4ire);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, r_w / 3, p_h);
> +    x += r_w / 3;
> +    ff_draw_color(&smpte->draw, &color, black);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, r_w / 3, p_h);
> +    x += r_w / 3;
> +    ff_draw_color(&smpte->draw, &color, pos4ire);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, r_w / 3, p_h);
> +    x += r_w / 3;
> +    ff_draw_color(&smpte->draw, &color, black);
> +    ff_fill_rectangle(&smpte->draw, &color, picref->data,
> picref->linesize, x, r_h + w_h, smpte->w - x, p_h);
> +}
> +
> +static int request_frame(AVFilterLink *outlink)
> +{
> +    SMPTEBarsContext *smpte = outlink->src->priv;
> +    AVFilterBufferRef *picref;
> +
> +    picref = ff_get_video_buffer(outlink, AV_PERM_WRITE, smpte->w,
> smpte->h);
> +    picref->video->sample_aspect_ratio = smpte->sar;
> +
> +    ff_start_frame(outlink, avfilter_ref_buffer(picref, ~0));
> +    draw_smptebars(outlink->src, picref);
> +    ff_draw_slice(outlink, 0, picref->video->h, 1);
> +    ff_end_frame(outlink);
> +    avfilter_unref_buffer(picref);
> +    return 0;
> +}
> +
> +static int config_props(AVFilterLink *outlink)
> +{
> +    AVFilterContext *ctx = outlink->src;
> +    SMPTEBarsContext *smpte = ctx->priv;
> +
> +    ff_draw_init(&smpte->draw, outlink->format, 0);
> +
> +    if (av_image_check_size(smpte->w, smpte->h, 0, ctx) < 0)
> +        return AVERROR(EINVAL);
> +
> +    outlink->w = smpte->w;
> +    outlink->h = smpte->h;
> +    outlink->sample_aspect_ratio = smpte->sar;
> +    outlink->time_base = smpte->time_base;
> +
> +    return 0;
> +}
> +
> +AVFilter avfilter_vsrc_smptebars = {
> +    .name      = "smptebars",
> +    .description = NULL_IF_CONFIG_SMALL("Generate SMPTE color bars."),
> +    .priv_size = sizeof(SMPTEBarsContext),
> +    .init      =  smptebars_init,
> +
> +    .query_formats   = smptebars_query_formats,
> +
> +    .inputs    = (const AVFilterPad[]) {{ .name = NULL}},
> +
> +    .outputs   = (const AVFilterPad[]) {{ .name = "default",
> +                                    .type = AVMEDIA_TYPE_VIDEO,
> +                                    .request_frame = request_frame,
> +                                    .config_props  = config_props, },
> +                                  { .name = NULL }},
> +};
> --
> 1.7.7
>
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel at ffmpeg.org
> http://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>


More information about the ffmpeg-devel mailing list