← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/graph.py

  • Committer: John Arbash Meinel
  • Date: 2005-09-15 21:35:53 UTC
  • mfrom: (907.1.57)
  • mto: (1393.2.1)
  • mto: This revision was merged to the branch mainline in revision 1396.
  • Revision ID: john@arbash-meinel.com-20050915213552-a6c83a5ef1e20897
(broken) Transport work is merged in. Tests do not pass yet.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/graph.py
1
 
# Copyright (C) 2005 Canonical Ltd
2
 
#
 
1
# (C) 2005 Canonical
 
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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
 
18
 
from bzrlib.tsort import topo_sort
19
 
 
20
 
 
21
17
def max_distance(node, ancestors, distances, root_descendants):
22
18
    """Calculate the max distance to an ancestor.  
23
19
    Return None if not all possible ancestors have known distances"""
34
30
            continue
35
31
        if ancestor not in distances:
36
32
            return None
37
 
        if best is None or distances[ancestor]+1 > best:
 
33
        if best is None or distances[ancestor] > best:
38
34
            best = distances[ancestor] + 1
39
35
    return best
40
36
 
71
67
        lines = new_lines
72
68
    return distances
73
69
 
74
 
def nodes_by_distance(distances):
75
 
    """Return a list of nodes sorted by distance"""
 
70
def farthest_nodes(graph, ancestors, start):
76
71
    def by_distance(n):
77
72
        return distances[n],n
78
73
 
 
74
    distances = node_distances(graph, ancestors, start)
79
75
    node_list = distances.keys()
80
76
    node_list.sort(key=by_distance, reverse=True)
81
77
    return node_list
82
78
 
83
 
def select_farthest(distances, common):
84
 
    """Return the farthest common node, or None if no node qualifies."""
85
 
    node_list = nodes_by_distance(distances)
86
 
    for node in node_list:
87
 
        if node in common:
88
 
            return node
89
 
 
90
79
def all_descendants(descendants, start):
91
 
    """Produce a set of all descendants of the start node.
92
 
    The input is a map of node->list of descendants for a graph encompassing
93
 
    start.
94
 
    """
95
80
    result = set()
96
81
    lines = set([start])
97
82
    while len(lines) > 0:
106
91
                new_lines.add(descendant)
107
92
        lines = new_lines
108
93
    return result
109
 
 
110
 
 
111
 
class Graph(object):
112
 
    """A graph object which can memoise and cache results for performance."""
113
 
 
114
 
    def __init__(self):
115
 
        super(Graph, self).__init__()
116
 
        self.roots = set([])
117
 
        self.ghosts = set([])
118
 
        self._graph_ancestors = {}
119
 
        self._graph_descendants = {}
120
 
 
121
 
    def add_ghost(self, node_id):
122
 
        """Add a ghost to the graph."""
123
 
        self.ghosts.add(node_id)
124
 
        self._ensure_descendant(node_id)
125
 
 
126
 
    def add_node(self, node_id, parent_ids):
127
 
        """Add node_id to the graph with parent_ids as its parents."""
128
 
        if parent_ids == []:
129
 
            self.roots.add(node_id)
130
 
        self._graph_ancestors[node_id] = list(parent_ids)
131
 
        self._ensure_descendant(node_id)
132
 
        for parent in parent_ids:
133
 
            self._ensure_descendant(parent)
134
 
            self._graph_descendants[parent][node_id] = 1
135
 
        
136
 
    def _ensure_descendant(self, node_id):
137
 
        """Ensure that a descendant lookup for node_id will work."""
138
 
        if not node_id in self._graph_descendants:
139
 
            self._graph_descendants[node_id] = {}
140
 
 
141
 
    def get_ancestors(self):
142
 
        """Return a dictionary of graph node:ancestor_list entries."""
143
 
        return dict(self._graph_ancestors.items())
144
 
 
145
 
    def get_ancestry(self, node_id, topo_sorted=True):
146
 
        """Return the inclusive ancestors of node_id in topological order."""
147
 
        # maybe optimise this ?
148
 
        result = {}
149
 
        pending = set([node_id])
150
 
        while len(pending):
151
 
            current = pending.pop()
152
 
            parents = self._graph_ancestors[current]
153
 
            parents = [parent for parent in parents if parent not in self.ghosts]
154
 
            result[current] = parents
155
 
            for parent in parents:
156
 
                if parent not in result and parent not in pending:
157
 
                    pending.add(parent)
158
 
        if not topo_sorted:
159
 
            return result.keys()
160
 
        return topo_sort(result.items())
161
 
 
162
 
    def get_descendants(self):
163
 
        """Return a dictionary of graph node:child_node:distance entries."""
164
 
        return dict(self._graph_descendants.items())