OOPIF support for layout test pixel dumps.

content/shell/browser/layout_test/blink_test_controller.h

 // Copyright 2013 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_SHELL_BROWSER_LAYOUT_TEST_BLINK_TEST_CONTROLLER_H_
 #define CONTENT_SHELL_BROWSER_LAYOUT_TEST_BLINK_TEST_CONTROLLER_H_
 
 #include <map>
 #include <memory>
 #include <ostream>
 #include <set>
 #include <string>
 
 #include "base/cancelable_callback.h"
 #include "base/files/file_path.h"
 #include "base/macros.h"
 #include "base/scoped_observer.h"
 #include "base/sequence_checker.h"
 #include "base/synchronization/lock.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "content/public/browser/bluetooth_chooser.h"
 #include "content/public/browser/gpu_data_manager_observer.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
+#include "content/public/browser/readback_types.h"
 #include "content/public/browser/render_process_host_observer.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/common/web_preferences.h"
 #include "content/shell/common/layout_test.mojom.h"
 #include "content/shell/common/leak_detection_result.h"
 #include "ui/gfx/geometry/size.h"
 
 #if defined(OS_ANDROID)
 #include "base/threading/thread_restrictions.h"
 #endif
@@ skipping 159 matching lines @@
 
   Shell* SecondaryWindow();
   void LoadDevToolsJSTest();
   void DiscardMainWindow();
 
   // Message handlers.
   void OnAudioDump(const std::vector<unsigned char>& audio_dump);
   void OnImageDump(const std::string& actual_pixel_hash, const SkBitmap& image);
   void OnTextDump(const std::string& dump);
   void OnInitiateLayoutDump();
+  void OnInitiatePixelsDump();
+  void TriggerCompositingIfAllChildrenHaveBeenComposited();
+  void TriggerCompositing(RenderFrameHost* frame);
+  void OnTriggerCompositingResponse(int frame_tree_node_id, int process_id);
   void OnDumpFrameLayoutResponse(int frame_tree_node_id,
                                  const std::string& dump);
+  void OnPixelsDumpCaptured(const SkBitmap& dump, ReadbackResponse response);
   void OnPrintMessageToStderr(const std::string& message);
   void OnPrintMessage(const std::string& message);
   void OnOverridePreferences(const WebPreferences& prefs);
   void OnSetPopupBlockingEnabled(bool block_popups);
   void OnTestFinished();
   void OnClearDevToolsLocalStorage();
   void OnShowDevTools(const std::string& settings,
                       const std::string& frontend_url);
   void OnEvaluateInDevTools(int call_id, const std::string& script);
   void OnCloseDevTools();
@@ skipping 56 matching lines @@
   bool should_override_prefs_;
   WebPreferences prefs_;
 
   NotificationRegistrar registrar_;
 
   const bool is_leak_detection_enabled_;
   bool crash_when_leak_found_;
 
   std::unique_ptr<LayoutTestBluetoothChooserFactory> bluetooth_chooser_factory_;
 
-  // Map from frame_tree_node_id into frame-specific dumps.
+  // Map from frame_tree_node_id into frame-specific text/layout dumps.
   std::map<int, std::string> frame_to_layout_dump_map_;
   // Number of LayoutTestControl.DumpFrameLayout responses we are waiting for.
   int pending_layout_dumps_;
 
+  // Stores frame tree node ids of already composited frames (frames for which
+  // OnTriggerCompositingResponse was already called).
+  std::set<int> composited_frames_;
+
   // Renderer processes are observed to detect crashes.
   ScopedObserver<RenderProcessHost, RenderProcessHostObserver>
       render_process_host_observer_;
   std::set<RenderProcessHost*> all_observed_render_process_hosts_;
   std::set<RenderProcessHost*> main_window_render_process_hosts_;
 
   // Changes reported by OnLayoutTestRuntimeFlagsChanged that have accumulated
   // since PrepareForLayoutTest (i.e. changes that need to be send to a fresh
   // renderer created while test is in progress).
   base::DictionaryValue accumulated_layout_test_runtime_flags_changes_;
 
   // Map from one frame to one mojo pipe.
   std::map<RenderFrameHost*, mojom::LayoutTestControlAssociatedPtr>
       layout_test_control_map_;
 #if defined(OS_ANDROID)
   // Because of the nested message pump implementation, Android needs to allow
   // waiting on the UI thread while layout tests are being ran.
   ScopedAllowWaitForAndroidLayoutTests reduced_restrictions_;
 #endif
 
   SEQUENCE_CHECKER(sequence_checker_);
 
   DISALLOW_COPY_AND_ASSIGN(BlinkTestController);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_SHELL_BROWSER_LAYOUT_TEST_BLINK_TEST_CONTROLLER_H_