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

时间:2019-06-27 23:38:56

标签: c++ templates sfinae std-function template-deduction

我正在尝试包装几个使用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<int...替换function<void...,但是编译器说调用不明确。

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

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

1 个答案:

答案 0 :(得分:0)

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

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

致电时

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

您有一个编译器无法在voidint版本之间进行选择的原因,因为EnumVec没有收到std::function的值。并且没有收到std::function,因为编译器无法在voidint版本之间进行选择。

可能的解决方案是显式创建正确的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));

但由于未推导而{@ {1}}类型名重复使用。