如何基于编译期对象状态实现函数重载或分支选择？

问题背景 #

原类NotAWriter实现如下，会根据自身on状态决定是否输出整数到cout：

#include <iostream>

class NotAWriter{
public:
    NotAWriter& operator<<(const int& arg) {
        if (on) std::cout << arg;
        return *this;
    }
    void toggle() {on=!on;}
private:
    bool on = false;
};

已知对NotAWriter.toggle()的调用次数是编译期常量，希望在编译期就根据on的状态选择是否执行输出逻辑，但尝试的两种方式均失败：

• 方式一用if constexpr触发错误：error: use of ‘this’ in a constant expression，原因是on是成员变量，常量表达式中无法通过this访问运行期成员。
• 方式二用std::enable_if不可行，因为on不是编译期可见的常量表达式，无法作为模板参数使用。

解决方案：将状态转为编译期模板参数 #

要实现编译期分支选择，核心是把on的状态从运行期成员变量转为编译期模板参数。由于toggle调用次数是编译期常量，我们可以通过模板类的参数变化模拟状态切换，每次"toggle"对应生成一个不同模板参数的类实例。

基础实现代码 #

#include <iostream>

// 模板类，用模板参数On表示编译期状态
template<bool On = false>
class CompileTimeWriter {
public:
    // 根据On的编译期值决定输出逻辑，无用分支会在编译期被剔除
    CompileTimeWriter& operator<<(const int& arg) {
        if constexpr (On) {
            std::cout << arg;
        }
        return *this;
    }

    // 返回状态翻转后的新实例，类型在编译期确定
    CompileTimeWriter<!On> toggle() const {
        return {};
    }
};

使用示例 #

因为toggle次数是编译期常量，编译器可以直接推导每个实例的状态类型：

int main() {
    auto writer = CompileTimeWriter<>{}; // 初始状态On=false
    auto writer_toggled = writer.toggle(); // 状态变为On=true，类型为CompileTimeWriter<true>
    auto writer_toggled_twice = writer_toggled.toggle(); // 状态回到On=false，类型为CompileTimeWriter<false>

    writer << 123; // 编译期优化：无输出代码
    writer_toggled << 456; // 编译期优化：仅保留输出456的代码
    writer_toggled_twice << 789; // 编译期优化：无输出代码

    return 0;
}

进阶方案：通过计数追踪状态 #

如果需要更直观地关联toggle次数与状态，可以用整数模板参数记录调用次数，通过奇偶性判断最终状态：

#include <iostream>
#include <type_traits>

template<size_t ToggleCount = 0>
class ToggledWriter {
public:
    // 编译期计算当前状态：奇数次toggle则开启输出
    static constexpr bool is_on = (ToggleCount % 2) != 0;

    ToggledWriter& operator<<(const int& arg) {
        if constexpr (is_on) {
            std::cout << arg;
        }
        return *this;
    }

    // 返回toggle次数+1的新实例
    ToggledWriter<ToggleCount + 1> toggle() const {
        return {};
    }
};

// 使用示例
int main() {
    auto w0 = ToggledWriter<>{}; // ToggleCount=0 → is_on=false
    auto w1 = w0.toggle(); // ToggleCount=1 → is_on=true
    auto w2 = w1.toggle(); // ToggleCount=2 → is_on=false

    w0 << 1; // 无输出
    w1 << 2; // 输出2
    w2 << 3; // 无输出
    return 0;
}

核心原理 #

• 模板参数是编译期常量，if constexpr会直接在编译期剔除未选中的分支，完全不会生成无用代码。
• toggle函数返回的实例类型在编译期确定，编译器可以对每个状态的operator<<进行极致优化。
• 该方案严格符合题目中"toggle调用次数为编译期常量"的前提，所有状态推导都在编译阶段完成。

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