使用一个宏创建两个不同的代码块

时间:2014-04-29 19:36:24

标签: c++ macros c-preprocessor

我想编写一个有一个参数的宏,并将此参数添加到枚举中并创建一个具有相同名称的类。在C ++中有可能吗?如果是,那怎么样?

为了更清楚,这里是我想要做的一个例子:

我像这样调用宏:

REGISTER(A);
REGISTER(B);
REGISTER(C);

或者像这样调用宏:

REGISTER(A, B, C);

我想让预处理器生成如下代码:

enum X { E_A, E_B, E_C };
class A {};
class B {};
class C {};

1 个答案:

答案 0 :(得分:3)

这样的事情可能有所帮助:

#define FE_1(WHAT, X) WHAT(X) 
#define FE_2(WHAT, X, ...) WHAT(X)FE_1(WHAT, __VA_ARGS__)
#define FE_3(WHAT, X, ...) WHAT(X)FE_2(WHAT, __VA_ARGS__)
#define FE_4(WHAT, X, ...) WHAT(X)FE_3(WHAT, __VA_ARGS__)
#define FE_5(WHAT, X, ...) WHAT(X)FE_4(WHAT, __VA_ARGS__)
//... repeat as needed

#define GET_MACRO(_1,_2,_3,_4,_5,NAME,...) NAME 
#define FOR_EACH(action,...) \
  GET_MACRO(__VA_ARGS__,FE_5,FE_4,FE_3,FE_2,FE_1)(action,__VA_ARGS__)

#define REGISTER_ENUM_VALUE(X) E_##X,
#define REGISTER_ENUM(...) enum X { FOR_EACH(REGISTER_ENUM_VALUE, __VA_ARGS__) }
#define REGISTER_CLASS(X) ; class X {}
#define REGISTER(...) REGISTER_ENUM(__VA_ARGS__) FOR_EACH(REGISTER_CLASS, __VA_ARGS__)

REGISTER(A, B, C);

请注意,您无法获得真正的可变参数FOR_EACH,它将始终限制为您正在计划并手动实现的特定最大参数量。至少没有人找到解决办法:)

Live example

致谢:某些代码从here“被盗”。