Added operator<<() and print() functions to Vector

Author: dcdillon

inst/include/Rcpp/vector/Vector.h

@@ -582,6 +582,14 @@ class Vector :
         return -1;
     }
 
+    inline void print(const std::string & comment = "", std::ostream & stream = Rcout) const {
+        if (comment.length() > 0) {
+            stream << comment << std::endl;
+        }
+
+        stream << (*this);
+    }
+
 
 protected:
     inline int* dims() const {
@@ -1080,6 +1088,44 @@ class Vector :
 
 } ; /* Vector */
 
+template <int RTYPE, template <class> class StoragePolicy >
+inline std::ostream &operator<<(std::ostream & s, const Vector<RTYPE, StoragePolicy> & rhs) {
+    typedef Vector<RTYPE, StoragePolicy> VECTOR;
+
+    typename VECTOR::iterator i = const_cast<VECTOR &>(rhs).begin();
+    typename VECTOR::iterator iend = const_cast<VECTOR &>(rhs).end();
+
+    if (i != iend) {
+        s << (*i);
+        ++i;
+
+        for ( ; i != iend; ++i) {
+            s << " " << (*i);
+        }
+    }
+
+    return s;
+}
+
+template<template <class> class StoragePolicy >
+inline std::ostream &operator<<(std::ostream & s, const Vector<STRSXP, StoragePolicy> & rhs) {
+    typedef Vector<STRSXP, StoragePolicy> VECTOR;
+
+    typename VECTOR::iterator i = const_cast<VECTOR &>(rhs).begin();
+    typename VECTOR::iterator iend = const_cast<VECTOR &>(rhs).end();
+
+    if (i != iend) {
+        s << "\"" << (*i) << "\"";
+        ++i;
+
+        for ( ; i != iend; ++i) {
+            s << " \"" << (*i) << "\"";
+        }
+    }
+
+    return s;
+}
+
 }
 
 #endif

inst/unitTests/cpp/Vector.cpp

@@ -21,6 +21,8 @@
 
 #include <climits>
 #include <Rcpp.h>
+#include <sstream>
+
 using namespace Rcpp ;
 
 inline double square( double x){ return x*x; }
@@ -782,3 +784,46 @@ int noprotect_matrix( Matrix<REALSXP, NoProtectStorage> x){
 int vec_access_with_bounds_checking(const IntegerVector x, int index) {
     return x.at(index);
 }
+
+// [[Rcpp::export]]
+String vec_print_numeric(NumericVector v) {
+    std::ostringstream buf;
+    buf << v;
+    return buf.str();
+}
+
+// [[Rcpp::export]]
+String vec_print_numeric_member(NumericVector v) {
+    std::ostringstream buf;
+    v.print("", buf);
+    return buf.str();
+}
+
+// [[Rcpp::export]]
+String vec_print_character(CharacterVector v) {
+    std::ostringstream buf;
+    buf << v;
+    return buf.str();
+}
+
+// [[Rcpp::export]]
+String vec_print_character_member(CharacterVector v) {
+    std::ostringstream buf;
+    v.print("", buf);
+    return buf.str();
+}
+
+// [[Rcpp::export]]
+String vec_print_integer(IntegerVector v) {
+    std::ostringstream buf;
+    buf << v;
+    return buf.str();
+}
+
+// [[Rcpp::export]]
+String vec_print_integer_member(IntegerVector v) {
+    std::ostringstream buf;
+    v.print("", buf);
+    return buf.str();
+}
+

inst/unitTests/runit.Vector.R

@@ -690,5 +690,41 @@ if (.runThisTest) {
         checkException(vec_access_with_bounds_checking(x, 5) , msg = "index out of bounds not detected" )
         checkException(vec_access_with_bounds_checking(x, -1) , msg = "index out of bounds not detected" )
     }
+
+    test.NumericVector.print <- function() {
+        v <- c(1.1, 2.2, 3.3, 4.4)
+        s <- vec_print_numeric(v)
+        checkEquals(s, "1.1 2.2 3.3 4.4")
+    }
+
+    test.NumericVector.print.member <- function() {
+        v <- c(1.1, 2.2, 3.3, 4.4)
+        s <- vec_print_numeric_member(v)
+        checkEquals(s, "1.1 2.2 3.3 4.4")
+    }
+
+    test.IntegerVector.print <- function() {
+        v <- c(1, 2, 3, 4)
+        s <-vec_print_integer(v)
+        checkEquals(s, "1 2 3 4")
+    }
+
+    test.IntegerVector.print.member <- function() {
+        v <- c(1, 2, 3, 4)
+        s <- vec_print_integer_member(v)
+        checkEquals(s, "1 2 3 4")
+    }
+
+    test.CharacterVector.print <- function() {
+        v <- c("a", "b", "c", "d")
+        s <- vec_print_character(v)
+        checkEquals(s, '"a" "b" "c" "d"')
+    }
+
+    test.CharacterVector.print.member <- function() {
+        v <- c("a", "b", "c", "d")
+        s <- vec_print_character_member(v)
+        checkEquals(s, '"a" "b" "c" "d"')
+    }
 }