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},
  {"global-alpha", 'g',
        0,
        G_OPTION_ARG_DOUBLE, &g_global_alpha,
      "global-alpha value", NULL},
  {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;
  GstVaapiSurfaceProxy *proxy;
  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");

  if (g_global_alpha != 1.0)
    flags |= GST_VIDEO_OVERLAY_FORMAT_FLAG_GLOBAL_ALPHA;

  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");

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

  surface = gst_vaapi_surface_proxy_get_surface (proxy);

  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;

  {
    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);
  }
  if (!overlay)
    g_error ("could not create video overlay");
  gst_buffer_unref (buffer);

  if (flags & GST_VIDEO_OVERLAY_FORMAT_FLAG_GLOBAL_ALPHA)
    gst_video_overlay_rectangle_set_global_alpha (overlay, g_global_alpha);

  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_surface_proxy_unref (proxy);
  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;
}