1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
|
#' @export
#' @importFrom methods new is
ScaledMatrixSeed <- function(x, center=NULL, scale=NULL) {
if (missing(x)) {
x <- matrix(0, 0, 0)
} else if (is(x, "ScaledMatrixSeed")) {
return(x)
}
use_center <- !is.null(center)
use_scale <- !is.null(scale)
new("ScaledMatrixSeed", .matrix=x, center=as.numeric(center), scale=as.numeric(scale), use_center=use_center, use_scale=use_scale, transposed=FALSE)
}
#' @importFrom S4Vectors setValidity2
setValidity2("ScaledMatrixSeed", function(object) {
msg <- character(0)
# Checking scalars.
if (length(use_center(object))!=1L) {
msg <- c(msg, "'use_center' must be a logical scalar")
}
if (length(use_scale(object))!=1L) {
msg <- c(msg, "'use_scale' must be a logical scalar")
}
if (length(is_transposed(object))!=1L) {
msg <- c(msg, "'transposed' must be a logical scalar")
}
# Checking vectors.
if (use_center(object) && length(get_center(object))!=ncol(object)) {
msg <- c(msg, "length of 'center' must equal 'ncol(object)'")
}
if (use_scale(object) && length(get_scale(object))!=ncol(object)) {
msg <- c(msg, "length of 'scale' must equal 'ncol(object)'")
}
if (length(msg)) {
return(msg)
}
return(TRUE)
})
#' @export
#' @importFrom methods show
setMethod("show", "ScaledMatrixSeed", function(object) {
cat(sprintf("%i x %i ScaledMatrixSeed object", nrow(object), ncol(object)),
sprintf("representation: %s", class(get_matrix2(object))),
sprintf("centering: %s", if (use_center(object)) "yes" else "no"),
sprintf("scaling: %s", if (use_scale(object)) "yes" else "no"),
sep="\n")
})
###################################
# Internal getters.
get_matrix2 <- function(x) x@.matrix
get_center <- function(x) x@center
get_scale <- function(x) x@scale
use_center <- function(x) x@use_center
use_scale <- function(x) x@use_scale
is_transposed <- function(x) x@transposed
###################################
# DelayedArray support utilities.
#' @export
setMethod("dim", "ScaledMatrixSeed", function(x) {
d <- dim(get_matrix2(x))
if (is_transposed(x)) { d <- rev(d) }
d
})
#' @export
setMethod("dimnames", "ScaledMatrixSeed", function(x) {
d <- dimnames(get_matrix2(x))
if (is_transposed(x)) { d <- rev(d) }
d
})
#' @export
#' @importFrom DelayedArray extract_array
setMethod("extract_array", "ScaledMatrixSeed", function(x, index) {
x2 <- subset_ScaledMatrixSeed(x, index[[1]], index[[2]])
realize_ScaledMatrixSeed(x2)
})
###################################
# Other utilities.
rename_ScaledMatrixSeed <- function(x, value) {
if (is_transposed(x)) value <- rev(value)
dimnames(x@.matrix) <- value
x
}
transpose_ScaledMatrixSeed <- function(x) {
x@transposed <- !is_transposed(x)
x
}
#' @importFrom Matrix t as.matrix
#' @importFrom methods is
realize_ScaledMatrixSeed <- function(x, ...) {
out <- get_matrix2(x)
if (use_scale(x) || use_center(x)) {
if (is(out, "ScaledMatrix")) {
# Any '-' and '/' would collapse this to a DelayedArray,
# which would then call extract_array, which would then
# call realize_ScaledMatrixSeed, forming an infinite loop.
# So we might as well realize it now.
out <- realize_ScaledMatrixSeed(seed(out))
}
out <- t(out)
if (use_center(x)) {
out <- out - get_center(x)
}
if (use_scale(x)) {
out <- out / get_scale(x)
}
if (!is_transposed(x)) out <- t(out)
} else {
if (is_transposed(x)) out <- t(out)
}
as.matrix(out)
}
subset_ScaledMatrixSeed <- function(x, i, j) {
if (is_transposed(x)) {
x2 <- transpose_ScaledMatrixSeed(x)
x2 <- subset_ScaledMatrixSeed(x2, i=j, j=i)
return(transpose_ScaledMatrixSeed(x2))
}
if (!is.null(i)) {
x@.matrix <- get_matrix2(x)[i,,drop=FALSE]
}
if (!is.null(j)) {
if (is.character(j)) {
j <- match(j, colnames(x))
}
x@.matrix <- get_matrix2(x)[,j,drop=FALSE]
if (use_scale(x)) {
x@scale <- get_scale(x)[j]
}
if (use_center(x)) {
x@center <- get_center(x)[j]
}
}
return(x)
}
|