Add fromSurface optional parameter to devtools Page.CaptureScreenshot

content/browser/renderer_host/render_widget_host_impl.cc

 // Copyright (c) 2012 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 "content/browser/renderer_host/render_widget_host_impl.h"
 
 #include <math.h>
 
 #include <set>
 #include <tuple>
@@ skipping 1401 matching lines @@
   // factor).
   WasResized();
 
   if (touch_emulator_) {
     touch_emulator_->SetDeviceScaleFactor(
         view_.get() ? content::GetScaleFactorForView(view_.get()) : 1.0f);
   }
 }
 
 void RenderWidgetHostImpl::GetSnapshotFromBrowser(
-    const GetSnapshotFromBrowserCallback& callback) {
+    const GetSnapshotFromBrowserCallback& callback,
+    bool from_surface) {
   int id = next_browser_snapshot_id_++;
+  if (from_surface) {
+    pending_surface_browser_snapshots_.insert(std::make_pair(id, callback));
+    ui::LatencyInfo latency_info;
+    latency_info.AddLatencyNumber(ui::WINDOW_SNAPSHOT_FRAME_NUMBER_COMPONENT, 0,
+                                  id);
+    Send(new ViewMsg_ForceRedraw(GetRoutingID(), latency_info));
+    return;
+  }
 
 #if defined(OS_MACOSX)
   // MacOS version of underlying GrabViewSnapshot() blocks while
   // display/GPU are in a power-saving mode, so make sure display
   // does not go to sleep for the duration of reading a snapshot.
   if (pending_browser_snapshots_.empty()) {
     DCHECK(!power_save_blocker_);
     power_save_blocker_.reset(new device::PowerSaveBlocker(
         device::PowerSaveBlocker::kPowerSaveBlockPreventDisplaySleep,
         device::PowerSaveBlocker::kReasonOther, "GetSnapshot",
@@ skipping 902 matching lines @@
                                   is_running_navigation_hint_task);
 }
 
 void RenderWidgetHostImpl::DidReceiveRendererFrame() {
   view_->DidReceiveRendererFrame();
 }
 
 void RenderWidgetHostImpl::WindowSnapshotReachedScreen(int snapshot_id) {
   DCHECK(base::MessageLoopForUI::IsCurrent());
 
+  if (!pending_surface_browser_snapshots_.empty()) {
+    GetView()->CopyFromSurface(
+        gfx::Rect(), gfx::Size(),
+        base::Bind(&RenderWidgetHostImpl::OnSnapshotFromSurfaceReceived,
+                   weak_factory_.GetWeakPtr(), snapshot_id, 0),
+        kN32_SkColorType);
+  }
+
+  if (!pending_browser_snapshots_.empty()) {
 #if defined(OS_ANDROID)
-  // On Android, call sites should pass in the bounds with correct offset
-  // to capture the intended content area.
-  gfx::Rect snapshot_bounds(GetView()->GetViewBounds());
-  snapshot_bounds.Offset(0, GetView()->GetNativeView()->content_offset().y());
+    // On Android, call sites should pass in the bounds with correct offset
+    // to capture the intended content area.
+    gfx::Rect snapshot_bounds(GetView()->GetViewBounds());
+    snapshot_bounds.Offset(0, GetView()->GetNativeView()->content_offset().y());
 #else
-  gfx::Rect snapshot_bounds(GetView()->GetViewBounds().size());
+    gfx::Rect snapshot_bounds(GetView()->GetViewBounds().size());
 #endif
 
-  gfx::Image image;
-  if (ui::GrabViewSnapshot(GetView()->GetNativeView(), snapshot_bounds,
-                           &image)) {
-    OnSnapshotReceived(snapshot_id, image);
+    gfx::Image image;
+    if (ui::GrabViewSnapshot(GetView()->GetNativeView(), snapshot_bounds,
+                             &image)) {
+      OnSnapshotReceived(snapshot_id, image);
+      return;
+    }
+
+    ui::GrabViewSnapshotAsync(
+        GetView()->GetNativeView(), snapshot_bounds,
+        base::Bind(&RenderWidgetHostImpl::OnSnapshotReceived,
+                   weak_factory_.GetWeakPtr(), snapshot_id));
+  }
+}
+
+void RenderWidgetHostImpl::OnSnapshotFromSurfaceReceived(
+    int snapshot_id,
+    int retry_count,
+    const SkBitmap& bitmap,
+    ReadbackResponse response) {
+  static const int kMaxRetries = 5;
+  if (response != READBACK_SUCCESS && retry_count < kMaxRetries) {
+    GetView()->CopyFromSurface(
+        gfx::Rect(), gfx::Size(),
+        base::Bind(&RenderWidgetHostImpl::OnSnapshotFromSurfaceReceived,
+                   weak_factory_.GetWeakPtr(), snapshot_id, retry_count + 1),
+        kN32_SkColorType);
     return;
   }
-
-  ui::GrabViewSnapshotAsync(
-      GetView()->GetNativeView(), snapshot_bounds,
-      base::Bind(&RenderWidgetHostImpl::OnSnapshotReceived,
-                 weak_factory_.GetWeakPtr(), snapshot_id));
+  // If all retries have failed, we return an empty image.
+  gfx::Image image;
+  if (response == READBACK_SUCCESS)
+    image = gfx::Image::CreateFrom1xBitmap(bitmap);
+  // Any pending snapshots with a lower ID than the one received are considered
+  // to be implicitly complete, and returned the same snapshot data.
+  PendingSnapshotMap::iterator it = pending_surface_browser_snapshots_.begin();
+  while (it != pending_surface_browser_snapshots_.end()) {
+    if (it->first <= snapshot_id) {
+      it->second.Run(image);
+      pending_surface_browser_snapshots_.erase(it++);
+    } else {
+      ++it;
+    }
+  }
 }
 
 void RenderWidgetHostImpl::OnSnapshotReceived(int snapshot_id,
                                               const gfx::Image& image) {
   // Any pending snapshots with a lower ID than the one received are considered
   // to be implicitly complete, and returned the same snapshot data.
   PendingSnapshotMap::iterator it = pending_browser_snapshots_.begin();
   while (it != pending_browser_snapshots_.end()) {
     if (it->first <= snapshot_id) {
       it->second.Run(image);
@@ skipping 121 matching lines @@
     // different from the receiver's.
     file_system_file.url =
         GURL(storage::GetIsolatedFileSystemRootURIString(
                  file_system_url.origin(), filesystem_id, std::string())
                  .append(register_name));
     file_system_file.filesystem_id = filesystem_id;
   }
 }
 
 }  // namespace content