如何检查模板参数是否是 2 的幂

How do I check if a template parameter is a power of two?

本文关键字：参数 是否是 的幂 何检查 检查      更新时间：2023-10-16

我想创建一个静态分配 2^N 字节数组的结构，但我不希望此结构的用户将此大小指定为指数。例：

my_stupid_array<char, 32> a1; // I want this!
my_stupid_array<char, 5> a2; // And not this...

如何检查此模板参数是否是 2 的幂，并用一条很好的消息警告用户？

我已经能够使用一个简单的模板来检查这一点：

template<int N>
struct is_power_of_two {
    enum {val = (N >= 1) & !(N & (N - 1))};
};

但是，我无法通过理智的消息警告用户。有什么想法吗？

编辑

修复了模棱两可的示例。

编辑

1 确实是 2 的幂。修复了！:)

编辑

使用 BOOST_STATIC_ASSERT，我在 GCC 中收到此代码的编译错误：

template<int N>
struct is_power_of_two {
    enum {val = (N >= 1) & !(N & (N - 1))};
    BOOST_STATIC_ASSERT(val);
};

错误

..main.cpp:29:1: error: invalid application of 'sizeof' to incomplete type 'boost::STATIC_ASSERTION_FAILURE<false>'

http://ideone.com/cMfEf

编辑

哦，我明白了。这就是断言失败时我应该得到的消息。但这无法给用户一些理智的信息。:(

这些天，有了constexpr和一些小技巧，你可以

constexpr bool is_powerof2(int v) {
    return v && ((v & (v - 1)) == 0);
}

static_assert救援(仅限C++11，取消注释BOOST_STATIC_ASSERT C++03(：

#include<iostream>
// #include <boost/static_assert.hpp>
template<int N>
struct is_power_of_two {
    enum {val = N && !(N & (N - 1))};
    static_assert(val, "should use a power of 2 as template parameter");
    // BOOST_STATIC_ASSERT(val); // without C++11 support, won't take a string message
};
int main()
{
        std::cout << is_power_of_two<2>::val << "n";
        std::cout << is_power_of_two<3>::val << "n";
}

用于 C++11 的 Ideone 输出

用于 C++03 的 Ideone 输出

UPDATE1：其他想法(我知道你不想要这个，但对于大指数来说要容易得多(：

template<int N>
make_power_of_two
{
    enum { val = 1 << N };
};
my_stupid_array<char, make_power_of_two<5>::val > a1; // size 2^5 = 32

UPDATE2：根据聊天中@sehe的评论，您也可以为constexpr功能执行此操作

constexpr bool is_power_of_two(int x)
{
    return x && ((x & (x-1)) == 0);
}
您可以使用

static_assert提供错误消息：

template<int N>
struct is_power_of_two {
    static_assert((N > 1) & !(N & (N - 1)), "Template parameter must be a power of two.");
};
相关文章：