R/bracket.r

In float: 32-Bit Floats

#' Extract
#' 
#' Extract subsets of a float vector/matrix.
#' 
#' @param x
#' A float vector/matrix.
#' @param i,j,...
#' The indices.  Most combinations of integer/double/logical values will be
#' treated the same as R does. One major difference is that \code{NA} values
#' will not be tolerated.
#' @param drop
#' Logical. If TRUE, single column matrices will be treated as one-dimensional
#' vectors.
#' @param value
#' The replacement value.
#' 
#' @return
#' A float vector/matrix.
#' 
#' @examples
#' \dontrun{
#' library(float)
#' 
#' s = flrunif(10, 3)
#' s[, -1]
#' s[c(1, 3, 5, 7), 1:2]
#' }
#' 
#' @name bracket
#' @rdname bracket
NULL



bracket_float32_vec = function(x, i, drop=TRUE)
{
  DATA(x)[i, drop=drop]
}

bracket_float32_mat = function(x, i, j, drop=TRUE)
{
  if (missing(i))
    DATA(x)[, j, drop=drop]
  else if (missing(j))
  {
    if (is.matrix(i))
      DATA(x)[i, drop=drop]
    else
      DATA(x)[i, , drop=drop]
  }
  else
    DATA(x)[i, j, drop=drop]
}

bracket_float32 = function(x, i, j, drop=TRUE)
{
  if (missing(i) && missing(j))
    return(x)
  
  if (isavec(x))
  {
    if (missing(i))
      stop("incorrect number of dimensions")
    
    d = bracket_float32_vec(x, i, drop=drop)
  }
  else
    d = bracket_float32_mat(x, i, j, drop=drop)
  
  if (!is.null(rownames(x)))
    rownames(d) = rownames(x)[i]
  if (!is.null(names(x)))
    names(d) = names(x)[j]
  
  float32(d)
}

bracket_replace_float32 = function(x, i, j, ..., value)
{
  if (is.double(value))
  {
    x = dbl(x)
    if (missing(i) && missing(j))
      x[, ] = value
    else if (missing(i))
      x[, j] = value
    else if (missing(j))
    {
      if (!is.matrix(x) || is.matrix(i))
        x[i] = value
      else
        x[i, ] = value
    }
    else
      x[i, j] = value
    
    return(x)
  }
  else if (is.integer(value))
    value = fl(value)
  
  
  if (missing(i) && missing(j))
    x@Data[, ] = DATA(value)
  else if (missing(i))
    x@Data[, j] = DATA(value)
  else if (missing(j))
  {
    if (isavec(x) || is.matrix(i))
      x@Data[i] = DATA(value)
    else
      x@Data[i, ] = DATA(value)
  }
  else
    x@Data[i, j] = DATA(value)
  
  x
}



#' @rdname bracket
#' @export
setMethod("[", signature(x="float32"), bracket_float32)

#' @rdname bracket
#' @export
setReplaceMethod("[", signature(x="float32", value="ANY"), bracket_replace_float32)