// Copyright 2019 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/performance_manager/graph/graph_impl_operations.h"

#include "chrome/browser/performance_manager/graph/process_node_impl.h"

namespace performance_manager {

// static
base::flat_set<PageNodeImpl*> GraphImplOperations::GetAssociatedPageNodes(
    const ProcessNodeImpl* process) {
  base::flat_set<PageNodeImpl*> page_nodes;
  for (auto* frame_node : process->frame_nodes())
    page_nodes.insert(frame_node->page_node());
  return page_nodes;
}

// static
base::flat_set<ProcessNodeImpl*> GraphImplOperations::GetAssociatedProcessNodes(
    const PageNodeImpl* page) {
  base::flat_set<ProcessNodeImpl*> process_nodes;
  VisitFrameTreePreOrder(page,
                         [&process_nodes](FrameNodeImpl* frame_node) -> bool {
                           if (auto* process_node = frame_node->process_node())
                             process_nodes.insert(process_node);
                           return true;
                         });
  return process_nodes;
}

// static
std::vector<FrameNodeImpl*> GraphImplOperations::GetFrameNodes(
    const PageNodeImpl* page) {
  std::vector<FrameNodeImpl*> frame_nodes;
  frame_nodes.reserve(20);  // This is in the 99.9th %ile of frame tree sizes.

  for (auto* main_frame_node : page->main_frame_nodes())
    frame_nodes.push_back(main_frame_node);

  for (size_t i = 0; i < frame_nodes.size(); ++i) {
    auto* parent_frame_node = frame_nodes[i];
    for (auto* frame_node : parent_frame_node->child_frame_nodes())
      frame_nodes.push_back(frame_node);
  }

  return frame_nodes;
}

// static
bool GraphImplOperations::HasFrame(const PageNodeImpl* page,
                                   FrameNodeImpl* frame) {
  bool has_frame = false;
  VisitFrameTreePreOrder(page, [&has_frame, frame](FrameNodeImpl* f) -> bool {
    if (f == frame) {
      has_frame = true;
      return false;
    }
    return true;
  });
  return has_frame;
}

}  // namespace performance_manager