mercurial/pure/parsers.py
changeset 51969 22da1dc97281
parent 51863 f4733654f144
child 51979 609700e5d8df
--- a/mercurial/pure/parsers.py	Thu Sep 26 00:50:21 2024 +0200
+++ b/mercurial/pure/parsers.py	Wed Sep 25 17:18:40 2024 +0200
@@ -696,6 +696,22 @@
             p = p[revlog_constants.INDEX_HEADER.size :]
         return p
 
+    def headrevs(self, excluded_revs=None):
+        count = len(self)
+        if not count:
+            return [nullrev]
+        # we won't iter over filtered rev so nobody is a head at start
+        ishead = [0] * (count + 1)
+        revs = range(count)
+        if excluded_revs is not None:
+            revs = (r for r in revs if r not in excluded_revs)
+
+        for r in revs:
+            ishead[r] = 1  # I may be an head
+            e = self[r]
+            ishead[e[5]] = ishead[e[6]] = 0  # my parent are not
+        return [r for r, val in enumerate(ishead) if val]
+
 
 class IndexObject(BaseIndexObject):
     def __init__(self, data):