如何实现仅接受2的幂输入的C++ constexpr logBase2函数？

需求描述 #

实现一个编译期求值的logBase2函数，仅当输入为2的正整数次幂时允许编译并返回对数结果，否则直接触发编译错误。期望行为如下：

constexpr int a = logBase2(16); // 正常编译，返回4
constexpr int b = logBase2(42); // 编译失败：输入不是2的幂
constexpr int c = logBase2(-1); // 编译失败：输入为负数

问题分析 #

最初尝试的代码无法通过编译，核心原因是constexpr函数的参数不强制要求是编译期常量，因此static_assert无法直接使用该参数做编译期验证：

constexpr bool isPowerOf2(int x) {
    return (x > 0) && ((x & (x - 1)) == 0);
}

constexpr int logBase2(int x) {
    static_assert(isPowerOf2(x), "Input must be a power of 2"); // 错误：x不是编译期常量表达式

    int result = 0;
    while ((x & 1) == 0) {
        x >>= 1;
        result++;
    }

    return result;
}

而constexpr int x作为函数参数的写法并非合法C++语法，因此需要通过其他方式强制参数为编译期常量并完成验证。

可行实现方案 #

方案1：C++20+ 使用consteval关键字 #

consteval关键字定义的函数必须在编译期完成求值，其参数天然是编译期常量，因此可以直接结合static_assert做验证：

constexpr bool isPowerOf2(int x) {
    return (x > 0) && ((x & (x - 1)) == 0);
}

consteval int logBase2(int x) {
    static_assert(isPowerOf2(x), "Input must be a positive power of 2");

    int result = 0;
    int tmp = x;
    while ((tmp & 1) == 0) {
        tmp >>= 1;
        result++;
    }
    return result;
}

调用方式完全符合最初的需求，只要输入不符合条件或不是编译期常量，编译器会直接报错。

方案2：兼容C++11+ 使用非类型模板参数 #

如果需要兼容更早的C++标准，可以将输入作为非类型模板参数（模板参数本身就是编译期常量），在模板内部完成静态断言：

template<int x>
constexpr int logBase2() {
    static_assert(x > 0 && ((x & (x - 1)) == 0), "Input must be a positive power of 2");

    int result = 0;
    int tmp = x;
    while ((tmp & 1) == 0) {
        tmp >>= 1;
        result++;
    }
    return result;
}

调用时需要调整为模板参数形式：

constexpr int a = logBase2<16>(); // 返回4
constexpr int b = logBase2<42>(); // 编译失败
constexpr int c = logBase2<-1>(); // 编译失败

优化方案：C++20+ 结合标准库位运算函数 #

C++20标准库提供了专门的位运算工具，可以更简洁高效地实现需求：

#include <bit>

consteval int logBase2(int x) {
    static_assert(x > 0 && std::has_single_bit(static_cast<unsigned int>(x)), 
                  "Input must be a positive power of 2");
    // std::bit_width返回无符号数的二进制位数，减1即为log₂结果
    return std::bit_width(static_cast<unsigned int>(x)) - 1;
}

std::has_single_bit专门用于判断是否为2的幂，std::bit_width返回二进制位数，两者均为编译期可用的标准库函数，无需手动实现循环或位运算逻辑。

内容的提问来源于stack exchange，提问作者puff_the_magic_dragon