使用函数参数返回类型解决模板重载

Question

我正在尝试包装几个使用C样式的“调用-分配-再次调用”模式的第三方库函数（对此应该有一个更好的名称）。 例如：

int EnumerateFoo(float f, uint32_t* count, float* buf) {
    if (!buf) {
        *count = 3;
        return 0;
    }
    if (*count < 3) {
        cout << "buffer too small\n";
        return -1;
    }
    buf[0] = f;
    buf[1] = f + f;
    buf[2] = f * f;
    return 0;
}

// ...

uint32_t count = 0;
int ret = EnumerateFoo(3.14f, &count, nullptr);
if (ret) return ret;
float* buf = new float[count];
ret = EnumerateFoo(3.14f, &count, buf);
if (ret) return ret;

我想将其包装起来，以便可以更简洁地调用此类函数。 理想情况下，我可以将其称为：

vector<float> vec = WrapEnumerate(EnumerateFoo, 3.14f);

我得到的最接近的是以下内容（使用std::bind因为count / buffer参数并不总是在相同的参数索引处）：

template<class T>
vector<T> EnumToVec(function<int(uint32_t*,T*)> fn) {
    vector<T> ret;
    uint32_t count = 0;
    if(fn(&count, nullptr))
        return vector<T>();
    ret.resize(count);
    if(fn(&count, ret.data()))
        return vector<T>();
    return ret;
}

// ...

auto vec = EnumToVec<float>(bind(
    EnumerateFoo, 3.14f, placeholder::_1, placeholder::_2));

这足够好用，但是不幸的是，一些库函数的返回类型为void而不是int 。 我尝试创建EnumToVec的重载，用function<void...替换function<int... ，但是编译器说调用是模棱两可的。

如何创建仅由function参数的return-Type消除歧义的重载？ 请注意，这不是仅基于函数返回类型的重载，我知道您不能这样做。 我知道我可以创建一个EnumToVecNoReturn替代方法，但我希望有一种更简单的方法。 我怀疑SFINAE可能会在这里申请，但是我不熟悉那些技术。

实验在这里： https ： //ideone.com/IDygru

Answer 1

问题是std::bind返回的值可以转换为std::function但不是std::function 。

这是种鸡蛋和鸡肉的问题。

你打电话时

EnumToVec<float>(std::bind(EnumerateFoo, 3.14f, _1, _2));

您可能会因为EnumVec没有收到std::function值而void编译器无法在void版本和int版本之间进行选择。 并且没有收到std::function因为编译器无法在void版本和int版本之间进行选择。

一个可能的解决方案是显式创建正确的std::function并调用EnumToVec

std::function<int(uint32_t *, float *)> ef { std::bind(EnumerateFoo, 3.14f, _1, _2) };
auto vecFoo = EnumToVec(ef);

请注意，由于可以通过ef推导，因此可以避免显式显示float模板参数。

另一个可能的解决方案是放弃std::function ，以通用类型名接收可执行文件，而SFINAE根据功能返回的类型启用/禁用这两个功能

像

template <typename T, typename F>
auto EnumToVec (F const & fn)
   -> std::enable_if_t<std::is_same_v<
         decltype(fn(std::declval<std::uint32_t*>(), std::declval<T*>())),
         int>, std::vector<T>>
 { //....^^^ int here
   vector<T> ret;
   uint32_t count = 0;
   if(fn(&count, nullptr))
      return vector<T>();
   ret.resize(count);
   if(fn(&count, ret.data()))
      return vector<T>();
   return ret;
 }


template <typename T, typename F>
auto EnumToVec (F const & fn)
   -> std::enable_if_t<std::is_same_v<
         decltype(fn(std::declval<std::uint32_t*>(), std::declval<T*>())),
         void>, std::vector<T>>
 { // ...^^^^  void here
   vector<T> ret;
   uint32_t count = 0;
   fn(&count, nullptr);
   ret.resize(count);
   fn(&count, ret.data());
   return ret;
 }

所以你可以打电话

auto vecFoo = EnumToVec<float>(std::bind(EnumerateFoo, 3.14f, _1, _2));

但由于未推断出T类型名称，因此进行了说明。