chromium/ash/wm/switchable_windows.cc

// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "ash/wm/switchable_windows.h"

#include <array>

#include "ash/public/cpp/shell_window_ids.h"
#include "ash/wm/desks/desks_util.h"
#include "base/containers/contains.h"
#include "ui/aura/window.h"

namespace ash {

namespace {

constexpr std::array<int, 3> kSwitchableContainers = {
    kShellWindowId_AlwaysOnTopContainer,
    kShellWindowId_FloatContainer,
    kShellWindowId_PipContainer,
};

std::vector<int> GetSwitchableContainerIds() {
  std::vector<int> ids = desks_util::GetDesksContainersIds();
  for (const int id : kSwitchableContainers)
    ids.emplace_back(id);

  return ids;
}

}  // namespace

std::vector<aura::Window*> GetSwitchableContainersForRoot(
    aura::Window* root,
    bool active_desk_only) {
  DCHECK(root);
  DCHECK(root->IsRootWindow());

  std::vector<aura::Window*> containers;
  if (active_desk_only) {
    containers.push_back(desks_util::GetActiveDeskContainerForRoot(root));
    containers.push_back(
        root->GetChildById(kShellWindowId_AlwaysOnTopContainer));
    return containers;
  }

  for (const auto& id : GetSwitchableContainerIds()) {
    auto* container = root->GetChildById(id);
    DCHECK(container);
    containers.push_back(container);
  }

  return containers;
}

// TODO(afakhry): Rename this to a better name.
bool IsSwitchableContainer(const aura::Window* window) {
  if (!window)
    return false;

  return base::Contains(GetSwitchableContainerIds(), window->GetId());
}

}  // namespace ash