Added assignment operators to newDatetimeVector and newDateVector. Fixed a small bug in their copy construtors.

Author: dcdillon

inst/include/Rcpp.h

@@ -56,7 +56,7 @@
 #include <Rcpp/Formula.h>
 #include <Rcpp/DataFrame.h>
 
-//  #define RCPP_NEW_DATE_DATETIME_VECTORS 1
+#define RCPP_NEW_DATE_DATETIME_VECTORS 1
 #include <Rcpp/date_datetime/date_datetime.h>
 
 #include <Rcpp/Na_Proxy.h>

inst/include/Rcpp/date_datetime/newDateVector.h

@@ -29,7 +29,9 @@ namespace Rcpp {
     class newDateVector : public NumericVector {
     public:
         template <int RTYPE, bool NA, typename VEC>
-        newDateVector(const VectorBase<RTYPE,NA,VEC>& vec) : NumericVector(vec) {}
+        newDateVector(const VectorBase<RTYPE,NA,VEC>& vec) : NumericVector(vec) {
+            setClass();
+        }
 
         newDateVector(SEXP vec) : NumericVector(vec) { setClass(); }
         newDateVector(int n) : NumericVector(n) { setClass(); }
@@ -41,6 +43,14 @@ namespace Rcpp {
                 v[i] = (*this)[i];
             return v;
         }
+        
+        inline newDateVector &operator=(const newDateVector &rhs) {
+            if (this != &rhs) {
+                NumericVector::operator=(rhs);
+            }
+            
+            return *this;
+        }
 
     private:
 

inst/include/Rcpp/date_datetime/newDatetimeVector.h

@@ -30,7 +30,9 @@ namespace Rcpp {
     public:
         template <int RTYPE, bool NA, typename VEC>
         newDatetimeVector(const VectorBase<RTYPE,NA,VEC>& other) :
-            NumericVector(other) {}
+            NumericVector(other) {
+                setClass(other.attr("tzone"));
+        }
 
         newDatetimeVector(SEXP vec, const char* tz = "") :
             NumericVector(vec) {
@@ -49,6 +51,15 @@ namespace Rcpp {
                 v[i] = (*this)[i];
             return v;
         }
+        
+        inline newDatetimeVector &operator=(const newDatetimeVector &rhs) {
+            if (this != &rhs) {
+                NumericVector::operator=(rhs);
+                this->attr("tzone") = rhs.attr("tzone");
+            }
+            
+            return *this;
+        }
 
     private:
 

inst/unitTests/cpp/dates.cpp

@@ -158,3 +158,17 @@ SEXP DatetimeVector_ctor(DatetimeVector d) {
     DatetimeVector dt = DatetimeVector(d);
     return wrap(dt);
 }
+
+// [[Rcpp::export]]
+DatetimeVector DatetimeVector_assignment(DatetimeVector v1,
+    DatetimeVector v2) {
+
+    v1 = v2;
+    return v1;
+}
+
+// [[Rcpp::export]]
+DateVector DateVector_assignment(DateVector v1, DateVector v2) {
+    v1 = v2;
+    return v1;
+}

inst/unitTests/runit.Date.R

@@ -181,5 +181,19 @@ if (.runThisTest) {
             checkEquals(fun(vec), c(now, rep(posixtNA, 3), now+2.345), msg = "Datetime.ctor.NA.NaN.Inf.set")
         }
     }
+    
+    test.DatetimeVector.assignment <- function() {
+        now <- Sys.time()
+        v1 <- c(now, now + 1, now + 2)
+        v2 <- c(now + 3, now + 4, now + 5)
+        checkEquals(v2, DatetimeVector_assignment(v1, v2))
+    }
+    
+    test.DateVector.assignment <- function() {
+        now <- Sys.Date()
+        v1 <- c(now, now + 1, now + 2)
+        v2 <- c(now + 3, now + 4, now + 5)
+        checkEquals(v2, DateVector_assignment(v1, v2))
+    }
 
 }