/* * test-subpicture.c - Test GstVaapiSubpicture * * Copyright (C) <2011-2013> Intel Corporation * Copyright (C) <2011> Collabora Ltd. * Copyright (C) <2011> Thibault Saunier * * 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 #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_object_unref (decoder); gst_object_unref (window); gst_object_unref (display); g_free (g_codec_str); video_output_exit (); return 0; }