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
", a);
    return;
}

void __SDK_V2_TEST(int a) {
    fprintf (stdout, "version %d
", 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
    {0, 0}    // this is the end of array
};

const char *get_enum_name(int v) {
    int idx = 0;
    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函数测试下效果，我就不写了，只是抛砖引玉下。

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