如何在C ++函数中传递Lambda函数?

时间:2019-04-16 19:01:24

标签: lambda rcpp rcppparallel

我在Rccp和Rccpparallel上还很新,我很难弄清楚哪里出错了。因此,我想创建一个在矩阵中并行执行幂元素的函数。我正在关注rcppParallel示例。

在一个内核上,代码可以编译并正常工作,但是当我尝试将n传递给下面的函子时,出现了以下错误。

capture of non-variable "Power::n"
"this" was not captured for this lambda function
invalid use of non-static data member "Power::n"

如果我在下面的函子中交换n,它将编译并正常工作。我想念什么? R代码:

library(Rcpp)
library(RcppParallel)
Sys.setenv("PKG_CXXFLAGS"="-std=c++11")
sourceCpp("lambdaPower.cpp")

lambdaPower.cpp

#include <Rcpp.h>
using namespace Rcpp;

#include <cmath>
#include <algorithm>

// [[Rcpp::export]]
NumericMatrix matrixPower(NumericMatrix orig, double n)
{

    // allocate the matrix we will return
    NumericMatrix mat(orig.nrow(), orig.ncol());

    // transform it
    std::transform(orig.begin(), orig.end(), mat.begin(), [n](double x) { return pow(x, n); });

    // return the new matrix
    return mat;
}

// [[Rcpp::depends(RcppParallel)]]
#include <RcppParallel.h>
using namespace RcppParallel;

struct Power : public Worker
{
    // source matrix
    const RMatrix<double> input;

    // destination matrix
    RMatrix<double> output;

    //power
    double n;


    // initialize with source and destination
    Power(const NumericMatrix input, NumericMatrix output, double n)
        : input(input), output(output), n(n){}
    // take the n power of the range of elements requested
    void operator()(std::size_t begin, std::size_t end)
    {
        std::transform(input.begin() + begin,
                       input.begin() + end,
                       output.begin() + begin,
                       [n](double x) { return pow(x,n); }); // why n doesn work?
                    // If i swap n with fixed number it compiles and works.
                    // [](double x) { return pow(x,2); }); compiles and works
    }
};

// [[Rcpp::export]]
NumericMatrix parallelMatrixPower(NumericMatrix x, double n)
{

    // allocate the output matrix
    NumericMatrix output(x.nrow(), x.ncol());

    // power functor (pass input and output matrixes)
    Power power(x, output, n);

    // call parallelFor to do the work
    parallelFor(0, x.length(), power);

    // return the output matrix
    return output;
}

非常感谢。

1 个答案:

答案 0 :(得分:3)

如果将n复制到定义lambda的作用域,则代码会编译:

....
  void operator()(std::size_t begin, std::size_t end)
  {
    auto _n = n;
    std::transform(input.begin() + begin,
                   input.begin() + end,
                   output.begin() + begin,
                   [_n](double x) { return pow(x,_n); });
  }
....

我不太擅长对此进行解释,但是您可以阅读Scott Meyers的“ Effective Modern C ++”的“ Item 31:避免默认捕获模式”中的详细信息。

顺便说一句,我会在C ++代码中使用Sys.setenv("PKG_CXXFLAGS"="-std=c++11")代替R代码中的// [[Rcpp::plugins(cpp11)]]