时间戳

首页 > 解决方案 > 使用函数参数返回类型解决模板重载

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

问题描述

我正在尝试包装几个使用 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...with的重载，function<void...但编译器说该调用不明确。

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

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

标签： c++templatessfinaestd-functiontemplate-argument-deduction

解决方案

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

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

你打电话时

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

您认为编译器无法在void和int版本之间进行选择，因为EnumVec没有收到std::function值；并且没有收到 astd::function因为编译器无法在 avoid和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类型名，因为没有推导出来。

推荐阅读