R/create_ce_tree_df.R

In n8thangreen/CEdecisiontree: Cost-effectiveness decision tree analysis

#' Create cost-effectiveness decision tree data frame input
#'
#' Using look up table, rather than separate lists,
#' of values and labels.
#' Can prepare these external to R and read them in.
#'
#' @param label_branch_tab Look up table
#' @param label_val_tab Look up table
#' @param tree_struc List of parent child branches
#'
#' @return tibble in long format
#' @importFrom stats setNames
#' @import dplyr
#' @export
#'
create_ce_tree_df <- function(label_branch_tab,
                              label_val_tab,
                              tree_struc) {

  # data frame version of tree structure list
  tree_struc_long <-
    melt(tree_struc) |>
    setNames(c("to", "from")) |>
    relocate(from)

  merge(label_branch_tab, label_val_tab,
        by = c("unit", "label")) |>
    dcast(from + to ~ unit, value.var = "val") |>
    merge(tree_struc_long, all = TRUE) |>
    arrange(from, to) |>
    fill_complementary_probs()
}