C语言宏定义使用总结与递归宏

C语言宏定义使用总结与递归宏

C语言的宏可以用来做宏定义、条件编译和文件包含，本文主要总结宏定义#define的用法。

以下例子通过Xcode12.0测试，gnu99标准。

特殊符号#和##

在一个宏参数前面使用#号，则此参数会变为字符串：

#define LOG(X) printf(#X)
LOG(abc);   // printf("abc");

##是连接符号，可在宏参数前后使用：

#define DefineValue(NAME, TYPE, VAL) TYPE NAME##_##TYPE = VAL;

DefineValue(aaa, int, 91) // int aaa_int = 91;
DefineValue(aaa, float, 3.26)  // float aaa_float = 3.26;
printf("%d--%f\n", aaa_int, aaa_float); // 91--3.260000

变长参数__VA_ARGS__和... 

#define PrintStderr(format, ...) fprintf(stderr, format, ##__VA_ARGS__)

int aa = 1;
int bb = 2;
PrintStderr("%d--%d\n", aa, bb); // fprintf(stderr, "%d--%d\n", aa, bb)
PrintStderr("Huimao Chen\n"); // fprintf(stderr, "Huimao Chen\n");

简单来说，...表示所有剩下的参数，__VA_ARGS__被宏定义中的...参数所替换。

需要注意的是，上面例子用##连接逗号和后面的__VA_ARGS__，这在c语言的GNU扩展语法里是一个特殊规则：当__VA_ARGS__为空时，会消除前面这个逗号。如果上面例子的宏定义去掉##号，第一个例子无影响，但第二个例子则会替换成fprintf(stderr, "Huimao Chen", );多出的一个逗号导致编译失败。

do{}while(0)

宏定义里可以有多行语句，do{}while(0)就能保证了这个宏成为独立的语法单元。
如果有这样一个宏定义 #define SWAP(A, B) int tmp = A; A = B; B = tmp;，虽然如下代码能正常执行：

int aa = 1;
int bb = 2;
SWAP(aa, bb);
printf("%d--%d\n", aa, bb); // 2--1

但是，下面的代码就有问题了，编译报错：

int aa = 1;
int bb = 2;
if (aa < bb)
    SWAP(aa, bb);
printf("%d--%d\n", aa, bb);

此时把这个宏改成如下这种形式就能正常运行：
#define SWAP(A, B) do {int tmp = A; A = B; B = tmp;} while(0)

参数用括号保护

宏定义只是简单的替换，通过替换可能会导致运算优先级不符合预期，此时需要用括号保护参数。

#define MAX(A, B) A > B ? A : B
int aa = 2;
int bb = 3;
printf("%d\n", 2 * MAX(aa, bb));
// printf("%d\n", 2 * aa > bb ? aa : bb);
// printf("%d\n", 2 * 2 > 3 ? 2 : 3);
// printf("%d\n", 4 > 3 ? 2 : 3);
// 2

上面的例子最后输出的是2，与预期的结果6不符，此时把宏定义改为如下形式就能解决问题：

#define MAX(A, B) ((A) > (B) ? (A) : (B))
int aa = 2;
int bb = 3;
printf("%d\n", 2 * MAX(aa, bb));
// printf("%d\n", 2 * ((aa) > (bb) ? (aa) : (bb)));
// printf("%d\n", 2 * ((2) > (3) ? (2) : (3)));
// printf("%d\n", 2 * (3));
// 6

({})包裹语句

GNU扩展语法。有时候，宏的参数可以是个复合结构，而参数可能有多次取值。如果传入的宏参数是一个函数，则这个函数会有多次调用：

#define MAX(A, B) ((A) > (B) ? (A) : (B))
int aa = 5;
int bb = MAX(2, foo(aa)); // 函数foo被调用了两次

为了防止此类副作用，可以改写为如下形式：

#define MAX(A, B) ({ __typeof__(A) __a = (A); \
                     __typeof__(B) __b = (B); \
                     __a > __b ? __a : __b; })

({})在顺序执行语句之后，返回最后一条表达式的值，这也是其区别于do{}while(0)的地方。

嵌套使用宏

在使用了# 或##的宏中，如果宏的参数是另一个宏，则会阻止另一个宏展开。为了保证参数优先展开，需要多嵌套一层宏定义。具体可以看如下例子：

#define Stringify(A) _Stringify(A)
#define _Stringify(A) #A

#define Concat(A, B) _Concat(A, B)
#define _Concat(A, B) A##B

printf("%s\n", Stringify(Concat(Hel, lo))); // 输出:Hello
// printf("%s\n", Stringify(Hello));
// printf("%s\n", _Stringify(Hello));
// printf("%s\n", "Hello");
// Hello

printf("%s\n", _Stringify(Concat(Hel, lo))); // 输出:Concat(Hel, lo)
// printf("%s\n", "Concat(Hel, lo)");
// Concat(Hel, lo)

宏的递归展开

虽然宏定义只是简单替换，但也有令人眼前一亮的小技巧，如模式匹配、参数检测、递归宏等等。这里只介绍递归宏，只要看懂了这篇文章的递归宏，遇到其他宏理解起来也是小意思。以下例子参考了我的开源框架HMLog，带上了前缀HM。

在介绍递归宏之前，先来介绍一个获取宏参数个数的技巧。

获取宏参数个数

这是一个常见的宏，其构建思维广泛使用于各种宏功能。下面的宏适用于1到10个参数，最后一个例子给出了解释：

#define HMMacroArgCount(...) _HMMacroArgCount(__VA_ARGS__, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1)
#define _HMMacroArgCount(_0, _1, _2, _3, _4, _5, _6, _7, _8, _9, COUNT, ...) COUNT

HMMacroArgCount(a); // 1;
HMMacroArgCount(a, a); // 2;
HMMacroArgCount(a, b, c, d); // 4;
// MacroArgCount(a, b, c, d); >>> _MacroArgCount(a, b, c, d, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1);
// _MacroArgCount定义里是固定取第11个参数，这里命名为COUNT，而上面第11个参数就是4，故最终展开结果为4;

举个实际应用的例子：

double average(int num, ...) {
    va_list valist;
    double sum = 0.0;
    va_start(valist, num);
    for (int i = 0; i < num; ++i) {
       sum += va_arg(valist, int);
    }
    va_end(valist);
    return sum / num;
}
#define HMAverage(...) average(HMMacroArgCount(__VA_ARGS__), __VA_ARGS__)

double result = average(5, 10, 20, 30, 40, 50);
printf("%f\n", result); // 30.000000

// 可以少输入一个总数5，预编译期间就替换为double result2 = average(5, 10, 20, 30, 40, 50);
double result2 = HMAverage(10, 20, 30, 40, 50);
printf("%f\n", result2); // 30.000000

average是个可变参数的函数，计算输入整数的平均值。直接调用可变参数函数往往需要传入参数的长度。使用宏HMAverage，则省略了这个长度参数，在函数调用频繁的情况下大大降低了出错概率，而且是在预编译期间完成替换，并不影响实际运行速度。

此时HMMacroArgCount并不支持0个参数的情况，其实根据前面总结的规律稍作修改就可以支持0个参数，留给读者思考。

接下来正式介绍递归宏，这里给出两种方法。

1. 连接宏的参数个数，定义一系列结构相似的宏。

我需要一个HMPrint宏，输入任意个整数（这个例子是5个以内），就能省略格式化参数，按照指定格式打印出来。

#define HMMacroArgCount(...) _HMMacroArgCount(__VA_ARGS__, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1)
#define _HMMacroArgCount(_0, _1, _2, _3, _4, _5, _6, _7, _8, _9, COUNT, ...) COUNT

#define HMStringify(A) _HMStringify(A)
#define _HMStringify(A) #A

#define HMConcat(A, B) _HMConcat(A, B)
#define _HMConcat(A, B) A##B

#define HMPrint(...) printf(HMStringify(_HMFormat(__VA_ARGS__)), __VA_ARGS__)
#define _HMFormat(...) HMConcat(_HMFormat, HMMacroArgCount(__VA_ARGS__))(__VA_ARGS__)

#define _HMFormat1(_0)                  _0->%d\n
#define _HMFormat2(_0, _1)              _HMFormat1(_0)_1->%d\n
#define _HMFormat3(_0, _1, _2)          _HMFormat2(_0, _1)_2->%d\n
#define _HMFormat4(_0, _1, _2, _3)      _HMFormat3(_0, _1, _2)_3->%d\n
#define _HMFormat5(_0, _1, _2, _3, _4)  _HMFormat4(_0, _1, _2, _3)_4->%d\n

int a = 1991, b = 3, c = 26;
HMPrint(a, b, c); // 预编译时被替换为 printf("a->%d\nb->%d\nc->%d\n", a, b, c);
//a->1991
//b->3
//c->26

根据定义，HMPrint展开后就是printf函数，后面的参数部分保持不变。前面格式化宏_HMFormat用连接符##把_HMFormat和HMMacroArgCount(__VA_ARGS__)连接起来，后者返回参数的个数，如果HMPrint传入3个参数，连接后变为_HMFormat3并传入原始参数。把_HMFormat3前两个参数传递给_HMFormat2，第3个参数替换为c->%d\n，继续就是_HMFormat2展开，依次类推，直到格式化部分为HMStringify(a->%d\nb->%d\nc->%d\n)，最终变为"a->%d\nb->%d\nc->%d\n"。

为了帮助理解，我这里给出展开的过程，只需让依次让参数优先展开，就能得到想要的结果：

// 依次替换展开宏，参数优先展开
HMPrint(a, b, c);
printf(HMStringify(_HMFormat(a, b, c)), a, b, c);
printf(HMStringify(HMConcat(_HMFormat, HMMacroArgCount(a, b, c))(a, b, c)), a, b, c);
printf(HMStringify(HMConcat(_HMFormat, 3)(a, b, c)), a, b, c);
printf(HMStringify(_HMFormat3(a, b, c)), a, b, c);
printf(HMStringify(_HMFormat2(a, b)c->%d\n), a, b, c);
printf(HMStringify(_HMFormat1(a)b->%d\nc->%d\n), a, b, c);
printf(HMStringify(a->%d\nb->%d\nc->%d\n), a, b, c);
printf(_HMStringify(a->%d\nb->%d\nc->%d\n), a, b, c);
printf("a->%d\nb->%d\nc->%d\n", a, b, c);

_HMFormat也可以写成这种方式，更容易理解：

#define _HMFormat1(_0)                  _0->%d\n
#define _HMFormat2(_0, _1)              _0->%d\n_1->%d\n
#define _HMFormat3(_0, _1, _2)          _0->%d\n_1->%d\n_2->%d\n
#define _HMFormat4(_0, _1, _2, _3)      _0->%d\n_1->%d\n_2->%d\n_3->%d\n
#define _HMFormat5(_0, _1, _2, _3, _4)  _0->%d\n_1->%d\n_2->%d\n_3->%d\n_4->%d\n

// 再次给出这种方式下展开的过程，可以看到_HMFormat3一次到位替换为需要的格式
HMPrint(a, b, c);
printf(HMStringify(_HMFormat(a, b, c)), a, b, c);
printf(HMStringify(HMConcat(_HMFormat, HMMacroArgCount(a, b, c))(a, b, c)), a, b, c);
printf(HMStringify(HMConcat(_HMFormat, 3)(a, b, c)), a, b, c);
printf(HMStringify(_HMFormat3(a, b, c)), a, b, c);
printf(HMStringify(a->%d\nb->%d\nc->%d\n), a, b, c);
printf(_HMStringify(a->%d\nb->%d\nc->%d\n), a, b, c);
printf("a->%d\nb->%d\nc->%d\n", a, b, c);

不建议用后面这种方式，一是递归写法更加简洁统一；二是结合HMMacroArgCount这个宏一起可以扩展成支持10个参数的HMPrint，此时只需要测试最多参数的例子，没有出错就几乎保证了所有这类宏都没写错。再次强调一点，宏的递归展开只发生在预编译期间，这种递归并不影响运行时效率。

2. 利用宏的延迟展开和多次扫描

这种方法较难理解，还是用HMPrint的例子：

#define HMStringify(A) _HMStringify(A)
#define _HMStringify(A) #A

#define HMConcat(A, B) _HMConcat(A, B)
#define _HMConcat(A, B) A##B

#define HMMacroArgCheck(...) _HMMacroArgCheck(__VA_ARGS__, N, N, N, N, N, N, N, N, N, 1)
#define _HMMacroArgCheck(_0, _1, _2, _3, _4, _5, _6, _7, _8, _9, TARGET, ...) TARGET

#define HMPrint(...) printf(HMStringify(HMExpand(HMForeach(_HMFormat, __VA_ARGS__))), __VA_ARGS__)
#define _HMFormat(A) A->%d\n

#define HMForeach(MACRO, ...) HMConcat(_HMForeach, HMMacroArgCheck(__VA_ARGS__)) (MACRO, __VA_ARGS__)
#define _HMForeach() HMForeach
#define _HMForeach1(MACRO, A) MACRO(A)
#define _HMForeachN(MACRO, A, ...) MACRO(A)HMDefer(_HMForeach)() (MACRO, __VA_ARGS__)

#define HMEmpty()
#define HMDefer(ID) ID HMEmpty()

#define HMExpand(...)   _HMExpand1(_HMExpand1(_HMExpand1(__VA_ARGS__)))
#define _HMExpand1(...) _HMExpand2(_HMExpand2(_HMExpand2(__VA_ARGS__)))
#define _HMExpand2(...) _HMExpand3(_HMExpand3(_HMExpand3(__VA_ARGS__)))
#define _HMExpand3(...) __VA_ARGS__

int a = 1991, b = 3, c = 26;
HMPrint(a, b, c); // 预编译时被替换为 printf("a->%d\nb->%d\nc->%d\n", a, b, c);
//a->1991
//b->3
//c->26

int a1 = 11, a2 = 22, a3 = 33, a4 = 44, a5 = 55, a6 = 66, a7 = 77, a8 = 88, a9 = 99, a10 = 100;
HMPrint(a1, a2, a3, a4, a5, a6, a7, a8, a9, a10);
//a1->11
//a2->22
//a3->33
//a4->44
//a5->55
//a6->66
//a7->77
//a8->88
//a9->99
//a10->100

HMMacroArgCheck用于检测参数数量，如果是1个参数则返回1，当参数大于1个，且小于等于10个的情况下返回N。HMDefer用于延迟展开，而HMExpand是为了多次扫描宏，理解这种技巧需要知道宏展开的一般规则，可以阅读这个系列的文章，本文不再赘述。HMForeach(MACRO, ...)这个宏的用处是每个参数都会被传递给MACRO宏，为HMForeach(MACRO, ...)举个简化的例子用于理解用处：

#define Increase(X) X += 1; // 定义一个宏，准备用来处理每一个参数。注意最后有分号
int aa = 10, bb = 20, cc = 30;
// 使用HMForeach需要有HMExpand包裹起来，以便多次扫描顺利展开
HMExpand(HMForeach(Increase, aa, bb, cc))
// 相当于：Increase(aa)Increase(bb)Increase(cc)
// 最后变为：aa += 1;bb += 1;cc += 1;
    
printf("%d--%d--%d", aa, bb, cc); // 输出：11--21--31

这种方法不需要去写_HMFormat1、_HMFormat2、_HMFormat3等这一类相似结构的宏，支持参数个数取决于HMMacroArgCheck，所以增加支持的参数数量变得轻而易举，当参数比较多的情况使用这种方式更加方便。不足之处是只能对每个参数做相同的处理，而第1种方式是可以对每个参数做不同处理的。

最后，我同样给出展开的过程，但这并非实际展开过程，比如忽略了HMExpand展开的时机，仅在最后直接消除：

// 这并非实际展开过程，比如忽略了HMExpand展开的时机，仅在最后直接消除
HMPrint(a, b, c);
printf(HMStringify(HMExpand(HMForeach(_HMFormat, a, b, c))), a, b, c);
printf(HMStringify(HMExpand(HMConcat(_HMForeach, HMMacroArgCheck(a, b, c)) (_HMFormat, a, b, c))), a, b, c);
printf(HMStringify(HMExpand(HMConcat(_HMForeach, N) (_HMFormat, a, b, c))), a, b, c);
printf(HMStringify(HMExpand(_HMForeachN (_HMFormat, a, b, c))), a, b, c);
printf(HMStringify(HMExpand(_HMFormat(a)HMDefer(_HMForeach)() (_HMFormat, b, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\n_HMForeach() (_HMFormat, b, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nHMForeach (_HMFormat, b, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nHMConcat(_HMForeach, HMMacroArgCheck(b, c)) (_HMFormat, b, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nHMConcat(_HMForeach, N) (_HMFormat, b, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\n_HMForeachN (_HMFormat, b, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\n_HMFormat(b)HMDefer(_HMForeach)() (_HMFormat, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nb->%d\n_HMForeach() (_HMFormat, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nb->%d\nHMForeach (_HMFormat, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nb->%d\nHMConcat(_HMForeach, HMMacroArgCheck(c)) (_HMFormat, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nb->%d\nHMConcat(_HMForeach, 1) (_HMFormat, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nb->%d\n_HMForeach1 (_HMFormat, c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nb->%d\n_HMFormat(c))), a, b, c);
printf(HMStringify(HMExpand(a->%d\nb->%d\nc->%d\n)), a, b, c);
printf(HMStringify(a->%d\nb->%d\nc->%d\n), a, b, c);
printf(_HMStringify(a->%d\nb->%d\nc->%d\n), a, b, c);
printf("a->%d\nb->%d\nc->%d\n", a, b, c);

利用宏能写出很多有意思的代码，如果你是iOS开发者，强烈建议看看我写的一个最佳实践HMLog（有且仅有一个HMLog.h文件），另外也可以看libextobjc库对宏的使用。关于宏更多的使用技巧，可以查看p99或Boost preprocessor。

©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 204,684评论 6赞 478
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 87,143评论 2赞 381
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 151,214评论 0赞 337
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 54,788评论 1赞 277
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 63,796评论 5赞 368
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 48,665评论 1赞 281
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 38,027评论 3赞 399
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 36,679评论 0赞 258
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 41,346评论 1赞 299
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,664评论 2赞 321
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,766评论 1赞 331
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 33,412评论 4赞 321
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 39,015评论 3赞 307
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,974评论 0赞 19
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1赞 260
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 45,073评论 2赞 350
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 42,501评论 2赞 343