colnames and rownames getters and setters for matrices. closes #210

romainfrancois · romainfrancois · commit 486d1ecbc8f5 · 2014-11-10T12:32:49.000+01:00
diff --git a/NEWS.md b/NEWS.md
@@ -2,6 +2,8 @@
 
 * New `wrap` implementation for `std::tuple<Args...>` (#195)
 
+* `colnames` and `rownames` setters for matrices (#210). 
+
 # Rcpp11 3.1.1
 
 * sugar `sum` now supports complex sugar vector expressions
diff --git a/inst/include/Rcpp.h b/inst/include/Rcpp.h
@@ -234,6 +234,8 @@ namespace Rcpp{
 
 #include <Rcpp/Timer.h>
 
+#include <Rcpp/vector/dimnames.h>
+
 namespace Rcpp11 = Rcpp ;
 
 #endif
diff --git a/inst/include/Rcpp/vector/dimnames.h b/inst/include/Rcpp/vector/dimnames.h
@@ -0,0 +1,69 @@
+#ifndef Rcpp__Matrix_Dimnames_h
+#define Rcpp__Matrix_Dimnames_h
+
+namespace Rcpp {
+
+    template <typename Data>
+    class RowNamesProxy {
+    public:
+        
+        RowNamesProxy( Data& data_ ) : data(data_){}
+        
+        inline RowNamesProxy& operator=( const CharacterVector& names ){
+            Language call( "rownames<-", data, names ) ;
+            data = call.eval() ;
+            return *this ;    
+        }
+        
+        inline RowNamesProxy& operator=( std::initializer_list<const char*> names ){
+            return this->operator=( CharacterVector(names) ) ;    
+        }
+        
+        inline operator CharacterVector() const {
+            Language call( "rownames", data ) ;
+            return call.eval() ;
+        }
+        
+    private:
+        Data& data ;
+    } ;
+    
+    template <typename Data>
+    class ColNamesProxy {
+    public:
+        
+        ColNamesProxy( Data& data_ ) : data(data_){}
+        
+        inline ColNamesProxy& operator=( const CharacterVector& names ){
+            Language call( "colnames<-", data, names ) ;
+            data = call.eval() ;
+            return *this ;    
+        }
+        
+        inline ColNamesProxy& operator=( std::initializer_list<const char*> names ){
+            return this->operator=( CharacterVector(names) ) ;    
+        }
+        
+        inline operator CharacterVector() const {
+            Language call( "colnames", data ) ;
+            return call.eval() ;
+        }
+        
+    private:
+        Data& data ;
+    } ;
+    
+    
+    template <int RTYPE, typename Storage>
+    inline RowNamesProxy<Matrix<RTYPE, Storage>> rownames( const Matrix<RTYPE, Storage>& m ){
+        return RowNamesProxy<Matrix<RTYPE, Storage>>(const_cast<Matrix<RTYPE,Storage>&>(m)) ;    
+    }
+    
+    template <int RTYPE, typename Storage>
+    inline ColNamesProxy<Matrix<RTYPE, Storage>> colnames( const Matrix<RTYPE, Storage>& m ){
+        return ColNamesProxy<Matrix<RTYPE, Storage>>(const_cast<Matrix<RTYPE,Storage>&>(m)) ;    
+    }
+    
+}
+
+#endif
