~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to graph.py

  • Committer: Aaron Bentley
  • Date: 2006-12-21 05:41:05 UTC
  • mfrom: (476.1.1 bzrtools)
  • Revision ID: aaron.bentley@utoronto.ca-20061221054105-r6a8njsyixw9998r
graph-ancestry shows branch nick

Show diffs side-by-side

added added

removed removed

Lines of Context:
201
201
            d_node.node_style.append('dotted')
202
202
 
203
203
        return d_node
204
 
       
205
 
    def get_relations(self, collapse=False, max_distance=None):
 
204
        
 
205
    def get_relations(self, collapse=False):
206
206
        dot_nodes = {}
207
207
        node_relations = []
208
208
        num = 0
211
211
                                                  self.ancestors, (self.base,))
212
212
        else:
213
213
            visible_ancestors = self.ancestors
214
 
        if max_distance is not None:
215
 
            min_distance = max(self.distances.values()) - max_distance
216
 
            visible_ancestors = dict((n, p) for n, p in visible_ancestors.iteritems() if
217
 
                    self.distances[n] >= min_distance)
218
214
        for node, parents in visible_ancestors.iteritems():
219
215
            if node not in dot_nodes:
220
216
                dot_nodes[node] = self.dot_node(node, num)
235
231
 
236
232
 
237
233
def write_ancestry_file(branch, filename, collapse=True, antialias=True,
238
 
                        merge_branch=None, ranking="forced", max_distance=None):
 
234
                        merge_branch=None, ranking="forced"):
239
235
    b = Branch.open_containing(branch)[0]
240
236
    if merge_branch is not None:
241
237
        m = Branch.open_containing(merge_branch)[0]
247
243
            m.lock_read()
248
244
        try:
249
245
            grapher = Grapher(b, m)
250
 
            relations = grapher.get_relations(collapse, max_distance)
 
246
            relations = grapher.get_relations(collapse)
251
247
        finally:
252
248
            if m is not None:
253
249
                m.unlock()