Implementation of window_snapshot_aura

chrome/browser/ui/window_snapshot/window_snapshot_aura.cc

Index: chrome/browser/ui/window_snapshot/window_snapshot_aura.cc
diff --git a/chrome/browser/ui/window_snapshot/window_snapshot_aura.cc b/chrome/browser/ui/window_snapshot/window_snapshot_aura.cc
index 3a652136ec15054d37a47e04d344303925ecd94d..bbf0253c723d38cfc43ad62645e0c77b0ce8fe40 100644
--- a/chrome/browser/ui/window_snapshot/window_snapshot_aura.cc
+++ b/chrome/browser/ui/window_snapshot/window_snapshot_aura.cc
@@ -5,16 +5,38 @@
 #include "chrome/browser/ui/window_snapshot/window_snapshot.h"
 
 #include "base/logging.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "ui/aura/desktop.h"
+#include "ui/gfx/codec/png_codec.h"
+#include "ui/gfx/compositor/compositor.h"
 #include "ui/gfx/rect.h"
 
 namespace browser {
 
-bool GrabWindowSnapshot(gfx::NativeWindow window_handle,
+bool GrabWindowSnapshot(gfx::NativeWindow window,
                              std::vector<unsigned char>* png_representation,
                              const gfx::Rect& snapshot_bounds) {
-  // TODO(saintlou): Stub for Aura.
-  NOTIMPLEMENTED();
-  return false;
+  ui::Compositor* compositor = aura::Desktop::GetInstance()->compositor();

[sky, 2011/11/18 19:03:44]

Get the compositor from window's layer.

+
+  gfx::Rect desktop_snapshot_bounds = gfx::Rect(
+      snapshot_bounds.origin().Add(window->bounds().origin()),
+      snapshot_bounds.size());
+
+  DCHECK_LE(desktop_snapshot_bounds.right() , compositor->size().width());
+  DCHECK_LE(desktop_snapshot_bounds.bottom(), compositor->size().height());
+
+  SkBitmap bitmap;
+  if (!compositor->ReadPixels(&bitmap, desktop_snapshot_bounds))

[sky, 2011/11/18 19:03:44]

I believe the coordinates passed to the compositor are wrong. You're passing
coordinates relative to the window when they need to be relative to the root.

[jonathan.backer, 2011/11/18 19:37:38]

Are you missing a file? or is this based off of another CL? My pull from this
morning on has ReadPixels taking one parameter.

+    return false;
+
+  unsigned char* pixels = reinterpret_cast<unsigned char*>(bitmap.getPixels());
+
+  gfx::PNGCodec::Encode(pixels, gfx::PNGCodec::FORMAT_BGRA,
+                        snapshot_bounds.size(),
+                        bitmap.rowBytes(), true,
+                        std::vector<gfx::PNGCodec::Comment>(),
+                        png_representation);
+  return true;
 }
 
 }  // namespace browser