如何编写一个模板，可以推导类型使用函数的参数类型？

Question

void foo(int *p) {}

template<typename T, void (*F)(T*)>
struct bar
{
    bar(T* t)
    {
        F(t);
    }
}

int *p;
bar<int, foo> b(p); // both int and foo are required here

bar<foo> b(p);

Answer 1

如果您可以将C++17与它的auto模板参数一起使用（就像@n.m.在注释中所说的那样），那么您可以将它用作模板参数，然后将类型T与类型traits一起使用。

首先，我们需要标准类型特征和一个类型特征来获得一元函数（您的foo）的参数，我们可以这样编写：

#include <type_traits>

// Trait to get the argument type of a unary function
template<typename T>
struct unary_func_arg;

template<typename R, typename T>
struct unary_func_arg< R(*)(T) >
{
    using type = T;
};

如果将函数指针以外的任何内容放入其中，这将产生一个错误，因为主专门化没有声明。

template< auto F >
struct bar
{
    // Assert it's a function pointer
    static_assert( std::is_pointer_v<decltype(F)> );
    static_assert( std::is_function_v< std::remove_pointer_t<decltype(F)> > );

    // Get the parameter type
    using T = typename unary_func_arg< decltype(F) >::type;

    bar(T t)
    {
        F(t);
    }
};

int* p;
bar<foo> b(p);

// Trait to get the argument type of a unary function
template<typename T>
struct unary_func_arg_pointer;

template<typename R, typename T>
struct unary_func_arg_pointer< R(*)(T*) >
{
    using type = T;
};