OLD | NEW |
(Empty) | |
| 1 // Copyright 2013 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 |
| 5 #include "content/browser/renderer_host/frame_tree.h" |
| 6 |
| 7 #include <queue> |
| 8 |
| 9 #include "base/bind.h" |
| 10 #include "base/callback.h" |
| 11 #include "content/browser/renderer_host/frame_tree_node.h" |
| 12 #include "content/browser/renderer_host/render_frame_host_impl.h" |
| 13 |
| 14 namespace content { |
| 15 |
| 16 namespace { |
| 17 // Used with FrameTree::ForEach() to search for the FrameTreeNode |
| 18 // corresponding to |frame_id|. |
| 19 bool FrameTreeNodeForId(int64 frame_id, FrameTreeNode** out_node, |
| 20 FrameTreeNode* node) { |
| 21 if (node->frame_id() == frame_id) { |
| 22 *out_node = node; |
| 23 // Terminate iteration once the node has been found. |
| 24 return false; |
| 25 } |
| 26 return true; |
| 27 } |
| 28 |
| 29 } // namespace |
| 30 |
| 31 FrameTree::FrameTree() |
| 32 : root_(new FrameTreeNode(FrameTreeNode::kInvalidFrameId, std::string(), |
| 33 scoped_ptr<RenderFrameHostImpl>())) { |
| 34 } |
| 35 |
| 36 FrameTree::~FrameTree() { |
| 37 } |
| 38 |
| 39 FrameTreeNode* FrameTree::FindByID(int64 frame_id) { |
| 40 FrameTreeNode* node = NULL; |
| 41 ForEach(base::Bind(&FrameTreeNodeForId, frame_id, &node)); |
| 42 return node; |
| 43 } |
| 44 |
| 45 void FrameTree::ForEach( |
| 46 const base::Callback<bool(FrameTreeNode*)>& on_node) const { |
| 47 std::queue<FrameTreeNode*> queue; |
| 48 queue.push(root_.get()); |
| 49 |
| 50 while (!queue.empty()) { |
| 51 FrameTreeNode* node = queue.front(); |
| 52 queue.pop(); |
| 53 if (!on_node.Run(node)) |
| 54 break; |
| 55 |
| 56 for (size_t i = 0; i < node->child_count(); ++i) |
| 57 queue.push(node->child_at(i)); |
| 58 } |
| 59 } |
| 60 |
| 61 bool FrameTree::IsFirstNavigationAfterSwap() const { |
| 62 return root_->frame_id() == FrameTreeNode::kInvalidFrameId; |
| 63 } |
| 64 |
| 65 void FrameTree::OnFirstNavigationAfterSwap(int main_frame_id) { |
| 66 root_->set_frame_id(main_frame_id); |
| 67 } |
| 68 |
| 69 void FrameTree::AddFrame(int render_frame_host_id, int64 parent_frame_id, |
| 70 int64 frame_id, const std::string& frame_name) { |
| 71 FrameTreeNode* parent = FindByID(parent_frame_id); |
| 72 parent->AddChild(CreateNode(frame_id, frame_name, render_frame_host_id, |
| 73 parent->render_frame_host()->GetProcess())); |
| 74 } |
| 75 |
| 76 void FrameTree::RemoveFrame(int64 parent_frame_id, int64 frame_id) { |
| 77 FrameTreeNode* parent = FindByID(parent_frame_id); |
| 78 parent->RemoveChild(frame_id); |
| 79 if (!on_frame_removed_.is_null()) |
| 80 on_frame_removed_.Run(frame_id); |
| 81 } |
| 82 |
| 83 void FrameTree::SetFrameUrl(int64 frame_id, const GURL& url) { |
| 84 FrameTreeNode* node = FindByID(frame_id); |
| 85 if (node) |
| 86 node->set_current_url(url); |
| 87 } |
| 88 |
| 89 void FrameTree::SwapMainFrame(RenderFrameHostImpl* render_frame_host) { |
| 90 return root_->ResetForMainFrame(render_frame_host); |
| 91 } |
| 92 |
| 93 RenderFrameHostImpl* FrameTree::GetMainFrame() const { |
| 94 return root_->render_frame_host(); |
| 95 } |
| 96 |
| 97 void FrameTree::SetFrameRemoveListener( |
| 98 const base::Callback<void(int64)>& on_frame_removed) { |
| 99 on_frame_removed_ = on_frame_removed; |
| 100 } |
| 101 |
| 102 scoped_ptr<FrameTreeNode> FrameTree::CreateNode( |
| 103 int64 frame_id, const std::string& frame_name, int render_frame_host_id, |
| 104 RenderProcessHost* render_process_host) { |
| 105 scoped_ptr<RenderFrameHostImpl> render_frame_host( |
| 106 new RenderFrameHostImpl(render_process_host, this, render_frame_host_id, |
| 107 false)); |
| 108 |
| 109 return make_scoped_ptr(new FrameTreeNode(frame_id, frame_name, |
| 110 render_frame_host.Pass())); |
| 111 } |
| 112 |
| 113 } // namespace content |
OLD | NEW |