mirror of
https://github.com/genodelabs/genode.git
synced 2024-12-24 15:56:41 +00:00
157 lines
3.5 KiB
C++
157 lines
3.5 KiB
C++
/*
|
|
* \brief Allocator using bitmaps
|
|
* \author Alexander Boettcher
|
|
* \author Stefan Kalkowski
|
|
* \date 2012-06-14
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2012-2017 Genode Labs GmbH
|
|
*
|
|
* This file is part of the Genode OS framework, which is distributed
|
|
* under the terms of the GNU Affero General Public License version 3.
|
|
*/
|
|
|
|
#ifndef _INCLUDE__UTIL__BIT_ARRAY_H_
|
|
#define _INCLUDE__UTIL__BIT_ARRAY_H_
|
|
|
|
#include <util/string.h>
|
|
#include <base/exception.h>
|
|
#include <base/stdint.h>
|
|
|
|
namespace Genode {
|
|
|
|
class Bit_array_base;
|
|
template <unsigned> class Bit_array;
|
|
}
|
|
|
|
|
|
class Genode::Bit_array_base
|
|
{
|
|
public:
|
|
|
|
class Invalid_bit_count : public Exception {};
|
|
class Invalid_index_access : public Exception {};
|
|
class Invalid_clear : public Exception {};
|
|
class Invalid_set : public Exception {};
|
|
|
|
private:
|
|
|
|
enum {
|
|
BITS_PER_BYTE = 8UL,
|
|
BITS_PER_WORD = sizeof(addr_t) * BITS_PER_BYTE
|
|
};
|
|
|
|
unsigned _bit_cnt;
|
|
unsigned _word_cnt;
|
|
addr_t *_words;
|
|
|
|
addr_t _word(addr_t index) const {
|
|
return index / BITS_PER_WORD; }
|
|
|
|
void _check_range(addr_t const index,
|
|
addr_t const width) const
|
|
{
|
|
if ((index >= _word_cnt * BITS_PER_WORD) ||
|
|
width > _word_cnt * BITS_PER_WORD ||
|
|
_word_cnt * BITS_PER_WORD - width < index)
|
|
throw Invalid_index_access();
|
|
}
|
|
|
|
addr_t _mask(addr_t const index, addr_t const width,
|
|
addr_t &rest) const
|
|
{
|
|
addr_t const shift = index - _word(index) * BITS_PER_WORD;
|
|
|
|
rest = width + shift > BITS_PER_WORD ?
|
|
width + shift - BITS_PER_WORD : 0;
|
|
|
|
return (width >= BITS_PER_WORD) ? ~0UL << shift
|
|
: ((1UL << width) - 1) << shift;
|
|
}
|
|
|
|
void _set(addr_t index, addr_t width, bool free)
|
|
{
|
|
_check_range(index, width);
|
|
|
|
addr_t rest, word, mask;
|
|
do {
|
|
word = _word(index);
|
|
mask = _mask(index, width, rest);
|
|
|
|
if (free) {
|
|
if ((_words[word] & mask) != mask)
|
|
throw Invalid_clear();
|
|
_words[word] &= ~mask;
|
|
} else {
|
|
if (_words[word] & mask)
|
|
throw Invalid_set();
|
|
_words[word] |= mask;
|
|
}
|
|
|
|
index = (_word(index) + 1) * BITS_PER_WORD;
|
|
width = rest;
|
|
} while (rest);
|
|
}
|
|
|
|
public:
|
|
|
|
Bit_array_base(unsigned bits, addr_t *addr, bool clear)
|
|
: _bit_cnt(bits),
|
|
_word_cnt(_bit_cnt / BITS_PER_WORD),
|
|
_words(addr)
|
|
{
|
|
if (!bits || bits % BITS_PER_WORD) throw Invalid_bit_count();
|
|
|
|
if (clear) memset(_words, 0, sizeof(addr_t)*_word_cnt);
|
|
}
|
|
|
|
/**
|
|
* Return true if at least one bit is set between
|
|
* index until index + width - 1
|
|
*/
|
|
bool get(addr_t index, addr_t width) const
|
|
{
|
|
_check_range(index, width);
|
|
|
|
bool used = false;
|
|
addr_t rest, mask;
|
|
do {
|
|
mask = _mask(index, width, rest);
|
|
used = _words[_word(index)] & mask;
|
|
index = (_word(index) + 1) * BITS_PER_WORD;
|
|
width = rest;
|
|
} while (!used && rest);
|
|
|
|
return used;
|
|
}
|
|
|
|
void set(addr_t const index, addr_t const width) {
|
|
_set(index, width, false); }
|
|
|
|
void clear(addr_t const index, addr_t const width) {
|
|
_set(index, width, true); }
|
|
};
|
|
|
|
|
|
template <unsigned BITS>
|
|
class Genode::Bit_array : public Bit_array_base
|
|
{
|
|
private:
|
|
|
|
static constexpr size_t _WORDS = BITS / BITS_PER_WORD;
|
|
|
|
static_assert(BITS % BITS_PER_WORD == 0,
|
|
"Count of bits need to be word aligned!");
|
|
|
|
addr_t _array[_WORDS];
|
|
|
|
public:
|
|
|
|
Bit_array() : Bit_array_base(BITS, _array, true) { }
|
|
Bit_array(const Bit_array & o) : Bit_array_base(BITS, _array, false) {
|
|
memcpy(&_array, &o._array, sizeof(_array)); }
|
|
};
|
|
|
|
#endif /* _INCLUDE__UTIL__BIT_ARRAY_H_ */
|