← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/intset.py

Merge bzr.dev (and fix NEWS)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/intset.py
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
from __future__ import absolute_import
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
16
 
19
17
# Author: Martin Pool <mbp@canonical.com>
20
18
 
62
60
    True
63
61
    >>> list(a)
64
62
    [10]
65
 
 
 
63
    
66
64
    """
67
65
    __slots__ = ['_val']
68
66
 
73
71
            If specified, an initial collection of values.
74
72
        """
75
73
        self._val = bitmask
76
 
        if values is not None:
 
74
        if values != None:
77
75
            self.update(values)
78
76
 
79
77
 
82
80
 
83
81
        >>> bool(IntSet())
84
82
        False
85
 
 
 
83
        
86
84
        >>> bool(IntSet([0]))
87
85
        True
88
86
        """
131
129
        """
132
130
        if not isinstance(other, IntSet):
133
131
            raise NotImplementedError(type(other))
134
 
        return IntSet(bitmask=(self._val | other._val))
 
132
        return IntSet(bitmask=(self._val | other._val))        
135
133
 
136
134
 
137
135
    def __eq__(self, other):
151
149
 
152
150
 
153
151
    def __contains__(self, i):
 
152
        assert i >= 0
154
153
        return self._val & (1L << i)
155
154
 
156
155
 
171
170
            v = v >> 1
172
171
            o = o + 1
173
172
 
174
 
 
 
173
        
175
174
    def update(self, to_add):
176
175
        """Add all the values from the sequence or intset to_add"""
177
176
        if isinstance(to_add, IntSet):
178
177
            self._val |= to_add._val
179
178
        else:
180
179
            for i in to_add:
 
180
                assert i >= 0
181
181
                self._val |= (1L << i)
182
182
 
183
183
 
184
184
    def add(self, to_add):
 
185
        assert 0 <= to_add
185
186
        self._val |= (1L << to_add)
186
187
 
187
188
 
202
203
        >>> not a
203
204
        True
204
205
        """
 
206
        assert 0 <= to_remove
205
207
        m = 1L << to_remove
206
208
        if not self._val & m:
207
209
            raise KeyError(to_remove)