C/C++ 宏操作小技巧

Abstract

之前写了一个非常mini的log库（也不算库把，自己瞎jb写的），里面几乎都是宏的实现。这里打算趁热打铁，把自己知道的几下子都贴出来，后续如果有新的收获会更新这个博文。

文笔拙劣，主要是给自己做个提醒。

1. 运行时检测大小端

一目了然，不做解释。

#define __ENDIAN() ({ \

        short _a = 0x1234; \

        *((char*)&_a) == 0x12 ? 1 : 0; \

    })

#define big_endian()  (__ENDIAN() == 1)

#define little_endian()  (__ENDIAN() == 0)

当然还有另一种方式去确定字节序。下面是编译时确定字节序的方式。

只要通过引入头文件<endian.h>便可以在编译时通过宏判断字节序了。参考 ‘/usr/include/linux/tcp.h’ 你会发现另一种写法（自己去看下）。

#if __BYTE_ORDER == __LITTLE_ENDIAN

    // do_sth();

#elif __BYTE_ORDER == __BIG_ENDIAN

    // do_sth();

#else

#error "Unknown byte order"

#endif

2. max函数与min函数

max与min函数是最常用的。我们可以有很多方式去实现它，宏，inline，函数等。

由于这两个函数都很小，一般不建议

#define max(a, b) ({ \

        typeof(a) _a = a; \

        typeof(b) _b = b; \

        _a >= _b ? _a : _b; \

    })

3. SDK 中的函数名拼接

#include <stdio.h>

#define test(__DOC, args) __SDK_##__DOC##_TEST(args)

void __SDK_V1_TEST(int a) {

    fprintf (stdout, "version %d\n", a);

    return;

}

void __SDK_V2_TEST(int a) {

    fprintf (stdout, "version %d\n", a);

    return;

}

int main(int argc, char **argv) {

    test(V1, 1);

    test(V2, 2);

    return 0;

}

4. 获取枚举变量的名字

枚举定义通常具有更好的可读性，比如

enum enWeek_t {MONDAY, TUESDAY};

中MONDAY与TUSDAY比枚举值0，1有更好的可读性。事实上对于开发者，一个可读性更好的名称比这个名称的实际值加友好。如果是日志上能给出‘MONDAY’而不是 0这样的字眼，更能帮助我们理解。所以获取枚举类型的名称或许是个更好的选择。

一个示例：

enum_name.h

 #ifndef __ENUM_NAME_H__

 #define __ENUM_NAME_H__

 // MACRO_HELPER

 #define MACRO_HELPER(v) v,

 #define WEEKDAYS \

     MACRO_HELPER(MONDAY)

     MACRO_HELPER(TUESDAY)

     MACRO_HELPER(WEDNESDAY)

 // 这里宏替换会换成 enum {MONDAY, TUESDAY, WEDNESDAY,};

 enum {

     WEEKDAYS

 };

16 #undef MACRO_HELPER(v)

 #endif //__ENUM_NAME_H__

enum_name.c

 #include "enum_name.h"

 // 这里重新定义宏，利用编译器的特性 #v会把v原封不动的保存为一个字符串类型（## 两个#才是拼接，参考上一篇log函数的实现）

 #define MACRO_HELPER(v) {v, #v},

 struct week_name_t {

     int val;

     const char *name;

 }; // 结构体的定义中成员顺序，必须和宏的定义一致

 // 这里宏会替换成 struct week_names[] = {

 // {MONDAY, "MONDAY"}, {TUESDAY, "TUESDAY"}, {WEDNESDAY, "WEDNESDAY"}, {0, 0}}

 struct week_names[] = {

     WEEKDAYS

     {, }    // this is the end of array

 };

 const char *get_enum_name(int v) {

     int idx = ;

     while (week_names[idx].name) {

         if (idx == v)

             return week_names[idx].name;

     }

     return "Unknow";

 }

这里我们对外的唯一接口就是 get_enum_name 这个函数，只需要传入指定的数值，我们就能获得对应星期的名字。主要依赖宏替换的效果，如果想追加‘星期四’， ‘星期五’，我们只需要在头文件的 WEEKDAYS 添加 MACRO_HELPER(THURSDAY) 与 MACRO_HELPER(FRIDAY) 就行了，不要要其他的变动。接下来可以写个main函数测试下效果，我就不写了，只是抛砖引玉下。

当然你也可以编写一个函数把所有星期的名称字符串对应输出，这是你的自由，不在这个博问的讨论范围。