/
select.R
39 lines (32 loc) · 1.06 KB
/
select.R
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
################################################################################
#' @inherit dplyr::select title description return
#'
#' @param .data A [FDF][FDF-class].
#' @inheritParams tidyselect::vars_select
#'
#' @importFrom dplyr select
#' @importFrom tidyselect vars_select vars_select_helpers
#' @export
#' @method select FDF
#'
#' @rdname select
#'
#' @examples
#' (test <- FDF(datasets::iris))
#' select(test, 1:4)
#' select(test, -5)
#' select(test, -Species)
#' select(test, c("Sepal.Length", "Sepal.Width"))
#' select(test, Sepal.Length, Sepal.Width)
#' select(test, starts_with("Sepal"))
select.FDF <- function(.data, ...) {
dots <- lapply(quos(...), rlang::env_bury, !!!vars_select_helpers)
ind_vars <- .data$ind_col
var_names <- vars_select(names(ind_vars), !!!dots)
.data$copy(ind_col = ind_vars[var_names])
}
################################################################################
#' @exportMethod select
#' @rdname select
setGeneric("select", dplyr::select)
################################################################################