Extract sheet names

tidyverse · Mar 16, 2015 · 6def875 · 6def875
1 parent 9530301
commit 6def875
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 0 deletions.
diff --git a/R/RcppExports.R b/R/RcppExports.R
@@ -9,3 +9,7 @@ xls_formats <- function(path) {
     .Call('exell_xls_formats', PACKAGE = 'exell', path)
 }
 
+xls_sheets <- function(path) {
+    .Call('exell_xls_sheets', PACKAGE = 'exell', path)
+}
+
diff --git a/src/RcppExports.cpp b/src/RcppExports.cpp
@@ -26,3 +26,14 @@ BEGIN_RCPP
     return __result;
 END_RCPP
 }
+// xls_sheets
+std::vector<std::string> xls_sheets(std::string path);
+RcppExport SEXP exell_xls_sheets(SEXP pathSEXP) {
+BEGIN_RCPP
+    Rcpp::RObject __result;
+    Rcpp::RNGScope __rngScope;
+    Rcpp::traits::input_parameter< std::string >::type path(pathSEXP);
+    __result = Rcpp::wrap(xls_sheets(path));
+    return __result;
+END_RCPP
+}
diff --git a/src/h-xls.cpp b/src/h-xls.cpp
@@ -129,3 +129,26 @@ bool is_datetime(int id, formatMap formats) {
   return false;
 }
 
+
+// [[Rcpp::export]]
+std::vector<std::string> xls_sheets(std::string path) {
+  xls::xlsWorkBook* pWB = xls::xls_open(path.c_str(), "UTF8");
+
+  if (pWB == NULL)
+    Rcpp::stop("Failed to open %s", path);
+
+  int n = pWB->sheets.count;
+
+  std::vector<std::string> sheets;
+  sheets.reserve(n);
+
+  for (int i = 0; i < n; ++i) {
+    std::string name((char*) pWB->sheets.sheet[i].name);
+    sheets.push_back(name);
+  }
+
+  // Cleanup
+  xls_close(pWB);
+
+  return sheets;
+}