1
// libTorrent - BitTorrent library
2
// Copyright (C) 2005, Jari Sundell
4
// This program is free software; you can redistribute it and/or modify
5
// it under the terms of the GNU General Public License as published by
6
// the Free Software Foundation; either version 2 of the License, or
7
// (at your option) any later version.
9
// This program is distributed in the hope that it will be useful,
10
// but WITHOUT ANY WARRANTY; without even the implied warranty of
11
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
// GNU General Public License for more details.
14
// You should have received a copy of the GNU General Public License
15
// along with this program; if not, write to the Free Software
16
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
// In addition, as a special exception, the copyright holders give
19
// permission to link the code of portions of this program with the
20
// OpenSSL library under certain conditions as described in each
21
// individual source file, and distribute linked combinations
24
// You must obey the GNU General Public License in all respects for
25
// all of the code used other than OpenSSL. If you modify file(s)
26
// with this exception, you may extend this exception to your version
27
// of the file(s), but you are not obligated to do so. If you do not
28
// wish to do so, delete this exception statement from your version.
29
// If you delete this exception statement from all source files in the
30
// program, then also delete it here.
32
// Contact: Jari Sundell <jaris@ifi.uio.no>
35
// 3185 Skoppum, NORWAY
37
#ifndef LIBTORRENT_UTILS_BITFIELD_H
38
#define LIBTORRENT_UTILS_BITFIELD_H
47
typedef uint32_t size_t;
48
typedef uint8_t data_t;
49
typedef const uint8_t c_data_t;
50
typedef data_t* iterator;
51
typedef const data_t* const_iterator;
59
BitField(const BitField& bf);
61
~BitField() { delete [] m_start; m_start = NULL; }
63
void clear() { if (m_start) std::memset(m_start, 0, size_bytes()); }
65
size_t size_bits() const { return m_size; }
66
size_t size_bytes() const { return m_end - m_start; }
68
size_t position(const_iterator itr) const { return (itr - m_start) * 8; }
73
// Clear the last byte's padding.
75
if (m_start && m_size % 8)
76
*(m_end - 1) &= ~(data_t)0 << 8 - m_size % 8;
79
void set(size_t i, bool s) {
81
m_start[i / 8] |= 1 << 7 - i % 8;
83
m_start[i / 8] &= ~(1 << 7 - i % 8);
86
void set(size_t begin, size_t end, bool s);
88
bool get(size_t i) const { return m_start[i / 8] & (1 << 7 - i % 8); }
90
// Mark all in this not in b2.
91
BitField& not_in(const BitField& bf);
93
bool all_zero() const;
96
data_t* begin() { return m_start; }
97
c_data_t* begin() const { return m_start; }
98
data_t* end() { return m_end; }
99
c_data_t* end() const { return m_end; }
101
bool operator [] (size_t i) const { return get(i); }
103
BitField& operator = (const BitField& bf);
114
#endif // LIBTORRENT_BITFIELD_H