cc : Store surface layer ids on LayerTreeHost and push them at commit

cc/layers/surface_layer.cc

Index: cc/layers/surface_layer.cc
diff --git a/cc/layers/surface_layer.cc b/cc/layers/surface_layer.cc
index 573f97b9b13280a3c9f5806052f0c7d81e52bd74..703e029fa6acbf703b8e894c7cc9be9b19788455 100644
--- a/cc/layers/surface_layer.cc
+++ b/cc/layers/surface_layer.cc
@@ -64,22 +64,38 @@ SurfaceLayer::~SurfaceLayer() {
 }
 
 void SurfaceLayer::SetPrimarySurfaceInfo(const SurfaceInfo& surface_info) {
+  const SurfaceId* old_surface_id = GetSurfaceId();
   RemoveReference(std::move(primary_reference_returner_));
   primary_surface_info_ = surface_info;
   if (layer_tree_host()) {
     primary_reference_returner_ = ref_factory_->CreateReference(
         layer_tree_host(), primary_surface_info_.id());
+    const SurfaceId* new_surface_id = GetSurfaceId();
+    if (old_surface_id != new_surface_id) {
+      if (old_surface_id)
+        layer_tree_host()->RemoveSurfaceLayerId(*old_surface_id);
+      if (new_surface_id)
+        layer_tree_host()->AddSurfaceLayerId(*new_surface_id);
+    }
   }
   UpdateDrawsContent(HasDrawableContent());
   SetNeedsCommit();
 }
 
 void SurfaceLayer::SetFallbackSurfaceInfo(const SurfaceInfo& surface_info) {
+  const SurfaceId* old_surface_id = GetSurfaceId();
   RemoveReference(std::move(fallback_reference_returner_));
   fallback_surface_info_ = surface_info;
   if (layer_tree_host()) {
     fallback_reference_returner_ = ref_factory_->CreateReference(
         layer_tree_host(), fallback_surface_info_.id());
+    const SurfaceId* new_surface_id = GetSurfaceId();
+    if (old_surface_id != new_surface_id) {
+      if (old_surface_id)
+        layer_tree_host()->RemoveSurfaceLayerId(*old_surface_id);
+      if (new_surface_id)
+        layer_tree_host()->AddSurfaceLayerId(*new_surface_id);
+    }
   }
   SetNeedsCommit();
 }
@@ -99,11 +115,41 @@ bool SurfaceLayer::HasDrawableContent() const {
   return primary_surface_info_.is_valid() && Layer::HasDrawableContent();
 }
 
+const SurfaceId* SurfaceLayer::GetSurfaceId(LayerTreeHost* new_host) {
+  LayerTreeHost* old_host = layer_tree_host();
+  if (!old_host && !new_host)
+    return nullptr;
+  DCHECK(!old_host || !new_host ||
+         old_host->GetSettings().enable_surface_synchronization ==
+             old_host->GetSettings().enable_surface_synchronization);
+
+  bool enable_surface_synchronization;
+  if (old_host) {
+    enable_surface_synchronization =
+        old_host->GetSettings().enable_surface_synchronization;
+  } else {
+    enable_surface_synchronization =
+        new_host->GetSettings().enable_surface_synchronization;
+  }
+
+  if (enable_surface_synchronization) {
+    if (fallback_surface_info_.is_valid())
+      return &fallback_surface_info_.id();
+  } else if (primary_surface_info_.is_valid()) {
+    return &primary_surface_info_.id();
+  }
+  return nullptr;
+}
+
 void SurfaceLayer::SetLayerTreeHost(LayerTreeHost* host) {
   if (layer_tree_host() == host) {
     Layer::SetLayerTreeHost(host);
     return;
   }
+  const SurfaceId* surface_id = GetSurfaceId(host);
+
+  if (layer_tree_host() && surface_id)
+    layer_tree_host()->RemoveSurfaceLayerId(*surface_id);
   RemoveReference(std::move(primary_reference_returner_));
   RemoveReference(std::move(fallback_reference_returner_));
   Layer::SetLayerTreeHost(host);
@@ -116,6 +162,8 @@ void SurfaceLayer::SetLayerTreeHost(LayerTreeHost* host) {
       fallback_reference_returner_ = ref_factory_->CreateReference(
           layer_tree_host(), fallback_surface_info_.id());
     }
+    if (surface_id)
+      layer_tree_host()->AddSurfaceLayerId(*surface_id);
   }
 }