我正在开发一个C ++迭代器,该迭代器在每次取消引用迭代器时都会评估lambda表达式。
因此,我实现了一个自定义的迭代器类,并重载了解除引用运算符,以这种方式在每次解除引用时执行lambda表达式,从而可以将所需的表达式放入lambda函数中。
一切正常,但是它不是类型通用的,因为在下面代码的第9行,我假设lambda表达式将返回一个double。如何获得Lambda的返回类型?我尝试了std::result_of<Lambda>
,但是没有用。
这是我写的代码:
#include <iostream>
#include <vector>
#include <utility>
#include <cmath>
template <class Iterator, class Lambda>
struct LambdaIterator: Iterator {
public:
using value_type = double; // <--- How do I get the return type of the lambda?
LambdaIterator(Iterator&& begin, Lambda&& lambda) :
Iterator(std::forward<Iterator>(begin)),
lamdaexpr(std::forward<Lambda>(lambda)) {}
value_type operator*() {
return lamdaexpr(*static_cast<Iterator&>(*this));
}
private:
Lambda lamdaexpr;
};
// Helper function, only needed to easily instantiate the iterator
template <class Iterator, class Lambda>
LambdaIterator<Iterator, Lambda> MakeLambdaIterator(Iterator&& begin, Lambda&& lambda) {
return {std::forward<Iterator>(begin), std::forward<Lambda>(lambda)};
}
template<class BeginIterator, class EndIterator>
typename BeginIterator::value_type Sum(BeginIterator begin, EndIterator end) {
typename BeginIterator::value_type sum = 0;
while (begin != end) {
sum += *begin;
begin++;
}
return sum;
}
int main() {
std::vector<double> vec;
for (int i = 0; i < 10; i++)
vec.push_back(i); // Puts some data into the vector
std::cout << Sum(vec.begin(), vec.end()) << std::endl; // Works fine
std::cout << Sum(MakeLambdaIterator(vec.begin(),
[](double item) -> double { // Works as long as it returns a double
return sqrt(item); // Prints the sum of the square roots
} ), vec.end()) << std::endl;
std::cout << Sum(MakeLambdaIterator(vec.begin(),
[](double item) -> double {
return pow(item, 2); // Prints the sum of the squares
} ), vec.end()) << std::endl;
}
答案 0 :(得分:2)
有几种方法可以实现它,最简单的方法如下:
decltype(auto) operator*() {
return lamdaexpr(*static_cast<Iterator&>(*this));
}
答案 1 :(得分:2)
如何获取Lambda的返回类型?
考虑到您知道参数的类型,在我看来,您正在decltype()
内部寻找执行。
我的意思是
using value_type = decltype( std::declval<Lambda>()(*std::declval<Iterator>()) );
这也适用于通用lambda。
主题外:如果要对LambdaIterator()
构造函数使用正向语义,则必须将其转换为模板&&
的通用引用而不是r值引用。
我的意思是:您应该这样写
template <typename I, typename L>
LambdaIterator (I && begin, L && lambda)
: Iterator{std::forward<I>(begin)}, lamdaexpr{std::forward<L>(lambda)}
{ }