在写C/C++语言时我们经常会用到宏定义,宏函数就是带参数的宏定义(blablabla……省去背景介绍)。
有时候我们会需要一个可以有多种参数版本的宏定义,例如:
|
#define MACRO_FUN(a) doSomethingWithOneArg(a)
#define MACRO_FUN(a, b) doSomethingWithTwoArgs(a, b)
|
但是宏定义不像正规函数那样可以重载,一旦试图重复定义一个已经定义的宏,就会把之前的定义覆盖掉。那么怎么正确的重载宏函数呢?
首先需要了解可变参数宏__VA_ARGS__,__VA_ARGS__可以将宏中的参数…展开,例如:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
#define F(...) f(__VA_ARGS__)
#include <cstdio>
int
f
(
int
a
,
int
b
,
int
c
)
{
return
a
+
b
+
c
;
}
int
main
(
void
)
{
printf
(
"%d"
,
F
(
1
,
2
,
3
)
)
;
// outputs 6
return
0
;
}
|
现在,见证奇迹的时候到了,以重载可以有一个或者两个参数的宏函数为例:
|
#define OneArgument(a) // ...action with one argument
#define TwoArguments(a, b) // ...action with two arguments
#define GetMacro(_1, _2, NAME, ...) NAME
#define Macro(...) GetMacro(__VA_ARGS__, TwoArguments, OneArgument, ...)(__VA_ARGS__)
// usage:
Macro
(
a
)
;
// OneArument(a) is called
Macro
(
a
,
b
)
;
// TwoArguments(a, b) is called
|
tada!是不是没有反应过来这有什么用呢?我们来测试一下这个宏定义是如何工作的,首先来看一个参数的情况:
Macro(a)
=>GetMacro(a, TwoArguments, OneArgument)
=>OneArgument(a)
两个参数的情况:
Macro(a, b)
=>GetMacro(a, b, TwoArguments, OneArgument)
=>TwoArgument(a, b)
很巧妙的实现了宏函数的重载功能,那么怎么实现三个参数的重载呢,同理:
|
#define OneArgument(a) // ...action with one argument
#define TwoArguments(a, b) // ...action with two arguments
#define ThreeArguments(a, b, c) // ...action with three arguments
#define GetMacro(_1, _2, _3, NAME, ...) NAME
#define Macro(...) GetMacro(__VA_ARGS__, ThreeArguments, TwoArguments, OneArgument, ...)(__VA_ARGS__)
// usage:
Macro
(
a
)
;
// OneArument(a) is called
Macro
(
a
,
b
)
;
// TwoArguments(a, b) is called
Macro
(
a
,
b
,
c
)
;
// ThreeArguments(a, b, c) is called
|
博客源地址:http://blog.nemoge.com/trick-for-overload-macro-function-in-c/