Add GrabWindowSnapshot method to mac_util class, and a unit test.

BUG= none
TEST= none
Review URL: http://codereview.chromium.org/333018

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@30199 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 18 insertions, 1 deletions

diff --git a/base/mac_util.mm b/base/mac_util.mm
index 772fe16b..3ffab82 100644
--- a/base/mac_util.mm
+++ b/base/mac_util.mm
@@ -1,4 +1,4 @@
-// Copyright (c) 2008 The Chromium Authors. All rights reserved.
+// Copyright (c) 2008-2009 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
@@ -161,4 +161,21 @@ void ReleaseFullScreen() {
     SetSystemUIMode(kUIModeNormal, 0);
 }
 
+void GrabWindowSnapshot(NSWindow* window,
+    std::vector<unsigned char>* png_representation) {
+  // Make sure to grab the "window frame" view so we get current tab +
+  // tabstrip.
+  NSView* view = [[window contentView] superview];
+  NSBitmapImageRep* rep =
+      [view bitmapImageRepForCachingDisplayInRect:[view bounds]];
+  [view cacheDisplayInRect:[view bounds] toBitmapImageRep:rep];
+  NSData* data = [rep representationUsingType:NSPNGFileType properties:nil];
+  const unsigned char* buf = static_cast<const unsigned char*>([data bytes]);
+  NSUInteger length = [data length];
+  if (buf != NULL && length > 0){
+    png_representation->assign(buf, buf + length);
+    DCHECK(png_representation->size() > 0);
+  }
+}
+
 }  // namespace mac_util