Pileus Git - ~andy/gtk/blob - docs/tools/shooter.c

   1 #include <gdk/gdk.h>
   2 #include <gtk/gtk.h>
   3 #include <gdkx.h>
   4 #include <stdio.h>
   5 #include <errno.h>
   6 #include <sys/wait.h>
   7 #include <unistd.h>
   8 #include <X11/extensions/shape.h>
   9
  10 #include <gdk-pixbuf/gdk-pixbuf.h>
  11 #include <sys/types.h>
  12 #include <sys/stat.h>
  13 #include <sys/wait.h>
  14 #include <signal.h>
  15 #include <unistd.h>
  16 #include <stdlib.h>
  17 #include <fcntl.h>
  18 #include <errno.h>
  19 #include <locale.h>
  20 #include "widgets.h"
  21 #include "shadow.h"
  22
  23 #define MAXIMUM_WM_REPARENTING_DEPTH 4
  24 #ifndef _
  25 #define _(x) (x)
  26 #endif
  27
  28 static Window
  29 find_toplevel_window (Window xid)
  30 {
  31   Window root, parent, *children;
  32   guint nchildren;
  33
  34   do
  35     {
  36       if (XQueryTree (GDK_DISPLAY_XDISPLAY (gdk_display_get_default ()), xid, &root,
  37                       &parent, &children, &nchildren) == 0)
  38         {
  39           g_warning ("Couldn't find window manager window");
  40           return 0;
  41         }
  42
  43       if (root == parent)
  44         return xid;
  45
  46       xid = parent;
  47     }
  48   while (TRUE);
  49 }
  50
  51 static GdkPixbuf *
  52 add_border_to_shot (GdkPixbuf *pixbuf)
  53 {
  54   GdkPixbuf *retval;
  55
  56   retval = gdk_pixbuf_new (GDK_COLORSPACE_RGB, TRUE, 8,
  57                            gdk_pixbuf_get_width (pixbuf) + 2,
  58                            gdk_pixbuf_get_height (pixbuf) + 2);
  59
  60   /* Fill with solid black */
  61   gdk_pixbuf_fill (retval, 0xFF);
  62   gdk_pixbuf_copy_area (pixbuf,
  63                         0, 0,
  64                         gdk_pixbuf_get_width (pixbuf),
  65                         gdk_pixbuf_get_height (pixbuf),
  66                         retval, 1, 1);
  67
  68   return retval;
  69 }
  70
  71 static GdkPixbuf *
  72 remove_shaped_area (GdkPixbuf *pixbuf,
  73                     Window     window)
  74 {
  75   GdkPixbuf *retval;
  76   XRectangle *rectangles;
  77   int rectangle_count, rectangle_order;
  78   int i;
  79
  80   retval = gdk_pixbuf_new (GDK_COLORSPACE_RGB, TRUE, 8,
  81                            gdk_pixbuf_get_width (pixbuf),
  82                            gdk_pixbuf_get_height (pixbuf));
  83
  84   gdk_pixbuf_fill (retval, 0);
  85   rectangles = XShapeGetRectangles (GDK_DISPLAY_XDISPLAY (gdk_display_get_default ()), window,
  86                                     ShapeBounding, &rectangle_count, &rectangle_order);
  87
  88   for (i = 0; i < rectangle_count; i++)
  89     {
  90       int y, x;
  91
  92       for (y = rectangles[i].y; y < rectangles[i].y + rectangles[i].height; y++)
  93         {
  94           guchar *src_pixels, *dest_pixels;
  95
  96           src_pixels = gdk_pixbuf_get_pixels (pixbuf) +
  97             y * gdk_pixbuf_get_rowstride (pixbuf) +
  98             rectangles[i].x * (gdk_pixbuf_get_has_alpha (pixbuf) ? 4 : 3);
  99           dest_pixels = gdk_pixbuf_get_pixels (retval) +
 100             y * gdk_pixbuf_get_rowstride (retval) +
 101             rectangles[i].x * 4;
 102
 103           for (x = rectangles[i].x; x < rectangles[i].x + rectangles[i].width; x++)
 104             {
 105               *dest_pixels++ = *src_pixels ++;
 106               *dest_pixels++ = *src_pixels ++;
 107               *dest_pixels++ = *src_pixels ++;
 108               *dest_pixels++ = 255;
 109
 110               if (gdk_pixbuf_get_has_alpha (pixbuf))
 111                 src_pixels++;
 112             }
 113         }
 114     }
 115
 116   return retval;
 117 }
 118
 119 static GdkPixbuf *
 120 take_window_shot (Window   child,
 121                   gboolean include_decoration)
 122 {
 123   GdkWindow *window;
 124   Display *disp;
 125   Window w, xid;
 126   gint x_orig, y_orig;
 127   gint x = 0, y = 0;
 128   gint width, height;
 129
 130   GdkPixbuf *tmp, *tmp2;
 131   GdkPixbuf *retval;
 132
 133   disp = GDK_DISPLAY_XDISPLAY (gdk_display_get_default ());
 134   w = GDK_ROOT_WINDOW ();
 135
 136   if (include_decoration)
 137     xid = find_toplevel_window (child);
 138   else
 139     xid = child;
 140
 141   window = gdk_window_foreign_new (xid);
 142
 143   width = gdk_window_get_width (window);
 144   height = gdk_window_get_height (window);
 145   gdk_window_get_origin (window, &x_orig, &y_orig);
 146
 147   if (x_orig < 0)
 148     {
 149       x = - x_orig;
 150       width = width + x_orig;
 151       x_orig = 0;
 152     }
 153
 154   if (y_orig < 0)
 155     {
 156       y = - y_orig;
 157       height = height + y_orig;
 158       y_orig = 0;
 159     }
 160
 161   if (x_orig + width > gdk_screen_width ())
 162     width = gdk_screen_width () - x_orig;
 163
 164   if (y_orig + height > gdk_screen_height ())
 165     height = gdk_screen_height () - y_orig;
 166
 167   tmp = gdk_pixbuf_get_from_window (window,
 168                                     x, y, width, height);
 169
 170   if (include_decoration)
 171     tmp2 = remove_shaped_area (tmp, xid);
 172   else
 173     tmp2 = add_border_to_shot (tmp);
 174
 175   retval = create_shadowed_pixbuf (tmp2);
 176   g_object_unref (tmp);
 177   g_object_unref (tmp2);
 178
 179   return retval;
 180 }
 181
 182 int main (int argc, char **argv)
 183 {
 184   GList *toplevels;
 185   GdkPixbuf *screenshot = NULL;
 186   GList *node;
 187
 188   /* If there's no DISPLAY, we silently error out.  We don't want to break
 189    * headless builds. */
 190   if (! gtk_init_check (&argc, &argv))
 191     return 0;
 192
 193   toplevels = get_all_widgets ();
 194
 195   for (node = toplevels; node; node = g_list_next (node))
 196     {
 197       GtkAllocation allocation;
 198       GdkWindow *window;
 199       WidgetInfo *info;
 200       XID id;
 201       char *filename;
 202
 203       info = node->data;
 204
 205       gtk_widget_show (info->window);
 206
 207       window = gtk_widget_get_window (info->window);
 208       gtk_widget_get_allocation (info->window, &allocation);
 209
 210       gtk_widget_show_now (info->window);
 211       gtk_widget_queue_draw_area (info->window,
 212                                   allocation.x, allocation.y,
 213                                   allocation.width, allocation.height);
 214       gdk_window_process_updates (window, TRUE);
 215
 216       while (gtk_events_pending ())
 217         {
 218           gtk_main_iteration ();
 219         }
 220       sleep (1);
 221
 222       while (gtk_events_pending ())
 223         {
 224           gtk_main_iteration ();
 225         }
 226
 227       id = gdk_x11_drawable_get_xid (GDK_DRAWABLE (window));
 228       screenshot = take_window_shot (id, info->include_decorations);
 229       filename = g_strdup_printf ("./%s.png", info->name);
 230       gdk_pixbuf_save (screenshot, filename, "png", NULL, NULL);
 231       g_free(filename);
 232       gtk_widget_hide (info->window);
 233     }
 234
 235   return 0;
 236 }