从boost dynamic_bitset中提取子集

时间:2010-09-12 06:27:09

标签: c++ boost

我需要从给定的boost dynamic_bitset中提取和解码位(idx,idx + 1,... idx + n_bits)。 我创建了以下解决方案:

boost::dynamic_bitset<> mybitset(...);
// build mask 2^{idx+n_bits} - 2^{idx}
const boost::dynamic_bitset<> mask(mybitset.size(), (1 << idx+n_bits) - (1 << idx));
// shift the masked result idx times and get long
unsigned long u = ((mybitset & mask) >> idx ).to_ulong();

它运行良好,但由于此代码对我的应用程序的性能至关重要,我很好奇是否有更好的方法来实现这一目标?

1 个答案:

答案 0 :(得分:8)

解决方案很简单:

#include <tuple>
    using std::get;
    using std::tuple;
    using std::make_tuple;
#include <boost/dynamic_bitset.hpp>
    using boost::dynamic_bitset;

template <typename Block, typename Allocator>
unsigned block_index(const boost::dynamic_bitset<Block, Allocator>& b, unsigned pos)
{ return pos / b.bits_per_block; }

namespace boost {
template <>
inline void
to_block_range(const dynamic_bitset<>& b, tuple<unsigned, unsigned,  unsigned long&> param)
{

    {
        unsigned beg = get<0>(param);
        unsigned len = get<1>(param);
        unsigned block1 = block_index(b, beg);
        unsigned block2 = block_index(b, beg + len -1);
        unsigned bit_index = beg % b.bits_per_block;
        unsigned long bitmask = (1 << len) - 1;
        get<2>(param) = ((b.m_bits[block1] >> bit_index) |
                               (b.m_bits[block2] << (b.bits_per_block - bit_index)  )) &
                                bitmask;
        return;
    }
}
}


unsigned long res;
to_block_range(bits, make_tuple(pos, len, std::ref(res)));

致电:

boost::dynamic_bitset<> bits;
unsigned long result;
to_block_range(bits, t_extract_range{begin_bit, length_bits, result});

dynamic_bitset中没有直接的原生支持。

要获得一系列位,您必须进入dynamic_bitset,访问底层存储,并自行提取位。

执行此操作的代码很简单,但数据(dynamic_bitset::m_bits)位于类的私有部分中。破解私人墙有三种方法:

  1. 假装您的编译器不符合要求 #define BOOST_DYNAMIC_BITSET_DONT_USE_FRIENDS。这会通过更改private
  2. public更改为BOOST_DYNAMIC_BITSET_PRIVATE
  3. 黑客dynamic_bitset.hpp标题以公开m_bits
  4. 第三种解决方案是解决当前的代码。
  5. (1)和(2)是脆弱的正面攻击,这将是一场维护噩梦。

    幸运的是(3),有frienddynamic_bitset的模板函数。我们可以通过接管(专门化)这个模板来替换我们自己的函数来进行自己的提取。

    template <typename Block, typename Allocator, typename BlockOutputIterator>
    inline void
    to_block_range(const dynamic_bitset<Block, Allocator>& b,
                   BlockOutputIterator result)
    {
        std::copy(b.m_bits.begin(), b.m_bits.end(), result);
    }
    

    规范模板函数将整个位集复制到迭代器BlockOutputIterator,这是我们想要的。

    我们将使用单个自定义类型代替boost::to_block_range来专门化BlockOutputIterator,它将包含所有3个i / o参数:即

    • begin_bit
    • length_of_range
    • 目的地。

    如果您使用必需类型调用to_block_range,它将调用您自己的函数而不是标准模板,但也可以完全访问内部函数。你基本上颠覆了c ++访问规范系统!

    N.B。示例代码不进行错误检查。没有尝试确保

    • 范围适合无符号长或
    • 该范围不超过bitset的范围或
    • bitset在内部使用unsigned longs。