Implement CompositorCC::ReadPixels

ui/gfx/compositor/compositor_cc.cc

 // Copyright (c) 2011 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.
 
 #include "ui/gfx/compositor/compositor_cc.h"
 
+#include "third_party/skia/include/images/SkImageEncoder.h"
+#include "third_party/skia/include/core/SkBitmap.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebCompositor.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFloatPoint.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebRect.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebSize.h"
 #include "ui/gfx/compositor/layer.h"
 #include "ui/gfx/gl/gl_context.h"
 #include "ui/gfx/gl/gl_surface.h"
 #include "ui/gfx/gl/gl_implementation.h"
 #include "webkit/glue/webthread_impl.h"
 #include "webkit/gpu/webgraphicscontext3d_in_process_impl.h"
 
 namespace {
 webkit_glue::WebThreadImpl* g_compositor_thread = NULL;
@@ skipping 148 matching lines @@
 
 void CompositorCC::OnRootLayerChanged() {
   root_web_layer_.removeAllChildren();
   root_web_layer_.addChild(root_layer()->web_layer());
 }
 
 void CompositorCC::DrawTree() {
   host_.composite();
 }
 
-void CompositorCC::ReadPixels(SkBitmap* bitmap) {
-  NOTIMPLEMENTED();
+bool CompositorCC::ReadPixels(SkBitmap* bitmap) {
+  bitmap->setConfig(SkBitmap::kARGB_8888_Config,
+                    size().width(), size().height());
+  bitmap->allocPixels();
+  SkAutoLockPixels lock_image(*bitmap);
+  unsigned char* pixels = static_cast<unsigned char*>(bitmap->getPixels());
+  if (host_.compositeAndReadback(pixels, gfx::Rect(size()))) {
+    SwizzleRGBAToBGRAAndFlip(pixels, size());
+    return true;
+  }
+  return false;
 }
 
 void CompositorCC::animateAndLayout(double frameBeginTime) {
 }
 
 void CompositorCC::applyScrollAndScale(const WebKit::WebSize& scrollDelta,
                                        float scaleFactor) {
 }
 
 void CompositorCC::applyScrollDelta(const WebKit::WebSize&) {
@@ skipping 20 matching lines @@
   ScheduleDraw();
 }
 
 Compositor* Compositor::Create(CompositorDelegate* owner,
                                gfx::AcceleratedWidget widget,
                                const gfx::Size& size) {
   return new CompositorCC(owner, widget, size);
 }
 
 }  // namespace ui