← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/deprecated_graph.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-08-30 22:49:20 UTC
  • mfrom: (5397.1.6 jam-integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100830224920-w9zw1vhsd5oiyljv
(vila, jam) Get PQM running correctly again (bug #626667),
        skip test_bzr_connect_to_bzr_ssh (bug #626876)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/deprecated_graph.py
1
 
# (C) 2005 Canonical
2
 
 
 
1
# Copyright (C) 2005, 2008 Canonical Ltd
 
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
 
 
7
#
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
 
 
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
16
17
 
17
18
def max_distance(node, ancestors, distances, root_descendants):
18
 
    """Calculate the max distance to an ancestor.  
 
19
    """Calculate the max distance to an ancestor.
19
20
    Return None if not all possible ancestors have known distances"""
20
21
    best = None
21
22
    if node in distances:
30
31
            continue
31
32
        if ancestor not in distances:
32
33
            return None
33
 
        if best is None or distances[ancestor] > best:
 
34
        if best is None or distances[ancestor]+1 > best:
34
35
            best = distances[ancestor] + 1
35
36
    return best
36
37
 
37
 
    
 
38
 
38
39
def node_distances(graph, ancestors, start, root_descendants=None):
39
40
    """Produce a list of nodes, sorted by distance from a start node.
40
41
    This is an algorithm devised by Aaron Bentley, because applying Dijkstra
56
57
        new_lines = set()
57
58
        for line in lines:
58
59
            line_descendants = graph[line]
59
 
            assert line not in line_descendants, "%s refers to itself" % line
60
60
            for descendant in line_descendants:
61
61
                distance = max_distance(descendant, ancestors, distances,
62
62
                                        root_descendants)
67
67
        lines = new_lines
68
68
    return distances
69
69
 
70
 
def farthest_nodes(graph, ancestors, start):
 
70
def nodes_by_distance(distances):
 
71
    """Return a list of nodes sorted by distance"""
71
72
    def by_distance(n):
72
73
        return distances[n],n
73
74
 
74
 
    distances = node_distances(graph, ancestors, start)
75
75
    node_list = distances.keys()
76
76
    node_list.sort(key=by_distance, reverse=True)
77
77
    return node_list
78
78
 
 
79
def select_farthest(distances, common):
 
80
    """Return the farthest common node, or None if no node qualifies."""
 
81
    node_list = nodes_by_distance(distances)
 
82
    for node in node_list:
 
83
        if node in common:
 
84
            return node
 
85
 
79
86
def all_descendants(descendants, start):
 
87
    """Produce a set of all descendants of the start node.
 
88
    The input is a map of node->list of descendants for a graph encompassing
 
89
    start.
 
90
    """
80
91
    result = set()
81
92
    lines = set([start])
82
93
    while len(lines) > 0:
91
102
                new_lines.add(descendant)
92
103
        lines = new_lines
93
104
    return result
 
105
 
 
106
 
 
107
class Graph(object):
 
108
    """A graph object which can memoise and cache results for performance."""
 
109
 
 
110
    def __init__(self):
 
111
        super(Graph, self).__init__()
 
112
        self.roots = set([])
 
113
        self.ghosts = set([])
 
114
        self._graph_ancestors = {}
 
115
        self._graph_descendants = {}
 
116
 
 
117
    def add_ghost(self, node_id):
 
118
        """Add a ghost to the graph."""
 
119
        self.ghosts.add(node_id)
 
120
        self._ensure_descendant(node_id)
 
121
 
 
122
    def add_node(self, node_id, parent_ids):
 
123
        """Add node_id to the graph with parent_ids as its parents."""
 
124
        if len(parent_ids) == 0:
 
125
            self.roots.add(node_id)
 
126
        self._graph_ancestors[node_id] = list(parent_ids)
 
127
        self._ensure_descendant(node_id)
 
128
        for parent in parent_ids:
 
129
            self._ensure_descendant(parent)
 
130
            self._graph_descendants[parent][node_id] = 1
 
131
 
 
132
    def _ensure_descendant(self, node_id):
 
133
        """Ensure that a descendant lookup for node_id will work."""
 
134
        if not node_id in self._graph_descendants:
 
135
            self._graph_descendants[node_id] = {}
 
136
 
 
137
    def get_ancestors(self):
 
138
        """Return a dictionary of graph node:ancestor_list entries."""
 
139
        return dict(self._graph_ancestors.items())
 
140
 
 
141
    def get_ancestry(self, node_id, topo_sorted=True):
 
142
        """Return the inclusive ancestors of node_id in topological order."""
 
143
        # maybe optimise this ?
 
144
        from bzrlib.tsort import topo_sort
 
145
        result = {}
 
146
        pending = set([node_id])
 
147
        while len(pending):
 
148
            current = pending.pop()
 
149
            parents = self._graph_ancestors[current]
 
150
            parents = [parent for parent in parents if parent not in self.ghosts]
 
151
            result[current] = parents
 
152
            for parent in parents:
 
153
                if parent not in result and parent not in pending:
 
154
                    pending.add(parent)
 
155
        if not topo_sorted:
 
156
            return result.keys()
 
157
        return topo_sort(result.items())
 
158
 
 
159
    def get_descendants(self):
 
160
        """Return a dictionary of graph node:child_node:distance entries."""
 
161
        return dict(self._graph_descendants.items())