/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to branchview/linegraph.py

  • Committer: Jelmer Vernooij
  • Date: 2008-06-29 15:47:30 UTC
  • mto: This revision was merged to the branch mainline in revision 519.
  • Revision ID: jelmer@samba.org-20080629154730-xfsotoxwkiytf0ph
Pass graph object rather than full repository to linegraph.

Show diffs side-by-side

added added

removed removed

Lines of Context:
branchview/linegraph.py
9
9
__copyright__ = "Copyright © 2005 Canonical Ltd."
10
10
__author__    = "Scott James Remnant <scott@ubuntu.com>"
11
11
 
 
12
from bzrlib.revision import NULL_REVISION
12
13
from bzrlib.tsort import merge_sort
13
14
 
14
 
def linegraph(repository, start_revs, maxnum, broken_line_length = None,
 
15
def linegraph(graph, start_revs, maxnum, broken_line_length = None,
15
16
              graph_data = True, mainline_only = False):
16
17
    """Produce a directed graph of a bzr repository.
17
18
 
42
43
    curved, kinked, etc.) and to pick the actual colours for each index.
43
44
    """
44
45
    
45
 
    graph = repository.get_graph()
46
46
    graph_parents = {}
 
47
    ghosts = set()
47
48
    graph_children = {}
48
49
    for (revid, parent_revids) in graph.iter_ancestry(start_revs):
49
 
        graph_parents[revid] = parent_revids
50
 
        graph_children[revid] = []
51
 
 
 
50
        if parent_revids is None:
 
51
            ghosts.add(revid)
 
52
            continue
 
53
        if parent_revids == (NULL_REVISION,):
 
54
            graph_parents[revid] = ()
 
55
        else:
 
56
            graph_parents[revid] = parent_revids
 
57
        for parent in parent_revids:
 
58
            graph_children.setdefault(parent, []).append(revid)
 
59
        graph_children.setdefault(revid, [])
 
60
    for ghost in ghosts:
 
61
        for ghost_child in graph_children[ghost]:
 
62
            graph_parents[ghost_child] = [p for p in graph_parents[ghost_child]
 
63
                                          if p not in ghosts]
52
64
    graph_parents["top:"] = start_revs
53
65
 
54
66
    if len(graph_parents)>0:
90
102
        revid_index[revid] = rev_index
91
103
        
92
104
        parents = graph_parents[revid]
93
 
        for parent_revid in parents:
94
 
            graph_children[parent_revid].append(revid)
95
 
        
96
105
        linegraph.append([revid,
97
106
                          None,
98
107
                          [],