1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
|
// Copyright 2014 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 "components/keyed_service/core/dependency_graph.h"
#include <algorithm>
#include <deque>
#include <iterator>
DependencyGraph::DependencyGraph() {}
DependencyGraph::~DependencyGraph() {}
void DependencyGraph::AddNode(DependencyNode* node) {
all_nodes_.push_back(node);
construction_order_.clear();
}
void DependencyGraph::RemoveNode(DependencyNode* node) {
all_nodes_.erase(std::remove(all_nodes_.begin(), all_nodes_.end(), node),
all_nodes_.end());
// Remove all dependency edges that contain this node.
EdgeMap::iterator it = edges_.begin();
while (it != edges_.end()) {
EdgeMap::iterator temp = it;
++it;
if (temp->first == node || temp->second == node)
edges_.erase(temp);
}
construction_order_.clear();
}
void DependencyGraph::AddEdge(DependencyNode* depended,
DependencyNode* dependee) {
edges_.insert(std::make_pair(depended, dependee));
construction_order_.clear();
}
bool DependencyGraph::GetConstructionOrder(
std::vector<DependencyNode*>* order) {
if (construction_order_.empty() && !BuildConstructionOrder())
return false;
*order = construction_order_;
return true;
}
bool DependencyGraph::GetDestructionOrder(std::vector<DependencyNode*>* order) {
if (construction_order_.empty() && !BuildConstructionOrder())
return false;
*order = construction_order_;
// Destroy nodes in reverse order.
std::reverse(order->begin(), order->end());
return true;
}
bool DependencyGraph::BuildConstructionOrder() {
// Step 1: Build a set of nodes with no incoming edges.
std::deque<DependencyNode*> queue;
std::copy(all_nodes_.begin(), all_nodes_.end(), std::back_inserter(queue));
std::deque<DependencyNode*>::iterator queue_end = queue.end();
for (EdgeMap::const_iterator it = edges_.begin(); it != edges_.end(); ++it) {
queue_end = std::remove(queue.begin(), queue_end, it->second);
}
queue.erase(queue_end, queue.end());
// Step 2: Do the Kahn topological sort.
std::vector<DependencyNode*> output;
EdgeMap edges(edges_);
while (!queue.empty()) {
DependencyNode* node = queue.front();
queue.pop_front();
output.push_back(node);
std::pair<EdgeMap::iterator, EdgeMap::iterator> range =
edges.equal_range(node);
EdgeMap::iterator it = range.first;
while (it != range.second) {
DependencyNode* dest = it->second;
EdgeMap::iterator temp = it;
it++;
edges.erase(temp);
bool has_incoming_edges = false;
for (EdgeMap::iterator jt = edges.begin(); jt != edges.end(); ++jt) {
if (jt->second == dest) {
has_incoming_edges = true;
break;
}
}
if (!has_incoming_edges)
queue.push_back(dest);
}
}
if (!edges.empty()) {
// Dependency graph has a cycle.
return false;
}
construction_order_ = output;
return true;
}
std::string DependencyGraph::DumpAsGraphviz(
const std::string& toplevel_name,
const base::Callback<std::string(DependencyNode*)>& node_name_callback)
const {
std::string result("digraph {\n");
// Make a copy of all nodes.
std::deque<DependencyNode*> nodes;
std::copy(all_nodes_.begin(), all_nodes_.end(), std::back_inserter(nodes));
// State all dependencies and remove |second| so we don't generate an
// implicit dependency on the top level node.
std::deque<DependencyNode*>::iterator nodes_end(nodes.end());
result.append(" /* Dependencies */\n");
for (EdgeMap::const_iterator it = edges_.begin(); it != edges_.end(); ++it) {
result.append(" ");
result.append(node_name_callback.Run(it->second));
result.append(" -> ");
result.append(node_name_callback.Run(it->first));
result.append(";\n");
nodes_end = std::remove(nodes.begin(), nodes_end, it->second);
}
nodes.erase(nodes_end, nodes.end());
// Every node that doesn't depend on anything else will implicitly depend on
// the top level node.
result.append("\n /* Toplevel attachments */\n");
for (std::deque<DependencyNode*>::const_iterator it = nodes.begin();
it != nodes.end();
++it) {
result.append(" ");
result.append(node_name_callback.Run(*it));
result.append(" -> ");
result.append(toplevel_name);
result.append(";\n");
}
result.append("\n /* Toplevel node */\n");
result.append(" ");
result.append(toplevel_name);
result.append(" [shape=box];\n");
result.append("}\n");
return result;
}
|