上一篇我们通过MAX(a,b)这个例子给大家简单的介绍了宏定义的那些坑。同时在最后也给大家留下了两个问题。
- 如何使得宏能同时处理整型和浮点型。
- 两个不同类型但兼容的数据给MAX(),比如一个3.14和一个100,它应该如何得出正确结果。
首先,我们先来解决第一个问题,要让我们的宏能同时处理整型和浮点型。因此我们不能在宏定义里固定为整型,而是要让程序自动获得宏参数的类型,代码修改如下:
#include <stdio.h>
#define MAX(a,b) \
({\
typeof(a) _a = a;\
typeof(b) _b = b;\
((_a)>(_b) ? (_a):(_b));\
})
注意:typeof这个关键字也是GUN的扩展语法,可以用它来取得一个数据的类型,这样一来,不管传递给宏的是整型还是浮点型数据,宏都可以自动处理。这样一来我们的第一个问题也就解决了。
接下来的第二个问题是:如果两个不同类型但是兼容的数据给MAX(),比如一个3.14和一个100,它应该怎么得出正确结果呢?基于第一题的解题思路,修改得出以下代码:
#include <stdio.h>
#define MAX(a,b) \
({\
typeof(a) _a = a;\
typeof(b) _b = b;\
(void)(&_a == &_b);\
((_a)>(_b) ? (_a):(_b));\
})
执行代码可以得出正确结果。
注意 (void)(&_a == &_b);\ ,使用一个判断来强迫编译器对比两个变量_a和_b的地址类型(不能直接比较_a和_b的值,因为即使浮点和整型不同但仍是兼容的),由此来触发由于类型不同而产生的警告。前面的强类型转换(void)的目的是要让编译器认为后面的比较语句是有作用的,从而不会误以为没有实际作用而报出其他警告。
实际上,这个MAX()宏定义代码出自Linux内核源码,是世界上最聪明的linux黑客写的,这些代码与其说是为了实现一些苛刻的要求,不如说更像一个展现C语言运用的舞台。