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

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 因为计数/缓冲区参数并不总是在相同的参数索引处):

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<int...与 function<void...但是编译器说调用不明确。

如何创建仅由函数参数的返回类型消除歧义的重载？ 请注意，这与仅基于函数返回类型的重载无关，我知道你做不到。 我知道我可以创建一个 EnumToVecNoReturn备用，但我希望有一个更简单的方法。我怀疑 SFINAE 可能适用于此，但我不熟悉这些技术。

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

Answer 1

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

这是一种先有鸡还是先有蛋的问题。

当你打电话时

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

编译器无法在 void 和 int 版本之间进行选择，因为 EnumVec 没有收到 std::function 值；并且没有收到 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);

请注意，您可以避免显式显示 float 模板参数，因为可以通过 ef 推导出来。

另一种可能的解决方案是放弃 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 类型名称，因为未推导。