#!/usr/bin/r -t
#
# Copyright (C) 2010 - 2012 Dirk Eddelbuettel and Romain Francois
#
# This file is part of Rcpp.
#
# Rcpp is 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/>.
.runThisTest <- Sys.getenv("RunAllRcppTests") == "yes"
if (.runThisTest) {
definitions <- function(){
list(
"matrix_numeric" = list(
signature(x = "matrix" ), '
NumericMatrix m(x) ;
double trace = 0.0 ;
for( size_t i=0 ; i<4; i++){
trace += m(i,i) ;
}
return wrap( trace ) ;
'
),
"matrix_character" = list(
signature(x = "matrix" ), '
CharacterMatrix m(x) ;
std::string trace ;
for( size_t i=0 ; i<4; i++){
trace += m(i,i) ;
}
return wrap( trace ) ;
'
),
"matrix_generic" = list(
signature(x = "matrix" ), '
GenericMatrix m(x) ;
List output( m.ncol() ) ;
for( size_t i=0 ; i<4; i++){
output[i] = m(i,i) ;
}
return output ;
'
),
"matrix_integer_diag" = list(
signature(),
'return IntegerMatrix::diag( 5, 1 ) ; '
),
"matrix_character_diag" = list(
signature(),
'return CharacterMatrix::diag( 5, "foo" ) ;'
),
"matrix_numeric_ctor1" = list(
signature(),
'
NumericMatrix m(3);
return m;
'
),
"matrix_numeric_ctor2" = list(
signature(), '
NumericMatrix m(3,3);
return m;
'
),
"integer_matrix_indexing"=list(
signature(x = "integer" ),
'IntegerVector m(x) ;
int trace = 0.0 ;
for( size_t i=0 ; i<4; i++){
trace += m(i,i) ;
}
return wrap( trace ) ;'
),
"integer_matrix_indexing_lhs"=list(
signature(x = "integer" ),
'IntegerVector m(x) ;
for( size_t i=0 ; i<4; i++){
m(i,i) = 2 * i ;
}
return m ; '
),
"runit_NumericMatrix_row" = list(
signature(x = "matrix" ),
'
NumericMatrix m(x) ;
NumericMatrix::Row first_row = m.row(0) ;
return wrap( std::accumulate( first_row.begin(), first_row.end(), 0.0 ) ) ;
'),
"runit_CharacterMatrix_row" = list(
signature(x = "matrix" ),
'
CharacterMatrix m(x) ;
CharacterMatrix::Row first_row = m.row(0) ;
std::string res(
std::accumulate(
first_row.begin(), first_row.end(), std::string() ) ) ;
return wrap(res) ;
'
),
"runit_GenericMatrix_row" = list(
signature(x = "matrix" ),
'
GenericMatrix m(x) ;
GenericMatrix::Row first_row = m.row(0) ;
IntegerVector out( first_row.size() ) ;
std::transform(
first_row.begin(), first_row.end(),
out.begin(),
unary_call<SEXP,int>( Function("length" ) ) ) ;
return wrap(out) ;
'
),
"runit_NumericMatrix_column" = list(
signature(x = "matrix" ),
'
NumericMatrix m(x) ;
NumericMatrix::Column col = m.column(0) ;
return wrap( std::accumulate( col.begin(), col.end(), 0.0 ) ) ;
'
),
"runit_NumericMatrix_cumsum" = list(
signature(x = "matrix" ),
'
NumericMatrix input( x ) ;
int nr = input.nrow(), nc = input.ncol() ;
NumericMatrix output(nr, nc) ;
NumericVector tmp( nr );
for( int i=0; i<nc; i++){
tmp = tmp + input.column(i) ;
NumericMatrix::Column target( output, i ) ;
std::copy( tmp.begin(), tmp.end(), target.begin() ) ;
}
return output ;
'
),
"runit_CharacterMatrix_column" = list(
signature(x = "matrix" ),
'
CharacterMatrix m(x) ;
CharacterMatrix::Column col = m.column(0) ;
std::string res(
std::accumulate(
col.begin(), col.end(), std::string() ) ) ;
return wrap(res) ;
'
),
"runit_GenericMatrix_column" = list(
signature(x = "matrix" ),
'
GenericMatrix m(x) ;
GenericMatrix::Column col = m.column(0) ;
IntegerVector out( col.size() ) ;
std::transform(
col.begin(), col.end(),
out.begin(),
unary_call<SEXP,int>( Function("length" ) ) ) ;
return wrap(out) ;
'
),
"runit_Row_Column_sugar" = list(
signature( x_ = "matrix" ),
'
NumericMatrix x( x_) ;
NumericVector r0 = x.row(0) ;
NumericVector c0 = x.column(0) ;
return List::create(
r0,
c0,
x.row(1),
x.column(1),
x.row(1) + x.column(1)
) ;
'
),
"runit_NumericMatrix_colsum" = list(
signature( x = "matrix" ),
'
NumericMatrix input( x ) ;
int nc = input.ncol() ;
NumericMatrix output = clone<NumericMatrix>( input ) ;
for( int i=1; i<nc; i++){
output(_,i) = output(_,i-1) + input(_,i) ;
}
return output ;
'
),
"runit_NumericMatrix_rowsum" = list(
signature( x = "matrix" ),
'
NumericMatrix input( x ) ;
int nr = input.nrow();
NumericMatrix output = clone<NumericMatrix>( input ) ;
for( int i=1; i<nr; i++){
output(i,_) = output(i-1,_) + input(i,_) ;
}
return output ;
'
),
"runit_SubMatrix" = list(
signature(),
'
NumericMatrix xx(4, 5);
xx(0,0) = 3;
xx(0,1) = 4;
xx(0,2) = 5;
xx(1,_) = xx(0,_);
xx(_,3) = xx(_,2);
SubMatrix<REALSXP> yy = xx( Range(0,2), Range(0,3) ) ;
NumericMatrix res = yy ;
return res;
'
)
)
}
cxxargs <- function(){
ifelse(Rcpp:::capabilities()[["initializer lists"]],"-std=c++0x","")
}
.setUp <- function() {
tests <- ".rcpp.Matrix"
if( ! exists( tests, globalenv() )) {
fun <- Rcpp:::compile_unit_tests(
definitions(),
cxxargs = cxxargs()
)
assign( tests, fun, globalenv() )
}
}
test.List.column <- function(){
funx <- .rcpp.Matrix$runit_Row_Column_sugar
x <- matrix( 1:16+.5, nc = 4 )
res <- funx( x )
target <- list(
x[1,],
x[,1],
x[2,],
x[,2],
x[2,] + x[,2]
)
checkEquals( res, target, msg = "column and row as sugar" )
}
test.NumericMatrix <- function(){
funx <- .rcpp.Matrix$matrix_numeric
x <- matrix( 1:16 + .5, ncol = 4 )
checkEquals( funx(x), sum(diag(x)), msg = "matrix indexing" )
y <- as.vector( x )
checkException( funx(y) , msg = "not a matrix" )
}
test.CharacterMatrix <- function(){
funx <- .rcpp.Matrix$matrix_character
x <- matrix( letters[1:16], ncol = 4 )
checkEquals( funx(x), paste( diag(x), collapse = "" ) )
}
test.GenericMatrix <- function( ){
funx <- .rcpp.Matrix$matrix_generic
g <- function(y){
sapply( y, function(x) seq(from=x, to = 16) )
}
x <- matrix( g(1:16), ncol = 4 )
checkEquals( funx(x), g(diag(matrix(1:16,ncol=4))), msg = "GenericMatrix" )
}
test.IntegerMatrix.diag <- function(){
funx <- .rcpp.Matrix$matrix_integer_diag
expected <- matrix( 0L, nrow = 5, ncol = 5 )
diag( expected ) <- 1L
checkEquals( funx(), expected, msg = "IntegerMatrix::diag" )
}
test.CharacterMatrix.diag <- function(){
funx <- .rcpp.Matrix$matrix_character_diag
expected <- matrix( "", nrow = 5, ncol = 5 )
diag( expected ) <- "foo"
checkEquals( funx(), expected, msg = "CharacterMatrix::diag" )
}
test.NumericMatrix.Ctors <- function(){
funx <- .rcpp.Matrix$matrix_numeric_ctor1
x <- matrix(0, 3, 3)
checkEquals( funx(), x, msg = "matrix from single int" )
funx <- .rcpp.Matrix$matrix_numeric_ctor2
x <- matrix(0, 3, 3)
checkEquals( funx(), x, msg = "matrix from two int" )
}
test.IntegerVector.matrix.indexing <- function(){
fun <- .rcpp.Matrix$integer_matrix_indexing
x <- matrix( 1:16, ncol = 4 )
checkEquals( fun(x), sum(diag(x)), msg = "matrix indexing" )
fun <- .rcpp.Matrix$integer_matrix_indexing_lhs
checkEquals( diag(fun(x)), 2*0:3, msg = "matrix indexing lhs" )
y <- as.vector( x )
checkException( fun(y) , msg = "not a matrix" )
}
test.NumericMatrix.row <- function(){
funx <- .rcpp.Matrix$runit_NumericMatrix_row
x <- matrix( 1:16 + .5, ncol = 4 )
checkEquals( funx( x ), sum( x[1,] ), msg = "iterating over a row" )
}
test.CharacterMatrix.row <- function(){
funx <- .rcpp.Matrix$runit_CharacterMatrix_row
m <- matrix( letters, ncol = 2 )
checkEquals( funx(m), paste( m[1,], collapse = "" ), msg = "CharacterVector::Row" )
}
test.List.row <- function(){
funx <- .rcpp.Matrix$runit_GenericMatrix_row
m <- lapply( 1:16, function(i) seq(from=1, to = i ) )
dim( m ) <- c( 4, 4 )
checkEquals( funx( m ), 1 + 0:3*4, msg = "List::Row" )
}
test.NumericMatrix.column <- function(){
funx <- .rcpp.Matrix$runit_NumericMatrix_column
x <- matrix( 1:16 + .5, ncol = 4 )
checkEquals( funx( x ), sum( x[,1] ) , msg = "iterating over a column" )
}
test.NumericMatrix.cumsum <- function(){
funx <- .rcpp.Matrix$runit_NumericMatrix_cumsum
x <- matrix( 1:8 + .5, ncol = 2 )
checkEquals( funx( x ), t(apply(x, 1, cumsum)) , msg = "cumsum" )
}
test.CharacterMatrix.column <- function(){
funx <- .rcpp.Matrix$runit_CharacterMatrix_column
m <- matrix( letters, ncol = 2 )
checkEquals( funx(m), paste( m[,1], collapse = "" ), msg = "CharacterVector::Column" )
}
test.List.column <- function(){
funx <- .rcpp.Matrix$runit_GenericMatrix_column
m <- lapply( 1:16, function(i) seq(from=1, to = i ) )
dim( m ) <- c( 4, 4 )
checkEquals( funx( m ), 1:4, msg = "List::Column" )
}
test.NumericMatrix.colsum <- function( ){
funx <- .rcpp.Matrix$runit_NumericMatrix_colsum
probs <- matrix(1:12,nrow=3)
checkEquals( funx( probs ), t(apply(probs,1,cumsum)) )
}
test.NumericMatrix.rowsum <- function( ){
funx <- .rcpp.Matrix$runit_NumericMatrix_rowsum
probs <- matrix(1:12,nrow=3)
checkEquals( funx( probs ), apply(probs,2,cumsum) )
}
test.NumericMatrix.SubMatrix <- function( ){
funx <- .rcpp.Matrix$runit_SubMatrix
target <- rbind( c(3,4,5,5), c(3,4,5,5), 0 )
checkEquals( funx(), target, msg = "SubMatrix" )
}
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.