Add "stylus-tools" Wayland protocol.

chrome/browser/chromeos/arc/voice_interaction/arc_voice_interaction_framework_service.cc

 // Copyright 2017 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.
 
 #include "chrome/browser/chromeos/arc/voice_interaction/arc_voice_interaction_framework_service.h"
 
 #include <utility>
 #include <vector>
 
 #include "ash/accelerators/accelerator_controller.h"
 #include "ash/public/cpp/shell_window_ids.h"
 #include "ash/shell.h"
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/containers/flat_set.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted.h"
 #include "base/task_scheduler/post_task.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chromeos/chromeos_switches.h"
 #include "components/arc/arc_bridge_service.h"
 #include "components/arc/instance_holder.h"
+#include "components/exo/surface.h"
 #include "content/public/browser/browser_thread.h"
+#include "ui/aura/client/aura_constants.h"
 #include "ui/aura/window.h"
 #include "ui/compositor/layer.h"
 #include "ui/compositor/layer_owner.h"
 #include "ui/compositor/layer_tree_owner.h"
 #include "ui/gfx/image/image.h"
 #include "ui/gfx/image/image_util.h"
 #include "ui/gfx/native_widget_types.h"
 #include "ui/snapshot/snapshot.h"
 #include "ui/snapshot/snapshot_aura.h"
 #include "ui/wm/core/window_util.h"
 #include "ui/wm/public/activation_client.h"
 
 namespace arc {
 
 namespace {
 
+using LayerSet = base::flat_set<const ui::Layer*>;
+
 void ScreenshotCallback(
     const mojom::VoiceInteractionFrameworkHost::CaptureFocusedWindowCallback&
         callback,
     scoped_refptr<base::RefCountedMemory> data) {
   if (data.get() == nullptr) {
     callback.Run(std::vector<uint8_t>{});
     return;
   }
   std::vector<uint8_t> result(data->front(), data->front() + data->size());
   callback.Run(result);
 }
 
+bool IsMetalayerWindow(aura::Window* window) {
+  exo::Surface* surface = exo::Surface::AsSurface(window);
+  return surface != nullptr && surface->IsStylusOnly();
+}
+
+void CollectLayers(LayerSet& layers,
+                   aura::Window* root_window,
+                   base::Callback<bool(aura::Window*)> matcher_func) {
+  if (matcher_func.Run(root_window))
+    layers.insert(root_window->layer());
+
+  aura::Window::Windows children = root_window->children();
+  for (aura::Window::Windows::iterator iter = children.begin();
+       iter != children.end(); ++iter) {
+    CollectLayers(layers, *iter, matcher_func);
+  }
+}
+
 std::unique_ptr<ui::LayerTreeOwner> CreateLayerTreeForSnapshot(
     aura::Window* root_window) {
-  using LayerSet = base::flat_set<const ui::Layer*>;
   LayerSet blocked_layers;
   for (auto* browser : *BrowserList::GetInstance()) {
     if (browser->profile()->IsOffTheRecord())
       blocked_layers.insert(browser->window()->GetNativeWindow()->layer());
   }
 
   LayerSet excluded_layers;
-  // For the best UX the metalayer has to be excluded from the snapshot.
-  // It is currently impossible to identify the metalayer among others layers
-  // under kShellWindowId_SystemModalContainer. Other layers in this container
-  // are not relevant for this kind of snapshot, so it is safe to exclude all
-  // of them.
-  aura::Window* modal_container = ash::Shell::GetContainer(
-      root_window, ash::kShellWindowId_SystemModalContainer);
-  if (modal_container != nullptr)
-    excluded_layers.insert(modal_container->layer());
+  CollectLayers(excluded_layers, root_window, base::Bind(IsMetalayerWindow));
 
   auto layer_tree_owner = ::wm::RecreateLayersWithClosure(
       root_window, base::BindRepeating(
                        [](LayerSet blocked_layers, LayerSet excluded_layers,
                           ui::LayerOwner* owner) -> std::unique_ptr<ui::Layer> {
                          // Parent layer is excluded meaning that it's pointless
                          // to clone current child and all its descendants. This
                          // reduces the number of layers to create.
                          if (blocked_layers.count(owner->layer()->parent()))
                            return nullptr;
@@ skipping 193 matching lines @@
           SetMetalayerVisibility);
   if (!framework_instance) {
     if (!metalayer_closed_callback_.is_null())
       base::ResetAndReturn(&metalayer_closed_callback_).Run();
     return;
   }
   framework_instance->SetMetalayerVisibility(visible);
 }
 
 }  // namespace arc