← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/graph.py

Unfuck performance.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/graph.py
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"""
138
142
        """Return a dictionary of graph node:ancestor_list entries."""
139
143
        return dict(self._graph_ancestors.items())
140
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
 
141
160
    def get_descendants(self):
142
161
        """Return a dictionary of graph node:child_node:distance entries."""
143
162
        return dict(self._graph_descendants.items())