如何在C++编译期自动定义枚举类的笛卡尔积枚举类？

可以通过预处理器+标准C++特性在编译期自动生成两个枚举类的笛卡尔积枚举类，完全满足你用于switch穷举的需求。以下是具体实现步骤：

1. 用宏定义基础枚举的元素列表 #

先通过宏保存枚举的元素名称，后续复用这些名称生成乘积枚举：

// 定义Color枚举的元素列表
#define COLOR_ITEMS \
    X(RED) \
    X(GREEN)

// 定义Shape枚举的元素列表
#define SHAPE_ITEMS \
    X(CIRCLE) \
    X(TRIANGLE)

2. 生成基础枚举类 #

基于上面的宏定义基础枚举，同时添加COUNT成员用于获取元素数量：

enum class Color {
    COLOR_ITEMS,
    COUNT  // 标记枚举元素总数
};

enum class Shape {
    SHAPE_ITEMS,
    COUNT
};

3. 自动生成乘积枚举类ComboResult #

通过嵌套宏遍历两个枚举的元素列表，拼接生成笛卡尔积的枚举项名称：

// 辅助宏：拼接两个枚举项名称
#define COMBINE_ENUM_NAMES(a, b) a##_##b

enum class ComboResult {
    // 遍历Color的每个元素，再遍历Shape的每个元素，生成组合枚举项
    #define X_COLOR(c) \
        #define X_SHAPE(s) COMBINE_ENUM_NAMES(c, s), \
        SHAPE_ITEMS \
        #undef X_SHAPE
    COLOR_ITEMS
    #undef X_COLOR
    COUNT  // 标记组合枚举的元素总数
};

这段代码会自动生成你需要的RED_CIRCLE、RED_TRIANGLE、GREEN_CIRCLE、GREEN_TRIANGLE枚举项。

4. 实现枚举间的转换函数 #

添加constexpr转换函数，方便在基础枚举和组合枚举之间双向转换：

#include <cstddef>
#include <utility>

// 获取枚举的元素数量
template <typename EnumType>
constexpr std::size_t get_enum_size() {
    return static_cast<std::size_t>(EnumType::COUNT);
}

// 将Color和Shape转换为ComboResult
constexpr ComboResult to_combo(Color color, Shape shape) {
    return static_cast<ComboResult>(
        static_cast<std::size_t>(color) * get_enum_size<Shape>() + static_cast<std::size_t>(shape)
    );
}

// 将ComboResult拆分为Color和Shape
constexpr std::pair<Color, Shape> from_combo(ComboResult combo) {
    const auto idx = static_cast<std::size_t>(combo);
    const auto shape_idx = idx % get_enum_size<Shape>();
    const auto color_idx = idx / get_enum_size<Shape>();
    return {static_cast<Color>(color_idx), static_cast<Shape>(shape_idx)};
}

5. 在switch中使用组合枚举 #

生成的ComboResult可以直接用于switch语句，实现全穷举：

#include <iostream>

void handle_combo(ComboResult combo) {
    switch (combo) {
        case ComboResult::RED_CIRCLE:
            std::cout << "处理红色圆形\n";
            break;
        case ComboResult::RED_TRIANGLE:
            std::cout << "处理红色三角形\n";
            break;
        case ComboResult::GREEN_CIRCLE:
            std::cout << "处理绿色圆形\n";
            break;
        case ComboResult::GREEN_TRIANGLE:
            std::cout << "处理绿色三角形\n";
            break;
        case ComboResult::COUNT:
            // 无需处理，仅用于标记数量
            break;
    }
}

int main() {
    handle_combo(to_combo(Color::RED, Shape::CIRCLE));
    handle_combo(to_combo(Color::GREEN, Shape::TRIANGLE));
    return 0;
}

注意事项 #

• 该方案依赖C预处理器和constexpr特性，兼容C11及以上版本。
• 如果需要扩展更多枚举类的笛卡尔积，只需修改宏定义的元素列表即可。

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