← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/intset.py

  • Committer: John Arbash Meinel
  • Date: 2011-01-12 16:45:49 UTC
  • mto: This revision was merged to the branch mainline in revision 5599.
  • Revision ID: john@arbash-meinel.com-20110112164549-6buxl1yr8vosoh1t
Add to What's New

Show diffs side-by-side

added added

removed removed

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