--- /dev/null
+/*
+ * test-vaapicontext.c - Testsuite for VAAPI app context
+ *
+ * Copyright (C) 2017 Intel Corporation
+ *
+ * This library 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.
+ *
+ * This library 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 this library; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
+ * Boston, MA 02110-1301 USA
+ */
+
+#include <gst/gst.h>
+#include <gst/video/videooverlay.h>
+
+#include <va/va.h>
+#include <gtk/gtk.h>
+
+#include <X11/Xlib.h>
+#include <va/va_x11.h>
+#ifdef GDK_WINDOWING_X11
+#include <gdk/gdkx.h>
+#else
+#error "X11 is not supported in GTK+"
+#endif
+
+static gboolean g_multisink;
+static gchar *g_filepath;
+
+static GOptionEntry g_options[] = {
+ {"multi", 'm', 0, G_OPTION_ARG_NONE, &g_multisink, "test multiple vaapisink",
+ NULL},
+ {"file", 'f', 0, G_OPTION_ARG_STRING, &g_filepath,
+ "file path to play (only mp4/h264)", NULL},
+ {NULL,}
+};
+
+typedef struct _CustomData
+{
+ GtkWidget *main_window;
+ VADisplay va_display;
+ GstElement *pipeline;
+ guintptr videoarea_handle[2];
+} AppData;
+
+static void
+delete_event_cb (GtkWidget * widget, GdkEvent * event, gpointer data)
+{
+ AppData *app = data;
+
+ gst_element_set_state (app->pipeline, GST_STATE_NULL);
+ gtk_main_quit ();
+}
+
+static void
+button_rotate_cb (GtkWidget * widget, GstElement * elem)
+{
+ static gint counter = 0;
+ const static gint tags[] = { 90, 180, 270, 0 };
+
+ g_object_set (elem, "rotation", tags[counter++ % G_N_ELEMENTS (tags)], NULL);
+}
+
+static Display *
+get_x11_window_display (AppData * app)
+{
+#if defined(GDK_WINDOWING_X11)
+ GdkDisplay *gdk_display;
+ Display *x11_display;
+
+ gdk_display = gtk_widget_get_display (app->main_window);
+ x11_display = gdk_x11_display_get_xdisplay (gdk_display);
+ return x11_display;
+#endif
+ g_error ("Running in a non-X11 environment");
+}
+
+static VADisplay
+ensure_va_display (AppData * app)
+{
+ if (app->va_display)
+ return app->va_display;
+ app->va_display = vaGetDisplay (get_x11_window_display (app));
+ /* There's no need to call vaInitialize() since element does it
+ * internally */
+ return app->va_display;
+}
+
+static GstContext *
+create_vaapi_app_display_context (AppData * app, gboolean new_va_display)
+{
+ GstContext *context;
+ GstStructure *s;
+ VADisplay va_display;
+ Display *x11_display;
+
+ x11_display = get_x11_window_display (app);
+
+ if (new_va_display)
+ va_display = vaGetDisplay (x11_display);
+ else
+ va_display = ensure_va_display (app);
+
+ context = gst_context_new ("gst.vaapi.app.Display", TRUE);
+ s = gst_context_writable_structure (context);
+ gst_structure_set (s, "va-display", G_TYPE_POINTER, va_display, NULL);
+ gst_structure_set (s, "x11-display", G_TYPE_POINTER, x11_display, NULL);
+
+ return context;
+}
+
+static GstBusSyncReply
+bus_sync_handler (GstBus * bus, GstMessage * msg, gpointer data)
+{
+ AppData *app = data;
+
+ switch (GST_MESSAGE_TYPE (msg)) {
+ case GST_MESSAGE_NEED_CONTEXT:{
+ const gchar *context_type;
+ gboolean new_va_disp;
+ GstContext *context;
+
+ gst_message_parse_context_type (msg, &context_type);
+ gst_println ("Got need context %s from %s", context_type,
+ GST_MESSAGE_SRC_NAME (msg));
+
+ if (g_strcmp0 (context_type, "gst.vaapi.app.Display") != 0)
+ break;
+
+ /* create a new VA display *only* for the second video sink */
+ new_va_disp = (g_strcmp0 (GST_MESSAGE_SRC_NAME (msg), "sink2") == 0);
+
+ context = create_vaapi_app_display_context (app, new_va_disp);
+ gst_element_set_context (GST_ELEMENT (GST_MESSAGE_SRC (msg)), context);
+ gst_context_unref (context);
+ break;
+ }
+ case GST_MESSAGE_ELEMENT:{
+ if (!gst_is_video_overlay_prepare_window_handle_message (msg))
+ break;
+
+ if (g_strcmp0 (GST_MESSAGE_SRC_NAME (msg), "sink2") == 0)
+ gst_video_overlay_set_window_handle (GST_VIDEO_OVERLAY
+ (GST_MESSAGE_SRC (msg)), app->videoarea_handle[1]);
+ else
+ gst_video_overlay_set_window_handle (GST_VIDEO_OVERLAY
+ (GST_MESSAGE_SRC (msg)), app->videoarea_handle[0]);
+ break;
+ }
+ case GST_MESSAGE_EOS:
+ gtk_main_quit ();
+ break;
+ default:
+ break;
+ }
+
+ return GST_BUS_PASS;
+}
+
+static void
+realize_cb (GtkWidget * widget, gpointer data)
+{
+ AppData *app = data;
+ GdkWindow *window;
+ static guint counter = 0;
+
+#if defined(GDK_WINDOWING_X11)
+ window = gtk_widget_get_window (widget);
+
+ if (!gdk_window_ensure_native (window))
+ g_error ("Couldn't create native window needed for GstXOverlay!");
+
+ app->videoarea_handle[counter++ % 2] = GDK_WINDOW_XID (window);
+#endif
+}
+
+static GtkWidget *
+create_video_box (AppData * app)
+{
+ GtkWidget *video_area;
+
+ video_area = gtk_drawing_area_new ();
+ gtk_widget_set_size_request (video_area, 640, 480);
+ g_signal_connect (video_area, "realize", G_CALLBACK (realize_cb), app);
+ return video_area;
+}
+
+static GtkWidget *
+create_rotate_button (AppData * app, const gchar * name)
+{
+ GtkWidget *rotate;
+ GstElement *sink;
+
+ sink = gst_bin_get_by_name (GST_BIN (app->pipeline), name);
+ g_assert (sink);
+
+ rotate = gtk_button_new_with_label ("Rotate");
+ g_signal_connect (rotate, "clicked", G_CALLBACK (button_rotate_cb), sink);
+ gst_object_unref (sink);
+
+ return rotate;
+}
+
+static void
+build_ui (AppData * app)
+{
+ GtkWidget *mainwin, *vbox, *pane, *bbox;
+
+ mainwin = gtk_window_new (GTK_WINDOW_TOPLEVEL);
+ gtk_window_set_title (GTK_WINDOW (mainwin), "VAAPI display context test");
+ gtk_window_set_resizable (GTK_WINDOW (mainwin), FALSE);
+ g_signal_connect (mainwin, "delete-event", G_CALLBACK (delete_event_cb), app);
+ app->main_window = mainwin;
+
+ vbox = gtk_box_new (GTK_ORIENTATION_VERTICAL, 0);
+ gtk_container_add (GTK_CONTAINER (mainwin), vbox);
+
+ pane = gtk_paned_new (GTK_ORIENTATION_HORIZONTAL);
+ gtk_box_pack_start (GTK_BOX (vbox), pane, TRUE, TRUE, 0);
+
+ /* first video box */
+ gtk_paned_pack1 (GTK_PANED (pane), create_video_box (app), TRUE, TRUE);
+
+ /* rotate buttons */
+ bbox = gtk_button_box_new (GTK_ORIENTATION_HORIZONTAL);
+ gtk_button_box_set_layout (GTK_BUTTON_BOX (bbox), GTK_BUTTONBOX_SPREAD);
+ gtk_box_pack_end (GTK_BOX (vbox), bbox, TRUE, TRUE, 0);
+
+ gtk_box_pack_start (GTK_BOX (bbox), create_rotate_button (app, "sink1"), TRUE,
+ TRUE, 0);
+
+ if (g_multisink) {
+ /* second video box */
+ gtk_paned_pack2 (GTK_PANED (pane), create_video_box (app), TRUE, TRUE);
+
+ gtk_box_pack_start (GTK_BOX (bbox), create_rotate_button (app, "sink2"),
+ TRUE, TRUE, 0);
+ }
+
+ gtk_widget_show_all (mainwin);
+}
+
+int
+main (gint argc, gchar ** argv)
+{
+ AppData app = { 0, };
+ GstBus *bus;
+ GOptionContext *ctx;
+ GError *error = NULL;
+
+ XInitThreads ();
+
+ ctx = g_option_context_new ("- test options");
+ if (!ctx)
+ return -1;
+
+ g_option_context_add_group (ctx, gtk_get_option_group (TRUE));
+ g_option_context_add_group (ctx, gst_init_get_option_group ());
+ g_option_context_add_main_entries (ctx, g_options, NULL);
+ if (!g_option_context_parse (ctx, &argc, &argv, NULL))
+ return -1;
+ g_option_context_free (ctx);
+
+ if (g_multisink) {
+ app.pipeline = gst_parse_launch ("videotestsrc ! tee name=t ! queue ! "
+ "vaapisink name=sink1 t. ! queue ! vaapisink name=sink2", &error);
+ } else if (!g_filepath) {
+ app.pipeline = gst_parse_launch ("videotestsrc ! vaapih264enc ! "
+ "vaapidecodebin ! vaapisink name=sink1", &error);
+ } else {
+ app.pipeline = gst_parse_launch ("filesrc name=src ! qtdemux ! h264parse ! "
+ "vaapidecodebin ! vaapisink name=sink1", &error);
+ }
+
+ if (error) {
+ gst_printerrln ("failed to parse pipeline: %s", error->message);
+ g_error_free (error);
+ return -1;
+ }
+
+ if (!g_multisink && g_filepath) {
+ GstElement *src;
+
+ src = gst_bin_get_by_name (GST_BIN (app.pipeline), "src");
+ g_assert (src);
+ g_object_set (src, "location", g_filepath, NULL);
+ gst_object_unref (src);
+ }
+
+ build_ui (&app);
+
+ bus = gst_element_get_bus (app.pipeline);
+ gst_bus_set_sync_handler (bus, bus_sync_handler, (gpointer) & app, NULL);
+ gst_object_unref (bus);
+
+ gst_element_set_state (app.pipeline, GST_STATE_PLAYING);
+ gst_println ("Now playing…");
+
+ gtk_main ();
+
+ gst_object_unref (app.pipeline);
+ /* there is no need to call vaTerminate() because it is done by the
+ * vaapi elements */
+ return 0;
+}
projects / platform / upstream / gstreamer.git / commitdiff