R/LPJmLGridData_subset.R

In lpjmlkit: Toolkit for Basic LPJmL Handling

# subset method roxygen documentation in LPJmlData.R
LPJmLGridData$set("private",
                  ".subset_space",
                  function(subset_list) {

    if (any(c("coords", "coordinates") %in% names(subset_list))) {
      # Get term used for subsetting ("coords" or "coordinates" legit)
      coords <- c("coords", "coordinates")[
        c("coords", "coordinates") %in% names(subset_list)
      ]

      self$.__set_data__(
        subset_array_pair(x = private$.data,
                          pair = subset_list[[coords]])
      )

      subset_list$lat <- subset_list[[coords]]$lat
      subset_list$lon <- subset_list[[coords]]$lon
      subset_list[[coords]] <- NULL

    } else {

      self$.__set_data__(
        subset_array(
          private$.data,
          subset_list,
          drop = FALSE
        )
      )
    }

    # New dimnames attribute of cell dimension needs to be passed to the
    # `.__update_subset__` method in LPJmLMetaData to update the number of
    # cells as well as the (new) firstcell if subsetting by cell
    if (private$.meta$._space_format_ == "cell") {
      cell_dimnames <- self$dimnames()$cell

    } else {
      cell_dimnames <- sort(private$.data) %>%
        format(trim = TRUE, scientific = FALSE, justify = "none")
    }

    private$.meta$.__update_subset__(
      subset_list,
      cell_dimnames = cell_dimnames
    )

    return(invisible(self))
  }
)