← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/intset.py

  • Committer: Robert Collins
  • Date: 2007-03-08 04:06:06 UTC
  • mfrom: (2323.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 2442.
  • Revision ID: robertc@robertcollins.net-20070308040606-84gsniv56huiyjt4
Merge bzr.dev.

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
 
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
 
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)