comparison mercurial/revset.py @ 25131:adfe4d9680bf

revset: extract addset._iterordered to free function

It never uses self, so let's make it less dependent on variables.

25130:a94ef7f517e0

        return None

    def __repr__(self):
        return '<%s %r>' % (type(self).__name__, self._subset)

class addset(abstractsmartset):
    """Represent the addition of two sets

    Wrapper structure for lazily adding two structures without losing much
    performance on the __contains__ method

        if iter2 is None:
            # let's avoid side effect (not sure it matters)
            iter2 = iter(sorted(self._r2, reverse=not self._ascending))
        else:
            iter2 = iter2()
        return self._iterordered(self._ascending, iter1, iter2)

    def _trysetasclist(self):
        """populate the _asclist attribute if possible and necessary"""
        if self._genlist is not None and self._asclist is None:
            self._asclist = sorted(self._genlist)

            return self._asclist.__iter__
        iter1 = self._r1.fastasc
        iter2 = self._r2.fastasc
        if None in (iter1, iter2):
            return None
        return lambda: self._iterordered(True, iter1(), iter2())

    @property
    def fastdesc(self):
        self._trysetasclist()
        if self._asclist is not None:
            return self._asclist.__reversed__
        iter1 = self._r1.fastdesc
        iter2 = self._r2.fastdesc
        if None in (iter1, iter2):
            return None
        return lambda: self._iterordered(False, iter1(), iter2())

    def _iterordered(self, ascending, iter1, iter2):
        """produce an ordered iteration from two iterators with the same order

        The ascending is used to indicated the iteration direction.
        """
        choice = max
        if ascending:
            choice = min

        val1 = None
        val2 = None
        try:
            # Consume both iterators in an ordered way until one is
            # empty
            while True:
                if val1 is None:
                    val1 = iter1.next()
                if val2 is None:
                    val2 = iter2.next()
                next = choice(val1, val2)
                yield next
                if val1 == next:
                    val1 = None
                if val2 == next:
                    val2 = None
        except StopIteration:
            # Flush any remaining values and consume the other one
            it = iter2
            if val1 is not None:
                yield val1
                it = iter1
            elif val2 is not None:
                # might have been equality and both are empty
                yield val2
            for val in it:
                yield val

    def __contains__(self, x):
        return x in self._r1 or x in self._r2

    def sort(self, reverse=False):

25131:adfe4d9680bf

        return None

    def __repr__(self):
        return '<%s %r>' % (type(self).__name__, self._subset)

def _iterordered(ascending, iter1, iter2):
    """produce an ordered iteration from two iterators with the same order

    The ascending is used to indicated the iteration direction.
    """
    choice = max
    if ascending:
        choice = min

    val1 = None
    val2 = None
    try:
        # Consume both iterators in an ordered way until one is empty
        while True:
            if val1 is None:
                val1 = iter1.next()
            if val2 is None:
                val2 = iter2.next()
            next = choice(val1, val2)
            yield next
            if val1 == next:
                val1 = None
            if val2 == next:
                val2 = None
    except StopIteration:
        # Flush any remaining values and consume the other one
        it = iter2
        if val1 is not None:
            yield val1
            it = iter1
        elif val2 is not None:
            # might have been equality and both are empty
            yield val2
        for val in it:
            yield val

class addset(abstractsmartset):
    """Represent the addition of two sets

    Wrapper structure for lazily adding two structures without losing much
    performance on the __contains__ method

        if iter2 is None:
            # let's avoid side effect (not sure it matters)
            iter2 = iter(sorted(self._r2, reverse=not self._ascending))
        else:
            iter2 = iter2()
        return _iterordered(self._ascending, iter1, iter2)

    def _trysetasclist(self):
        """populate the _asclist attribute if possible and necessary"""
        if self._genlist is not None and self._asclist is None:
            self._asclist = sorted(self._genlist)

            return self._asclist.__iter__
        iter1 = self._r1.fastasc
        iter2 = self._r2.fastasc
        if None in (iter1, iter2):
            return None
        return lambda: _iterordered(True, iter1(), iter2())

    @property
    def fastdesc(self):
        self._trysetasclist()
        if self._asclist is not None:
            return self._asclist.__reversed__
        iter1 = self._r1.fastdesc
        iter2 = self._r2.fastdesc
        if None in (iter1, iter2):
            return None
        return lambda: _iterordered(False, iter1(), iter2())

    def __contains__(self, x):
        return x in self._r1 or x in self._r2

    def sort(self, reverse=False):