Nothing
R/kronecker.R
In Matrix: Sparse and Dense Matrix Classes and Methods
Defines functions
kroneckerDN
## METHODS FOR GENERIC: kronecker
## ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
kroneckerDN <- function(dnX, dX = lengths(dnX, FALSE),
dnY, dY = lengths(dnX, FALSE),
sep = ":") {
dnr <- list(NULL, NULL)
if(identical(dnX, dnr) && identical(dnY, dnr))
return(NULL)
for(i in 1:2) {
if(have.sX <- (nY <- dY[i]) && !is.null(sX <- dnX[[i]]))
sX <- rep (sX, each = nY)
if(have.sY <- (nX <- dX[i]) && !is.null(sY <- dnY[[i]]))
sY <- rep.int(sY, times = nX)
dnr[[i]] <-
if(have.sX && have.sY)
paste0(sX, sep, sY)
else if(have.sX)
paste0(sX, sep )
else if(have.sY)
paste0( sep, sY)
else if(nX && nY)
rep.int(sep, nX * nY)
}
dnr
}
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
r <- new("ddiMatrix")
r@Dim <- dX * dY
uX <- X@diag != "N"
uY <- Y@diag != "N"
if(uX && uY)
r@diag <- "U"
else {
if(!uX) {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
}
if(!uY) {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
}
r@x <-
if(uX)
rep.int(as.double(Y.ii), dX[1L])
else if(uY)
rep(as.double(X.ii), each = dY[1L])
else rep(X.ii, each = dY[1L]) * Y.ii
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY)))
r@Dimnames <- dnr
r
})
if(FALSE) { # --NOT YET--
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "denseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- X@diag != "N"
uY <- FALSE
shape <- .M.shape(Y)
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
uplo <- r@uplo <- Y@uplo
if(shape == "t")
uY <- Y@diag != "N"
}
if(!all(dr)) {
r@p <- integer(dr[2L] + 1)
if(uX && uY)
r@diag <- "U"
} else {
m <- dY[1L]
nX <- dX[1L]
nY <- length(y <- Y@x)
if(shape != "g" && nY > 1L && nY == prod(dY)) {
Y <- pack(Y)
nY <- length(y <- Y@x)
}
if(as.double(nX) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
if(!uX && uY) {
diag(Y) <- TRUE
nY <- length(y <- Y@x)
}
if(is.logical(y) && .M.kind(Y) == "n")
y <- y | is.na(y)
if(shape == "g") {
r@p <- seq.int(0L, by = m, length.out = dr[2L] + 1)
r@i <-
rep(seq.int(0L, by = m, length.out = nX),
each = nY) +
0:(m-1L)
} else if(uplo == "U") {
r@p <- c(0L, cumsum(rep.int(1:m, nX)))
r@i <-
rep(seq.int(0L, by = m, length.out = nX),
each = nY) +
sequence.default(nvec = 1:m, from = 0L)
} else {
r@p <- c(0L, cumsum(rep.int(m:1, nX)))
r@i <-
rep(seq.int(0L, by = m, length.out = nX),
each = nY) +
sequence.default(nvec = m:1, from = 0:(m-1L))
}
r@x <-
if(uX)
rep.int(as.double(y), nX)
else {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
as.double(y) * rep(X.ii, each = nY)
}
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "denseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
shape <- .M.shape(X)
uX <- FALSE
uY <- Y@diag != "N"
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
uplo <- r@uplo <- X@uplo
if(shape == "t")
uX <- X@diag != "N"
}
if(!all(dr)) {
r@p <- integer(dr[2L] + 1)
if(uX && uY)
r@diag <- "U"
} else {
m <- dX[1L]
n <- dX[2L]
nX <- length(x <- X@x)
nY <- dY[1L]
if(shape != "g" && nX > 1L && nX == prod(dX)) {
X <- pack(X)
nX <- length(x <- X@x)
}
if(as.double(nX) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
if(uX && !uY) {
diag(X) <- TRUE
nX <- length(x <- X@x)
}
if(is.logical(x) && .M.kind(X) == "n")
x <- x | is.na(x)
if(shape == "g") {
x. <- function() {
j <- rep(1:n, each = nY)
as.vector(`dim<-`(as.double(x), dX)[, j])
}
r@p <- seq.int(0L, by = m, length.out = dr[2L] + 1)
r@i <- rep.int(sequence.default(nvec = rep.int(m, nY),
from = 0:(nY-1L),
by = nY),
n)
r@x <-
if(uY)
x.()
else {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
x.() * rep(Y.ii, each = m)
}
} else if(uplo == "U") {
rep.1.n <- rep(1:n, each = nY)
s <- sequence.default(
nvec = rep.1.n,
from = rep(1L + cumsum(0:(n-1L)), each = nY),
by = 1L)
r@p <- c(0L, cumsum(rep.1.n))
r@i <- sequence.default(nvec = rep.1.n,
from = rep.int(0:(nY-1L), n),
by = nY)
r@x <-
if(uY)
as.double(x)[s]
else
as.double(x)[s] *
rep.int(rep.int(Y@x, n), rep.1.n)
} else {
rep.n.1 <- rep(n:1, each = nY)
s <- sequence.default(
nvec = rep.n.1,
from = rep(1L + cumsum(c(0L, if(n > 1L) n:2)),
each = nY),
by = 1L)
r@p <- c(0L, cumsum(rep.n.1))
r@i <- sequence.default(nvec = rep.n.1,
from = 0:(n*nY-1L),
by = nY)
r@x <-
if(uY)
as.double(x)[s]
else
as.double(x)[s] *
rep.int(rep.int(Y@x, n), rep.n.1)
}
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
} # --NOT YET--
setMethod("kronecker", signature(X = "denseMatrix", Y = "denseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
shape <- switch(.M.shape(X),
g = "g",
t = if(.M.shape(Y) == "t" && X@uplo == Y@uplo)
"t"
else "g",
s = if(.M.shape(Y) == "s")
"s"
else "g")
r <- new(switch(shape,
g = "dgeMatrix",
t = "dtrMatrix",
s = "dsyMatrix"))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && X@diag != "N" && Y@diag != "N")
r@diag <- "U"
}
if(all(dr)) {
X <- .M2m(X)
Y <- .M2m(Y)
storage.mode(X) <- "double"
storage.mode(Y) <- "double"
r@x <-
as.vector(X[rep(seq_len(dX[1L]), each = dY[1L]),
rep(seq_len(dX[2L]), each = dY[2L])]) *
as.vector(Y[rep.int(seq_len(dY[1L]), dX[1L]), ])
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "CsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- X@diag != "N"
uY <- FALSE
shape <- .M.shape(Y)
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- Y@uplo
if(shape == "t" && (uX & (uY <- Y@diag != "N")))
r@diag <- "U"
}
if(!all(dr))
r@p <- integer(dr[2L] + 1)
else {
if(!uX && uY)
Y <- ..diagU2N(Y)
if((nY <- (p <- Y@p)[length(p)]) == 0L)
r@p <- integer(dr[2L] + 1)
else if(as.double(nX <- dX[1L]) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
else {
head. <-
if(length(Y@i) > nY)
function(x) x[seq_len(nY)]
else identity
r@p <- c(0L, cumsum(rep.int(p[-1L] - p[-length(p)], nX)))
r@i <-
rep(seq.int(0L, by = dY[1L], length.out = nX),
each = nY) +
head.(Y@i)
r@x <-
if(uX) {
if(.M.kind(Y) == "n")
rep.int(1, nX * nY)
else rep.int(as.double(head.(Y@x)), nX)
} else {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
if(.M.kind(Y) == "n")
rep(as.double(X.ii), each = nY)
else rep(as.double(X.ii), each = nY) *
as.double(head.(Y@x))
}
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "CsparseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- FALSE
uY <- Y@diag != "N"
shape <- .M.shape(X)
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && (uX <- X@diag != "N") && uY)
r@diag <- "U"
}
if(!all(dr))
r@p <- integer(dr[2L] + 1)
else {
if(uX && !uY)
X <- ..diagU2N(X)
if((nX <- (p <- X@p)[length(p)]) == 0L)
r@p <- integer(dr[2L] + 1)
else if(as.double(nY <- dY[1L]) * nX > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
else {
dp <- p[-1L] - p[-length(p)]
j. <- which(dp > 0L)
nj. <- length(j.)
rep.dp <- rep(dp[j.], each = nY)
s. <- sequence.default(
nvec = rep.dp,
from = rep(1L + p[j.], each = nY),
by = 1L)
r@p <- c(0L, cumsum(rep(dp, each = nY)))
r@i <- (nY * X@i)[s.] +
rep.int(rep.int(0:(nY-1L), nj.), rep.dp)
r@x <-
if(uY) {
if(.M.kind(X) == "n")
rep.int(1, nX * nY)
else as.double(X@x)[s.]
} else {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
if(.M.kind(X) == "n")
rep.int(rep.int(as.double(Y.ii), nj.), rep.dp)
else rep.int(rep.int(as.double(Y.ii), nj.), rep.dp) *
as.double(X@x)[s.]
}
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "CsparseMatrix", Y = "CsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- uY <- FALSE
if((sX <- .M.shape(X)) == "t")
uX <- X@diag != "N"
if((sY <- .M.shape(Y)) == "t")
uY <- Y@diag != "N"
shape <-
switch(sX,
g = "g",
t = if(sY == "t" && X@uplo == Y@uplo) "t" else "g",
s = if(sY == "s") "s" else "g")
cl <- "d.CMatrix"
if(!all(dr <- dX * dY)) {
substr(cl, 2L, 2L) <- shape
r <- new(cl)
r@Dim <- dr
r@p <- integer(dr[2L] + 1)
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && uX && uY)
r@diag <- "U"
}
} else {
substr(cl, 2L, 2L) <- if(shape == "s") "g" else shape
r <- new(cl)
r@Dim <- dr
if(uX)
X <- ..diagU2N(X)
if(uY)
Y <- ..diagU2N(Y)
if(sY == "s")
Y <- .M2gen(Y)
else if(sX == "s")
X <- .M2gen(X)
if((nX <- (pX <- X@p)[length(pX)]) == 0L ||
(nY <- (pY <- Y@p)[length(pY)]) == 0L)
r@p <- integer(dr[2L] + 1)
else if(as.double(nX) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
else {
dpX <- pX[-1L] - (pX. <- pX[-length(pX)])
dpY <- pY[-1L] - (pY. <- pY[-length(pY)])
rep.dpX <- rep(dpX, each = dY[2L])
rep.dpY <- rep.int(dpY, dX[2L])
t1 <- rep.int(rep.dpY, rep.dpX)
s1 <- sequence.default(
nvec = rep.dpX,
from = rep(1L + pX., each = dY[2L]),
by = 1L)
s2 <- sequence.default(
nvec = t1,
from = rep.int(rep.int(1L + pY., dX[2L]), rep.dpX),
by = 1L)
r@p <- c(0L, cumsum(rep.dpX * dpY))
r@i <- rep.int((dY[1L] * X@i)[s1], t1) + Y@i[s2]
r@x <-
if(.M.kind(X) == "n") {
if(.M.kind(Y) == "n")
rep.int(1, nX * nY)
else as.double(Y@x)[s2]
} else {
if(.M.kind(Y) == "n")
rep.int(as.double(X@x)[s1], t1)
else rep.int(as.double(X@x)[s1], t1) *
as.double(Y@x)[s2]
}
}
if(shape == "t") {
r@uplo <- X@uplo
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
} else if(shape == "s")
r <- .Call(R_sparse_force_symmetric, r, X@uplo)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "RsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
.tCRT(kronecker(t(X), .tCRT(Y), FUN, make.dimnames, ...)))
setMethod("kronecker", signature(X = "RsparseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
.tCRT(kronecker(.tCRT(X), t(Y), FUN, make.dimnames, ...)))
setMethod("kronecker", signature(X = "RsparseMatrix", Y = "RsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
.tCRT(kronecker(.tCRT(X), .tCRT(Y), FUN, make.dimnames, ...)))
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "TsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- X@diag != "N"
uY <- FALSE
shape <- .M.shape(Y)
r <- new(`substr<-`("d.TMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- Y@uplo
if(shape == "t" && (uX & (uY <- Y@diag != "N")))
r@diag <- "U"
}
if(all(dr)) {
if(any((kind <- .M.kind(Y)) == c("n", "l")))
Y <- aggregateT(Y)
if(!uX && uY)
Y <- ..diagU2N(Y)
nX <- dX[1L]
nY <- length(Y@i)
r@i <-
rep(seq.int(0L, by = dY[1L], length.out = nX),
each = nY) +
Y@i
r@j <-
rep(seq.int(0L, by = dY[2L], length.out = nX),
each = nY) +
Y@j
r@x <-
if(uX) {
if(kind == "n")
rep.int(1, as.double(nX) * nY)
else rep.int(as.double(Y@x), nX)
} else {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
if(kind == "n")
rep(as.double(X.ii), each = nY)
else rep(as.double(X.ii), each = nY) * as.double(Y@x)
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "TsparseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- FALSE
uY <- Y@diag != "N"
shape <- .M.shape(X)
r <- new(`substr<-`("d.TMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && (uX <- X@diag != "N") && uY)
r@diag <- "U"
}
if(all(dr)) {
if(any((kind <- .M.kind(X)) == c("n", "l")))
X <- aggregateT(X)
if(uX && !uY)
X <- ..diagU2N(X)
nX <- length(X@i)
nY <- dY[1L]
r@i <- rep(nY * X@i, each = nY) + 0:(nY-1L)
r@j <- rep(nY * X@j, each = nY) + 0:(nY-1L)
r@x <-
if(uY) {
if(kind == "n")
rep.int(1, as.double(nX) * nY)
else rep(as.double(X@x), each = nY)
} else {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
if(kind == "n")
rep.int(as.double(Y.ii), nX)
else rep(as.double(X@x), each = nY) * as.double(Y.ii)
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "TsparseMatrix", Y = "TsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- uY <- FALSE
if((sX <- .M.shape(X)) == "t")
uX <- X@diag != "N"
if((sY <- .M.shape(Y)) == "t")
uY <- Y@diag != "N"
shape <-
switch(sX,
g = "g",
t = if(sY == "t" && X@uplo == Y@uplo) "t" else "g",
s = if(sY == "s") "s" else "g")
cl <- "d.TMatrix"
if(!all(dr <- dX * dY)) {
substr(cl, 2L, 2L) <- shape
r <- new(cl)
r@Dim <- dr
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && uX && uY)
r@diag <- "U"
}
} else {
substr(cl, 2L, 2L) <- if(shape == "s") "g" else shape
r <- new(cl)
r@Dim <- dr
if(uX)
X <- ..diagU2N(X)
if(uY)
Y <- ..diagU2N(Y)
if(sY == "s")
Y <- .M2gen(Y)
else if(sX == "s")
X <- .M2gen(X)
nY <- length(Y@i)
r@i <- i. <- rep(dY[1L] * X@i, each = nY) + Y@i
r@j <- rep(dY[2L] * X@j, each = nY) + Y@j
r@x <-
if(.M.kind(X) == "n") {
if(.M.kind(Y) == "n")
rep.int(1, length(i.))
else rep.int(as.double(X@x), nY)
} else {
if(.M.kind(Y) == "n")
rep(as.double(X@x), each = nY)
else rep(as.double(X@x), each = nY) * as.double(Y@x)
}
if(shape == "t") {
r@uplo <- X@uplo
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
} else if(shape == "s")
r <- .Call(R_sparse_force_symmetric, r, X@uplo)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "indMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2kind(Y, "n"), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "indMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.M2kind(X, "n"), Y, FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "indMatrix", Y = "indMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if((margin <- X@margin) != Y@margin)
kronecker(.M2C(X), .M2C(Y), FUN, make.dimnames, ...)
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
r <- new("indMatrix")
r@Dim <- dX * dY
r@perm <-
if(margin == 1L)
rep(dY[2L] * (X@perm - 1L), each = dY[1L]) +
rep.int(Y@perm, dX[1L])
else {
r@margin <- 1L
rep(dY[1L] * (X@perm - 1L), each = dY[2L]) +
rep.int(Y@perm, dX[2L])
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY)))
r@Dimnames <- dnr
r
})
## Catch everything else with these:
for(.cl in c("vector", "matrix")) {
setMethod("kronecker", signature(X = "Matrix", Y = .cl),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .m2dense(Y, ",ge"), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = .cl, Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.m2dense(X, ",ge"), Y, FUN, make.dimnames, ...))
}
rm(.cl)
setMethod("kronecker", signature(X = "denseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.M2C(X), Y, FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "CsparseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2C(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "RsparseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2R(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "TsparseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2T(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2C(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "indMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.M2kind(X, "n"), Y, FUN, make.dimnames, ...))
Try the Matrix package in your browser
Any scripts or data that you put into this service are public.
Matrix documentation built on Jan. 19, 2024, 1:11 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Defines functions kroneckerDN
## METHODS FOR GENERIC: kronecker
## ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
kroneckerDN <- function(dnX, dX = lengths(dnX, FALSE),
dnY, dY = lengths(dnX, FALSE),
sep = ":") {
dnr <- list(NULL, NULL)
if(identical(dnX, dnr) && identical(dnY, dnr))
return(NULL)
for(i in 1:2) {
if(have.sX <- (nY <- dY[i]) && !is.null(sX <- dnX[[i]]))
sX <- rep (sX, each = nY)
if(have.sY <- (nX <- dX[i]) && !is.null(sY <- dnY[[i]]))
sY <- rep.int(sY, times = nX)
dnr[[i]] <-
if(have.sX && have.sY)
paste0(sX, sep, sY)
else if(have.sX)
paste0(sX, sep )
else if(have.sY)
paste0( sep, sY)
else if(nX && nY)
rep.int(sep, nX * nY)
}
dnr
}
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
r <- new("ddiMatrix")
r@Dim <- dX * dY
uX <- X@diag != "N"
uY <- Y@diag != "N"
if(uX && uY)
r@diag <- "U"
else {
if(!uX) {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
}
if(!uY) {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
}
r@x <-
if(uX)
rep.int(as.double(Y.ii), dX[1L])
else if(uY)
rep(as.double(X.ii), each = dY[1L])
else rep(X.ii, each = dY[1L]) * Y.ii
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY)))
r@Dimnames <- dnr
r
})
if(FALSE) { # --NOT YET--
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "denseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- X@diag != "N"
uY <- FALSE
shape <- .M.shape(Y)
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
uplo <- r@uplo <- Y@uplo
if(shape == "t")
uY <- Y@diag != "N"
}
if(!all(dr)) {
r@p <- integer(dr[2L] + 1)
if(uX && uY)
r@diag <- "U"
} else {
m <- dY[1L]
nX <- dX[1L]
nY <- length(y <- Y@x)
if(shape != "g" && nY > 1L && nY == prod(dY)) {
Y <- pack(Y)
nY <- length(y <- Y@x)
}
if(as.double(nX) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
if(!uX && uY) {
diag(Y) <- TRUE
nY <- length(y <- Y@x)
}
if(is.logical(y) && .M.kind(Y) == "n")
y <- y | is.na(y)
if(shape == "g") {
r@p <- seq.int(0L, by = m, length.out = dr[2L] + 1)
r@i <-
rep(seq.int(0L, by = m, length.out = nX),
each = nY) +
0:(m-1L)
} else if(uplo == "U") {
r@p <- c(0L, cumsum(rep.int(1:m, nX)))
r@i <-
rep(seq.int(0L, by = m, length.out = nX),
each = nY) +
sequence.default(nvec = 1:m, from = 0L)
} else {
r@p <- c(0L, cumsum(rep.int(m:1, nX)))
r@i <-
rep(seq.int(0L, by = m, length.out = nX),
each = nY) +
sequence.default(nvec = m:1, from = 0:(m-1L))
}
r@x <-
if(uX)
rep.int(as.double(y), nX)
else {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
as.double(y) * rep(X.ii, each = nY)
}
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "denseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
shape <- .M.shape(X)
uX <- FALSE
uY <- Y@diag != "N"
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
uplo <- r@uplo <- X@uplo
if(shape == "t")
uX <- X@diag != "N"
}
if(!all(dr)) {
r@p <- integer(dr[2L] + 1)
if(uX && uY)
r@diag <- "U"
} else {
m <- dX[1L]
n <- dX[2L]
nX <- length(x <- X@x)
nY <- dY[1L]
if(shape != "g" && nX > 1L && nX == prod(dX)) {
X <- pack(X)
nX <- length(x <- X@x)
}
if(as.double(nX) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
if(uX && !uY) {
diag(X) <- TRUE
nX <- length(x <- X@x)
}
if(is.logical(x) && .M.kind(X) == "n")
x <- x | is.na(x)
if(shape == "g") {
x. <- function() {
j <- rep(1:n, each = nY)
as.vector(`dim<-`(as.double(x), dX)[, j])
}
r@p <- seq.int(0L, by = m, length.out = dr[2L] + 1)
r@i <- rep.int(sequence.default(nvec = rep.int(m, nY),
from = 0:(nY-1L),
by = nY),
n)
r@x <-
if(uY)
x.()
else {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
x.() * rep(Y.ii, each = m)
}
} else if(uplo == "U") {
rep.1.n <- rep(1:n, each = nY)
s <- sequence.default(
nvec = rep.1.n,
from = rep(1L + cumsum(0:(n-1L)), each = nY),
by = 1L)
r@p <- c(0L, cumsum(rep.1.n))
r@i <- sequence.default(nvec = rep.1.n,
from = rep.int(0:(nY-1L), n),
by = nY)
r@x <-
if(uY)
as.double(x)[s]
else
as.double(x)[s] *
rep.int(rep.int(Y@x, n), rep.1.n)
} else {
rep.n.1 <- rep(n:1, each = nY)
s <- sequence.default(
nvec = rep.n.1,
from = rep(1L + cumsum(c(0L, if(n > 1L) n:2)),
each = nY),
by = 1L)
r@p <- c(0L, cumsum(rep.n.1))
r@i <- sequence.default(nvec = rep.n.1,
from = 0:(n*nY-1L),
by = nY)
r@x <-
if(uY)
as.double(x)[s]
else
as.double(x)[s] *
rep.int(rep.int(Y@x, n), rep.n.1)
}
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
} # --NOT YET--
setMethod("kronecker", signature(X = "denseMatrix", Y = "denseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
shape <- switch(.M.shape(X),
g = "g",
t = if(.M.shape(Y) == "t" && X@uplo == Y@uplo)
"t"
else "g",
s = if(.M.shape(Y) == "s")
"s"
else "g")
r <- new(switch(shape,
g = "dgeMatrix",
t = "dtrMatrix",
s = "dsyMatrix"))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && X@diag != "N" && Y@diag != "N")
r@diag <- "U"
}
if(all(dr)) {
X <- .M2m(X)
Y <- .M2m(Y)
storage.mode(X) <- "double"
storage.mode(Y) <- "double"
r@x <-
as.vector(X[rep(seq_len(dX[1L]), each = dY[1L]),
rep(seq_len(dX[2L]), each = dY[2L])]) *
as.vector(Y[rep.int(seq_len(dY[1L]), dX[1L]), ])
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "CsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- X@diag != "N"
uY <- FALSE
shape <- .M.shape(Y)
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- Y@uplo
if(shape == "t" && (uX & (uY <- Y@diag != "N")))
r@diag <- "U"
}
if(!all(dr))
r@p <- integer(dr[2L] + 1)
else {
if(!uX && uY)
Y <- ..diagU2N(Y)
if((nY <- (p <- Y@p)[length(p)]) == 0L)
r@p <- integer(dr[2L] + 1)
else if(as.double(nX <- dX[1L]) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
else {
head. <-
if(length(Y@i) > nY)
function(x) x[seq_len(nY)]
else identity
r@p <- c(0L, cumsum(rep.int(p[-1L] - p[-length(p)], nX)))
r@i <-
rep(seq.int(0L, by = dY[1L], length.out = nX),
each = nY) +
head.(Y@i)
r@x <-
if(uX) {
if(.M.kind(Y) == "n")
rep.int(1, nX * nY)
else rep.int(as.double(head.(Y@x)), nX)
} else {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
if(.M.kind(Y) == "n")
rep(as.double(X.ii), each = nY)
else rep(as.double(X.ii), each = nY) *
as.double(head.(Y@x))
}
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "CsparseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- FALSE
uY <- Y@diag != "N"
shape <- .M.shape(X)
r <- new(`substr<-`("d.CMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && (uX <- X@diag != "N") && uY)
r@diag <- "U"
}
if(!all(dr))
r@p <- integer(dr[2L] + 1)
else {
if(uX && !uY)
X <- ..diagU2N(X)
if((nX <- (p <- X@p)[length(p)]) == 0L)
r@p <- integer(dr[2L] + 1)
else if(as.double(nY <- dY[1L]) * nX > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
else {
dp <- p[-1L] - p[-length(p)]
j. <- which(dp > 0L)
nj. <- length(j.)
rep.dp <- rep(dp[j.], each = nY)
s. <- sequence.default(
nvec = rep.dp,
from = rep(1L + p[j.], each = nY),
by = 1L)
r@p <- c(0L, cumsum(rep(dp, each = nY)))
r@i <- (nY * X@i)[s.] +
rep.int(rep.int(0:(nY-1L), nj.), rep.dp)
r@x <-
if(uY) {
if(.M.kind(X) == "n")
rep.int(1, nX * nY)
else as.double(X@x)[s.]
} else {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
if(.M.kind(X) == "n")
rep.int(rep.int(as.double(Y.ii), nj.), rep.dp)
else rep.int(rep.int(as.double(Y.ii), nj.), rep.dp) *
as.double(X@x)[s.]
}
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "CsparseMatrix", Y = "CsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- uY <- FALSE
if((sX <- .M.shape(X)) == "t")
uX <- X@diag != "N"
if((sY <- .M.shape(Y)) == "t")
uY <- Y@diag != "N"
shape <-
switch(sX,
g = "g",
t = if(sY == "t" && X@uplo == Y@uplo) "t" else "g",
s = if(sY == "s") "s" else "g")
cl <- "d.CMatrix"
if(!all(dr <- dX * dY)) {
substr(cl, 2L, 2L) <- shape
r <- new(cl)
r@Dim <- dr
r@p <- integer(dr[2L] + 1)
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && uX && uY)
r@diag <- "U"
}
} else {
substr(cl, 2L, 2L) <- if(shape == "s") "g" else shape
r <- new(cl)
r@Dim <- dr
if(uX)
X <- ..diagU2N(X)
if(uY)
Y <- ..diagU2N(Y)
if(sY == "s")
Y <- .M2gen(Y)
else if(sX == "s")
X <- .M2gen(X)
if((nX <- (pX <- X@p)[length(pX)]) == 0L ||
(nY <- (pY <- Y@p)[length(pY)]) == 0L)
r@p <- integer(dr[2L] + 1)
else if(as.double(nX) * nY > .Machine$integer.max)
stop(gettextf("number of nonzero entries cannot exceed %s", "2^31-1"),
domain = NA)
else {
dpX <- pX[-1L] - (pX. <- pX[-length(pX)])
dpY <- pY[-1L] - (pY. <- pY[-length(pY)])
rep.dpX <- rep(dpX, each = dY[2L])
rep.dpY <- rep.int(dpY, dX[2L])
t1 <- rep.int(rep.dpY, rep.dpX)
s1 <- sequence.default(
nvec = rep.dpX,
from = rep(1L + pX., each = dY[2L]),
by = 1L)
s2 <- sequence.default(
nvec = t1,
from = rep.int(rep.int(1L + pY., dX[2L]), rep.dpX),
by = 1L)
r@p <- c(0L, cumsum(rep.dpX * dpY))
r@i <- rep.int((dY[1L] * X@i)[s1], t1) + Y@i[s2]
r@x <-
if(.M.kind(X) == "n") {
if(.M.kind(Y) == "n")
rep.int(1, nX * nY)
else as.double(Y@x)[s2]
} else {
if(.M.kind(Y) == "n")
rep.int(as.double(X@x)[s1], t1)
else rep.int(as.double(X@x)[s1], t1) *
as.double(Y@x)[s2]
}
}
if(shape == "t") {
r@uplo <- X@uplo
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
} else if(shape == "s")
r <- .Call(R_sparse_force_symmetric, r, X@uplo)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "RsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
.tCRT(kronecker(t(X), .tCRT(Y), FUN, make.dimnames, ...)))
setMethod("kronecker", signature(X = "RsparseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
.tCRT(kronecker(.tCRT(X), t(Y), FUN, make.dimnames, ...)))
setMethod("kronecker", signature(X = "RsparseMatrix", Y = "RsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
.tCRT(kronecker(.tCRT(X), .tCRT(Y), FUN, make.dimnames, ...)))
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "TsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- X@diag != "N"
uY <- FALSE
shape <- .M.shape(Y)
r <- new(`substr<-`("d.TMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- Y@uplo
if(shape == "t" && (uX & (uY <- Y@diag != "N")))
r@diag <- "U"
}
if(all(dr)) {
if(any((kind <- .M.kind(Y)) == c("n", "l")))
Y <- aggregateT(Y)
if(!uX && uY)
Y <- ..diagU2N(Y)
nX <- dX[1L]
nY <- length(Y@i)
r@i <-
rep(seq.int(0L, by = dY[1L], length.out = nX),
each = nY) +
Y@i
r@j <-
rep(seq.int(0L, by = dY[2L], length.out = nX),
each = nY) +
Y@j
r@x <-
if(uX) {
if(kind == "n")
rep.int(1, as.double(nX) * nY)
else rep.int(as.double(Y@x), nX)
} else {
X.ii <- X@x
if(.M.kind(X) == "n" && anyNA(X.ii))
X.ii <- X.ii | is.na(X.ii)
if(kind == "n")
rep(as.double(X.ii), each = nY)
else rep(as.double(X.ii), each = nY) * as.double(Y@x)
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "TsparseMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- FALSE
uY <- Y@diag != "N"
shape <- .M.shape(X)
r <- new(`substr<-`("d.TMatrix", 2L, 2L, shape))
r@Dim <- dr <- dX * dY
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && (uX <- X@diag != "N") && uY)
r@diag <- "U"
}
if(all(dr)) {
if(any((kind <- .M.kind(X)) == c("n", "l")))
X <- aggregateT(X)
if(uX && !uY)
X <- ..diagU2N(X)
nX <- length(X@i)
nY <- dY[1L]
r@i <- rep(nY * X@i, each = nY) + 0:(nY-1L)
r@j <- rep(nY * X@j, each = nY) + 0:(nY-1L)
r@x <-
if(uY) {
if(kind == "n")
rep.int(1, as.double(nX) * nY)
else rep(as.double(X@x), each = nY)
} else {
Y.ii <- Y@x
if(.M.kind(Y) == "n" && anyNA(Y.ii))
Y.ii <- Y.ii | is.na(Y.ii)
if(kind == "n")
rep.int(as.double(Y.ii), nX)
else rep(as.double(X@x), each = nY) * as.double(Y.ii)
}
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "TsparseMatrix", Y = "TsparseMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
uX <- uY <- FALSE
if((sX <- .M.shape(X)) == "t")
uX <- X@diag != "N"
if((sY <- .M.shape(Y)) == "t")
uY <- Y@diag != "N"
shape <-
switch(sX,
g = "g",
t = if(sY == "t" && X@uplo == Y@uplo) "t" else "g",
s = if(sY == "s") "s" else "g")
cl <- "d.TMatrix"
if(!all(dr <- dX * dY)) {
substr(cl, 2L, 2L) <- shape
r <- new(cl)
r@Dim <- dr
if(shape != "g") {
r@uplo <- X@uplo
if(shape == "t" && uX && uY)
r@diag <- "U"
}
} else {
substr(cl, 2L, 2L) <- if(shape == "s") "g" else shape
r <- new(cl)
r@Dim <- dr
if(uX)
X <- ..diagU2N(X)
if(uY)
Y <- ..diagU2N(Y)
if(sY == "s")
Y <- .M2gen(Y)
else if(sX == "s")
X <- .M2gen(X)
nY <- length(Y@i)
r@i <- i. <- rep(dY[1L] * X@i, each = nY) + Y@i
r@j <- rep(dY[2L] * X@j, each = nY) + Y@j
r@x <-
if(.M.kind(X) == "n") {
if(.M.kind(Y) == "n")
rep.int(1, length(i.))
else rep.int(as.double(X@x), nY)
} else {
if(.M.kind(Y) == "n")
rep(as.double(X@x), each = nY)
else rep(as.double(X@x), each = nY) * as.double(Y@x)
}
if(shape == "t") {
r@uplo <- X@uplo
if(uX && uY)
r <- ..diagN2U(r, sparse = TRUE)
} else if(shape == "s")
r <- .Call(R_sparse_force_symmetric, r, X@uplo)
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY))) {
if(shape == "s" && !isSymmetricDN(dnr))
r <- .M2gen(r)
r@Dimnames <- dnr
}
r
})
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "indMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2kind(Y, "n"), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "indMatrix", Y = "diagonalMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.M2kind(X, "n"), Y, FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "indMatrix", Y = "indMatrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...) {
if((margin <- X@margin) != Y@margin)
kronecker(.M2C(X), .M2C(Y), FUN, make.dimnames, ...)
if(!(missing(FUN) || identical(FUN, "*")))
stop(gettextf("'%s' method must use default %s=\"%s\"",
"kronecker", "FUN", "*"),
domain = NA)
if(any(as.double(dX <- X@Dim) * (dY <- Y@Dim) >
.Machine$integer.max))
stop(gettextf("dimensions cannot exceed %s", "2^31-1"),
domain = NA)
r <- new("indMatrix")
r@Dim <- dX * dY
r@perm <-
if(margin == 1L)
rep(dY[2L] * (X@perm - 1L), each = dY[1L]) +
rep.int(Y@perm, dX[1L])
else {
r@margin <- 1L
rep(dY[1L] * (X@perm - 1L), each = dY[2L]) +
rep.int(Y@perm, dX[2L])
}
if(make.dimnames &&
!is.null(dnr <- kroneckerDN(dimnames(X), dX, dimnames(Y), dY)))
r@Dimnames <- dnr
r
})
## Catch everything else with these:
for(.cl in c("vector", "matrix")) {
setMethod("kronecker", signature(X = "Matrix", Y = .cl),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .m2dense(Y, ",ge"), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = .cl, Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.m2dense(X, ",ge"), Y, FUN, make.dimnames, ...))
}
rm(.cl)
setMethod("kronecker", signature(X = "denseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.M2C(X), Y, FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "CsparseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2C(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "RsparseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2R(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "TsparseMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2T(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "diagonalMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(X, .M2C(Y), FUN, make.dimnames, ...))
setMethod("kronecker", signature(X = "indMatrix", Y = "Matrix"),
function(X, Y, FUN = "*", make.dimnames = FALSE, ...)
kronecker(.M2kind(X, "n"), Y, FUN, make.dimnames, ...))
Try the Matrix package in your browser
Any scripts or data that you put into this service are public.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.