comparison mercurial/revset.py @ 20703:e07b1fd30805

revset: added sort methods to generatorsets

Method needed to propagate sort calls amongst lazy structures.
The generated list (stored in the object) is sorted.

If the generated list did not contain all elements from the generator, we
take care of that before sorting the list.

20696:77ab0abb08a0

        self._gen = gen
        self._iter = iter(gen)
        self._cache = {}
        self._genlist = baseset([])
        self._iterated = False

    def __contains__(self, x):
        if x in self._cache:
            return self._cache[x]

        for l in self:
            if l == x:
                return True

        self._cache[x] = False
        return False

    def __iter__(self):
        if self._iterated:

        for item in self._gen:
            self._cache[item] = True
            self._genlist.append(item)
            yield item

    def set(self):
        return self

class ascgeneratorset(generatorset):
    """ Same structure as generatorset but stops iterating after it goes past
    the value when asked for membership and the element is not contained
    """

20703:e07b1fd30805

        self._gen = gen
        self._iter = iter(gen)
        self._cache = {}
        self._genlist = baseset([])
        self._iterated = False
        self._finished = False

    def __contains__(self, x):
        if x in self._cache:
            return self._cache[x]

        for l in self:
            if l == x:
                return True

        self._finished = True
        self._cache[x] = False
        return False

    def __iter__(self):
        if self._iterated:

        for item in self._gen:
            self._cache[item] = True
            self._genlist.append(item)
            yield item

        self._finished = True

    def set(self):
        return self

    def sort(self, reverse=False):
        # Basic implementation to be changed in future patches
        if not self._finished:
            for i in self:
                continue
        self._genlist.sort(reverse=reverse)

class ascgeneratorset(generatorset):
    """ Same structure as generatorset but stops iterating after it goes past
    the value when asked for membership and the element is not contained
    """