C语言重载宏函数的小技巧

在写C/C++语言时我们经常会用到宏定义，宏函数就是带参数的宏定义（blablabla……省去背景介绍）。

有时候我们会需要一个可以有多种参数版本的宏定义，例如：

1

2

#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>

 

intf(inta,intb,intc)

{

returna+b+c;

}

 

intmain(void)

{

printf("%d",F(1,2,3));// outputs 6

 

return0;

}

现在，见证奇迹的时候到了，以重载可以有一个或者两个参数的宏函数为例：

1

2

3

4

5

6

7

8

9

#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)

很巧妙的实现了宏函数的重载功能，那么怎么实现三个参数的重载呢，同理：

1

2

3

4

5

6

7

8

9

10

11

#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/