branchmap: introduce branchheads() method
authorBrodie Rao <brodie@sf.io>
Mon, 16 Sep 2013 01:08:29 -0700
changeset 20188 3a3727829607
parent 20187 4d6d5ef88538
child 20189 1831993d0902
branchmap: introduce branchheads() method
mercurial/branchmap.py
--- a/mercurial/branchmap.py	Mon Sep 16 01:08:29 2013 -0700
+++ b/mercurial/branchmap.py	Mon Sep 16 01:08:29 2013 -0700
@@ -179,6 +179,12 @@
     def branchtip(self, branch):
         return self._branchtip(self[branch])[0]
 
+    def branchheads(self, branch, closed=False):
+        heads = self[branch]
+        if not closed:
+            heads = [h for h in heads if h not in self._closednodes]
+        return heads
+
     def copy(self):
         """return an deep copy of the branchcache object"""
         return branchcache(self, self.tipnode, self.tiprev, self.filteredhash,