gtk-app.c

   1 /*
   2 initially based on an example by Tristan Matthews
   3 http://tristanswork.blogspot.com/2008/09/fullscreen-video-in-gstreamer-with-gtk.html
   4 */
   5 #include <gst/gst.h>
   6 #include <gtk/gtk.h>
   7 #include <gst/interfaces/xoverlay.h>
   8 #include <gdk/gdk.h>
   9 #include <gdk/gdkx.h>
  10 #include "gtk-app.h"
  11
  12
  13 static void hide_mouse(GtkWidget *widget){
  14   GdkWindow *w = GDK_WINDOW(widget->window);
  15   GdkDisplay *display = gdk_display_get_default();
  16   GdkCursor *cursor = gdk_cursor_new_for_display(display, GDK_BLANK_CURSOR);
  17   gdk_window_set_cursor(w, cursor);
  18   gdk_cursor_unref (cursor);
  19 }
  20
  21 static void
  22 mjpeg_branch(GstPipeline *pipeline, GstElement *tee)
  23 {
  24   /* ! jpegenc ! avimux ! filesink location=mjpeg.avi */
  25   GstElement *queue = gst_element_factory_make("queue", NULL);
  26   GstElement *jpegenc = gst_element_factory_make("jpegenc", NULL);
  27   GstElement *avimux = gst_element_factory_make("avimux", NULL);
  28   GstElement *filesink = gst_element_factory_make("filesink", NULL);
  29   GstElement *cs = gst_element_factory_make("ffmpegcolorspace", NULL);
  30   g_object_set(G_OBJECT(filesink),
  31       "location", option_avi,
  32       NULL);
  33   gst_bin_add_many(GST_BIN(pipeline),
  34       queue,
  35       cs,
  36       jpegenc,
  37       avimux,
  38       filesink,
  39       NULL);
  40   gst_element_link_many(tee,
  41       queue,
  42       cs,
  43       jpegenc,
  44       avimux,
  45       filesink,
  46       NULL);
  47 }
  48
  49 static void
  50 post_tee_pipeline(GstPipeline *pipeline, GstElement *tee, GstElement *sink,
  51     int rngseed, int colour, int timer, int debug, char *save, char *reload){
  52   GstElement *queue = gst_element_factory_make("queue", NULL);
  53   GstElement *sparrow = gst_element_factory_make("sparrow", NULL);
  54   GstElement *caps_posteriori = gst_element_factory_make("capsfilter", NULL);
  55   GstElement *cs_posteriori = gst_element_factory_make("ffmpegcolorspace", NULL);
  56
  57   g_object_set(G_OBJECT(caps_posteriori), "caps",
  58       gst_caps_new_simple ("video/x-raw-rgb",
  59           COMMON_CAPS), NULL);
  60
  61   g_object_set(G_OBJECT(sparrow),
  62       "timer", timer,
  63       "debug", debug,
  64       "rngseed", rngseed,
  65       "colour", colour,
  66       NULL);
  67   if (reload){
  68     g_object_set(G_OBJECT(sparrow),
  69         "reload", reload,
  70         NULL);
  71   }
  72   if (save){
  73     g_object_set(G_OBJECT(sparrow),
  74         "save", save,
  75         NULL);
  76   }
  77
  78   gst_bin_add_many (GST_BIN(pipeline),
  79       queue,
  80       sparrow,
  81       caps_posteriori,
  82       cs_posteriori,
  83       sink,
  84       NULL);
  85
  86   gst_element_link_many(tee,
  87       queue,
  88       sparrow,
  89       caps_posteriori,
  90       cs_posteriori,
  91       sink,
  92       NULL);
  93 }
  94
  95 static GstElement *
  96 pre_tee_pipeline(GstPipeline *pipeline){
  97   if (pipeline == NULL){
  98     pipeline = GST_PIPELINE(gst_pipeline_new("sparrow_pipeline"));
  99   }
 100   GstElement *src = gst_element_factory_make("v4l2src", NULL);
 101   GstElement *caps_priori = gst_element_factory_make("capsfilter", NULL);
 102   GstElement *cs_priori = gst_element_factory_make("ffmpegcolorspace", NULL);
 103   GstElement *caps_interiori = gst_element_factory_make("capsfilter", NULL);
 104   GstElement *tee = gst_element_factory_make ("tee", NULL);
 105
 106   g_object_set(G_OBJECT(caps_priori), "caps",
 107       gst_caps_new_simple ("video/x-raw-yuv",
 108           "format", GST_TYPE_FOURCC, GST_MAKE_FOURCC('Y', 'U', 'Y', '2'),
 109           COMMON_CAPS), NULL);
 110
 111   g_object_set(G_OBJECT(caps_interiori), "caps",
 112       gst_caps_new_simple ("video/x-raw-rgb",
 113           COMMON_CAPS), NULL);
 114
 115   gst_bin_add_many(GST_BIN(pipeline),
 116       src,
 117       caps_priori,
 118       cs_priori,
 119       //caps_interiori,
 120       tee,
 121       NULL);
 122
 123   gst_element_link_many(src,
 124       caps_priori,
 125       cs_priori,
 126       //caps_interiori,
 127       tee,
 128       NULL);
 129   return tee;
 130 }
 131
 132
 133 static GstPipeline *
 134 make_multi_pipeline(windows_t *windows, int count)
 135 {
 136   GstPipeline *pipeline = GST_PIPELINE(gst_pipeline_new("sparrow_pipeline"));
 137   GstElement *tee = pre_tee_pipeline(pipeline);
 138   char *reload = NULL;
 139   char *save = NULL;
 140   int i;
 141   for (i = 0; i < count; i++){
 142     GstElement *sink = windows->sinks[i];
 143     //args are:
 144     //(pipeline, tee, sink, int rngseed, int colour, timer flag, int debug flag)
 145     /* timer should only run on one of them. colour >= 3 is undefined */
 146     int debug = option_debug == i;
 147     if (option_reload != NULL){
 148       if (option_reload[i] == NULL){
 149         g_critical("You can't reload some screens and not others!");
 150         exit(1);
 151       }
 152       reload = option_reload[i];
 153     }
 154     if (option_save && option_save[i]){
 155       save = option_save[i];
 156     }
 157     post_tee_pipeline(pipeline, tee, sink, i, i + 1, i == 0, debug, save, reload);
 158   }
 159   if (option_avi){
 160     /*add a branch saving the video to a file */
 161     mjpeg_branch(pipeline, tee);
 162   }
 163
 164   return pipeline;
 165 }
 166
 167
 168 static void
 169 bus_call(GstBus * bus, GstMessage *msg, gpointer data)
 170 {
 171   windows_t *windows = (windows_t *)data;
 172   if (GST_MESSAGE_TYPE(msg) == GST_MESSAGE_ELEMENT &&
 173       gst_structure_has_name(msg->structure, "prepare-xwindow-id")){
 174     g_print("Got prepare-xwindow-id msg. option screens: %d\n", option_screens);
 175     for (int i = 0; i < option_screens; i++){
 176       gst_x_overlay_set_xwindow_id(GST_X_OVERLAY(windows->sinks[i]),
 177           windows->xwindows[i]);
 178       g_print("connected sink %d to window %lu\n", i, windows->xwindows[i]);
 179       hide_mouse(windows->gtk_windows[i]);
 180     }
 181   }
 182 }
 183
 184 static void
 185 toggle_fullscreen(GtkWidget *widget){
 186   GdkWindowState state = gdk_window_get_state(GDK_WINDOW(widget->window));
 187   if (state == GDK_WINDOW_STATE_FULLSCREEN){
 188     gtk_window_unfullscreen(GTK_WINDOW(widget));
 189   }
 190   else{
 191     gtk_window_fullscreen(GTK_WINDOW(widget));
 192   }
 193 }
 194
 195 static gboolean
 196 key_press_event_cb(GtkWidget *widget, GdkEventKey *event, gpointer data)
 197 {
 198   g_print("got key %c\n", event->keyval);
 199   switch (event->keyval){
 200   case 'f':
 201     toggle_fullscreen(widget);
 202     break;
 203   case 'q':
 204     g_signal_emit_by_name(widget, "destroy");
 205     break;
 206   default:
 207     break;
 208   }
 209   return TRUE;
 210 }
 211
 212 void destroy_cb(GtkWidget * widget, gpointer data)
 213 {
 214   GMainLoop *loop = (GMainLoop*) data;
 215   g_print("Window destroyed\n");
 216   g_main_loop_quit(loop);
 217 }
 218
 219 static void
 220 video_widget_realize_cb(GtkWidget *widget, gpointer data)
 221 {
 222   windows_t *windows = (windows_t *)data;
 223   int r = windows->realised;
 224   if (r < MAX_SCREENS){
 225     windows->xwindows[r] = GDK_WINDOW_XID(GDK_WINDOW(widget->window));
 226     g_print("realised window %d with XID %lu\n", r, windows->xwindows[r]);
 227   }
 228   else {
 229     g_print("wtf, there seem to be %d windows!\n", r);
 230   }
 231   windows->realised++;
 232   hide_mouse(widget);
 233 }
 234
 235
 236 static void
 237 set_up_window(GMainLoop *loop, GtkWidget *window, int screen_no){
 238   static const GdkColor black = {0, 0, 0, 0};
 239   gtk_window_set_default_size(GTK_WINDOW(window), WIDTH, HEIGHT);
 240
 241   if (option_fullscreen){
 242     gtk_window_fullscreen(GTK_WINDOW(window));
 243   }
 244
 245   /*if more than one screen is requested, set the screen number.
 246     otherwise let it fall were it falls */
 247   if (option_screens > 1 || option_first_screen){
 248     /* "screen" is not the same thing as "monitor" */
 249     GdkScreen * screen = gdk_screen_get_default();
 250     int width = gdk_screen_get_width(screen);
 251     /* XXX window selection is specific to equally sized windows arranged
 252        horizontally.  This could be generalised, perhaps using trial and
 253        error */
 254     gtk_window_move(GTK_WINDOW(window),
 255         (width / 2 * screen_no + 200) % width, 50);
 256   }
 257
 258   // attach key press signal to key press callback
 259   gtk_widget_set_events(window, GDK_KEY_PRESS_MASK);
 260   g_signal_connect(G_OBJECT(window), "key-press-event", G_CALLBACK(key_press_event_cb), NULL);
 261   g_signal_connect(G_OBJECT(window), "destroy", G_CALLBACK(destroy_cb), loop);
 262
 263   gtk_widget_modify_bg(window, GTK_STATE_NORMAL, &black);
 264   gtk_widget_show_all(window);
 265   hide_mouse(window);
 266 }
 267
 268
 269 static GOptionEntry entries[] =
 270 {
 271   { "full-screen", 'f', 0, G_OPTION_ARG_NONE, &option_fullscreen, "run full screen", NULL },
 272   { "fps", 'p', 0, G_OPTION_ARG_INT, &option_fps, "speed (Frames per second)", "FPS" },
 273   { "screens", 's', 0, G_OPTION_ARG_INT, &option_screens, "Use this many screens", "S" },
 274   { "first-screen", 0, 0, G_OPTION_ARG_INT, &option_first_screen, "Start with this screen", "S" },
 275   { "debug", 'd', 0, G_OPTION_ARG_INT, &option_debug, "Save screen's debug images in /tmp", "SCREEN" },
 276   { "reload", 'r', 0, G_OPTION_ARG_FILENAME_ARRAY, &option_reload,
 277     "load calibration data from FILE (one per screen)", "FILE" },
 278   { "save", 'S', 0, G_OPTION_ARG_FILENAME_ARRAY, &option_save,
 279     "save calibration data to FILE (one per screen)", "FILE" },
 280   { "avi", 'a', 0, G_OPTION_ARG_FILENAME, &option_avi,
 281     "save mjpeg video to FILE", "FILE" },
 282   //  { "overlay", 'o', 0, G_OPTION_ARG_NONE, &option_overlay, "Use some kind of overlay", NULL },
 283   { NULL, 0, 0, 0, NULL, NULL, NULL }
 284 };
 285
 286
 287 gint main (gint argc, gchar *argv[])
 288 {
 289   //initialise threads before any gtk stuff (because not using gtk_init)
 290   g_thread_init(NULL);
 291   /*this is more complicated than plain gtk_init/gst_init, so that options from
 292     all over can be gathered and presented together.
 293    */
 294   GOptionGroup *gst_opts = gst_init_get_option_group();
 295   GOptionGroup *gtk_opts = gtk_get_option_group(TRUE);
 296   GOptionContext *ctx = g_option_context_new("...!");
 297   g_option_context_add_main_entries(ctx, entries, NULL);
 298   g_option_context_add_group(ctx, gst_opts);
 299   g_option_context_add_group(ctx, gtk_opts);
 300   GError *error = NULL;
 301   if (!g_option_context_parse(ctx, &argc, &argv, &error)){
 302     g_print ("Error initializing: %s\n", GST_STR_NULL(error->message));
 303     exit (1);
 304   }
 305   g_option_context_free(ctx);
 306
 307   GMainLoop *loop = g_main_loop_new(NULL, FALSE);
 308
 309   windows_t windows;
 310   windows.realised = 0;
 311
 312   int i;
 313   for (i = 0; i < option_screens; i++){
 314     GtkWidget *window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
 315     g_signal_connect(window, "realize",
 316         G_CALLBACK(video_widget_realize_cb), &windows);
 317     /* set up sink here */
 318     GstElement *sink = gst_element_factory_make("ximagesink", NULL);
 319     set_up_window(loop, window, i + option_first_screen);
 320     windows.gtk_windows[i] = window;
 321     windows.sinks[i] = sink;
 322   }
 323
 324   GstElement *pipeline = (GstElement *)make_multi_pipeline(&windows, option_screens);
 325
 326   GstBus *bus = gst_pipeline_get_bus(GST_PIPELINE(pipeline));
 327   gst_bus_add_watch(bus, (GstBusFunc)bus_call, &windows);
 328   gst_object_unref(bus);
 329
 330   gst_element_set_state(pipeline, GST_STATE_PLAYING);
 331
 332   g_main_loop_run(loop);
 333
 334   gst_element_set_state (pipeline, GST_STATE_NULL);
 335   gst_object_unref (pipeline);
 336   return 0;
 337 }