-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathrshrinklist.py
More file actions
36 lines (29 loc) · 1.03 KB
/
rshrinklist.py
File metadata and controls
36 lines (29 loc) · 1.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class AbstractShrinkList(object):
"""A mixin base class. You should subclass it and add a method
must_keep(). Behaves like a list with the method append(), and
you can read *for reading* the list of items by calling items().
The twist is that occasionally append() will throw away the
items for which must_keep() returns False. (It does so without
changing the order.)
See also rpython.rlib.rweaklist.
"""
_mixin_ = True
def __init__(self):
self._list = []
self._next_shrink = 16
def append(self, x):
self._do_shrink()
self._list.append(x)
def items(self):
return self._list
def _do_shrink(self):
if len(self._list) >= self._next_shrink:
rest = 0
for x in self._list:
if self.must_keep(x):
self._list[rest] = x
rest += 1
del self._list[rest:]
self._next_shrink = 16 + 2 * rest
def must_keep(self, x):
raise NotImplementedError