我正在尝试包装几个使用 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 可能适用于此,但我不熟悉这些技术。
最佳答案
问题是 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
类型名称,因为未推导。
关于c++ - 使用函数参数返回类型解决模板重载,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56799505/