R/differential-length-dim-names.r

In salad: Simple Automatic Differentiation

# --------------- length and dimension

length.differential <- function(x) length(unclass(x)[[1]])

dim.differential <- function(x) dim(unclass(x)[[1]])

`dim_differential<-` <- function(x, value) { 
  x <- unclass(x)
  for(i in seq_along(x)) dim(x[[i]]) <- value
  class(x) <- "differential"
  x
}

# --------------- dimnames

dimnames.differential <- function(x) dimnames(x[[1]])

`dimnames_differential<-` <- function(x, value) {
  x <- unclass(x)
  if(!is.null(dim(x[[1]]))) { # matrices
    for(i in seq_along(x)) dimnames(x[[i]]) <- value
  }
  class(x) <- "differential"
  x
}

# --------------- names 

names.differential <- function(x) names(x[[1]])

`names_differential<-` <- function(x, value) {
  x <- unclass(x)
  if(is.null(dim(x[[1]]))) { # vecteurs
    for(i in seq_along(x)) names(x[[i]]) <- value
  }
  class(x) <- "differential"
  x
}