宏内的变量参数

Question

我有两个函数 foo1(a,b) & foo2(a,b,c) 和一个宏

#define add(a,b) foo(a,b)

我需要重新定义宏来完成，

1.如果用2个参数调用add()，则调用foo1

1. 如果使用 3 个参数调用 add()，则调用 foo2

我是选项 VA_ARGS 的新手。我该怎么做呢

Answer 1

如果你只是想区分两个功能，下面的工作：

#define ADD(_1, _2, _3, X, ...) X
#define add(...) ADD(__VA_ARGS__, add3, add2, 0)(__VA_ARGS__)

辅助宏ADD 总是选择第四个参数：

add(a, b)    --> ADD(a, b, add3, add2, 0)    --> add2
add(a, b, c) --> ADD(a, b, c, add3, add2, 0) --> add3

缺点是当您不向函数提供两个或三个参数时，您会收到非常神秘的错误消息。

与可变参数函数相比，优势在于您获得了类型安全性。例如，如果您的函数在doubles 上运行，您仍然可以说add(1, 2)，并且整数参数将转换为doubles。可变参数函数需要一些关于实际参数数量的额外信息，所以这不是一个可行的解决方案，除非你在函数中指定被加数的数量。

附录：我更改了add 宏，使其不会将空的可变参数列表传递给ADD。一些编译器允许空列表，但它不是标准 C。

Answer 2

usual trick for counting arguments 可能适用于此：

#define ADD_EXPAND(...) \
        ADD_EXPAND_(__VA_ARGS__, EXPAND_ADD_FUNCS())

#define ADD_EXPAND_(...) \
        EXPAND_ADD_SEL_FUNC(__VA_ARGS__)

#define EXPAND_ADD_SEL_FUNC(first_, second_, third_, func, ...) func

#define EXPAND_ADD_FUNCS() foo2, foo, dummy

#define add(...) ADD_EXPAND(__VA_ARGS__)(__VA_ARGS__)

一旦你通过样板，它基本上只需将所有参数放在一行中，然后是函数标记，然后查看哪个函数突出。这就是EXPAND_ADD_SEL_FUNC 所做的。

你可以看到它live on coliru。

但我会重申我们在 cmets 中告诉你的内容。这可能是对适当功能的低于标准的解决方案。我还没有彻底测试过，所以很容易损坏。使用风险自负。

Answer 3

如果您必须使用可变参数宏，那么这里有一个技巧。

#define add(...) _Generic ( &(int[]){__VA_ARGS__}, \
                            int(*)[2]: add2,       \
                            int(*)[3]: add3) (__VA_ARGS__)

• 让宏创建一个复合文字数组。该数组的大小取决于参数的数量。
• 获取复合字面量的地址，获取数组指针类型。
• 让_Generic检查你得到的类型，然后根据它调用正确的函数。

这是 100% 标准的 C 语言，而且类型安全。

---

演示：

#include <stdio.h>


#define add(...) _Generic ( &(int[]){__VA_ARGS__}, \
                            int(*)[2]: add2,       \
                            int(*)[3]: add3) (__VA_ARGS__)

int add2 (int a, int b);
int add3 (int a, int b, int c);

int main (void)
{
  printf("%d\n", add(1, 2));
  printf("%d\n", add(1, 2, 3));
//printf("%d\n", add(1, 2, 3, 4)); Compiler error for this.
}


int add2 (int a, int b)
{
  return a + b;
}

int add3 (int a, int b, int c)
{
  return a + b + c;
}