R/math.R

Defines functions spam_Arith logical_Summary spam_Math

# HEADER ####################################################
# This is file spam/R/math.R.                               #
# It is part of the R package spam,                         #
#  --> https://CRAN.R-project.org/package=spam              #
#  --> https://CRAN.R-project.org/package=spam64            #
#  --> https://git.math.uzh.ch/reinhard.furrer/spam         #
# by Reinhard Furrer [aut, cre], Florian Gerber [aut],      #
#    Roman Flury [aut], Daniel Gerber [ctb],                #
#    Kaspar Moesinger [ctb]                                 #
# HEADER END ################################################



# `"Ops"":
#      `"+"", `"-"", `"*"", `"/"", `"^"", `"%%"", `"%/%""
#      `"&"", `"|"", `"!""
#      `"=="", `"!="", `"<"", `"<="", `">="", `">""


#     `Math" `"abs"", `"sign"", `"sqrt"", `"ceiling"", `"floor"",
#          `"trunc"", `"cummax"", `"cummin"", `"cumprod"", `"cumsum"",
#          `"log"", `"log10"", `"log2"", `"log1p"", `"acos"", `"acosh"",
#          `"asin"", `"asinh"", `"atan"", `"atanh"", `"exp"", `"expm1"",
#          `"cos"", `"cosh"", `"cospi"", `"sin"", `"sinh"", `"sinpi"",
#          `"tan"", `"tanh"", `"tanpi"", `"gamma"", `"lgamma"",
#          `"digamma"", `"trigamma""

#     `Math2" `"round"", `"signif""

#     `Summary" `"max"", `"min"", `"range"", `"prod"", `"sum"", `"any"", `"all""


##############
# Unary operators "+", "-" and "!" are handled with e2 missing...
#
# Currently, "+", "-" are handled...
setMethod("!",signature(x="spam"),    function(x){
    if(getOption("spam.structurebased")) {
        x@entries <- as.double(callGeneric(x@entries))
        x
    } else {
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(x)))
        spam(as.double( callGeneric(as.matrix(x))), nrow=dim(x)[1])
    }
})

setMethod("+",signature(e1="spam",e2="missing"), function(e1) e1 )
setMethod("-",signature(e1="spam",e2="missing"), function(e1) { e1@entries <- -e1@entries; e1} )

#     `Math2" :

setMethod("Math2",signature(x = "spam", digits = "ANY"),
          function(x, digits){ x@entries <- callGeneric(x@entries, digits = digits); x })

#     `Math" :

setMethod("Math","spam", function(x){
    if(getOption("spam.structurebased")) {
        x@entries <- callGeneric(x@entries)
        x
    }else{
        x@entries <- callGeneric(x@entries)
        as.spam.spam( x)
    }
})

#     `Math", where we pass to matrix first...

spam_Math <- function(x) {
    if(getOption("spam.structurebased")) {
        x@entries <- callGeneric(x@entries)
        x
    }else{
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(x)))
        as.spam(callGeneric(as.matrix(x)))
    }}


setMethod("exp","spam", spam_Math )
setMethod("log10","spam", spam_Math )
setMethod("log2","spam", spam_Math )
# from ?log: Do not set S4 methods on `logb" itself.
# special case to set base...
setMethod("log","spam", function(x,...) {
    if(getOption("spam.structurebased")) {
        x@entries <- callGeneric(x@entries,...)
        x
    }else{
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(x)))
        as.spam(callGeneric(as.matrix(x),...))
    }}
          )



setMethod("cos","spam", spam_Math )
#setMethod("cospi","spam", spam_Math )
setMethod("cosh","spam", spam_Math )
setMethod("acosh","spam", spam_Math )
setMethod("acos","spam", spam_Math )

setMethod("gamma","spam", spam_Math )
setMethod("digamma","spam", spam_Math )
setMethod("trigamma","spam", spam_Math )
setMethod("lgamma","spam", spam_Math )

setMethod("cummax","spam", spam_Math )
setMethod("cummin","spam", spam_Math )
setMethod("cumprod","spam", spam_Math )
setMethod("cumsum","spam", spam_Math )


#     `Summary" :
setMethod("Summary","spam", function(x,...,na.rm=FALSE){
    if(getOption("spam.structurebased")) {
        callGeneric(x@entries,...,na.rm=na.rm)
    }else{
        if ( prod( x@dimension) == length( x@entries)) {
            callGeneric(x@entries,...,na.rm=na.rm)
        } else {
            callGeneric(c(0,x@entries),...,na.rm=na.rm)
        }
    }
}
          )

logical_Summary <- function( x,...,na.rm=FALSE){
    if(getOption("spam.structurebased")) {
        callGeneric(as.logical(x@entries),...,na.rm=na.rm)
    }else{
        if ( prod( x@dimension) == length( x@entries)) {
            callGeneric(as.logical(x@entries),...,na.rm=na.rm)
        } else {
            callGeneric(as.logical(c(0,x@entries)),...,na.rm=na.rm)
        }
    }
}


setMethod("any","spam", logical_Summary)
setMethod("all","spam", logical_Summary)


################################################################################################################################################################################################################################################################################################
#     `Ops" `"Arith"", `"Compare"", `"Logic""



#     `Logic" `"&"", `"|"".


"spam_Logic_vectorspam" <- function(e1, e2) {
    if(getOption("spam.structurebased")) {
        if( isTRUE(all.equal(length(e2@entries), prod(e2@dimension))))  # spam is full
            return( as.spam( callGeneric(e1, as.matrix(e2))) )          # 3.0: no as.spam -> matrix
        if(identical(length(e1),1L) | identical(length(e1), length(e2@entries))) {
            e2@entries <- as.double( callGeneric(e1, e2@entries))
            return(e2)
        }
        stop(gettextf("incompatible lengths for %s operation.", sQuote(.Generic)))
    }  else {
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e2)))
        return( as.spam( callGeneric(e1, as.matrix(e2))) )
    }
}

"spam_Logic_spamvector" <- function(e1, e2)  {
    if(getOption("spam.structurebased")) {
        if( isTRUE(all.equal(length(e1@entries), prod(e1@dimension))))  # spam is full
            return( as.spam( callGeneric(as.matrix(e1), e2)) )   # 3.0: no as.spam -> matrix
        if(identical(length(e2),1L) | identical(length(e2), length(e1@entries))) {
            e1@entries <- as.double( callGeneric(e1@entries, e2))
            return(e1)
        }
        stop(gettextf("incompatible lengths for %s operation.", sQuote(.Generic)))
    }  else {
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e1)))
        return( as.spam( callGeneric(as.matrix(e1), e2)) )
    }
}
"spam_Logic_matrixspam" <- function(e1, e2){
  #    cat("spam_Arith_matrixspam")
  if( identical( e2@dimension, dim(e1))) {
    if(getOption("spam.structurebased")) {
      return( as.spam( callGeneric(e1, as.matrix(e2))) )          # 3.0: no as.spam -> matrix
    }  else {
      inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e2)))
      return( as.spam( callGeneric(e1, as.matrix(e2))) )
    }
  }
  stop(gettextf("incompatible dimensions for %s operation.", sQuote(.Generic)))
}
"spam_Logic_spammatrix" <- function(e1, e2){
  #    cat("spam_Arith_spammatrix")
  if( identical( e1@dimension, dim(e2))) {
    if(getOption("spam.structurebased")) {
      return( as.spam( callGeneric(as.matrix(e1), e2)) )          # 3.0: no as.spam -> matrix
    }  else {
      inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e1)))
      return( as.spam( callGeneric(as.matrix(e1), e2)) )
    }
  }
  stop(gettextf("incompatible dimensions for %s operation.", sQuote(.Generic)))
}


setMethod("|",signature(e1="spam",e2="spam"),
          function(e1,e2){ z <- spam_add(e1,e2);  z@entries <- rep(1,length(z@colindices));z})

setMethod("&",signature(e1="spam",e2="spam"),
          function(e1,e2){ z <- spam_mult(e1,e2); z@entries <- rep(1,length(z@colindices));z})
setMethod("Logic",signature(e1="spam",e2="vector"), spam_Logic_spamvector)
setMethod("Logic",signature(e1="vector",e2="spam"), spam_Logic_vectorspam)
setMethod("Logic",signature(e1="spam",e2="matrix"), spam_Logic_spammatrix)
setMethod("Logic",signature(e1="matrix",e2="spam"), spam_Logic_matrixspam)

##################################################################################################
#     `Compare" `"=="", `">"", `"<"", `"!="", `"<="", `">=""
"spam_Compare" <- function(e1,e2) {
    inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)),  max(prod(dim(e1)), prod(dim(e2))))
    as.spam( callGeneric( as.matrix(e1), as.matrix(e2))  )
}

"spam_Compare_spamvector" <- function(e1, e2){
    if(getOption("spam.structurebased")) {
        if( isTRUE(all.equal(length(e1@entries), prod(e1@dimension))))  # spam is full
            return( as.spam( callGeneric(as.matrix(e1), e2)) )   # 3.0: no as.spam -> matrix
        if(identical(length(e2),1L) | identical(length(e2), length(e1@entries))) {
            e1@entries <- as.double(callGeneric(e1@entries, e2))
            return(e1)
        }
        stop(gettextf("incompatible lengths for %s operation.", sQuote(.Generic)))
    }  else {
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e1)))
        return( as.spam( callGeneric(as.matrix(e1), e2)) )
    }
}
"spam_Compare_vectorspam" <- function(e1, e2) {
    if(getOption("spam.structurebased")) {
        if( isTRUE(all.equal(length(e2@entries), prod(e2@dimension))))  # spam is full
            return( as.spam( callGeneric(e1, as.matrix(e2))) )          # 3.0: no as.spam -> matrix
        if(identical(length(e1),1L) | identical(length(e1), length(e2@entries))) {
            e2@entries <- as.double( callGeneric(e1, e2@entries))
            return(e2)
        }
        stop(gettextf("incompatible lengths for %s operation.", sQuote(.Generic)))
     }  else {
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e2)))
        return( as.spam( callGeneric(e1, as.matrix(e2))) )
    }
}
"spam_Compare_matrixspam" <- function(e1, e2){
  #    cat("spam_Arith_matrixspam")
  if( identical( e2@dimension, dim(e1))) {
    if(getOption("spam.structurebased")) {
      return( as.spam( callGeneric(e1, as.matrix(e2))) )          # 3.0: no as.spam -> matrix
    }  else {
      inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e2)))
      return( as.spam( callGeneric(e1, as.matrix(e2))) )
    }
  }
  stop(gettextf("incompatible dimensions for %s operation.", sQuote(.Generic)))
}
"spam_Compare_spammatrix" <- function(e1, e2){
  #    cat("spam_Arith_spammatrix")
  if( identical( e1@dimension, dim(e2))) {
    if(getOption("spam.structurebased")) {
      return( as.spam( callGeneric(as.matrix(e1), e2)) )          # 3.0: no as.spam -> matrix
    }  else {
      inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e1)))
      return( as.spam( callGeneric(as.matrix(e1), e2)) )
    }
  }
  stop(gettextf("incompatible dimensions for %s operation.", sQuote(.Generic)))
}


setMethod("Compare",signature(e1="spam",e2="spam"),   spam_Compare )
setMethod("Compare",signature(e1="spam",e2="vector"), spam_Compare_spamvector )
setMethod("Compare",signature(e1="vector",e2="spam"), spam_Compare_vectorspam )
setMethod("Compare",signature(e1="spam",e2="matrix"), spam_Compare_spammatrix )
setMethod("Compare",signature(e1="matrix",e2="spam"), spam_Compare_matrixspam )


##################################################################################################
#     `Arith": `"+"", `"-"", `"*"", `"^"", `"%%"", `"%/%"", `"/""

"spam_Arith_vectorspam" <- function(e1, e2){
#    cat("spam_Arith_vectorspam")
    if(getOption("spam.structurebased")) {
        if( isTRUE(all.equal(length(e2@entries), prod(e2@dimension))))  # spam is full
            return( as.spam( callGeneric(e1, as.matrix(e2))) )          # 3.0: no as.spam -> matrix
        if(identical(length(e1),1L) | identical(length(e1), length(e2@entries))) {
            e2@entries <- callGeneric(e1, e2@entries)
            return(e2)
        }
        stop(gettextf("incompatible lengths for %s operation.", sQuote(.Generic)))
    }  else {
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e2)))
         return( as.spam( callGeneric(e1, as.matrix(e2))) )
    }
}
"spam_Arith_spamvector" <- function(e1, e2){
#    cat("spam_Arith_spamvector")
    if(getOption("spam.structurebased")) {
        if( isTRUE(all.equal(length(e1@entries), prod(e1@dimension))))  # spam is full
            return( as.spam( callGeneric(as.matrix(e1), e2)) )          # 3.0: no as.spam -> matrix
        if(identical(length(e2),1L) | identical(length(e2), length(e1@entries))) {
            e1@entries <- callGeneric(e1@entries, e2)
            return(e1)
        }
        stop(gettextf("incompatible lengths for %s operation.", sQuote(.Generic)))
     }  else {
        inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e1)))
        return( as.spam( callGeneric(as.matrix(e1), e2)) )
    }
}
"spam_Arith_matrixspam" <- function(e1, e2){
  #    cat("spam_Arith_matrixspam")
  if( identical( e2@dimension, dim(e1))) {
    if(getOption("spam.structurebased")) {
      return( as.spam( callGeneric(e1, as.matrix(e2))) )          # 3.0: no as.spam -> matrix
    }  else {
      inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e2)))
      return( as.spam( callGeneric(e1, as.matrix(e2))) )
    }
  }
  stop(gettextf("incompatible dimensions for %s operation.", sQuote(.Generic)))
}
"spam_Arith_spammatrix" <- function(e1, e2){
  #    cat("spam_Arith_spammatrix")
  if( identical( e1@dimension, dim(e2))) {
    if(getOption("spam.structurebased")) {
      return( as.spam( callGeneric(as.matrix(e1), e2)) )          # 3.0: no as.spam -> matrix
    }  else {
      inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)), prod(dim(e1)))
      return( as.spam( callGeneric(as.matrix(e1), e2)) )
    }
  }
  stop(gettextf("incompatible dimensions for %s operation.", sQuote(.Generic)))
}
spam_Arith <- function(e1,e2) {
    inefficiencywarning( gettextf("This %s operation may be inefficient",sQuote(.Generic)),  max(prod(dim(e1)), prod(dim(e2))))
    as.spam( callGeneric( as.matrix(e1), as.matrix(e2)))
}



setMethod("Arith",signature(e1="spam",e2="spam"),   spam_Arith )
setMethod("Arith",signature(e1="spam",e2="vector"), spam_Arith_spamvector)
setMethod("Arith",signature(e1="vector",e2="spam"), spam_Arith_vectorspam)
setMethod("Arith",signature(e1="spam",e2="matrix"), spam_Arith_spammatrix)
setMethod("Arith",signature(e1="matrix",e2="spam"), spam_Arith_matrixspam)

setMethod("/",signature(e1="spam",e2="spam"),   function(e1,e2){ "/"(e1,as.matrix(e2)) } )
setMethod("/",signature(e1="matrix",e2="spam"), function(e1,e2){ "/"(e1,as.matrix(e2)) } )
setMethod("^",signature(e1="spam",e2="spam"),   function(e1,e2){ "^"(e1,as.matrix(e2)) } )


######################################################################
# nz <- 128; ln <- nz^2; A <- spam(0,ln,ln); is <- sample(ln,nz); js <- sample(ln,nz);A[cbind(is,js)] <- 1:nz
# nz <- 128; ln <- nz^2; A <- spam(0,ln,ln); is <- sample(ln,ln); js <- sample(ln,ln);A[cbind(is,js)] <- 1:ln
# system.time(   spam:::.spam.addsparsefull(A,1)) ; system.time(   as.matrix.spam(A)+1.5)


#######################################################################
"spam_add" <- function(A, B, s=1)
{

                                        # cat("spam_add")
    nrow <- A@dimension[1]
    ncol <- A@dimension[2]
    if(ncol != B@dimension[2] || nrow != B@dimension[1])
        stop("non-conformable matrices")

    if( getOption("spam.force64") || .format.spam(A)$package == "spam64" || .format.spam(B)$package == "spam64")
        SS <- .format64()
    else
        SS <- .format32

    nzmax <- .C64("aplbdg",
                  ## subroutine aplbdg (nrow,ncol,ja,ia,jb,ib,ndegr,nnz,iw)
                  SIGNATURE = c(SS$signature, SS$signature,
                      SS$signature, SS$signature, SS$signature, SS$signature,
                      SS$signature, SS$signature, SS$signature),

                  nrow,
                  ncol,

                  A@colindices,
                  A@rowpointers,
                  B@colindices,
                  B@rowpointers,

                  vector_dc(SS$type, nrow),
                  nnz = vector_dc(SS$type, 1),
                  vector_dc(SS$type, ncol),

                  INTENT = c("r", "r",
                      "r", "r", "r", "r",
                      "w", "w", "w"),
                  NAOK=getOption("spam.NAOK"),
                  PACKAGE = SS$package)$nnz

    z <- .C64("aplsb1",
              ## subroutine aplsb1 (nrow,ncol,a,ja,ia,s,b,jb,ib,c,jc,ic,nzmax,ierr)
              SIGNATURE = c(SS$signature, SS$signature,
                  "double", SS$signature, SS$signature,
                  "double",
                  "double", SS$signature, SS$signature,
                  "double", SS$signature, SS$signature,
                  SS$signature, SS$signature),

              nrow,
              ncol,

              A@entries,
              A@colindices,
              A@rowpointers,

              s,

              B@entries,
              B@colindices,
              B@rowpointers,

              entries     = vector_dc("double", nzmax),
              colindices  = vector_dc(SS$type, nzmax),
              rowpointers = vector_dc(SS$type, nrow+1),

              nzmax+1,
              ierr = vector_dc(SS$type, 1),

              INTENT = c("r", "r",
                  "r", "r", "r",
                  "r",
                  "r", "r", "r",
                  "w", "w", "w",
                  "r", "w" ),
              NAOK=getOption("spam.NAOK"),
              PACKAGE = SS$package)

    if(z$ierr != 0) stop("insufficient space for sparse matrix addition")
    nz <- z$rowpointers[nrow+1]-1
    return(.newSpam(
        entries = z$entries[1:nz],
        colindices = z$colindices[1:nz],
        rowpointers = z$rowpointers,
        dimension =  c(nrow,ncol)
        ))
}

setMethod("+",signature(e1="spam",e2="spam"),  function(e1,e2){ spam_add(e1, e2)    })
setMethod("-",signature(e1="spam",e2="spam"),  function(e1,e2){ spam_add(e1, e2, -1)})



###############################################################################

"spam_mult" <- function(e1,e2)
{
#  if(is.vector(e1)) {
#    if(length(e1) == 1){
#      if(e1==0) return( spam(0,nrow(e2),ncol(e2)))
#      else{  # just a scalar
#        e2@entries <- e1*e2@entries
#        return(e2)
#      }
#    }  else if(length(e1) == nrow(e2))
#      return(diag.spam(e1) %*% e2)
#    else # length(e1) == ncol(e2) is not required
#      stop("e1 and e2 not conformable for efficient element-by-element multiplication")
#  }
#  else if(is.vector(e2)) {
#    if(length(e2) == 1){
#      if(e2==0)   return( spam(0,nrow(e1),ncol(e1)))
#      else {
#        e1@entries <- e2*e1@entries
#        return(e1)
#      }
#    }
#    else if(length(e2) == nrow(e1))
#      return(diag.spam(e2) %*% e1)
#    else
#      stop("e1 and e2 not conformable for efficient element-by-element multiplication")
#  }
#  if(is.matrix(e1))
#    e1 <- as.spam(e1)
#  else if(is.matrix(e2))
#    e2 <- as.spam(e2)
#  if(!(is.spam(e1) && is.spam(e2)))
#    stop("Arguments must be of class:  vector, matrix or spam")

  e1row <- e1@dimension[1]
  e1col <- e1@dimension[2]
  if(e1col != e2@dimension[2] | e1row != e2@dimension[1])
    stop("non-conformable matrices")
  nnzmax <- length(intersect(e1@colindices+e1col*(rep(1:e1row,diff(e1@rowpointers))-1),
                             e2@colindices+e2@dimension[2]*(rep(1:e2@dimension[1],diff(e2@rowpointers))-1)))+1

  if( getOption("spam.force64") || .format.spam(e1)$package == "spam64" || .format.spam(e2)$package == "spam64")
      SS <- .format64()
  else
      SS <- .format32

  z <- .C64("aemub",
     ##        subroutine aemub (nrow,ncol,a,ja,ia,amask,jmask,imask,
     ## *                  c,jc,ic,nzmax,ierr)
            SIGNATURE = c(SS$signature, SS$signature,
                "double", SS$signature, SS$signature,
                "double", SS$signature, SS$signature,
                "double", SS$signature, SS$signature,
                SS$signature, SS$signature),

            e1row,
            e1col,

            e1@entries, e1@colindices, e1@rowpointers,

            e2@entries, e2@colindices, e2@rowpointers,

            entries     = vector_dc("double", nnzmax),
            colindices  = vector_dc(SS$type, nnzmax),
            rowpointers = vector_dc(SS$type, e1row+1),

            nnzmax,
            ierr = vector_dc(SS$type,1),

            INTENT = c("r","r",
                "r", "r", "r",
                "r", "r", "r",
                "w", "w", "w",
                "r", "w"),
            NAOK=getOption("spam.NAOK"),
            PACKAGE = SS$package)
  ## z <- .Fortran("aemub",
  ##               as.integer(e1row),
  ##               as.integer(e1col),
  ##               as.double(e1@entries), as.integer(e1@colindices),  as.integer(e1@rowpointers),
  ##               as.double(e2@entries), as.integer(e2@colindices),  as.integer(e2@rowpointers),
  ##               entries     = vector("double",nnzmax),
  ##               colindices  = vector("integer",nnzmax),
  ##               rowpointers = vector("integer",e1row+1),
  ##               ## integer(e1col),
  ##               double(e1col),
  ##               as.integer(nnzmax),
  ##               ierr = vector("integer",1),
  ##               NAOK=getOption("spam.NAOK"),  PACKAGE = "spam")

    if(z$ierr != 0)      stop("insufficient space for element-wise sparse matrix multiplication")

    nnz <- z$rowpointers[e1row+1]-1

    if(identical(z$entries,0L)){#trap zero matrix
        z$colindices <- 1L
        z$rowpointers <- c(1L,rep(2L,e1row))
    }

    return(.newSpam(
        entries=z$entries[1:nnz],
        colindices=z$colindices[1:nnz],
        rowpointers=z$rowpointers,
        dimension=c(e1row,e1col)))
}


setMethod("*",signature(e1="spam",e2="spam"), spam_mult)


##########################################################################################################################################################################################################################################################################################################################################################################################################################################################################################################
"matrix_add_spammatrix" <- function(A,B){
                                        #    cat("matrix_add_spammatrix")
                                        # A is sparse, B is full
                                        #  if (missing(B)) return(A)
                                        #  if (!is.numeric(B)) stop("numeric argument expected")
    nrow <- A@dimension[1]
    ncol <- A@dimension[2]
    pdim <- prod(nrow,ncol)
                                        #  if (is.matrix(B)) {
    if(ncol != dim(B)[2] || nrow != dim(B)[1])
        stop("non-conformable matrices")
                                        #  } else {
                                        #    if(pdim%%length(B)!=0) {
                                        #      stop("longer object length
                                        #        is not a multiple of shorter object length")
                                        #    } else  B <- rep(B,pdim %/% length(B))
                                        #  }
    ## print("addsparsefull")
    if(  getOption("spam.force64") )
        SS <- .format64()
    else
        SS <- .format.spam(A)

    return(array(
        .C64("addsparsefull",
                 SIGNATURE = c(SS$signature, "double", SS$signature, SS$signature,
                               "double"),

                 nrow,
                 A@entries,
                 A@colindices,
                 A@rowpointers,
                 b = c(B),

                 INTENT = c("r", "r", "r", "r",
                            "rw"),
                 NAOK=getOption("spam.NAOK"),
                 PACKAGE = SS$package )$b,
        c(nrow,ncol)))
 }

"matrix_sub_spammatrix" <- function(A,B){
#    cat("matrix_sub_spammatrix")
  # A is sparse, B is full
#  if (missing(B)) {
#    A@entries <- -A@entries
#    return(A)
#  }
#  if (!is.numeric(B)) stop("numeric argument expected")
  nrow <- A@dimension[1]
  ncol <- A@dimension[2]
#  pdim <- prod(nrow,ncol)
#  if (is.matrix(B)) {
    if(ncol != dim(B)[2] || nrow != dim(B)[1])
      stop("non-conformable matrices")
#  } else {
#    if(pdim %% length(B)!=0) {
#      stop("longer object length
#        is not a multiple of shorter object length")
#    } else  B <- rep(B,pdim %/% length(B))
#  }
                                        #  if (!is.double(B[1]))  B <- as.double(B)
    ## print("subfullsparse")
    if(  getOption("spam.force64") )
        SS <- .format64()
    else
        SS <- .format.spam(A)

    return(array(
        .C64("subfullsparse",
             SIGNATURE = c( SS$signature, SS$signature,
                           "double", SS$signature, SS$signature,
                           "double"),

             nrow,
             ncol,

             A@entries,
             A@colindices,
             A@rowpointers,

             b = c(B),

             INTENT = c("r", "r",
                        "r", "r", "r",
                        "rw"),
             NAOK=getOption("spam.NAOK"),
             PACKAGE = SS$package)$b,
        c(nrow,ncol)))
}

"matrix_sub_matrixspam" <- function(B,A){
#    cat("matrix_sub_spammatrix")
  # A is sparse, B is full
  if (!is.numeric(B)) stop("numeric argument expected")
  nrow <- A@dimension[1]
  ncol <- A@dimension[2]
#  pdim <- prod(nrow,ncol)
#  if (is.matrix(B)) {
    if(ncol != dim(B)[2] || nrow != dim(B)[1])
      stop("non-conformable matrices")
#  } else {
#    if(pdim %% length(B)!=0) {
#      stop("longer object length
#        is not a multiple of shorter object length")
#    } else  B <- rep(B,pdim %/% length(B))
#  }
#  if (!is.double(B[1]))  B <- as.double(B)

    if(  getOption("spam.force64") )
        SS <- .format64()
    else
        SS <- .format.spam(A)

    return(array(
        .C64("subsparsefull",
             SIGNATURE = c(SS$signature, "double", SS$signature, SS$signature,
                           "double"),

             nrow,
             A@entries,
             A@colindices,
             A@rowpointers,

             b = c(B),

             INTENT = c("r", "r", "r", "r",
                        "rw"),
             NAOK=getOption("spam.NAOK"),
             PACKAGE = SS$package )$b,
        c(nrow,ncol)))
}
#
setMethod("+",signature(e1="spam",   e2="matrix"), function(e1,e2){ matrix_add_spammatrix(e1,e2)})
setMethod("+",signature(e1="matrix",   e2="spam"), function(e1,e2){ matrix_add_spammatrix(e2,e1)})
setMethod("-",signature(e1="matrix",   e2="spam"), function(e1,e2){ matrix_sub_matrixspam(e1,e2)})
setMethod("-",signature(e1="spam",   e2="matrix"), function(e1,e2){ matrix_sub_spammatrix(e1,e2)})


#"spam_division" <- function(e1,e2) { # Element-wise matrix division of two spams
#  if(is.numeric(e1) && length(e1) == 1)
#  { e2@entries <- e1/e2@entries
#    return(e2)
# } else if(is.numeric(e2) && length(e2) == 1) {
#    e1@entries <- e1@entries/e2
#    return(e1)
#  }
#  else if(is.spam(e1) || is.spam(e2) || is.matrix(e1) || is.matrix(e2)){
#        if(is.matrix(e1)) e1 <- as.spam(e1)
#        if(is.matrix(e2)) e2 <- as.spam(e2)
#        nrow <- e1@dimension[1]
#        ncol <- e1@dimension[2]
#        if(ncol != e2@dimension[2] | nrow != e2@dimension[1])
#          stop("matrices not conformable for element-by-element division")
#	nzmax <- length(unique(c(e1@colindices+ncol*(rep(1:nrow,diff(e1@rowpointers))-1),
#                                 e2@colindices+e2@dimension[2]*(rep(1:e2@dimension[1],diff(e2@rowpointers))-1))))+1
#        z <- .Fortran("_aedib_",   # does not order the colindicies upon return!
#                      nrow,
#                      ncol,
#                      as.integer(1),
#                      as.double(e1@entries), e1@colindices,  e1@rowpointers,
#                      as.double(e2@entries), e2@colindices,  e2@rowpointers,
#                      entries     = vector("double",nzmax),
#                      colindices  = vector("integer",nzmax),
#                      rowpointers = vector("integer",nrow+1),
#                      as.integer(nzmax),
#                      integer(ncol),
#                      double(ncol),
#                      ierr = vector("integer",1),
#                      NAOK=getOption("spam.NAOK"),PACKAGE = "spam"
#                      )
#        if(z$ierr != 0) stop("insufficient space for element-wise sparse matrix division")
#        nz <- z$rowpointers[nrow+1]-1
#        return(new("spam",entries=z$entries[1:nz],colindices=z$colindices[1:nz],rowpointers=z$rowpointers,
#                   dimension=c(nrow,ncol)))
#    }



##"spam_exponent" <- function(e1, e2)
#{
#    nrow <- e1@dimension[1]
#    ncol <- e1@dimension[2]
#    if(ncol != e2@dimension[2] | nrow != e2@dimension[1])
#        stop("matrices not conformable for element-wise exponentiation ")
#    nzmax <- length(unique(c(e1@colindices+ncol*(rep(1:nrow,diff(e1@rowpointers))-1),
#                             e2@colindices+e2@dimension[2]*(rep(1:e2@dimension[1],diff(e2@rowpointers))-1))))+1
#    z <- .Fortran("_aeexpb_", does not reorder col indices!
#                  as.integer(nrow), as.integer(ncol),
#                  1L,
#                  as.double(e1@entries),  as.integer(e1@colindices),  as.integer(e1@rowpointers),
#                  as.double(e2@entries),  as.integer(e2@colindices),  as.integer(e2@rowpointers),
#                  entries     = vector("double",nzmax),
#                  colindices  = vector("integer",nzmax),
#                  rowpointers = vector("integer",nrow+1),
#                  as.integer(nzmax),
#                  integer(ncol),       double(ncol),
#                  ierr = vector("integer",1),
#                  NAOK=getOption("spam.NAOK"),PACKAGE = "spam"
#                  )
#    if(z$ierr != 0) stop("insufficient space for element-wise exponentiation")
#    nz <- z$rowpointers[nrow+1]-1
#    return(new("spam",entries=z$entries[1:nz],colindices=z$colindices[1:nz],rowpointers=z$rowpointers,
#               dimension=c(nrow,ncol)))
#}

#############################
#getClass("numeric")
#  Extends: "vector"
#getClass("matrix")
#Extends: Class "vector", by class "array", distance 3, with explicit coerce
# Hence we use use vector, especially, to include the NA case that is not of type numeric!

Try the spam package in your browser

Any scripts or data that you put into this service are public.

spam documentation built on Oct. 23, 2023, 5:07 p.m.