获取lambda表达式的返回类型

时间:2019-01-22 19:12:21

标签: c++ lambda generic-programming return-type template-deduction

我正在开发一个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;
}

2 个答案:

答案 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)}
     { }