[英]Determining number of arguments of C++ template functor
假設我們有:
template<typename F, typename T1, typename T2>
void my_magic_method(F func, T1 t1, T2 t2)
{
if (???)
func(t1);
else
func(t1,t2);
}
什么可以幫助我確定:
參數數量
也許每個參數的類型
返回值類型
由於MSVS 2010,我無法使用可變參數模板...
更新
我的第一個解決方案:
template<typename F>
auto my_magic_func(F f) -> decltype(f(1))
{
return f(1);
}
template<typename F>
auto my_magic_func(F f, void * fake = NULL) -> decltype(f(2,3))
{
return f(2,3);
}
int main()
{
auto x1 = my_magic_func([](int a){ return a+100; });
auto x2 = my_magic_func([](int a, int b){ return a*b; });
// x1 == 1+100
// x2 == 2*3
}
那就是我的函數類型重載的方式。 它有效,但也許是更好的解決方案?
並非完全符合您的要求,但是如果我正確理解了您的意圖,則在VC ++ 2010中,可以通過基於Arity的簡單重載來實現(但很難做到):
#include <utility>
#include <string>
#include <iostream>
template<typename F, typename T1>
auto my_magic_method(F&& func, T1&& t1) ->
decltype(std::forward<F>(func)(std::forward<T1>(t1)))
{
return std::forward<F>(func)(std::forward<T1>(t1));
}
template<typename F, typename T1, typename T2>
auto my_magic_method(F&& func, T1&& t1, T2&& t2) ->
decltype(std::forward<F>(func)(std::forward<T1>(t1), std::forward<T2>(t2)))
{
return std::forward<F>(func)(std::forward<T1>(t1), std::forward<T2>(t2));
}
struct string_to_float_functor
{
float operator ()(std::string const& s) const
{
return std::stof(s);
}
};
int main()
{
auto a = my_magic_method([](std::string const& x) { return x + x; }, "foo");
auto b = my_magic_method([](double x, int y) { return x * y; }, 21.5, 3);
auto c = my_magic_method(string_to_float_functor(), "3.14159265");
std::cout << a << '\n' << b << '\n' << c << '\n';
}
這支持一元和二進制函子–繼續該模式,並根據需要為其他arities添加重載。
這里有一些方法; 都假設C ++ 11。 使用-std = c ++ 11在clang ++ 3.2上進行了測試。
//Taking a function pointer argument (template types inferred)
template <typename ret, typename ... Args>
constexpr int arg_count(ret (*f)(Args...)) {
return sizeof...(Args);
}
//Taking a function type (or related) directly
template <typename T>
struct ArgCount {
static const int value = 0;
};
template <typename Ret, typename ... Args>
struct ArgCount<Ret(Args...)> {
static const int value = sizeof...(Args);
};
template <typename Ret, typename ... Args>
struct ArgCount<Ret(*)(Args...)> {
static const int value = sizeof...(Args);
};
template <typename Ret, typename ... Args>
struct ArgCount<Ret(&)(Args...)> {
static const int value = sizeof...(Args);
};
//Using the latter for dispatch
template <int N>
struct helper {
template<typename F, typename T1, typename T2>
static void call(F func, T1 t1, T2 t2);
};
template <>
struct helper<1> {
template<typename F, typename T1, typename T2>
static void call(F func, T1 t1, T2 t2) {
func(t1);
}
};
template <>
struct helper<2> {
template<typename F, typename T1, typename T2>
static void call(F func, T1 t1, T2 t2) {
func(t1, t2);
}
};
template<typename F, typename T1, typename T2>
void my_magic_method(F func, T1 t1, T2 t2)
{
helper<ArgCount<F>::value>::call(func, t1, t2);
}
//Testing
#include <cstdio>
void a(int a, int b) { printf("%i\n", a + b); }
void b(int x) { printf("%i\n", x); }
int main() {
printf("%i %i\n", arg_count(a), arg_count(b));
printf("%i %i\n", ArgCount<decltype(a)>::value, ArgCount<decltype(b)>::value);
my_magic_method(a, 1, 2);
my_magic_method(b, 1, 2);
}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.