gstreamer/tests/test-subpicture.c

/*
 *  test-subpicture.c - Test GstVaapiSubpicture
 *
 *  Copyright (C) <2011-2013> Intel Corporation
 *  Copyright (C) <2011> Collabora Ltd.
 *  Copyright (C) <2011> Thibault Saunier <thibault.saunier@collabora.com>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */

#include "gst/vaapi/sysdeps.h"
#include <string.h>
#include <gst/vaapi/gstvaapisurface.h>
#include "decoder.h"
#include "output.h"
#include "test-subpicture-data.h"

static inline void pause(void)
{
    g_print("Press any key to continue...\n");
    getchar();
}

static gchar *g_codec_str;
static gdouble g_global_alpha = 1.0;

static GOptionEntry g_options[] = {
    { "codec", 'c',
      0,
      G_OPTION_ARG_STRING, &g_codec_str,
      "codec to test", NULL },
#ifdef HAVE_GST_VIDEO_OVERLAY_HWCAPS
    { "global-alpha", 'g',
      0,
      G_OPTION_ARG_DOUBLE, &g_global_alpha,
      "global-alpha value", NULL },
#endif
    { NULL, }
};

static void
upload_subpicture(GstBuffer *buffer, const VideoSubpictureInfo *subinfo)
{
    const guint32 * const src = subinfo->data;
    guint i, len = subinfo->data_size / 4;
    GstMapInfo map_info;
    guint32 *dst;

    if (!gst_buffer_map(buffer, &map_info, GST_MAP_WRITE))
        return;
    dst = (guint32 *)map_info.data;

    /* Convert from RGBA source to ARGB */
    for (i = 0; i < len; i++) {
        const guint32 rgba = src[i];
        dst[i] = (rgba >> 8) | (rgba << 24);
    }

    gst_buffer_unmap(buffer, &map_info);
}

int
main(int argc, char *argv[])
{
    GstVaapiDisplay      *display;
    GstVaapiWindow       *window;
    GstVaapiDecoder      *decoder;
    GstVaapiSurface      *surface;
    GstBuffer            *buffer;
    VideoSubpictureInfo   subinfo;
    GstVaapiRectangle     subrect;
    GstVideoOverlayRectangle *overlay;
    GstVideoOverlayComposition *compo;
    guint flags = 0;

    static const guint win_width  = 640;
    static const guint win_height = 480;

    if (!video_output_init(&argc, argv, g_options))
        g_error("failed to initialize video output subsystem");

#ifdef HAVE_GST_VIDEO_OVERLAY_HWCAPS
    if (g_global_alpha != 1.0)
        flags |= GST_VIDEO_OVERLAY_FORMAT_FLAG_GLOBAL_ALPHA;
#endif

    g_print("Test subpicture\n");

    display = video_output_create_display(NULL);
    if (!display)
        g_error("could not create VA display");

    window = video_output_create_window(display, win_width, win_height);
    if (!window)
        g_error("could not create window");

    decoder = decoder_new(display, g_codec_str);
    if (!decoder)
        g_error("could not create decoder");

    if (!decoder_put_buffers(decoder))
        g_error("could not fill decoder with sample data");

    surface = decoder_get_surface(decoder);
    if (!surface)
        g_error("could not get decoded surface");

    subpicture_get_info(&subinfo);
    buffer = gst_buffer_new_and_alloc(subinfo.data_size);
    upload_subpicture(buffer, &subinfo);

    /* We position the subpicture at the bottom center */
    subrect.x = (gst_vaapi_surface_get_width(surface) - subinfo.width) / 2;
    subrect.y = gst_vaapi_surface_get_height(surface) - subinfo.height - 10;
    subrect.height = subinfo.height;
    subrect.width = subinfo.width;

#if GST_CHECK_VERSION(1,0,0)
    {
        GstVideoMeta * const vmeta =
            gst_buffer_add_video_meta(buffer, GST_VIDEO_FRAME_FLAG_NONE,
                GST_VIDEO_OVERLAY_COMPOSITION_FORMAT_RGB,
                subinfo.width, subinfo.height);
        if (!vmeta)
            g_error("could not create video meta");

        overlay = gst_video_overlay_rectangle_new_raw(buffer,
            subrect.x, subrect.y, subrect.width, subrect.height, flags);
    }
#else
    overlay = gst_video_overlay_rectangle_new_argb(buffer,
        subinfo.width, subinfo.height, subinfo.width * 4,
        subrect.x, subrect.y, subrect.width, subrect.height, flags);
#endif
    if (!overlay)
        g_error("could not create video overlay");
    gst_buffer_unref(buffer);

#ifdef HAVE_GST_VIDEO_OVERLAY_HWCAPS
    if (flags & GST_VIDEO_OVERLAY_FORMAT_FLAG_GLOBAL_ALPHA)
        gst_video_overlay_rectangle_set_global_alpha(overlay, g_global_alpha);
#endif

    compo = gst_video_overlay_composition_new(overlay);
    if (!compo)
        g_error("could not create video overlay composition");
    gst_video_overlay_rectangle_unref(overlay);

    if (!gst_vaapi_surface_set_subpictures_from_composition(surface, compo,
            FALSE))
        g_error("could not create subpictures from video overlay compoition");

    gst_vaapi_window_show(window);

    if (!gst_vaapi_window_put_surface(window, surface, NULL, NULL,
            GST_VAAPI_PICTURE_STRUCTURE_FRAME))
        g_error("could not render surface");

    pause();

    gst_video_overlay_composition_unref(compo);
    gst_vaapi_decoder_unref(decoder);
    gst_vaapi_window_unref(window);
    gst_vaapi_display_unref(display);
    g_free(g_codec_str);
    video_output_exit();
    return 0;
}