a bit more just extending outputs to include absolute probabilities (as well as contrasts) - mainly changing results_stats(). Not quite finished but should work.

* need to return probabilities from ALD_stats()

Author: n8thangreen

R/IPD_stats.R

@@ -94,9 +94,10 @@ IPD_stat_factory <- function(ipd_fun) {
 
     out <- ipd_fun(strategy, ipd, ald, ...)
 
+    # relative treatment effect
     hat.delta.AC <- calculate_ate(out$mean_A, out$mean_C,
                                   effect = scale)
-
+    
     coef_est <- mean(hat.delta.AC)
 
     if (var_method == "sandwich") {
@@ -106,8 +107,17 @@ IPD_stat_factory <- function(ipd_fun) {
       var_est <- var(hat.delta.AC)
     }
 
-    list(mean = coef_est,
-         var = var_est)
+    p_est <- sapply(out, mean)
+    p_var <- sapply(out, var)
+    
+    list(
+      contrasts = list(
+        mean = coef_est,
+        var = var_est),
+      absolute = list(
+        mean = p_est,
+        var = p_var)
+    )
   }
 }
 

R/result_stats.R

@@ -8,34 +8,54 @@
 #' @keywords internal
 #'
 result_stats <- function(AC_stats,
-                           BC_stats,
-                           CI = 0.95) {
+                         BC_stats,
+                         CI = 0.95) {
   upper <- 0.5 + CI/2
   ci_range <- c(1-upper, upper)
   z_vals <- qnorm(ci_range)
 
+  AC_contrasts <- AC_stats$contrasts
+  AC_absolute <- AC_stats$absolute
+  
+  # contrasts
+  
   contrasts <- list(
-    AB = AC_stats$mean - BC_stats$mean,
-    AC = AC_stats$mean,
+    AB = AC_contrasts$mean - BC_stats$mean,
+    AC = AC_contrasts$mean,
     BC = BC_stats$mean)
 
   contrast_variances <- list(
-    AB = AC_stats$var + BC_stats$var,
-    AC = AC_stats$var,
+    AB = AC_contrasts$var + BC_stats$var,
+    AC = AC_contrasts$var,
     BC = BC_stats$var)
 
   contrast_ci <- list(
     AB = contrasts$AB + z_vals*as.vector(sqrt(contrast_variances$AB)),
     AC = contrasts$AC + z_vals*as.vector(sqrt(contrast_variances$AC)),
     BC = contrasts$BC + z_vals*as.vector(sqrt(contrast_variances$BC)))
 
+  # absolute values
+  
+  absolute <- list(
+    A = AC_absolute$mean["mean_A"],
+    # B = AB_absolute$mean["mean_B"],
+    C = AC_absolute$mean["mean_C"]
+  )
+  
+  absolute_var <- list(
+    A = AC_absolute$var["mean_A"],
+    # B = AB_absolute$var["mean_B"],
+    C = AC_absolute$var["mean_C"]
+  )
+  
   list(
     contrasts = list(
       means = contrasts,
       variances = contrast_variances,
       CI = contrast_ci),
     absolute = list(   ##TODO:
-      means = contrasts,
-      variances = contrast_variances,
-      CI = contrast_ci))
+      means = absolute,
+      variances = absolute_var
+      # CI = contrast_ci
+    ))
 }