[devtools] Support different encodings for Page.CaptureScreenshot.

content/browser/renderer_host/render_widget_host_impl.h

 // 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.
 
 #ifndef CONTENT_BROWSER_RENDERER_HOST_RENDER_WIDGET_HOST_IMPL_H_
 #define CONTENT_BROWSER_RENDERER_HOST_RENDER_WIDGET_HOST_IMPL_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
@@ skipping 30 matching lines @@
 #include "content/public/browser/render_widget_host.h"
 #include "content/public/common/page_zoom.h"
 #include "content/public/common/url_constants.h"
 #include "ipc/ipc_listener.h"
 #include "third_party/WebKit/public/platform/WebDisplayMode.h"
 #include "ui/base/ime/text_input_mode.h"
 #include "ui/base/ime/text_input_type.h"
 #include "ui/events/gesture_detection/gesture_provider_config_helper.h"
 #include "ui/events/latency_info.h"
 #include "ui/gfx/native_widget_types.h"
+#include "ui/snapshot/snapshot.h"
 
 struct FrameHostMsg_HittestData_Params;
 struct ViewHostMsg_SelectionBounds_Params;
 struct ViewHostMsg_UpdateRect_Params;
 
 namespace base {
 class RefCountedBytes;
 }
 
 namespace blink {
@@ skipping 141 matching lines @@
   void DragSourceEndedAt(const gfx::Point& client_pt,
                          const gfx::Point& screen_pt,
                          blink::WebDragOperation operation) override;
   void DragSourceSystemDragEnded() override;
   void FilterDropData(DropData* drop_data) override;
 
   // Notification that the screen info has changed.
   void NotifyScreenInfoChanged();
 
   // Forces redraw in the renderer and when the update reaches the browser
-  // grabs snapshot from the compositor. Returns PNG-encoded snapshot.
+  // grabs snapshot from the compositor. Returns encoded snapshot.
   using GetSnapshotFromBrowserCallback =
       base::Callback<void(const unsigned char*, size_t)>;
-  void GetSnapshotFromBrowser(const GetSnapshotFromBrowserCallback& callback);
+  void GetSnapshotFromBrowser(const GetSnapshotFromBrowserCallback& callback,
+                              ui::SnapshotEncoding encoding,
+                              ui::SnapshotQuality quality);
 
   const NativeWebKeyboardEvent* GetLastKeyboardEvent() const;
 
   // Sets the View of this RenderWidgetHost.
   void SetView(RenderWidgetHostViewBase* view);
 
   RenderWidgetHostDelegate* delegate() const { return delegate_; }
 
   bool empty() const { return current_size_.IsEmpty(); }
 
@@ skipping 467 matching lines @@
   void OnUnexpectedEventAck(UnexpectedEventAckType type) override;
 
   void OnSyntheticGestureCompleted(SyntheticGesture::Result result);
 
   // Called when there is a new auto resize (using a post to avoid a stack
   // which may get in recursive loops).
   void DelayedAutoResized();
 
   void WindowSnapshotReachedScreen(int snapshot_id);
 
-  void OnSnapshotDataReceived(int snapshot_id,
-                              const unsigned char* png,
+  void OnSnapshotDataReceived(const GetSnapshotFromBrowserCallback& callback,
+                              const unsigned char* data,
                               size_t size);
 
   void OnSnapshotDataReceivedAsync(
-      int snapshot_id,
-      scoped_refptr<base::RefCountedBytes> png_data);
+      const GetSnapshotFromBrowserCallback& callback,
+      scoped_refptr<base::RefCountedBytes> data);
 
   // 1. Grants permissions to URL (if any)
   // 2. Grants permissions to filenames
   // 3. Grants permissions to file system files.
   // 4. Register the files with the IsolatedContext.
   void GrantFileAccessFromDropData(DropData* drop_data);
 
   // true if a renderer has once been valid. We use this flag to display a sad
   // tab only when we lose our renderer and not if a paint occurs during
   // initialization.
@@ skipping 139 matching lines @@
   // |new_content_rendering_timeout_| is not needed. The renderer will send
   // both the FirstPaintAfterLoad and DidCommitProvisionalLoad messages after
   // any new page navigation, it doesn't matter which is received first, and
   // it should not be possible to interleave other navigations in between
   // receipt of those messages (unless FirstPaintAfterLoad is prevented from
   // being sent, in which case the timer should fire).
   bool received_paint_after_load_;
 
   RenderWidgetHostLatencyTracker latency_tracker_;
 
+  struct PendingSnapshotParams {
+    PendingSnapshotParams(const GetSnapshotFromBrowserCallback& callback,
+                          ui::SnapshotEncoding encoding,
+                          ui::SnapshotQuality quality);
+    PendingSnapshotParams(const PendingSnapshotParams& params);
+    ~PendingSnapshotParams();
+
+    GetSnapshotFromBrowserCallback callback;
+    ui::SnapshotEncoding encoding;
+    ui::SnapshotQuality quality;
+  };
+  using PendingSnapshotParamsMap = std::map<int, PendingSnapshotParams>;
+
   int next_browser_snapshot_id_;
-  using PendingSnapshotMap = std::map<int, GetSnapshotFromBrowserCallback>;
-  PendingSnapshotMap pending_browser_snapshots_;
+  PendingSnapshotParamsMap pending_browser_snapshot_params_;
+  int pending_browser_snapshots_;

[pfeldman, 2017/01/03 18:42:33]

not sure why you need it now.

[Eric Seckler, 2017/01/03 19:18:09]

Currently, OnSnapshotDataReceived can't remove from
pending_browser_snapshot_params_, because it may be called synchronously during
iteration on the map in WindowSnapshotReachedScreen. So I'm removing in
WindowSnapshotReachedScreen instead and using this count to figure out if
OnSnapshotDataReceived should reset the power saver blocker.

 
   // Indicates whether a RenderFramehost has ownership, in which case this
   // object does not self destroy.
   bool owned_by_render_frame_host_;
 
   // Indicates whether this RenderWidgetHost thinks is focused. This is trying
   // to match what the renderer process knows. It is different from
   // RenderWidgetHostView::HasFocus in that in that the focus request may fail,
   // causing HasFocus to return false when is_focused_ is true.
   bool is_focused_;
@@ skipping 25 matching lines @@
 #endif
 
   base::WeakPtrFactory<RenderWidgetHostImpl> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(RenderWidgetHostImpl);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_RENDERER_HOST_RENDER_WIDGET_HOST_IMPL_H_