2014-09-23 57 views
1

目前我有一個宏來檢查一個值是一種類型。編譯時間檢查針對C中的多種類型?

#define CHECK_TYPE_INLINE(val, type) \ 
    ((void)(((type)0) != (0 ? (val) : ((type)0)))) 

這對於能夠在某些情況下鍵入檢查宏參數非常有用。

但是如果我要檢查多種類型呢?例如,檢查它是否是struct Foo *struct Bar *

例,

static inline _insert_item(struct List *ls, void *item) { /* function body*/ } 

/* type-checked wrapper */ 
#define insert_item(ls, item) \ 
    (CHECK_TYPE_ANY(item, struct Foo *, struct Bar *), \ 
    _insert_item(ls, item)) 

有沒有做一些這方面的好辦法嗎?

+3

你能告訴你如何使用宏的例子嗎? – imreal 2014-09-23 06:39:28

+0

@imreal,好點,完成 – ideasman42 2014-09-23 06:47:08

回答

3

由於這是標記C11,你必須通過_Generic關鍵字此語言支持:

#define CHECK(x) _Generic((x), \ 
        int : 1, \ 
        float : 2, \ 
        default : 3) 

,其中1,2和3是如果傳遞的參數是一個特定類型的應該發生的事情。

Link with some good examples

+0

注意,在_Generic之前包含'(void)'可能會很好,以確保值永遠不會被使用。 (假設這隻用於檢查類型) – ideasman42 2014-09-23 06:56:57

0

要添加到@ Lundin的回答,這可以使用__VA_ARGS__

/* From: http://stackoverflow.com/questions/24836793/varargs-elem-macro-for-use-with-c/24837037#24837037 */ 

/* -------------------------------------------------------------------- */ 
/* varargs macros (keep first so others can use) */ 
/* --- internal helpers --- */ 
#define _VA_NARGS_GLUE(x, y) x y 
#define _VA_NARGS_RETURN_COUNT(\ 
    _1_, _2_, _3_, _4_, _5_, _6_, _7_, _8_, _9_, _10_, _11_, _12_, _13_, _14_, _15_, _16_, \ 
    _17_, _18_, _19_, _20_, _21_, _22_, _23_, _24_, _25_, _26_, _27_, _28_, _29_, _30_, _31_, _32_, \ 
    count, ...) count 
#define _VA_NARGS_EXPAND(args) _VA_NARGS_RETURN_COUNT args 
#define _VA_NARGS_COUNT_MAX32(...) _VA_NARGS_EXPAND((__VA_ARGS__, \ 
    32, 31, 30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, \ 
    15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)) 
#define _VA_NARGS_OVERLOAD_MACRO2(name, count) name##count 
#define _VA_NARGS_OVERLOAD_MACRO1(name, count) _VA_NARGS_OVERLOAD_MACRO2(name, count) 
#define _VA_NARGS_OVERLOAD_MACRO(name, count) _VA_NARGS_OVERLOAD_MACRO1(name, count) 
/* --- expose for re-use --- */ 
#define VA_NARGS_CALL_OVERLOAD(name, ...) \ 
    _VA_NARGS_GLUE(_VA_NARGS_OVERLOAD_MACRO(name, _VA_NARGS_COUNT_MAX32(__VA_ARGS__)), (__VA_ARGS__)) 

/* -------------------------------------------------------------------- */ 
/* CHECK_TYPE_ANY */ 

#define _VA_CHECK_TYPE_ELEM2(v, a) \ 
    ((void)_Generic((v), a: 0)) 
#define _VA_CHECK_TYPE_ELEM3(v, a, b) \ 
    ((void)_Generic((v), a: 0, b: 0)) 
#define _VA_CHECK_TYPE_ELEM4(v, a, b, c) \ 
    ((void)_Generic((v), a: 0, b: 0, c: 0)) 
#define _VA_CHECK_TYPE_ELEM5(v, a, b, c, d) \ 
    ((void)_Generic((v), a: 0, b: 0, c: 0, d: 0)) 
#define _VA_CHECK_TYPE_ELEM6(v, a, b, c, d, e) \ 
    ((void)_Generic((v), a: 0, b: 0, c: 0, d: 0, e: 0)) 
#define _VA_CHECK_TYPE_ELEM7(v, a, b, c, d, e, f) \ 
    ((void)_Generic((v), a: 0, b: 0, c: 0, d: 0, e: 0, f: 0)) 
#define _VA_CHECK_TYPE_ELEM8(v, a, b, c, d, e, f, g) \ 
    ((void)_Generic((v), a: 0, b: 0, c: 0, d: 0, e: 0, f: 0, g: 0)) 

/* For general use */ 
#define CHECK_TYPE_ANY(...) VA_NARGS_CALL_OVERLOAD(_VA_CHECK_TYPE_ELEM, __VA_ARGS__) 

使用例:

static inline _insert_item(struct List *ls, void *item) { /* function body*/ } 

/* type-checked wrapper */ 
#define insert_item(ls, item) \ 
    (CHECK_TYPE_ANY(item, struct Foo *, struct Bar *), \ 
    _insert_item(ls, item)) 

注意,使用Python腳本生成_VA_CHECK_TYPE_ELEM#

for i in range(32): 
    args = [(("" if c < 26 else "_") + chr(ord('a') + (c % 26))) for c in range(i + 1)] 
    print("#define _VA_CHECK_TYPE_ELEM%d(var, %s) \\" % (i + 2, ", ".join(args))) 
    print(" ((void)_Generic((var), %s))" % (": 0, ".join(args) + ": 0"))