#!/usr/bin/env r
#
# Copyright (C) 2010 - 2014  Dirk Eddelbuettel, Romain Francois and Kevin Ushey
#
# 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) {

    .setUp <- Rcpp:::unitTestSetup("Matrix.cpp")

    test.List.column <- function(){
	x <- matrix( 1:16+.5, nc = 4 )
	res <- runit_Row_Column_sugar( 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(){
	x <- matrix( 1:16 + .5, ncol = 4 )
	checkEquals( matrix_numeric(x), sum(diag(x)), msg = "matrix indexing" )

	y <- as.vector( x )
	checkException( matrix_numeric(y) , msg = "not a matrix" )

    }

    test.CharacterMatrix <- function(){
	x <- matrix( letters[1:16], ncol = 4 )
	checkEquals( matrix_character(x), paste( diag(x), collapse = "" ) )
    }

    test.GenericMatrix <- function( ){
	g <- function(y){
            sapply( y, function(x) seq(from=x, to = 16) )
	}
	x <- matrix( g(1:16), ncol = 4 )
	checkEquals( matrix_generic(x), g(diag(matrix(1:16,ncol=4))), msg = "GenericMatrix" )
    }

    test.IntegerMatrix.diag <- function(){
	expected <- matrix( 0L, nrow = 5, ncol = 5 )
	diag( expected ) <- 1L
	checkEquals( matrix_integer_diag(), expected, msg = "IntegerMatrix::diag" )
    }

    test.CharacterMatrix.diag <- function(){
	expected <- matrix( "", nrow = 5, ncol = 5 )
	diag( expected ) <- "foo"
	checkEquals( matrix_character_diag(), expected, msg = "CharacterMatrix::diag" )
    }

    test.NumericMatrix.Ctors <- function(){
	x <- matrix(0, 3, 3)
	checkEquals( matrix_numeric_ctor1(), x, msg = "matrix from single int" )

	x <- matrix(0, 3, 3)
	checkEquals( matrix_numeric_ctor2(), x, msg = "matrix from two int" )
    }

    test.IntegerVector.matrix.indexing <- function(){
        x <- matrix( 1:16, ncol = 4 )
        checkEquals( integer_matrix_indexing(x), sum(diag(x)), msg = "matrix indexing" )

        checkEquals( diag(integer_matrix_indexing_lhs(x)), 2*0:3, msg = "matrix indexing lhs" )

        y <- as.vector( x )
        checkException( integer_matrix_indexing_lhs(y) , msg = "not a matrix" )
    }

    test.NumericMatrix.row <- function(){
	x <- matrix( 1:16 + .5, ncol = 4 )
	checkEquals( runit_NumericMatrix_row( x ), sum( x[1,] ), msg = "iterating over a row" )
    }

    test.NumericMatrix.row.const <- function(){
	x <- matrix( 1:16 + .5, ncol = 4 )
	checkEquals( runit_NumericMatrix_row_const( x ), sum( x[1,] ), msg = "iterating over a row" )
    }

    test.CharacterMatrix.row <- function(){
	m <- matrix( letters, ncol = 2 )
	checkEquals( runit_CharacterMatrix_row(m), paste( m[1,], collapse = "" ), msg = "CharacterVector::Row" )
    }

    test.CharacterMatrix.row.const <- function(){
	m <- matrix( letters, ncol = 2 )
	checkEquals( runit_CharacterMatrix_row_const(m), paste( m[1,], collapse = "" ), msg = "CharacterVector::Row" )
    }

    test.List.row <- function(){
	m <- lapply( 1:16, function(i) seq(from=1, to = i ) )
	dim( m ) <- c( 4, 4 )
	checkEquals( runit_GenericMatrix_row( m ), 1 + 0:3*4, msg = "List::Row" )
    }

    test.List.row.const <- function(){
	m <- lapply( 1:16, function(i) seq(from=1, to = i ) )
	dim( m ) <- c( 4, 4 )
	checkEquals( runit_GenericMatrix_row_const( m ), 1 + 0:3*4, msg = "List::Row" )
    }

    test.NumericMatrix.column <- function(){
	x <- matrix( 1:16 + .5, ncol = 4 )
	checkEquals( runit_NumericMatrix_column( x ), sum( x[,1] ) , msg = "iterating over a column" )
    }

    test.NumericMatrix.column.const <- function(){
	x <- matrix( 1:16 + .5, ncol = 4 )
	checkEquals( runit_NumericMatrix_column_const( x ), sum( x[,1] ) , msg = "iterating over a column" )
    }

    test.NumericMatrix.cumsum <- function(){
	x <- matrix( 1:8 + .5, ncol = 2 )
	checkEquals( runit_NumericMatrix_cumsum( x ), t(apply(x, 1, cumsum)) , msg = "cumsum" )
    }

    test.CharacterMatrix.column <- function(){
	m <- matrix( letters, ncol = 2 )
	checkEquals( runit_CharacterMatrix_column(m), paste( m[,1], collapse = "" ), msg = "CharacterVector::Column" )
    }

    test.CharacterMatrix.column.const <- function(){
	m <- matrix( letters, ncol = 2 )
	checkEquals( runit_CharacterMatrix_column_const(m), paste( m[,1], collapse = "" ), msg = "CharacterVector::Column" )
    }

    test.List.column <- function(){
	m <- lapply( 1:16, function(i) seq(from=1, to = i ) )
	dim( m ) <- c( 4, 4 )
	checkEquals( runit_GenericMatrix_column( m ), 1:4, msg = "List::Column" )
    }

    test.List.column.const <- function(){
	m <- lapply( 1:16, function(i) seq(from=1, to = i ) )
	dim( m ) <- c( 4, 4 )
	checkEquals( runit_GenericMatrix_column_const( m ), 1:4, msg = "List::Column" )
    }

    test.NumericMatrix.colsum <- function( ){
        probs <- matrix(1:12,nrow=3)
        checkEquals( runit_NumericMatrix_colsum( probs ), t(apply(probs,1,cumsum)) )
    }

    test.NumericMatrix.rowsum <- function( ){
        probs <- matrix(1:12,nrow=3)
        checkEquals( runit_NumericMatrix_rowsum( probs ), apply(probs,2,cumsum) )
    }

    test.NumericMatrix.SubMatrix <- function( ){
        target <- rbind( c(3,4,5,5), c(3,4,5,5), 0 )
        checkEquals( runit_SubMatrix(), target, msg = "SubMatrix" )
    }

    test.NumericMatrix.opequals <- function() {
        m <- matrix(1:4, nrow=2)
        checkEquals(m, matrix_opequals(m))
    }

    test.NumericMatrix.rownames.colnames.proxy <- function() {
        m <- matrix(as.numeric(1:4), nrow = 2)
        runit_rownames_colnames_proxy(m, letters[1:2], LETTERS[1:2])
        checkEquals(rownames(m), letters[1:2])
        checkEquals(colnames(m), LETTERS[1:2])
        checkException(runit_rownames_colnames_proxy(m, letters[1:3], letters[1:3]))
        checkException(runit_rownames_colnames_proxy(m, letters[1:2], NULL))

        m <- matrix(as.numeric(1:9), nrow = 3)
        runit_rownames_proxy(m)
        checkEquals(rownames(m), c("A", "B", "C"))
        checkEquals(colnames(m), NULL)
    }

    test.NumericMatrix.no.init <- function() {
      m <- runit_no_init_matrix()
      checkEquals(m, matrix(c(0, 1, 2, 3), nrow = 2))
    }

    test.NumericMatrix.const.Column <- function(){
      m <- matrix(as.numeric(1:9), nrow = 3)
      res <- runit_const_Matrix_column(m)
      checkEquals( m[,1], m[,2] )
    }

    test.IntegerMatrix.accessor.with.bounds.checking <- function() {
        m <- matrix(seq(1L, 12, by=1L), nrow=4L, ncol=3L)
        checkEquals(mat_access_with_bounds_checking(m, 0, 0), 1)
        checkEquals(mat_access_with_bounds_checking(m, 1, 2), 10)
        checkEquals(mat_access_with_bounds_checking(m, 3, 2), 12)
        checkException(mat_access_with_bounds_checking(m, 4, 2) , msg = "index out of bounds not detected" )
        checkException(mat_access_with_bounds_checking(m, 3, 3) , msg = "index out of bounds not detected" )
        checkException(mat_access_with_bounds_checking(m, 3, -1) , msg = "index out of bounds not detected" )
        checkException(mat_access_with_bounds_checking(m, -1, 2) , msg = "index out of bounds not detected" )
        checkException(mat_access_with_bounds_checking(m, -1, -1) , msg = "index out of bounds not detected" )
    }

    test.IntegerMatrix.transpose <- function() {
        M <- matrix(1:12, 3, 4)
        checkEquals(transposeInteger(M), t(M), msg="integer transpose")
        rownames(M) <- letters[1:nrow(M)]
        checkEquals(transposeInteger(M), t(M), msg="integer transpose with rownames")
        colnames(M) <- LETTERS[1:ncol(M)]
        checkEquals(transposeInteger(M), t(M), msg="integer transpose with row and colnames")
    }

    test.NumericMatrix.transpose <- function() {
        M <- matrix(1.0 * (1:12), 3, 4)
        checkEquals(transposeNumeric(M), t(M), msg="numeric transpose")
        rownames(M) <- letters[1:nrow(M)]
        checkEquals(transposeNumeric(M), t(M), msg="numeric transpose with rownames")
        colnames(M) <- LETTERS[1:ncol(M)]
        checkEquals(transposeNumeric(M), t(M), msg="numeric transpose with row and colnames")
    }

    test.CharacterMatrix.transpose <- function() {
        M <- matrix(as.character(1:12), 3, 4)
        checkEquals(transposeCharacter(M), t(M), msg="character transpose")
        rownames(M) <- letters[1:nrow(M)]
        checkEquals(transposeCharacter(M), t(M), msg="character transpose with rownames")
        colnames(M) <- LETTERS[1:ncol(M)]
        checkEquals(transposeCharacter(M), t(M), msg="character transpose with row and colnames")
    }
    
    test.Matrix.Scalar.op <- function() {
        M <- matrix(c(1:12), 3, 4)
        checkEquals(matrix_scalar_plus(M, 2), M + 2, msg="matrix + scalar")
        checkEquals(matrix_scalar_plus2(M, 2), 2 + M, msg="scalar + matrix")
        checkEquals(matrix_scalar_divide(M, 2), M / 2, msg="matrix / scalar")
        checkEquals(matrix_scalar_divide2(M, 2), 2 / M, msg="scalar / matrix")
    }

    ## 23 October 2016
    ## eye function
    test.Matrix.eye <- function() {
        
        checkEquals(
            dbl_eye(3),
            diag(1.0, 3, 3),
            "eye - numeric"
        )
        
        checkEquals(
            int_eye(3),
            diag(1L, 3, 3),
            "eye - integer"
        )
        
        checkEquals(
            cx_eye(3),
            diag(1.0 + 0i, 3, 3),
            "eye - complex"
        )
        
        ## diag(TRUE, 3, 3) was registering as 
        ## a numeric matrix on Travis for some reason
        mat <- matrix(FALSE, 3, 3)
        diag(mat) <- TRUE
        checkEquals(
            lgl_eye(3),
            mat,
            "eye - logical"
        )
    }
    
    ## ones function
    test.Matrix.ones <- function() {
        
        checkEquals(
            dbl_ones(3),
            matrix(1.0, 3, 3),
            "ones - numeric"
        )
        
        checkEquals(
            int_ones(3),
            matrix(1L, 3, 3),
            "ones - integer"
        )
        
        checkEquals(
            cx_ones(3),
            matrix(1.0 + 0i, 3, 3),
            "ones - complex"
        )
        
        checkEquals(
            lgl_ones(3),
            matrix(TRUE, 3, 3),
            "ones - logical"
        )
    }
    
    ## zeros function
    test.Matrix.zeros <- function() {
        
        checkEquals(
            dbl_zeros(3),
            matrix(0.0, 3, 3),
            "zeros - numeric"
        )
        
        checkEquals(
            int_zeros(3),
            matrix(0L, 3, 3),
            "zeros - integer"
        )
        
        checkEquals(
            cx_zeros(3),
            matrix(0.0 + 0i, 3, 3),
            "zeros - complex"
        )
        
        checkEquals(
            lgl_zeros(3),
            matrix(FALSE, 3, 3),
            "zeros - logical"
        )
    }
    
    
    test.Matrix.diagfill <- function() {
        
        checkEquals(num_diag_fill(diag(1.0, 2, 4), 0.0),
                    matrix(0.0, 2, 4),
                    msg = "diagonal fill - case: n < p")
        
        checkEquals(num_diag_fill(diag(1.0, 4, 2), 0.0),
                    matrix(0.0, 4, 2),
                    msg = "diagonal fill - case: n > p")
        
        checkEquals(num_diag_fill(diag(1.0, 3, 3), 0.0),
                    matrix(0.0, 3, 3),
                    msg = "diagonal fill - case: n = p")
        
        m <- matrix("", 2, 4)
        diag(m) <- letters[1:2]
        
        checkEquals(char_diag_fill(m, ""), 
                    matrix("", 2, 4),
                    msg = "diagonal fill - char")
        
        
    }

}
