added Na_proxy. closes #60

Author: romainfrancois

inst/include/Rcpp.h

@@ -57,6 +57,8 @@
 #include <Rcpp/Datetime.h>
 #include <Rcpp/DatetimeVector.h>
 
+#include <Rcpp/NA_Proxy.h>
+
 #include <Rcpp/Module.h>
 #include <Rcpp/InternalFunction.h>
 

inst/include/Rcpp/Na_Proxy.h

@@ -0,0 +1,54 @@
+// Copyright (C) 2013 Romain Francois
+//
+// This file is part of Rcpp.
+//
+//    Rcppis free software: you can redistribute it and/or modify it
+// under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 2 of the License, or
+// (at your option) any later version.
+//
+// Rcpp is distributed in the hope that it will be useful, but
+// WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with Rcpp.  If not, see <http://www.gnu.org/licenses/>.
+
+#ifndef Rcpp_Na_Proxy_h
+#define Rcpp_Na_Proxy_h
+
+namespace Rcpp{
+    class Na_Proxy{} ;
+    static Na_Proxy NA ;    
+}
+
+inline bool operator==(double x      , Rcpp::Na_Proxy){ return Rcpp::traits::is_na<REALSXP>(x) ; }
+inline bool operator==(int x         , Rcpp::Na_Proxy){ return Rcpp::traits::is_na<INTSXP>(x) ; }
+inline bool operator==(Rcpp::String x, Rcpp::Na_Proxy){ return Rcpp::traits::is_na<STRSXP>(x.get_sexp()) ; }
+inline bool operator==(Rcomplex x    , Rcpp::Na_Proxy){ return Rcpp::traits::is_na<CPLXSXP>(x) ; }
+inline bool operator==(SEXP x        , Rcpp::Na_Proxy){ return TYPEOF(x)==CHARSXP && Rcpp::traits::is_na<STRSXP>(x) ; }
+inline bool operator==(std::string   , Rcpp::Na_Proxy){ return false ; }
+inline bool operator==(const char*   , Rcpp::Na_Proxy){ return false ; }
+inline bool operator==(Rcpp::internal::string_proxy<STRSXP> x, Rcpp::Na_Proxy){ 
+    return Rcpp::traits::is_na<STRSXP>(x.get()) ; 
+}
+inline bool operator==(Rcpp::internal::const_string_proxy<STRSXP> x, Rcpp::Na_Proxy){ 
+    return Rcpp::traits::is_na<STRSXP>(x.get()) ; 
+}
+
+inline bool operator==(Rcpp::Na_Proxy, double x       ){ return Rcpp::traits::is_na<REALSXP>(x) ; }
+inline bool operator==(Rcpp::Na_Proxy, int x          ){ return Rcpp::traits::is_na<INTSXP>(x) ; }
+inline bool operator==(Rcpp::Na_Proxy, Rcpp::String x ){ return Rcpp::traits::is_na<STRSXP>(x.get_sexp()) ; }
+inline bool operator==(Rcpp::Na_Proxy, SEXP x         ){ return TYPEOF(x)==CHARSXP && Rcpp::traits::is_na<STRSXP>(x) ; }
+inline bool operator==(Rcpp::Na_Proxy, Rcomplex x     ){ return Rcpp::traits::is_na<CPLXSXP>(x) ; }
+inline bool operator==(Rcpp::Na_Proxy, std::string    ){ return false ; }
+inline bool operator==(Rcpp::Na_Proxy, const char*    ){ return false ; }
+inline bool operator==(Rcpp::Na_Proxy, Rcpp::internal::string_proxy<STRSXP> x){ 
+    return Rcpp::traits::is_na<STRSXP>(x.get()) ; 
+}
+inline bool operator==(Rcpp::Na_Proxy, Rcpp::internal::const_string_proxy<STRSXP> x){ 
+    return Rcpp::traits::is_na<STRSXP>(x.get()) ; 
+}
+
+#endif

inst/unitTests/cpp/misc.cpp

@@ -107,3 +107,29 @@ void test_rcout(std::string tfile, std::string teststring){
     testfile.close();
 }
 
+// [[Rcpp::export]]
+LogicalVector na_proxy(){
+    CharacterVector s("foo") ;
+    return LogicalVector::create( 
+        NA_REAL    == NA, 
+        NA_INTEGER == NA,
+        NA_STRING  == NA,
+        true       == NA, 
+        false      == NA, 
+        1.2        == NA, 
+        12         == NA,
+        "foo"      == NA,
+        s[0]       == NA, 
+        
+        NA         == NA_REAL, 
+        NA         == NA_INTEGER,
+        NA         == NA_STRING,
+        NA         == true, 
+        NA         == false,
+        NA         == 1.2  , 
+        NA         == 12   ,
+        NA         == "foo", 
+        NA         == s[0]
+        ) ;
+}      
+

inst/unitTests/runit.misc.R

@@ -118,4 +118,12 @@ test.rcout <- function(){
     checkEquals( readLines(rcppfile), readLines(rfile), msg="Rcout output")    
 }
 
+test.na_proxy <- function(){
+    checkEquals( 
+        na_proxy(), 
+        rep(c(TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,FALSE,FALSE) , 2), 
+        msg = "Na_Proxy NA == handling"
+    )    
+}
+
 }