diff --git a/mercurial/dagop.py b/mercurial/dagop.py --- a/mercurial/dagop.py +++ b/mercurial/dagop.py @@ -715,3 +715,21 @@ for g in groups: for r in g[0]: yield r + +def headrevs(revs, parentsfn): + """Resolve the set of heads from a set of revisions. + + Receives an iterable of revision numbers and a callbable that receives a + revision number and returns an iterable of parent revision numbers, possibly + including nullrev. + + Returns a set of revision numbers that are DAG heads within the passed subset. + """ + headrevs = set() + + for rev in revs: + for prev in parentsfn(rev): + if prev != node.nullrev: + headrevs.discard(prev) + + return headrevs diff --git a/mercurial/dagutil.py b/mercurial/dagutil.py --- a/mercurial/dagutil.py +++ b/mercurial/dagutil.py @@ -10,6 +10,10 @@ from .node import nullrev +from . import ( + dagop, +) + class revlogdag(object): '''dag interface to a revlog''' @@ -31,21 +35,6 @@ return [prev2] return [] - def headsetofconnecteds(self, ixs): - if not ixs: - return set() - rlog = self._revlog - idx = rlog.index - headrevs = set(ixs) - for rev in ixs: - revdata = idx[rev] - for i in [5, 6]: - prev = revdata[i] - if prev != nullrev: - headrevs.discard(prev) - assert headrevs - return headrevs - def linearize(self, ixs): '''linearize and topologically sort a list of revisions @@ -56,7 +45,7 @@ parent, then adding the rev itself to the output list. ''' sorted = [] - visit = list(self.headsetofconnecteds(ixs)) + visit = list(dagop.headrevs(ixs, self.parents)) visit.sort(reverse=True) finished = set()