R/extract.R

In SticsRPacks/CroPlotR: A Package to Analyze Crop Model Simulations Outputs with Plots and Statistics

#' `[` method for `cropr_simulation`
#'
#' This method ensure keeping the `cropr_simulation` attribute when subsetting a
#' `cropr_simulation` list.
#'
#' @param x A `cropr_simulation` list
#' @param ... An index
#'
#' @return A subset of a `cropr_simulation`, keeping its attribute
#' @export
#'
#' @examples
#' \dontrun{
#' library(SticsRFiles)
#' sim <- SticsRFiles::get_sim(workspace = "inst/extdata/stics_example_1")
#' sim[[1]] # returns a `cropr_simulation` list
#' }
`[.cropr_simulation` <- function(x, ...) {
  l <- unclass(x)[...]
  attr_names <- names(attributes(x))
  attr_names <- attr_names[attr_names != "names"]
  attributes(l)[attr_names] <- attributes(x)[attr_names]
  return(l)
}