diff -r 11ff34956ee7 -r d8e3a98018cb hgext/graphlog.py --- a/hgext/graphlog.py Fri Jun 19 13:14:01 2009 +0200 +++ b/hgext/graphlog.py Fri Jun 19 13:14:45 2009 +0200 @@ -12,7 +12,7 @@ revision graph is also shown. ''' -import os +import os, sys from mercurial.cmdutil import revrange, show_changeset from mercurial.commands import templateopts from mercurial.i18n import _ @@ -252,17 +252,11 @@ ascii(ui, grapher(graphdag)) def graphrevs(repo, nodes, opts): - include = set(nodes) limit = cmdutil.loglimit(opts) - count = 0 - for node in reversed(nodes): - if count >= limit: - break - ctx = repo[node] - parents = [p.rev() for p in ctx.parents() if p.node() in include] - parents.sort() - yield (ctx, parents) - count += 1 + nodes.reverse() + if limit < sys.maxint: + nodes = nodes[:limit] + return graphmod.nodes(repo, nodes) def graphabledag(ui, repo, revdag, opts): showparents = [ctx.node() for ctx in repo[None].parents()]