| 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_LE(snapshot_bounds.right(), window_bounds.width()); |
| 41 DCHECK_LE(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 |