https://github.com/hadley/dplyr
Tip revision: c105a015e651a4b4cb75a8a46ae37b4a7677d9a9 authored by hadley on 20 February 2014, 15:53:17 UTC
Update R CMD check notes
Update R CMD check notes
Tip revision: c105a01
over.R
# Generate SQL expression for window function
# over("avg(x)", frame = c(-Inf, 0))
# over("avg(x)", order = "y")
over <- function(expr, partition = NULL, order = NULL, frame = NULL) {
args <- (!is.null(partition)) + (!is.null(order)) + (!is.null(frame))
if (args == 0) {
stop("Must supply at least one of partition, order, frame", call. = FALSE)
}
if (!is.null(partition)) {
partition <- build_sql("PARTITION BY ",
sql_vector(partition, collapse = ", "))
}
if (!is.null(order)) {
order <- build_sql("ORDER BY ", sql_vector(order, collapse = ", "))
}
if (!is.null(frame)) {
if (is.numeric(frame)) frame <- rows(frame[1], frame[2])
frame <- build_sql("ROWS ", frame)
}
over <- sql_vector(compact(list(partition, order, frame)), parens = TRUE)
build_sql(expr, " OVER ", over)
}
rows <- function(from = -Inf, to = 0) {
if (from >= to) stop("from must be less than to", call. = FALSE)
dir <- function(x) if (x < 0) "PRECEDING" else "FOLLOWING"
val <- function(x) if (is.finite(x)) as.integer(abs(x)) else "UNBOUNDED"
bound <- function(x) {
if (x == 0) return("CURRENT ROW")
paste(val(x), dir(x))
}
if (to == 0) {
sql(bound(from))
} else {
sql(paste0("BETWEEN ", bound(from), " AND ", bound(to)))
}
}