Mus+Ash: Unify CompositorFrameSinks

services/ui/ws/server_window_compositor_frame_sink.h

 // Copyright 2015 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 SERVICES_UI_WS_SERVER_WINDOW_COMPOSITOR_FRAME_SINK_H_
 #define SERVICES_UI_WS_SERVER_WINDOW_COMPOSITOR_FRAME_SINK_H_
 
 #include <set>
 
 #include "base/macros.h"
 #include "cc/ipc/compositor_frame.mojom.h"
 #include "cc/ipc/mojo_compositor_frame_sink.mojom.h"
+#include "cc/output/context_provider.h"
+#include "cc/surfaces/display.h"
+#include "cc/surfaces/display_client.h"
 #include "cc/surfaces/frame_sink_id.h"
 #include "cc/surfaces/surface_factory.h"
 #include "cc/surfaces/surface_factory_client.h"
 #include "cc/surfaces/surface_id.h"
 #include "cc/surfaces/surface_id_allocator.h"
 #include "mojo/public/cpp/bindings/binding.h"
 #include "services/ui/public/interfaces/window_tree.mojom.h"
 #include "services/ui/ws/ids.h"
 
+namespace gpu {
+class GpuMemoryBufferManager;
+}
+
 namespace ui {
 
 class DisplayCompositor;
 
 namespace ws {
 
 class ServerWindow;
 class ServerWindowCompositorFrameSinkManager;
 
 // Server side representation of a WindowSurface.
 class ServerWindowCompositorFrameSink
     : public cc::mojom::MojoCompositorFrameSink,
+      public cc::DisplayClient,
       public cc::SurfaceFactoryClient {
  public:
   ServerWindowCompositorFrameSink(
       ServerWindowCompositorFrameSinkManager* manager,
       const cc::FrameSinkId& frame_sink_id,
+      gfx::AcceleratedWidget widget,
+      gpu::GpuMemoryBufferManager* gpu_memory_buffer_manager,
+      scoped_refptr<cc::ContextProvider> context_provider,
       cc::mojom::MojoCompositorFrameSinkRequest request,
       cc::mojom::MojoCompositorFrameSinkClientPtr client);
 
   ~ServerWindowCompositorFrameSink() override;
 
   // cc::mojom::MojoCompositorFrameSink:
   void SetNeedsBeginFrame(bool needs_begin_frame) override;
   void SubmitCompositorFrame(cc::CompositorFrame frame) override;
 
  private:
+  void InitDisplay(gfx::AcceleratedWidget widget,
+                   gpu::GpuMemoryBufferManager* gpu_memory_buffer_manager,
+                   scoped_refptr<cc::ContextProvider> context_provider);
+
   void DidReceiveCompositorFrameAck();
 
+  // DisplayClient implementation.
+  void DisplayOutputSurfaceLost() override;
+  void DisplayWillDrawAndSwap(bool will_draw_and_swap,
+                              const cc::RenderPassList& render_passes) override;
+  void DisplayDidDrawAndSwap() override;
+
   // SurfaceFactoryClient implementation.
   void ReturnResources(const cc::ReturnedResourceArray& resources) override;
   void SetBeginFrameSource(cc::BeginFrameSource* begin_frame_source) override;
 
   const cc::FrameSinkId frame_sink_id_;
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
 
   ServerWindowCompositorFrameSinkManager* manager_;  // Owns this.
 
   gfx::Size last_submitted_frame_size_;
+  std::unique_ptr<cc::Display> display_;
 
   cc::LocalFrameId local_frame_id_;
   cc::SurfaceIdAllocator surface_id_allocator_;
   cc::SurfaceFactory surface_factory_;
 
   cc::mojom::MojoCompositorFrameSinkClientPtr client_;
   mojo::Binding<MojoCompositorFrameSink> binding_;
 
   DISALLOW_COPY_AND_ASSIGN(ServerWindowCompositorFrameSink);
 };
 
 }  // namespace ws
 
 }  // namespace ui
 
 #endif  // SERVICES_UI_WS_SERVER_WINDOW_COMPOSITOR_FRAME_SINK_H_