From 14fc721e783b32d353022cf1e82d18bd8d2c119d Mon Sep 17 00:00:00 2001
From: "Huhn, Monika" <Monika.Huhn@astrazeneca.com>
Date: Mon, 4 Mar 2024 23:08:32 +0000
Subject: [PATCH] Add forest plot calculation

---
 R/aaa.R                      |  3 +++
 R/internal.R                 | 48 +++++++++++++++++++++++++++++++++---
 R/maraca.R                   | 10 +++++---
 tests/testthat/test_maraca.R |  3 ++-
 4 files changed, 56 insertions(+), 8 deletions(-)

diff --git a/R/aaa.R b/R/aaa.R
index 9597d77..9377b05 100644
--- a/R/aaa.R
+++ b/R/aaa.R
@@ -53,3 +53,6 @@ utils::globalVariables("WIN_A")
 utils::globalVariables("WIN_P")
 utils::globalVariables("TIE_A")
 utils::globalVariables("linetype")
+utils::globalVariables("wins")
+utils::globalVariables("losses")
+utils::globalVariables("ties")
diff --git a/R/internal.R b/R/internal.R
index 61c8afc..a77716a 100644
--- a/R/internal.R
+++ b/R/internal.R
@@ -12,7 +12,9 @@
 }
 
 # Computes the win odds from the internal data.
-.compute_win_odds <- function(hce_dat, arm_levels) {
+.compute_win_odds <- function(hce_dat, arm_levels,
+                              step_outcomes, last_outcome) {
+
   hce_dat <- base::as.data.frame(hce_dat)
   hce_dat <- .with_ordered_column(hce_dat)
   fit <- hce::calcWO(x = hce_dat, AVAL = "ordered",
@@ -27,8 +29,45 @@
                                      ref = unname(arm_levels["control"]),
                                      GROUP = "outcome")
 
+  endpoints <- c(step_outcomes, last_outcome)
+  hce_dat <- hce_dat %>%
+    dplyr::mutate_at(dplyr::vars(outcome), factor, levels = c(endpoints, "X"))
+
+  calcs_lst <- lapply(seq_along(endpoints), function(x) {
+    idx <- !(hce_dat$outcome %in% endpoints[1:x])
+    hce_dat[idx, "outcome"] <- "X"
+    hce_dat[idx, "ordered"] <- 1000000
+    wins <- hce::calcWINS(hce_dat, AVAL = "ordered", TRTP = "arm",
+                          ref = unname(arm_levels["control"]),
+                          GROUP = "outcome")
+    wo <- hce::summaryWO(hce_dat, AVAL = "ordered", TRTP = "arm",
+                           ref = unname(arm_levels["control"]),
+                           GROUP = "outcome")
+    list("wins" = wins, "wo" = wo)
+  })
+
+  wins_forest <- do.call("rbind", lapply(calcs_lst, function(c_lst) {
+    wins <- c_lst$wins
+    nm <- c("value", "LCL", "UCL", "p value")
+    rbind(data.frame(setNames(wins$WO, nm), "method" = "win odds"),
+          data.frame(setNames(wins$WR1, nm), "method" = "win ratio"))
+  }))
+
+  wo_bar <- do.call("rbind", lapply(seq_along(calcs_lst), function(i) {
+    wo <- head(calcs_lst[[i]]$wo$summary, 1)
+    wo$outcome <- endpoints[i]
+    wo %>%
+      dplyr::rename(dplyr::all_of(c(wins = "WIN", losses = "LOSS",
+                                    ties = "TIE"))) %>%
+      tidyr::pivot_longer(cols = c(wins, losses, ties)) %>%
+      dplyr::mutate_at(dplyr::vars(name), factor,
+                       levels = c("wins", "losses", "ties"))
+  }))
+
   return(list("win_odds" = win_odds,
-              "win_odds_outcome" = win_odds_outcome))
+              "win_odds_outcome" = win_odds_outcome,
+              "wins_forest" = wins_forest,
+              "wo_bar" = wo_bar))
 
 }
 
@@ -333,8 +372,9 @@
                      x = base::sum(value, na.rm = TRUE),
                      average = 100 *
                        as.numeric(stats::prop.test(x, n)$estimate),
-                     se = abs(average - (100 *
-                       as.numeric(stats::prop.test(x, n)$conf.int)[1]))) %>%
+                     se = abs(average -
+                                (100 * as.numeric(
+                                  stats::prop.test(x, n)$conf.int)[1]))) %>%
     dplyr::ungroup()
 
   # To create ellipsis shape and avoid overlapping between both of them,
diff --git a/R/maraca.R b/R/maraca.R
index 71b00ef..5c3423d 100644
--- a/R/maraca.R
+++ b/R/maraca.R
@@ -196,9 +196,11 @@ maraca <- function(
     data_last_outcome <- NULL
   }
 
-  win_odds <- list("win_odds" = NULL, "win_odds_outcome" = NULL)
+  win_odds <- list("win_odds" = NULL, "win_odds_outcome" = NULL,
+                   "wins_forest" = NULL, "wo_bar" = NULL)
   if (compute_win_odds) {
-    win_odds <- .compute_win_odds(hce_dat, arm_levels)
+    win_odds <- .compute_win_odds(hce_dat, arm_levels,
+                                  step_outcomes, last_outcome)
   }
 
   return(
@@ -215,7 +217,9 @@ maraca <- function(
         ecdf_by_outcome = ecdf_by_outcome,
         data_last_outcome = data_last_outcome,
         win_odds = win_odds[["win_odds"]],
-        win_odds_outcome = win_odds[["win_odds_outcome"]]
+        win_odds_outcome = win_odds[["win_odds_outcome"]],
+        wins_forest = win_odds[["wins_forest"]],
+        wo_bar = win_odds[["wo_bar"]]
       ),
       class = c("maraca")
     )
diff --git a/tests/testthat/test_maraca.R b/tests/testthat/test_maraca.R
index 7bd314a..b6900d0 100644
--- a/tests/testthat/test_maraca.R
+++ b/tests/testthat/test_maraca.R
@@ -516,7 +516,8 @@ test_that("winOddsData", {
     arm_levels, column_names = column_names
   )
 
-  win_odds_list <- .compute_win_odds(data, arm_levels)
+  win_odds_list <- .compute_win_odds(data, arm_levels,
+                                     step_outcomes, last_outcome)
   win_odds <- win_odds_list[["win_odds"]]
 
   expect_equal(class(win_odds), "numeric")