R/stat_node_helpers.r

In ggsankeyfier: Create Sankey and Alluvial Diagrams Using 'ggplot2'

.compute_panel_statnodes <- function(self, data, params, scales) {

  if (missing(params)) params <- .setup_params_position(self, data)

  data |>
    .add_node_id() |>
    .group_across("PANEL", "x", "group", "connector", "node_id") |>
    dplyr::summarise(
      dplyr::across(-dplyr::any_of(c("y", "edge_id")), ~{
        if (is.numeric(.)) sum(.) else .[[1]]
      }),
      edge_id = list(.data[["edge_id"]]),
      y       = sum(.data[["y"]]), .groups = "keep") |>
    .group_across("PANEL", "x", "group") |>
    dplyr::mutate(node_size  = .data[["y"]]) |>
    dplyr::ungroup()
}

.add_node_id <- function(data) {
  data |>
    .group_across("x", "group") |>
    dplyr::mutate(node_id = dplyr::cur_group_id()) |>
    dplyr::ungroup()
}