[Rcpp-devel] NA_REAL in RcppParallel

Hadley Wickham h.wickham at gmail.com
Thu Jun 2 13:30:07 CEST 2016


NA_REAL is just a NaN with a special value in the payload, so it's fine to use.
Hadley

On Thu, Jun 2, 2016 at 7:07 AM, Jason Foster <jason.j.foster at gmail.com> wrote:
> Hi,
>
> My question is how can I return a thread safe NA_REAL using RcppParallel?
> I've read many posts about how it's not thread safe to call R or Rcpp APIs
> when using RcppParallel
> (http://lists.r-forge.r-project.org/pipermail/rcpp-devel/2016-January/009061.html),
> so in the past I've played it safe by returning NAN in C++ instead (as it's
> my understanding NA_REAL is specific to R).
>
> The following example is a modified version of "Transforming a Matrix in
> Parallel using RcppParallel"
> (http://gallery.rcpp.org/articles/parallel-matrix-transform/) that checks
> whether the input value is NA before taking the sqrt, otherwise NA is
> returned (note: this is the desired behavior and for illustrative purposes
> only):
>
> // [[Rcpp::depends(RcppParallel)]]
> #include <Rcpp.h>
> #include <RcppParallel.h>
> using namespace Rcpp;
> using namespace RcppParallel;
>
> struct SquareRoot : public Worker
> {
>   // source vector
>   const RVector<double> input;
>
>   // destination vector
>   RVector<double> output;
>
>   // initialize with source and destination
>   SquareRoot(const NumericVector input, NumericVector output)
>     : input(input), output(output) {}
>
>   // take the square root of the range of elements requested
>   void operator()(std::size_t begin, std::size_t end) {
>     for (std::size_t i = begin; i < end; i++) {
>       if (std::isnan(input[i])) {
>         output[i] = NA_REAL; // this is the desired output (instead of NAN)
>       } else {
>         output[i] = sqrt(input[i]);
>       }
>     }
>   }
> };
>
> // [[Rcpp::export]]
> NumericVector parallelVectorSqrt(NumericVector x) {
>
>   // allocate the output vector
>   NumericVector output(x.size());
>
>   // SquareRoot functor (pass input and output vectors)
>   SquareRoot squareRoot(x, output);
>
>   // call parallelFor to do the work
>   parallelFor(0, x.size(), squareRoot);
>
>   // return the output vector
>   return output;
>
> }
>
> Any guidance or insight would be much appreciated. Thanks!
>
>
>
>
>
> _______________________________________________
> Rcpp-devel mailing list
> Rcpp-devel at lists.r-forge.r-project.org
> https://lists.r-forge.r-project.org/cgi-bin/mailman/listinfo/rcpp-devel



-- 
http://hadley.nz


More information about the Rcpp-devel mailing list