comparison mercurial/revlog.py @ 51262:f20c4b307a5a

rust-index: add fast-path for getting a list of all heads as nodes

This avoids a lot of back-and-forth between Python and Rust. We forgo adding
a fast-path in the `filteredchangelog` case yet. If it shows up in profiling,
we might add the variant with a filter.

51261:9088c6d65ef6

            ishead[r] = 1  # I may be an head
            e = index[r]
            ishead[e[5]] = ishead[e[6]] = 0  # my parent are not
        return [r for r, val in enumerate(ishead) if val]

    def heads(self, start=None, stop=None):
        """return the list of all nodes that have no children

        if start is specified, only heads that are descendants of
        start will be returned

        as if they had no children
        """
        if start is None and stop is None:
            if not len(self):
                return [self.nullid]
            return [self.node(r) for r in self.headrevs()]

        if start is None:
            start = nullrev
        else:
            start = self.rev(start)

51262:f20c4b307a5a

            ishead[r] = 1  # I may be an head
            e = index[r]
            ishead[e[5]] = ishead[e[6]] = 0  # my parent are not
        return [r for r, val in enumerate(ishead) if val]

    def _head_node_ids(self):
        try:
            return self.index.head_node_ids()
        except AttributeError:
            return [self.node(r) for r in self.headrevs()]

    def heads(self, start=None, stop=None):
        """return the list of all nodes that have no children

        if start is specified, only heads that are descendants of
        start will be returned

        as if they had no children
        """
        if start is None and stop is None:
            if not len(self):
                return [self.nullid]
            return self._head_node_ids()
        if start is None:
            start = nullrev
        else:
            start = self.rev(start)