Перегрузка макроса по количеству аргументов


184

У меня есть два макроса FOO2и FOO3:

#define FOO2(x,y) ...
#define FOO3(x,y,z) ...

Я хочу определить новый макрос FOOследующим образом:

#define FOO(x,y) FOO2(x,y)
#define FOO(x,y,z) FOO3(x,y,z)

Но это не работает, потому что макросы не перегружают количество аргументов.

Без изменения FOO2и FOO3, есть какой - то способ определить макрос FOO( с использованием __VA_ARGS__или иным образом ) , чтобы получить тот же эффект , диспетчерские FOO(x,y)к FOO2, и FOO(x,y,z)к FOO3?


1
У меня очень сильное чувство, что об этом уже несколько раз спрашивали ... [обновить], например, здесь .
Kerrek SB

1
@KerrekSB: Это может быть связано, если это наверняка не обман.
Эндрю Томазос

Нет, может быть, не тот, но что-то подобное появляется примерно раз в месяц ...
Kerrek SB

То же самое для C ++: stackoverflow.com/questions/3046889/… Должно быть одинаковым, поскольку препроцессоры в основном одинаковы: stackoverflow.com/questions/5085533/…
Сиро Сантилли 郝海东 冠状 病 六四 事件 法轮功

Ответы:


265

Просто как:

#define GET_MACRO(_1,_2,_3,NAME,...) NAME
#define FOO(...) GET_MACRO(__VA_ARGS__, FOO3, FOO2)(__VA_ARGS__)

Итак, если у вас есть эти макросы:

FOO(World, !)         # expands to FOO2(World, !)
FOO(foo,bar,baz)      # expands to FOO3(foo,bar,baz)

Если вы хотите четвертый:

#define GET_MACRO(_1,_2,_3,_4,NAME,...) NAME
#define FOO(...) GET_MACRO(__VA_ARGS__, FOO4, FOO3, FOO2)(__VA_ARGS__)

FOO(a,b,c,d)          # expeands to FOO4(a,b,c,d)

Естественно, если вы определите FOO2,FOO3 и FOO4, выходные данные будут заменены теми из определенных макросов.


5
@ Uroc327 Добавление макроса с 0 аргументами в список возможно, смотрите мой ответ.
Авгурар

7
Не работает в Microsoft Visual Studio 2010, VA_ARGS, кажется, расширен в один аргумент макроса.
Этьен

9
Нашел этот ответ, чтобы заставить его работать под MSVC 2010.
Этьен

8
Если кто-то запутался в том, как использовать EXPANDупомянутое в ссылке @ Étienne, вы в основном вызываете его GET_MACROпримерно так, #define FOO(...) EXPAND(GET_MACRO(__VA_ARGS__, FOO3, FOO2, FOO1)(__VA_ARGS__))и он должен расшириться до нужного числа аргументов в msvc.
vexe

3
Обратите внимание , что на C ++ 11, вы получите предупреждение: ISO C++11 requires at least one argument for the "..." in a variadic macro. Чтобы это исправить, добавьте неиспользуемый аргумент (или даже просто запятую) после последнего параметра в определении FOO (...): #define FOO(...) GET_MACRO(__VA_ARGS__, FOO3, FOO2, UNUSED)(__VA_ARGS__)( смотрите, как он работает на Coliru ).
металл

49

Чтобы добавить ответ netcoder , вы МОЖЕТЕ сделать это с помощью макроса с 0 аргументами с помощью ##__VA_ARGS__расширения GCC :

#define GET_MACRO(_0, _1, _2, NAME, ...) NAME
#define FOO(...) GET_MACRO(_0, ##__VA_ARGS__, FOO2, FOO1, FOO0)(__VA_ARGS__)

1
Можно ли разрешить FOO1 и FOO2, но не FOO0, не делая #define FOO0 _Pragma("error FOO0 not allowed")?
noɥʇʎԀʎzɐɹƆ

FOO0не работает в qt + mingw32, вызов FOO0вызоветFOO1
JustWe

Очень перспективно и просто. Но не работает для FOO0 с -std = c ++ 11 ... :-(
leonp

1
Та же проблема, если вы делаете это в C и пытаетесь использовать -std=c99или -std=c11. Вы должны использовать -std=gnu99или -std=gnu11вместо этого
Майкл Мрозек

1
Похоже, что замена _0, ##__VA_ARGS__на _0 __VA_OPT__(,) __VA_ARGS__это новый способ сделать это.
Wrzlprmft

36

Вот более общее решение:

// get number of arguments with __NARG__
#define __NARG__(...)  __NARG_I_(__VA_ARGS__,__RSEQ_N())
#define __NARG_I_(...) __ARG_N(__VA_ARGS__)
#define __ARG_N( \
      _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,_33,_34,_35,_36,_37,_38,_39,_40, \
     _41,_42,_43,_44,_45,_46,_47,_48,_49,_50, \
     _51,_52,_53,_54,_55,_56,_57,_58,_59,_60, \
     _61,_62,_63,N,...) N
#define __RSEQ_N() \
     63,62,61,60,                   \
     59,58,57,56,55,54,53,52,51,50, \
     49,48,47,46,45,44,43,42,41,40, \
     39,38,37,36,35,34,33,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

// general definition for any function name
#define _VFUNC_(name, n) name##n
#define _VFUNC(name, n) _VFUNC_(name, n)
#define VFUNC(func, ...) _VFUNC(func, __NARG__(__VA_ARGS__)) (__VA_ARGS__)

// definition for FOO
#define FOO(...) VFUNC(FOO, __VA_ARGS__)

Определите свои функции:

#define FOO2(x, y) ((x) + (y))
#define FOO3(x, y, z) ((x) + (y) + (z))

// it also works with C functions:
int FOO4(int a, int b, int c, int d) { return a + b + c + d; }

Теперь вы можете использовать FOOс 2, 3 и 4 аргументами:

FOO(42, 42) // will use makro function FOO2
FOO(42, 42, 42) // will use makro function FOO3
FOO(42, 42, 42, 42) // will call FOO4 function

Ограничения

  • Только до 63 аргументов (но расширяемых)
  • Функция без аргументов только в GCC возможна

идеи

Используйте его для аргументов по умолчанию:

#define func(...) VFUNC(func, __VA_ARGS__)
#define func2(a, b) func4(a, b, NULL, NULL)
#define func3(a, b, c) func4(a, b, c, NULL)

// real function:
int func4(int a, int b, void* c, void* d) { /* ... */ }

Используйте его для функций с возможным бесконечным числом аргументов:

#define SUM(...) VFUNC(SUM, __VA_ARGS__)
#define SUM2(a, b) ((a) + (b))
#define SUM3(a, b, c) ((a) + (b) + (c))
#define SUM4(a, b, c) ((a) + (b) + (c) + (d))
// ...

PS: __NARG__скопировано с Laurent Deniau & Roland Illig здесь: https://groups.google.com/group/comp.std.c/browse_thread/thread/77ee8c8f92e4a3fb/346fc464319b1ee5?pli=1




Макрос __NARG_I_кажется совершенно ненужным и лишним. Это просто добавляет дополнительный шаг и путаницу. Я рекомендую удалить его полностью и просто определение , __NARG__а не как: #define __NARG__(...) __ARG_N(__VA_ARGS__,__RSEQ_N()).
Габриэль Стейплз

Или это как-то нарушит предварительную обработку? Я что-то упускаю?
Габриэль Стейплз

То же самое с _VFUNC_: просто удалите это. Затем определите _VFUNCкак: #define _VFUNC(name, n) name##nвместо #define _VFUNC(name, n) _VFUNC_(name, n).
Габриэль Стейплз

15

Я только что исследовал это сам, и я столкнулся с этим здесь . Автор добавил поддержку аргументов по умолчанию для функций C через макросы.

Я постараюсь кратко подвести итог статьи. По сути, вам нужно определить макрос, который может считать аргументы. Этот макрос вернет 2, 1, 0 или любой другой диапазон аргументов, которые он может поддерживать. Например:

#define _ARG2(_0, _1, _2, ...) _2
#define NARG2(...) _ARG2(__VA_ARGS__, 2, 1, 0)

При этом вам нужно создать другой макрос, который принимает переменное число аргументов, считает аргументы и вызывает соответствующий макрос. Я взял твой пример макроса и соединил его с примером статьи. У меня есть функция вызова FOO1 a () и функция вызова FOO2 a с аргументом b (очевидно, я предполагаю C ++ здесь, но вы можете изменить макрос на что угодно).

#define FOO1(a) a();
#define FOO2(a,b) a(b);

#define _ARG2(_0, _1, _2, ...) _2
#define NARG2(...) _ARG2(__VA_ARGS__, 2, 1, 0)

#define _ONE_OR_TWO_ARGS_1(a) FOO1(a)
#define _ONE_OR_TWO_ARGS_2(a, b) FOO2(a,b)

#define __ONE_OR_TWO_ARGS(N, ...) _ONE_OR_TWO_ARGS_ ## N (__VA_ARGS__)
#define _ONE_OR_TWO_ARGS(N, ...) __ONE_OR_TWO_ARGS(N, __VA_ARGS__)

#define FOO(...) _ONE_OR_TWO_ARGS(NARG2(__VA_ARGS__), __VA_ARGS__)

Так что если у вас есть

FOO(a)
FOO(a,b)

Препроцессор расширяет это до

a();
a(b);

Я определенно прочитал бы статью, которую я связал. Это очень информативно, и он упоминает, что NARG2 не будет работать на пустых аргументах. Он следует за этим здесь .


7

Вот более компактная версия ответа выше . С примером.

#include <iostream>
using namespace std;

#define OVERLOADED_MACRO(M, ...) _OVR(M, _COUNT_ARGS(__VA_ARGS__)) (__VA_ARGS__)
#define _OVR(macroName, number_of_args)   _OVR_EXPAND(macroName, number_of_args)
#define _OVR_EXPAND(macroName, number_of_args)    macroName##number_of_args

#define _COUNT_ARGS(...)  _ARG_PATTERN_MATCH(__VA_ARGS__, 9,8,7,6,5,4,3,2,1)
#define _ARG_PATTERN_MATCH(_1,_2,_3,_4,_5,_6,_7,_8,_9, N, ...)   N


//Example:
#define ff(...)     OVERLOADED_MACRO(ff, __VA_ARGS__)
#define ii(...)     OVERLOADED_MACRO(ii, __VA_ARGS__)

#define ff3(c, a, b) for (int c = int(a); c < int(b); ++c)
#define ff2(c, b)   ff3(c, 0, b)

#define ii2(a, b)   ff3(i, a, b)
#define ii1(n)      ii2(0, n)


int main() {
    ff (counter, 3, 5)
        cout << "counter = " << counter << endl;
    ff (abc, 4)
        cout << "abc = " << abc << endl;
    ii (3)
        cout << "i = " << i << endl;
    ii (100, 103)
        cout << "i = " << i << endl;


    return 0;
}

Бегать:

User@Table 13:06:16 /c/T
$ g++ test_overloaded_macros.cpp 

User@Table 13:16:26 /c/T
$ ./a.exe
counter = 3
counter = 4
abc = 0
abc = 1
abc = 2
abc = 3
i = 0
i = 1
i = 2
i = 100
i = 101
i = 102

Обратите внимание , что наличие как _OVRи _OVR_EXPANDможет выглядеть излишним, но это необходимо для препроцессора расширить _COUNT_ARGS(__VA_ARGS__)часть, которая в противном случае трактуется как строка.


Мне нравится это решение. Можно ли изменить его для обработки перегруженного макроса, который принимает нулевые аргументы?
Андрей


2

Вот ответ от Евгения Сергеева. Этот поддерживает нулевые перегрузки аргументов !

Я проверил это с GCC и MinGW. Он должен работать со старыми и новыми версиями C ++. Обратите внимание, что я не гарантирую это для MSVC ... Но с некоторыми изменениями, я уверен, что это может быть сделано для работы с этим тоже.

Я также отформатировал это, чтобы вставить в файл заголовка (который я назвал macroutil.h). Если вы сделаете это, вы можете просто включить этот заголовок в любую нужную вам функцию, а не смотреть на злобу, связанную с реализацией.

#ifndef MACROUTIL_H
#define MACROUTIL_H

//-----------------------------------------------------------------------------
// OVERLOADED_MACRO
//
// used to create other macros with overloaded argument lists
//
// Example Use:
// #define myMacro(...) OVERLOADED_MACRO( myMacro, __VA_ARGS__ )
// #define myMacro0() someFunc()
// #define myMacro1( arg1 ) someFunc( arg1 )
// #define myMacro2( arg1, arg2 ) someFunc( arg1, arg2 )
//
// myMacro();
// myMacro(1);
// myMacro(1,2);
//
// Note the numerical suffix on the macro names,
// which indicates the number of arguments.
// That is the REQUIRED naming convention for your macros.
//
//-----------------------------------------------------------------------------

// OVERLOADED_MACRO
// derived from: /programming/11761703/overloading-macro-on-number-of-arguments
// replaced use of _COUNT_ARGS macro with VA_NUM_ARGS defined below
// to support of zero argument overloads
#define OVERLOADED_MACRO(M, ...) _OVR(M, VA_NUM_ARGS(__VA_ARGS__)) (__VA_ARGS__)
#define _OVR(macroName, number_of_args)   _OVR_EXPAND(macroName, number_of_args)
#define _OVR_EXPAND(macroName, number_of_args)    macroName##number_of_args
//#define _COUNT_ARGS(...)  _ARG_PATTERN_MATCH(__VA_ARGS__, 15,14,13,12,11,10,9,8,7,6,5,4,3,2,1)
#define _ARG_PATTERN_MATCH(_1,_2,_3,_4,_5,_6,_7,_8,_9,_10,_11,_12,_13,_14,_15, N, ...)   N

// VA_NUM_ARGS
// copied from comments section of:
// http://efesx.com/2010/07/17/variadic-macro-to-count-number-of-arguments/
// which itself was derived from:
// https://gustedt.wordpress.com/2010/06/08/detect-empty-macro-arguments/
#define _ARG16(_0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, ...) _15
#define HAS_COMMA(...) _ARG16(__VA_ARGS__, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0)
#define HAS_NO_COMMA(...) _ARG16(__VA_ARGS__, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1)
#define _TRIGGER_PARENTHESIS_(...) ,

#define HAS_ZERO_OR_ONE_ARGS(...) \
    _HAS_ZERO_OR_ONE_ARGS( \
    /* test if there is just one argument, eventually an empty one */ \
    HAS_COMMA(__VA_ARGS__), \
    /* test if _TRIGGER_PARENTHESIS_ together with the argument adds a comma */ \
    HAS_COMMA(_TRIGGER_PARENTHESIS_ __VA_ARGS__), \
    /* test if the argument together with a parenthesis adds a comma */ \
    HAS_COMMA(__VA_ARGS__ (~)), \
    /* test if placing it between _TRIGGER_PARENTHESIS_ and the parenthesis adds a comma */ \
    HAS_COMMA(_TRIGGER_PARENTHESIS_ __VA_ARGS__ (~)) \
    )

#define PASTE5(_0, _1, _2, _3, _4) _0 ## _1 ## _2 ## _3 ## _4
#define _HAS_ZERO_OR_ONE_ARGS(_0, _1, _2, _3) HAS_NO_COMMA(PASTE5(_IS_EMPTY_CASE_, _0, _1, _2, _3))
#define _IS_EMPTY_CASE_0001 ,

#define _VA0(...) HAS_ZERO_OR_ONE_ARGS(__VA_ARGS__)
#define _VA1(...) HAS_ZERO_OR_ONE_ARGS(__VA_ARGS__)
#define _VA2(...) 2
#define _VA3(...) 3
#define _VA4(...) 4
#define _VA5(...) 5
#define _VA6(...) 6
#define _VA7(...) 7
#define _VA8(...) 8
#define _VA9(...) 9
#define _VA10(...) 10
#define _VA11(...) 11
#define _VA12(...) 12
#define _VA13(...) 13
#define _VA14(...) 14
#define _VA15(...) 15
#define _VA16(...) 16

#define VA_NUM_ARGS(...) VA_NUM_ARGS_IMPL(__VA_ARGS__, PP_RSEQ_N(__VA_ARGS__) )
#define VA_NUM_ARGS_IMPL(...) VA_NUM_ARGS_N(__VA_ARGS__)

#define VA_NUM_ARGS_N( \
    _1, _2, _3, _4, _5, _6, _7, _8, _9,_10, \
    _11,_12,_13,_14,_15,_16,N,...) N

#define PP_RSEQ_N(...) \
    _VA16(__VA_ARGS__),_VA15(__VA_ARGS__),_VA14(__VA_ARGS__),_VA13(__VA_ARGS__), \
    _VA12(__VA_ARGS__),_VA11(__VA_ARGS__),_VA10(__VA_ARGS__), _VA9(__VA_ARGS__), \
    _VA8(__VA_ARGS__),_VA7(__VA_ARGS__),_VA6(__VA_ARGS__),_VA5(__VA_ARGS__), \
    _VA4(__VA_ARGS__),_VA3(__VA_ARGS__),_VA2(__VA_ARGS__),_VA1(__VA_ARGS__), \
    _VA0(__VA_ARGS__)

//-----------------------------------------------------------------------------

#endif // MACROUTIL_H

2

Похоже, это хорошо работает на GCC, Clang и MSVC. Это исправленная версия некоторых ответов здесь

#define _my_BUGFX(x) x

#define _my_NARG2(...) _my_BUGFX(_my_NARG1(__VA_ARGS__,_my_RSEQN()))
#define _my_NARG1(...) _my_BUGFX(_my_ARGSN(__VA_ARGS__))
#define _my_ARGSN(_1,_2,_3,_4,_5,_6,_7,_8,_9,_10,N,...) N
#define _my_RSEQN() 10,9,8,7,6,5,4,3,2,1,0

#define _my_FUNC2(name,n) name ## n
#define _my_FUNC1(name,n) _my_FUNC2(name,n)
#define GET_MACRO(func,...) _my_FUNC1(func,_my_BUGFX(_my_NARG2(__VA_ARGS__))) (__VA_ARGS__)

#define FOO(...) GET_MACRO(FOO,__VA_ARGS__)

1
@RianQuinn Как настроить этот макрос так, чтобы он работал с нулевым аргументом #define func0() foo? Текущая версия, к сожалению, не обрабатывает этот случай.
Джерри Ма
Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.