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
|
#' @export
#' @importFrom dplyr arrange
arrange.tbl_graph <- function(.data, ...) {
.data <- unfocus(.data)
.register_graph_context(.data)
d_tmp <- as_tibble(.data)
check_reserved(d_tmp)
orig_ind <- seq_len(nrow(d_tmp))
d_tmp$.tbl_graph_index <- orig_ind
d_tmp <- arrange(d_tmp, ...)
switch(
active(.data),
nodes = permute_nodes(.data, d_tmp$.tbl_graph_index),
edges = permute_edges(.data, d_tmp$.tbl_graph_index)
) %gr_attr% .data
}
#' @export
#' @importFrom dplyr arrange
arrange.morphed_tbl_graph <- function(.data, ...) {
.data[] <- lapply(.data, arrange, ...)
.data
}
#' @export
dplyr::arrange
#' @importFrom igraph is_directed as_data_frame
permute_edges <- function(graph, order) {
graph_mod <- as_data_frame(graph, what = 'both')
graph_mod$edges <- graph_mod$edges[order, ]
as_tbl_graph(graph_mod, directed = is_directed(graph))
}
#' @importFrom igraph permute
permute_nodes <- function(graph, order) {
permute(graph, match(seq_along(order), order))
}
|
