graph_operations.cc 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. // Copyright 2019 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/performance_manager/public/graph/graph_operations.h"
  5. #include "components/performance_manager/graph/frame_node_impl.h"
  6. #include "components/performance_manager/graph/graph_impl_operations.h"
  7. #include "components/performance_manager/graph/graph_impl_util.h"
  8. #include "components/performance_manager/graph/page_node_impl.h"
  9. #include "components/performance_manager/graph/process_node_impl.h"
  10. namespace performance_manager {
  11. // static
  12. base::flat_set<const PageNode*> GraphOperations::GetAssociatedPageNodes(
  13. const ProcessNode* process) {
  14. return UpcastNodeSet<PageNode>(GraphImplOperations::GetAssociatedPageNodes(
  15. ProcessNodeImpl::FromNode(process)));
  16. }
  17. // static
  18. base::flat_set<const ProcessNode*> GraphOperations::GetAssociatedProcessNodes(
  19. const PageNode* page) {
  20. return UpcastNodeSet<ProcessNode>(
  21. GraphImplOperations::GetAssociatedProcessNodes(
  22. PageNodeImpl::FromNode(page)));
  23. }
  24. // static
  25. std::vector<const FrameNode*> GraphOperations::GetFrameNodes(
  26. const PageNode* page) {
  27. auto impls = GraphImplOperations::GetFrameNodes(PageNodeImpl::FromNode(page));
  28. return std::vector<const FrameNode*>(impls.begin(), impls.end());
  29. }
  30. // static
  31. bool GraphOperations::VisitFrameTreePreOrder(const PageNode* page,
  32. const FrameNodeVisitor& visitor) {
  33. return GraphImplOperations::VisitFrameTreePreOrder(
  34. PageNodeImpl::FromNode(page),
  35. [&visitor](FrameNodeImpl* frame_impl) -> bool {
  36. const FrameNode* frame = frame_impl;
  37. return visitor.Run(frame);
  38. });
  39. }
  40. // static
  41. bool GraphOperations::VisitFrameTreePostOrder(const PageNode* page,
  42. const FrameNodeVisitor& visitor) {
  43. return GraphImplOperations::VisitFrameTreePostOrder(
  44. PageNodeImpl::FromNode(page),
  45. [&visitor](FrameNodeImpl* frame_impl) -> bool {
  46. const FrameNode* frame = frame_impl;
  47. return visitor.Run(frame);
  48. });
  49. }
  50. // static
  51. bool GraphOperations::HasFrame(const PageNode* page, const FrameNode* frame) {
  52. return GraphImplOperations::HasFrame(PageNodeImpl::FromNode(page),
  53. FrameNodeImpl::FromNode(frame));
  54. }
  55. } // namespace performance_manager