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>   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 ; }

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

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/