introduce DateSubsetVectorVisitor. #1204

tidyverse · Aug 14, 2015 · 7a81db0 · 7a81db0
1 parent a90f778
commit 7a81db0
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 0 deletions.
diff --git a/inst/include/dplyr/SubsetVectorVisitorImpl.h b/inst/include/dplyr/SubsetVectorVisitorImpl.h
@@ -152,6 +152,59 @@ namespace dplyr {
 
     } ;
 
+    class DateSubsetVectorVisitor : public SubsetVectorVisitor {
+    public:
+
+        DateSubsetVectorVisitor( SEXP data ) : impl(0){
+          if( TYPEOF(data) == INTSXP ) {
+            impl  = new SubsetVectorVisitorImpl<INTSXP>(data) ;
+          } else if( TYPEOF(data) == REALSXP ) {
+            impl = new SubsetVectorVisitorImpl<REALSXP>(data) ;
+          } else {
+            stop( "" ) ;
+          }
+        }
+
+        ~DateSubsetVectorVisitor( ){
+          delete impl ;
+        }
+
+        virtual SEXP subset( const Rcpp::IntegerVector& index ) const {
+          return impl->subset( index ) ;
+        }
+
+        virtual SEXP subset( const std::vector<int>& index ) const {
+          return impl->subset( index ) ;
+        }
+
+        virtual SEXP subset( const ChunkIndexMap& index ) const {
+          return impl->subset(index) ;
+        }
+
+        virtual SEXP subset( const Rcpp::LogicalVector& index ) const {
+          return impl->subset( index ) ;
+        }
+
+        virtual SEXP subset( EmptySubset index ) const {
+          return impl->subset( index ) ;
+        }
+
+        virtual int size() const {
+          return impl->size() ;
+        }
+
+        virtual std::string get_r_type() const {
+          return impl->get_r_type() ;
+        }
+
+    private:
+        SubsetVectorVisitor* impl ;
+        DateSubsetVectorVisitor( const DateSubsetVectorVisitor& ) ;
+
+    } ;
+
+
+
 }
 
 #endif
diff --git a/inst/include/dplyr/subset_visitor.h b/inst/include/dplyr/subset_visitor.h
@@ -18,6 +18,10 @@ namespace dplyr {
             }
         }
 
+        if( Rf_inherits(vec, "Date") ){
+            return new DateSubsetVectorVisitor(vec) ;
+        }
+
         switch( TYPEOF(vec) ){
             case CPLXSXP:
                 return new SubsetVectorVisitorImpl<CPLXSXP>( vec ) ;