copy complex numbers/vectors/matrices from R to C++ and viceversa - replaces #427

Author: pachadotdev

cpp11test/src/grow.cpp

@@ -1,3 +1,4 @@
+#include "cpp11/complexes.hpp"
 #include "cpp11/doubles.hpp"
 
 [[cpp11::register]] cpp11::writable::doubles grow_(R_xlen_t n) {
@@ -9,3 +10,14 @@
 
   return x;
 }
+
+[[cpp11::register]] cpp11::writable::complexes grow_cplx_(R_xlen_t n) {
+  cpp11::writable::complexes x;
+  R_xlen_t i = 0;
+  while (i < n) {
+    x.push_back(std::complex<double>(i, i));
+    i++;
+  }
+
+  return x;
+}

cpp11test/src/sum.cpp

@@ -1,4 +1,5 @@
 #include <numeric>
+#include "cpp11/complexes.hpp"
 #include "cpp11/doubles.hpp"
 
 [[cpp11::register]] double sum_dbl_for_(cpp11::doubles x) {
@@ -58,3 +59,111 @@
   const cpp11::doubles x(x_sxp, false);
   return std::accumulate(x.cbegin(), x.cend(), 0.);
 }
+
+// Pacha: Functions for complex data type
+
+[[cpp11::register]] cpp11::r_complex sum_cplx_for_(cpp11::complexes x) {
+  std::complex<double> sum = {0.0, 0.0};
+  R_xlen_t n = x.size();
+  for (R_xlen_t i = 0; i < n; ++i) {
+    // sum.real(sum.real() + x[i].real());
+    // sum.imag(sum.imag() + x[i].imag());
+    sum += x[i];
+  }
+
+  return cpp11::r_complex(sum.real(), sum.imag());
+}
+
+[[cpp11::register]] cpp11::complexes sum_cplx_for_2_(cpp11::complexes x) {
+  std::complex<double> sum = {0.0, 0.0};
+  R_xlen_t n = x.size();
+  for (R_xlen_t i = 0; i < n; ++i) {
+    // sum.real(sum.real() + x[i].real());
+    // sum.imag(sum.imag() + x[i].imag());
+    sum += x[i];
+  }
+
+  cpp11::writable::complexes result(1);
+  // result[0] = cpp11::r_complex(sum.real(), sum.imag());
+  result[0] = sum;
+
+  return result;
+}
+
+[[cpp11::register]] std::complex<double> sum_cplx_for_3_(cpp11::complexes x_sxp) {
+  std::complex<double> sum = {0.0, 0.0};
+  const cpp11::complexes x(x_sxp, false);
+  R_xlen_t n = x.size();
+  for (R_xlen_t i = 0; i < n; ++i) {
+    // sum.real(sum.real() + x[i].real());
+    // sum.imag(sum.imag() + x[i].imag());
+    sum += x[i];
+  }
+
+  return sum;
+}
+
+[[cpp11::register]] std::complex<double> sum_cplx_for_4_(SEXP x_sxp) {
+  std::complex<double> sum = {0.0, 0.0};
+  const cpp11::complexes x(x_sxp, false);
+  R_xlen_t n = x.size();
+  for (R_xlen_t i = 0; i < n; ++i) {
+    // sum.real(sum.real() + x[i].real());
+    // sum.imag(sum.imag() + x[i].imag());
+    sum += x[i];
+  }
+
+  return sum;
+}
+
+[[cpp11::register]] SEXP sum_cplx_for_5_(SEXP x_sxp) {
+  std::complex<double> sum = {0.0, 0.0};
+  const cpp11::complexes x(x_sxp, false);
+  R_xlen_t n = x.size();
+  for (R_xlen_t i = 0; i < n; ++i) {
+    // sum.real(sum.real() + x[i].real());
+    // sum.imag(sum.imag() + x[i].imag());
+    sum += x[i];
+  }
+
+  return cpp11::as_sexp(sum);
+}
+
+[[cpp11::register]] cpp11::complexes sum_cplx_for_6_(SEXP x_sxp) {
+  std::complex<double> sum = {0.0, 0.0};
+  const cpp11::complexes x(x_sxp, false);
+  R_xlen_t n = x.size();
+  for (R_xlen_t i = 0; i < n; ++i) {
+    // sum.real(sum.real() + x[i].real());
+    // sum.imag(sum.imag() + x[i].imag());
+    sum += x[i];
+  }
+
+  return cpp11::as_sexp(sum);
+}
+
+[[cpp11::register]] std::complex<double> sum_cplx_foreach_(cpp11::complexes x) {
+  std::complex<double> sum = {0.0, 0.0};
+  for (const auto&& val : x) {
+    // sum.real(sum.real() + val.real());
+    // sum.imag(sum.imag() + val.imag());
+    sum += val;
+  }
+
+  return sum;
+}
+
+[[cpp11::register]] std::complex<double> sum_cplx_accumulate_(cpp11::complexes x) {
+  return std::accumulate(x.cbegin(), x.cend(), std::complex<double>(0.0, 0.0));
+}
+
+[[cpp11::register]] std::complex<double> sum_cplx_for2_(SEXP x_sxp) {
+  std::complex<double> sum = {0.0, 0.0};
+  const cpp11::complexes x(x_sxp);
+  R_xlen_t n = x.size();
+  for (R_xlen_t i = 0; i < n; ++i) {
+    sum += x[i];
+  }
+
+  return sum;
+}