emo/thin-provisioning-tools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[persistent-data/data-structures] add method to visit values and damage in a bitset

Browse Source
This commit is contained in:
Joe Thornber 2013-10-10 12:55:10 +01:00
parent 347de67e2d
commit 9b5bf559ec
2 changed files with 74 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
persistent-data/data-structures/bitset.cc
View File

@ -2,6 +2,7 @@
#include "persistent-data/data-structures/bitset.h"
#include "persistent-data/math_utils.h"
using namespace boost;
using namespace persistent_data;
using namespace persistent_data::bitset_detail;
using namespace std;
@ -75,7 +76,52 @@ namespace persistent_data {
void flush() {
}
void walk_bitset(bitset_visitor &v) const {
bit_visitor vv(v);
damage_visitor dv(v);
array_.visit_values(vv, dv);
}
private:
class bit_visitor {
public:
bit_visitor(bitset_visitor &v)
: v_(v) {
}
void visit(uint32_t word_index, uint64_t word) {
uint32_t bit_index = word_index * 64;
for (unsigned bit = 0; bit < 64; bit++, bit_index++)
v_.visit(bit_index, !!(word & (1 << bit)));
}
private:
bitset_visitor &v_;
};
class damage_visitor {
public:
damage_visitor(bitset_visitor &v)
: v_(v) {
}
void visit(array_detail::damage const &d) {
run<uint32_t> bits(lifted_mult64(d.lost_keys_.begin_),
lifted_mult64(d.lost_keys_.end_));
v_.visit(bits);
}
private:
optional<uint32_t> lifted_mult64(optional<uint32_t> const &m) {
if (!m)
return m;
return optional<uint32_t>(*m * 64);
}
bitset_visitor &v_;
};
void pad_last_block(bool default_value) {
// Set defaults in the final word
if (bit(nr_bits_)) {
@ -198,5 +244,11 @@ bitset::flush()
impl_->flush();
}
void
bitset::walk_bitset(bitset_visitor &v) const
{
impl_->walk_bitset(v);
}
//----------------------------------------------------------------

22
persistent-data/data-structures/bitset.h
View File

@ -19,11 +19,31 @@
#ifndef BITSET_H
#define BITSET_H
#include "persistent-data/run.h"
//----------------------------------------------------------------
namespace persistent_data {
namespace bitset_detail {
class bitset_impl;
class missing_bits {
public:
missing_bits(base::run<uint32_t> const &keys)
: keys_(keys) {
}
base::run<uint32_t> keys_;
};
class bitset_visitor {
public:
typedef boost::shared_ptr<bitset_visitor> ptr;
virtual ~bitset_visitor();
virtual void visit(uint32_t index, bool value);
virtual void visit(missing_bits const &d);
};
}
class bitset {
@ -42,6 +62,8 @@ namespace persistent_data {
void set(unsigned n, bool value);
void flush();
void walk_bitset(bitset_detail::bitset_visitor &v) const;
private:
boost::shared_ptr<bitset_detail::bitset_impl> impl_;
};