'make_table' helper for creating table from SEXP

kevinushey · kevinushey · commit c5ad3c2089c8 · 2016-01-28T14:54:52.000-08:00
diff --git a/inst/include/RcppNT2/convert/as.h b/inst/include/RcppNT2/convert/as.h
@@ -2,5 +2,4 @@
 #define RCPP_NT2_CONVERT_AS_H
 
 
-
 #endif /* RCPP_NT2_CONVERT_AS_H */
diff --git a/inst/include/RcppNT2/core/core.h b/inst/include/RcppNT2/core/core.h
@@ -1,6 +1,8 @@
 #ifndef RCPP_NT2_CORE_CORE_H
 #define RCPP_NT2_CORE_CORE_H
 
+#include <RcppNT2/core/create.h>
+
 namespace RcppNT2 {
 
 template <typename T>
diff --git a/inst/include/RcppNT2/core/create.h b/inst/include/RcppNT2/core/create.h
@@ -0,0 +1,54 @@
+#ifndef RCPP_NT2_CORE_CREATE_H
+#define RCPP_NT2_CORE_CREATE_H
+
+#define R_NO_REMAP
+#include <R.h>
+#include <Rinternals.h>
+
+namespace RcppNT2 {
+
+template <typename T, int RTYPE>
+nt2::table<T> make_table_vector_impl(SEXP data)
+{
+  int n = Rf_length(data);
+
+  T* ptr = reinterpret_cast<T*>(STRING_PTR(data));
+  return nt2::table<T>(nt2::of_size(n, 1), ptr, ptr + n);
+}
+
+template <typename T, int RTYPE>
+nt2::table<T> make_table_matrix_impl(SEXP data)
+{
+  int nr = Rf_nrows(data);
+  int nc = Rf_ncols(data);
+  int n = Rf_length(data);
+
+  T* ptr = reinterpret_cast<T*>(STRING_PTR(data));
+  return nt2::table<T>(nt2::of_size(nr, nc), ptr, ptr + n);
+}
+
+template <typename T, int RTYPE>
+nt2::table<T> make_table_impl(SEXP data)
+{
+  if (Rf_isMatrix(data))
+    return make_table_matrix_impl<T, RTYPE>(data);
+  else
+    return make_table_vector_impl<T, RTYPE>(data);
+}
+
+template <typename T>
+nt2::table<T> make_table(SEXP data)
+{
+  switch (TYPEOF(data))
+  {
+  case REALSXP: return make_table_impl<double, REALSXP>(data);
+  // case INTSXP:  return make_table_impl<int,    INTSXP>(data); // TODO: fails to compile?
+  default:
+    Rf_warning("can't create an 'nt2::table' from type '%s'", Rf_type2char(TYPEOF(data)));
+  return nt2::table<T>();
+  }
+}
+
+} // namespace RcppNT2
+
+#endif /* RCPP_NT2_CORE_CREATE_H */
diff --git a/inst/include/RcppNT2/traits/traits.h b/inst/include/RcppNT2/traits/traits.h
@@ -0,0 +1,8 @@
+#ifndef RCPP_NT2_TRAITS_TRAITS_h
+#define RCPP_NT2_TRAITS_TRAITS_h
+
+template <int RTYPE> struct storage_type {};
+template <> struct storage_type<REALSXP> { typedef double type; };
+template <> struct storage_type<INTSXP>  { typedef int type; };
+
+#endif /* RCPP_NT2_TRAITS_TRAITS_h */

Original file line number	Diff line number	Diff line change
`@@ -2,5 +2,4 @@`
`2`	`2`	`#define RCPP_NT2_CONVERT_AS_H`
`3`	`3`
`4`	`4`
`5`		`-`
`6`	`5`	`#endif /* RCPP_NT2_CONVERT_AS_H */`