← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/intset.py

  • Committer: Martin Pool
  • Date: 2005-07-22 22:37:53 UTC
  • Revision ID: mbp@sourcefrog.net-20050722223753-7dced4e32d3ce21d
- add the start of a test for inventory file-id matching

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/intset.py
 
1
#! /usr/bin/python
 
2
 
1
3
# Copyright (C) 2005 Canonical Ltd
2
 
#
 
4
 
3
5
# This program is free software; you can redistribute it and/or modify
4
6
# it under the terms of the GNU General Public License as published by
5
7
# the Free Software Foundation; either version 2 of the License, or
6
8
# (at your option) any later version.
7
 
#
 
9
 
8
10
# This program is distributed in the hope that it will be useful,
9
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
13
# GNU General Public License for more details.
12
 
#
 
14
 
13
15
# You should have received a copy of the GNU General Public License
14
16
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
18
 
17
19
# Author: Martin Pool <mbp@canonical.com>
18
20
 
60
62
    True
61
63
    >>> list(a)
62
64
    [10]
63
 
 
 
65
    
64
66
    """
65
67
    __slots__ = ['_val']
66
68
 
71
73
            If specified, an initial collection of values.
72
74
        """
73
75
        self._val = bitmask
74
 
        if values is not None:
 
76
        if values != None:
75
77
            self.update(values)
76
78
 
77
79
 
80
82
 
81
83
        >>> bool(IntSet())
82
84
        False
83
 
 
 
85
        
84
86
        >>> bool(IntSet([0]))
85
87
        True
86
88
        """
129
131
        """
130
132
        if not isinstance(other, IntSet):
131
133
            raise NotImplementedError(type(other))
132
 
        return IntSet(bitmask=(self._val | other._val))
 
134
        return IntSet(bitmask=(self._val | other._val))        
133
135
 
134
136
 
135
137
    def __eq__(self, other):
149
151
 
150
152
 
151
153
    def __contains__(self, i):
 
154
        assert i >= 0
152
155
        return self._val & (1L << i)
153
156
 
154
157
 
169
172
            v = v >> 1
170
173
            o = o + 1
171
174
 
172
 
 
 
175
        
173
176
    def update(self, to_add):
174
177
        """Add all the values from the sequence or intset to_add"""
175
178
        if isinstance(to_add, IntSet):
176
179
            self._val |= to_add._val
177
180
        else:
178
181
            for i in to_add:
 
182
                assert i >= 0
179
183
                self._val |= (1L << i)
180
184
 
181
185
 
182
186
    def add(self, to_add):
 
187
        assert 0 <= to_add
183
188
        self._val |= (1L << to_add)
184
189
 
185
190
 
200
205
        >>> not a
201
206
        True
202
207
        """
 
208
        assert 0 <= to_remove
203
209
        m = 1L << to_remove
204
210
        if not self._val & m:
205
211
            raise KeyError(to_remove)
206
212
        self._val ^= m
207
 
 
208
 
    def set_remove(self, to_remove):
209
 
        """Remove all values that exist in to_remove.
210
 
 
211
 
        >>> a = IntSet(range(10))
212
 
        >>> b = IntSet([2,3,4,7,12])
213
 
        >>> a.set_remove(b)
214
 
        >>> list(a)
215
 
        [0, 1, 5, 6, 8, 9]
216
 
        >>> a.set_remove([1,2,5])
217
 
        >>> list(a)
218
 
        [0, 6, 8, 9]
219
 
        """
220
 
        if not isinstance(to_remove, IntSet):
221
 
            self.set_remove(IntSet(to_remove))
222
 
            return
223
 
        intersect = self._val & to_remove._val
224
 
        self._val ^= intersect
225
 
 
 
213
        
 
214
        
 
215
            
 
216
    
 
217
 
 
218
if __name__ == '__main__':
 
219
    import doctest
 
220
    doctest.testmod()
 
221