← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/graph.py

  • Committer: Robey Pointer
  • Date: 2006-09-08 18:46:29 UTC
  • mto: This revision was merged to the branch mainline in revision 1996.
  • Revision ID: robey@lag.net-20060908184629-e3fc4c61ca21508c
pychecker is on crack; go back to using 'is None'.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/graph.py
1
1
# (C) 2005 Canonical
2
 
 
 
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
 
17
21
def max_distance(node, ancestors, distances, root_descendants):
18
22
    """Calculate the max distance to an ancestor.  
19
23
    Return None if not all possible ancestors have known distances"""
102
106
                new_lines.add(descendant)
103
107
        lines = new_lines
104
108
    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):
 
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
        return topo_sort(result.items())
 
159
 
 
160
    def get_descendants(self):
 
161
        """Return a dictionary of graph node:child_node:distance entries."""
 
162
        return dict(self._graph_descendants.items())