/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/intset.py

  • Committer: Robert Collins
  • Date: 2005-10-16 22:31:25 UTC
  • mto: This revision was merged to the branch mainline in revision 1458.
  • Revision ID: robertc@lifelesslap.robertcollins.net-20051016223125-26d4401cb94b7b82
Branch.relpath has been moved to WorkingTree.relpath.

WorkingTree no no longer takes an inventory, rather it takes an optional branch
parameter, and if None is given will open the branch at basedir implicitly.

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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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 is not None:
 
74
        if values != 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
152
153
        return self._val & (1L << i)
153
154
 
154
155
 
169
170
            v = v >> 1
170
171
            o = o + 1
171
172
 
172
 
 
 
173
        
173
174
    def update(self, to_add):
174
175
        """Add all the values from the sequence or intset to_add"""
175
176
        if isinstance(to_add, IntSet):
176
177
            self._val |= to_add._val
177
178
        else:
178
179
            for i in to_add:
 
180
                assert i >= 0
179
181
                self._val |= (1L << i)
180
182
 
181
183
 
182
184
    def add(self, to_add):
 
185
        assert 0 <= to_add
183
186
        self._val |= (1L << to_add)
184
187
 
185
188
 
200
203
        >>> not a
201
204
        True
202
205
        """
 
206
        assert 0 <= to_remove
203
207
        m = 1L << to_remove
204
208
        if not self._val & m:
205
209
            raise KeyError(to_remove)
206
210
        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