使用带有bitset的initializer_list

时间:2016-06-21 13:01:41

标签: c++ boolean bit-shift initializer-list bitset

有没有办法使用initializer_list来构建bitset

例如,我想这样做:

const auto msb = false;
const auto b = true;
const auto lsb = false;
const bitset<3> foo = {msb, b, lsb};

但是当我尝试这个时,我得到了:

  

错误:无法从'&lt;大括号括起来的初始列表&gt;'中转换{msb, b, lsb}const std::bitset<3u>

我是否必须使用轮班构建unsigned long来初始化foo,还是有办法做到这一点我不知道?

2 个答案:

答案 0 :(得分:3)

没有构造函数可以直接从初始化列表构造一个bitset。你需要一个功能:

#include <bitset>
#include <initializer_list>
#include <iostream>

auto to_bitset(std::initializer_list<bool> il)
{
    using ul = unsigned long;
    auto bits = ul(0);
    if (il.size())
    {
        auto mask = ul(1) << (il.size() - 1);

        for (auto b : il) {
            if (b) {
                bits |= mask;
            }
            mask >>= 1;
        }
    }
    return std::bitset<3> { bits };

}

int main()
{
    auto bs = to_bitset({true, false, true});

    std::cout << bs << std::endl;
}

预期结果:

101

如评论中所述,也可以使用可变版本。

#include <bitset>
#include <iostream>
#include <utility>

namespace detail {
    template<std::size_t...Is, class Tuple>
    auto to_bitset(std::index_sequence<Is...>, Tuple&& tuple)
    {
        static constexpr auto size = sizeof...(Is);
        using expand = int[];
        unsigned long bits = 0;
        void(expand {
            0,
            ((bits |= std::get<Is>(tuple) ? 1ul << (size - Is - 1) : 0),0)...
        });
        return std::bitset<size>(bits);
    }
}

template<class...Bools>
auto to_bitset(Bools&&...bools)
{
    return detail::to_bitset(std::make_index_sequence<sizeof...(Bools)>(),
                             std::make_tuple(bool(bools)...));
}

int main()
{
    auto bs = to_bitset(true, false, true);

    std::cout << bs << std::endl;
}

答案 1 :(得分:1)

使用bitset的字符串表示的简短示例。在某些情况下,这种实施可能不是最佳的。

template <typename ... args>
auto make_bitset(args... bools) -> std::bitset<sizeof...(bools)>
{
    char data[] = {bools ? '1':'0'...};
    return std::bitset<sizeof...(bools)>(data, sizeof...(bools));
}

int main()
{
  std::cout << make_bitset(1,1,0,0,1,0, 1) << "\n";
  std::cout << make_bitset(1,1,1) << "\n";
  std::cout << make_bitset(0,0,0,0) << "\n";
}

另一种使用“几乎递归”调用的实现:

template <size_t N, int pos>
void init_bitset(std::bitset<N>&){}

template <size_t N, int pos = 0, typename ...Bits>
void init_bitset(std::bitset<N>& bitset, bool head, Bits ...tail)
{   
   bitset.set(pos, head);
   init_bitset<N, pos + 1>(bitset, tail...);
}

template <typename ... args>
auto make_bitset(args... bools) -> std::bitset<sizeof...(bools)>
{
    std::bitset<sizeof...(bools)> bitset;
    init_bitset(bitset, bools...);
    return bitset;
}

int main()
{
  std::cout << make_bitset(1,1,0,0,1,0, 1) << "\n";
  std::cout << make_bitset(1,1,1) << "\n";
  std::cout << make_bitset(0,0,0,0) << "\n";
}

有关详细信息,请参阅compiler explorer https://godbolt.org/g/9z2HZa

使用C ++ 17的奖金版

template <typename ... args>
auto make_bitset(args... bools)
{
    unsigned n = 0;
    static_assert( sizeof...(bools) <= sizeof(n) * 8, "overflow" );
    ((n = n * 2 + static_cast<bool>(bools)), ...);
    return std::bitset<sizeof...(bools)>(n);
}