R/fill_complementary_probs.R

In n8thangreen/CEdecisiontree: Cost-effectiveness decision tree analysis

#' Fill Complementary Probabilities
#'
#' Only one of each pair of branches is assigned a probability
#' and then the other event probability is filled-in afterwards.
#' This is good because specify fewer input values and
#' if sampling probabilities we don't know the complementary
#' probability.
#'
#' Only works for binary trees or when one of n is missing.
#'
#' @template args-dat_long
#'
#' @return Long format tree object
#' @export
#'
#' @examples
#' dat <- tibble::tribble(~from, ~to, ~prob, ~vals,
#'                        1,  2,  NA,    10,
#'                        1,  3,  0.8,    1,
#'                        2,  4,  0.2,   10,
#'                        2,  5,  NA,     1,
#'                        3,  6,  0.2,   10,
#'                        3,  7,  0.8,    1)
#' fill_complementary_probs(dat)
#'
fill_complementary_probs <- function(dat_long) {

  dat_long |>
    group_by(.data$from) |>
    mutate(prob = ifelse(is.na(prob),
                         1 - sum(prob, na.rm = TRUE),
                         prob)) |>
    ungroup()
}