test-colorkey.c 7.81 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
/* GStreamer
 * Copyright (C) <2008> Stefan Kost <ensonic@users.sf.net>
 *
 * test-colorkey: test manual colorkey handling
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdlib.h>
27
28
#include <string.h>

29
#include <glib.h>
30
#include <gdk/gdkx.h>
31
32
#include <gtk/gtk.h>

33
#include <gst/gst.h>
34
#include <gst/interfaces/xoverlay.h>
35
#include <gst/interfaces/propertyprobe.h>
36

37
38
39
40
41
42
43
44
#if !GTK_CHECK_VERSION (2, 17, 7)
static void
gtk_widget_get_allocation (GtkWidget * w, GtkAllocation * a)
{
  *a = w->allocation;
}
#endif

45
46
static GtkWidget *video_window = NULL;
static GstElement *sink = NULL;
47
static gulong embed_xid = 0;
48
49
static GdkGC *trans_gc = NULL;

50
51
static void
redraw_overlay (GtkWidget * widget)
52
{
53
54
55
56
57
58
59
  GtkAllocation allocation;
  GdkWindow *window = gtk_widget_get_window (widget);
  GtkStyle *style = gtk_widget_get_style (widget);

  gtk_widget_get_allocation (widget, &allocation);
  gdk_draw_rectangle (window, style->white_gc, TRUE, 0, 0,
      allocation.width, allocation.height);
60
61
62

  if (trans_gc) {
    guint x, y;
63
    guint h = allocation.height * 0.75;
64

65
    gdk_draw_rectangle (window, trans_gc, TRUE, 0, 0, allocation.width, h);
66

67
68
    for (y = h; y < allocation.height; y++) {
      for (x = 0; x < allocation.width; x++) {
69
        if (((x & 1) || (y & 1)) && (x & 1) != (y & 1)) {
70
          gdk_draw_point (window, trans_gc, x, y);
71
72
73
74
        }
      }
    }
  }
75
76
77
78
79
80
81
82
83
84
85
86
87
}

static gboolean
handle_resize_cb (GtkWidget * widget, GdkEventConfigure * event, gpointer data)
{
  redraw_overlay (widget);
  return FALSE;
}

static gboolean
handle_expose_cb (GtkWidget * widget, GdkEventExpose * event, gpointer data)
{
  redraw_overlay (widget);
88
89
90
  return FALSE;
}

91
92
93
static void
realize_cb (GtkWidget * widget, gpointer data)
{
94
#if GTK_CHECK_VERSION(2,18,0)
95
96
97
98
99
100
101
102
  {
    GdkWindow *window = gtk_widget_get_window (widget);

    /* This is here just for pedagogical purposes, GDK_WINDOW_XID will call it
     * as well */
    if (!gdk_window_ensure_native (window))
      g_error ("Couldn't create native window needed for GstXOverlay!");
  }
103
#endif
104

105
106
107
108
109
110
  {
    GdkWindow *window = gtk_widget_get_window (video_window);

    embed_xid = GDK_WINDOW_XID (window);
    g_print ("Window realize: video window XID = %lu\n", embed_xid);
  }
111
112
}

113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
static void
msg_state_changed (GstBus * bus, GstMessage * message, GstPipeline * pipeline)
{
  const GstStructure *s;

  s = gst_message_get_structure (message);

  /* We only care about state changed on the pipeline */
  if (s && GST_MESSAGE_SRC (message) == GST_OBJECT_CAST (pipeline)) {
    GstState old, new, pending;
    gint color;

    gst_message_parse_state_changed (message, &old, &new, &pending);

    /* When state of the pipeline changes to paused or playing we start updating scale */
    switch (GST_STATE_TRANSITION (old, new)) {
129
130
131
      case GST_STATE_CHANGE_READY_TO_PAUSED:{
        GdkWindow *window = gtk_widget_get_window (video_window);

132
133
134
135
136
137
138
139
        g_object_get (G_OBJECT (sink), "colorkey", &color, NULL);
        if (color != -1) {
          GdkColor trans_color = { 0,
            (color & 0xff0000) >> 8,
            (color & 0xff00),
            (color & 0xff) << 8
          };

140
          trans_gc = gdk_gc_new (window);
141
142
143
144
          gdk_gc_set_rgb_fg_color (trans_gc, &trans_color);
        }
        handle_resize_cb (video_window, NULL, NULL);
        break;
145
      }
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
      default:
        break;
    }
  }
}

static void
window_closed (GtkWidget * widget, GdkEvent * event, gpointer user_data)
{
  GstElement *pipeline = user_data;

  g_print ("stopping\n");
  gtk_widget_hide_all (widget);
  gst_element_set_state (pipeline, GST_STATE_NULL);
  gtk_main_quit ();
}

Stefan Kost's avatar
Stefan Kost committed
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
static gboolean
start_pipeline (gpointer user_data)
{
  GstElement *pipeline = GST_ELEMENT (user_data);
  GstStateChangeReturn sret;

  sret = gst_element_set_state (pipeline, GST_STATE_PLAYING);
  if (sret == GST_STATE_CHANGE_FAILURE) {
    gst_element_set_state (pipeline, GST_STATE_NULL);
    gst_object_unref (pipeline);
    gtk_main_quit ();
  }
  return FALSE;
}

178
179
180
181
182
183
int
main (int argc, char **argv)
{
  GtkWidget *window;
  GstElement *pipeline, *src;
  GstBus *bus;
184
185
186
  GstStateChangeReturn sret;
  GstPropertyProbe *probe;
  GValueArray *arr;
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201

  if (!g_thread_supported ())
    g_thread_init (NULL);

  gst_init (&argc, &argv);
  gtk_init (&argc, &argv);

  /* prepare the pipeline */

  pipeline = gst_pipeline_new ("xvoverlay");
  src = gst_element_factory_make ("videotestsrc", NULL);
  sink = gst_element_factory_make ("xvimagesink", NULL);
  gst_bin_add_many (GST_BIN (pipeline), src, sink, NULL);
  gst_element_link (src, sink);

202
203
204
205
206
#define COLOR_GRAY 0x7F7F7F

  g_object_set (G_OBJECT (sink), "autopaint-colorkey", FALSE,
      "force-aspect-ratio", TRUE, "draw-borders", FALSE,
      "colorkey", COLOR_GRAY, NULL);
207
208
209
210

  /* check xvimagesink capabilities */
  sret = gst_element_set_state (pipeline, GST_STATE_READY);
  if (sret == GST_STATE_CHANGE_FAILURE) {
211
    g_printerr ("Can't set pipeline to READY\n");
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
    gst_object_unref (pipeline);
    return -1;
  }

  probe = GST_PROPERTY_PROBE (sink);
  if (!probe) {
    g_printerr ("Can't probe sink\n");
    gst_element_set_state (pipeline, GST_STATE_NULL);
    gst_object_unref (pipeline);
    return -1;
  }
  arr =
      gst_property_probe_probe_and_get_values_name (probe,
      "autopaint-colorkey");
  if (!arr || !arr->n_values) {
    g_printerr ("Can't disable autopaint-colorkey property\n");
    gst_element_set_state (pipeline, GST_STATE_NULL);
    gst_object_unref (pipeline);
    return -1;
  }
  if (arr)
    g_value_array_free (arr);
234
235
236
237
238
239
240
241
242
243
244
245

  bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline));
  gst_bus_add_signal_watch_full (bus, G_PRIORITY_HIGH);
  g_signal_connect (bus, "message::state-changed",
      G_CALLBACK (msg_state_changed), pipeline);
  gst_object_unref (bus);

  /* prepare the ui */

  window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
  g_signal_connect (G_OBJECT (window), "delete-event",
      G_CALLBACK (window_closed), (gpointer) pipeline);
246
247
248
  gtk_window_set_default_size (GTK_WINDOW (window), 320, 240);

  video_window = gtk_drawing_area_new ();
249
250
  g_signal_connect (G_OBJECT (video_window), "configure-event",
      G_CALLBACK (handle_resize_cb), NULL);
251
252
  g_signal_connect (G_OBJECT (video_window), "expose-event",
      G_CALLBACK (handle_expose_cb), NULL);
253
  g_signal_connect (video_window, "realize", G_CALLBACK (realize_cb), NULL);
254
255
  gtk_widget_set_double_buffered (video_window, FALSE);
  gtk_container_add (GTK_CONTAINER (window), video_window);
256

257
  /* show the gui and play */
258

259
  gtk_widget_show_all (window);
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274

  /* realize window now so that the video window gets created and we can
   * obtain its XID before the pipeline is started up and the videosink
   * asks for the XID of the window to render onto */
  gtk_widget_realize (window);

  /* we should have the XID now */
  g_assert (embed_xid != 0);

  /* we know what the video sink is in this case (xvimagesink), so we can
   * just set it directly here now (instead of waiting for a prepare-xwindow-id
   * element message in a sync bus handler and setting it there) */
  g_print ("setting XID %lu\n", embed_xid);
  gst_x_overlay_set_xwindow_id (GST_X_OVERLAY (sink), embed_xid);

Stefan Kost's avatar
Stefan Kost committed
275
  g_idle_add (start_pipeline, pipeline);
276
  gtk_main ();
Stefan Kost's avatar
Stefan Kost committed
277

278
279
280
281
  gst_object_unref (pipeline);

  return 0;
}