facebook/sapling
1
1
Fork 0
mirror of https://github.com/facebook/sapling.git synced 2024-10-08 07:49:11 +03:00
Code Issues Projects Releases Wiki Activity

dagop: factor out pfunc from revancestors() generator

Browse Source 
This generator will be reused for tracking descendants with depth limit.

  # ancestors(tip) using hg repo
  0) 0.065868
  1) 0.068976
This commit is contained in:
Yuya Nishihara 2017-06-24 23:22:45 +09:00
parent 1103437683
commit 568f49d319
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mercurial/dagop.py
View File

@ -38,6 +38,11 @@ def _genrevancestors(repo, revs, followfirst, startdepth, stopdepth):
raise error.ProgrammingError('negative stopdepth')
cl = repo.changelog
def pfunc(rev):
try:
return cl.parentrevs(rev)[:cut]
except error.WdirUnsupported:
return (pctx.rev() for pctx in repo[rev].parents()[:cut])
# load input revs lazily to heap so earlier revisions can be yielded
# without fully computing the input revs
@ -65,14 +70,9 @@ def _genrevancestors(repo, revs, followfirst, startdepth, stopdepth):
yield currev
pdepth = curdepth + 1
if foundnew and pdepth < stopdepth:
try:
for prev in cl.parentrevs(currev)[:cut]:
if prev != node.nullrev:
heapq.heappush(pendingheap, (-prev, pdepth))
except error.WdirUnsupported:
for pctx in repo[currev].parents()[:cut]:
if pctx.rev() != node.nullrev:
heapq.heappush(pendingheap, (-pctx.rev(), pdepth))
for prev in pfunc(currev):
if prev != node.nullrev:
heapq.heappush(pendingheap, (-prev, pdepth))
def revancestors(repo, revs, followfirst, startdepth=None, stopdepth=None):
"""Like revlog.ancestors(), but supports additional options, includes