Add/remove surface references via MojoCompositorFrameSink.

services/ui/ws/frame_generator.h

 // Copyright 2016 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_FRAME_GENERATOR_H_
 #define SERVICES_UI_WS_FRAME_GENERATOR_H_
 
 #include <memory>
 #include <unordered_map>
+#include <vector>
 
 #include "base/macros.h"
-#include "base/timer/timer.h"
-#include "cc/ipc/display_compositor.mojom.h"
 #include "cc/surfaces/frame_sink_id.h"
 #include "cc/surfaces/surface_id.h"
 #include "cc/surfaces/surface_id_allocator.h"
+#include "cc/surfaces/surface_reference.h"
 #include "services/ui/public/interfaces/window_tree_constants.mojom.h"
 #include "services/ui/ws/ids.h"
 #include "services/ui/ws/server_window_delegate.h"
 #include "services/ui/ws/server_window_tracker.h"
 #include "ui/gfx/geometry/rect.h"
 #include "ui/gfx/native_widget_types.h"
 
 namespace cc {
-class CompositorFrame;
 class RenderPass;
-class SurfaceId;
-}
-
-namespace gpu {
-class GpuChannelHost;
 }
 
 namespace ui {
-
-class DisplayCompositor;
-
 namespace ws {
 
 namespace test {
 class FrameGeneratorTest;
 }
 
 class FrameGeneratorDelegate;
 class ServerWindow;
 
 // Responsible for redrawing the display in response to the redraw requests by
@@ skipping 49 matching lines @@
 
   // Does work necessary for adding the first surface reference.
   void AddFirstReference(const cc::SurfaceId& surface_id, ServerWindow* window);
 
   // Finds all Surfaces with references from |old_surface_id| and adds a new
   // reference from |new_surface_id|. The caller should remove any references
   // to |old_surface_id| afterwards to finish cleanup.
   void AddNewParentReferences(const cc::SurfaceId& old_surface_id,
                               const cc::SurfaceId& new_surface_id);
 
+  // Adds all references to surfaces in |new_references_|.
+  void SendAddSurfaceReferences();
+
   // Removes all references to surfaces in |dead_references_|.
-  void RemoveDeadSurfaceReferences();
+  void SendRemoveSurfaceReferences();
 
   // Removes any retained references for the provided FrameSink.
   void RemoveFrameSinkReference(const cc::FrameSinkId& frame_sink_id);
 
   // Removes all retained references to surfaces.
   void RemoveAllSurfaceReferences();
 
-  cc::mojom::DisplayCompositor* GetDisplayCompositor();
-
   // ServerWindowObserver implementation.
   void OnWindowDestroying(ServerWindow* window) override;
 
   FrameGeneratorDelegate* delegate_;
   ServerWindow* const root_window_;
 
   gfx::Size last_submitted_frame_size_;
   cc::LocalFrameId local_frame_id_;
   cc::SurfaceIdAllocator id_allocator_;
   cc::mojom::MojoCompositorFrameSinkPtr compositor_frame_sink_;
 
-  // Represents the top level root surface id that should reference the display
-  // root surface. We don't know the actual value, because it's generated in
-  // another process, this is used internally as a placeholder.
-  const cc::SurfaceId top_level_root_surface_id_;
-
-  struct SurfaceReference {
-    cc::SurfaceId parent_id;
-    cc::SurfaceId child_id;
-  };
-
   // Active references held by this client to surfaces that could be embedded in
   // a CompositorFrame submitted from FrameGenerator.
-  std::unordered_map<cc::FrameSinkId, SurfaceReference, cc::FrameSinkIdHash>
+  std::unordered_map<cc::FrameSinkId, cc::SurfaceReference, cc::FrameSinkIdHash>
       active_references_;
 
   // References to surfaces that should be removed after a CompositorFrame has
   // been submitted and the surfaces are not being used.
-  std::vector<SurfaceReference> dead_references_;
+  std::vector<cc::SurfaceReference> dead_references_;
+
+  // References that should be added before the next CompositorFrame is
+  // submitted.
+  std::vector<cc::SurfaceReference> new_references_;
 
   // If a CompositorFrame for a child surface is submitted before the first
   // display root CompositorFrame, we can't add a reference from the unknown
   // display root SurfaceId. Track the child SurfaceId here and add a reference
   // to it when the display root SurfaceId is available.
   std::vector<cc::SurfaceId> waiting_for_references_;
 
   mojo::Binding<cc::mojom::MojoCompositorFrameSinkClient> binding_;
 
   base::WeakPtrFactory<FrameGenerator> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(FrameGenerator);
 };
 
 }  // namespace ws
 
 }  // namespace ui
 
 #endif  // SERVICES_UI_WS_FRAME_GENERATOR_H_