R/dplyr-arrange.R

In sa-lee/plyranges: A fluent interface for manipulating GenomicRanges

rng_arrange <- function(.data, dots) {
  rng_os <- overscope_ranges(.data)
  rng_list <- lapply(dots, rlang::eval_tidy, data = rng_os)

  if (length(rng_list) == 1L) {
    idx <- order(rng_list[[1]])
  } else {
    idx <- Reduce(order, rng_list)
  }
  .data[idx, ]
}
#' Sort a Ranges object
#'
#' @param .data A Ranges object.
#' @param ... Comma seperated list of variable names.
#'
#' @importFrom dplyr arrange
#' @examples
#' rng <- as_iranges(data.frame(start = 1:10, width = 10:1))
#' rng <- mutate(rng, score = runif(10))
#' arrange(rng, score)
#' # you can also use dplyr::desc to arrange by descending order
#' @rdname ranges-arrange
#' @return A sorted Ranges object
#' @method arrange Ranges
#' @export
arrange.Ranges <- function(.data, ...) {
  dots <- rlang::enquos(...)
  rng_arrange(.data, dots)
}

#' @method arrange DelegatingGenomicRanges
#' @export
arrange.DelegatingGenomicRanges <- function(.data, ...) {
  dots <- rlang::enquos(...)
  delegate <- .data@delegate
  .data@delegate <- rng_arrange(delegate, dots)
  .data
}

#' @method arrange DelegatingIntegerRanges
#' @export
arrange.DelegatingIntegerRanges <- arrange.DelegatingGenomicRanges
# -- TODO: allow sorting by group?