OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/ui/window_snapshot/window_snapshot.h" | 5 #include "chrome/browser/ui/window_snapshot/window_snapshot.h" |
6 | 6 |
7 #include <gdk/gdkx.h> | 7 #include <gdk/gdkx.h> |
8 #include <gtk/gtk.h> | 8 #include <gtk/gtk.h> |
9 | 9 |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 #include "ui/base/x/x11_util.h" | 11 #include "ui/base/x/x11_util.h" |
12 #include "ui/gfx/rect.h" | 12 #include "ui/gfx/rect.h" |
13 | 13 |
14 namespace browser { | 14 namespace browser { |
15 | 15 |
16 static cairo_status_t SnapshotCallback( | 16 static cairo_status_t SnapshotCallback( |
17 void *closure, const unsigned char *data, unsigned int length) { | 17 void *closure, const unsigned char *data, unsigned int length) { |
18 std::vector<unsigned char>* png_representation = | 18 std::vector<unsigned char>* png_representation = |
19 static_cast<std::vector<unsigned char>*>(closure); | 19 static_cast<std::vector<unsigned char>*>(closure); |
20 | 20 |
21 size_t old_size = png_representation->size(); | 21 size_t old_size = png_representation->size(); |
22 png_representation->resize(old_size + length); | 22 png_representation->resize(old_size + length); |
23 memcpy(&(*png_representation)[old_size], data, length); | 23 memcpy(&(*png_representation)[old_size], data, length); |
24 return CAIRO_STATUS_SUCCESS; | 24 return CAIRO_STATUS_SUCCESS; |
25 } | 25 } |
26 | 26 |
27 gfx::Rect GrabWindowSnapshot(gfx::NativeWindow gtk_window, | 27 bool GrabWindowSnapshot(gfx::NativeWindow window_handle, |
28 std::vector<unsigned char>* png_representation) { | 28 std::vector<unsigned char>* png_representation, |
29 GdkWindow* gdk_window = GTK_WIDGET(gtk_window)->window; | 29 const gfx::Rect snapshot_bounds) { |
| 30 GdkWindow* gdk_window = GTK_WIDGET(window_handle)->window; |
30 Display* display = GDK_WINDOW_XDISPLAY(gdk_window); | 31 Display* display = GDK_WINDOW_XDISPLAY(gdk_window); |
31 XID win = GDK_WINDOW_XID(gdk_window); | 32 XID win = GDK_WINDOW_XID(gdk_window); |
32 XWindowAttributes attr; | 33 |
33 if (XGetWindowAttributes(display, win, &attr) == 0) { | 34 gfx::Rect window_bounds; |
34 LOG(ERROR) << "Couldn't get window attributes"; | 35 if (ui::GetWindowRect(win, &window_bounds) == 0) { |
35 return gfx::Rect(); | 36 LOG(ERROR) << "Couldn't get window bounds"; |
| 37 return false; |
36 } | 38 } |
| 39 |
| 40 DCHECK(snapshot_bounds.right() <= window_bounds.width()); |
| 41 DCHECK(snapshot_bounds.bottom() <= window_bounds.height()); |
| 42 |
37 XImage* image = XGetImage( | 43 XImage* image = XGetImage( |
38 display, win, 0, 0, attr.width, attr.height, AllPlanes, ZPixmap); | 44 display, win, snapshot_bounds.x(), snapshot_bounds.y(), |
| 45 snapshot_bounds.width(), snapshot_bounds.height(), AllPlanes, ZPixmap); |
39 if (!image) { | 46 if (!image) { |
40 LOG(ERROR) << "Couldn't get image"; | 47 LOG(ERROR) << "Couldn't get image"; |
41 return gfx::Rect(); | 48 return false; |
42 } | 49 } |
43 if (image->depth != 24) { | 50 if (image->depth != 24) { |
44 LOG(ERROR)<< "Unsupported image depth " << image->depth; | 51 LOG(ERROR)<< "Unsupported image depth " << image->depth; |
45 return gfx::Rect(); | 52 return false; |
46 } | 53 } |
47 cairo_surface_t* surface = | 54 cairo_surface_t* surface = |
48 cairo_image_surface_create_for_data( | 55 cairo_image_surface_create_for_data( |
49 reinterpret_cast<unsigned char*>(image->data), | 56 reinterpret_cast<unsigned char*>(image->data), |
50 CAIRO_FORMAT_RGB24, | 57 CAIRO_FORMAT_RGB24, |
51 image->width, | 58 image->width, |
52 image->height, | 59 image->height, |
53 image->bytes_per_line); | 60 image->bytes_per_line); |
54 | 61 |
55 if (!surface) { | 62 if (!surface) { |
56 LOG(ERROR) << "Unable to create Cairo surface from XImage data"; | 63 LOG(ERROR) << "Unable to create Cairo surface from XImage data"; |
57 return gfx::Rect(); | 64 return false; |
58 } | 65 } |
59 cairo_surface_write_to_png_stream( | 66 cairo_surface_write_to_png_stream( |
60 surface, SnapshotCallback, png_representation); | 67 surface, SnapshotCallback, png_representation); |
61 cairo_surface_destroy(surface); | 68 cairo_surface_destroy(surface); |
62 | 69 |
63 return gfx::Rect(image->width, image->height); | 70 return true; |
64 } | 71 } |
65 | 72 |
66 } // namespace browser | 73 } // namespace browser |
OLD | NEW |