R/summary.CIMTx_ATT_posterior.R

In CIMTx: Causal Inference for Multiple Treatments with a Binary Outcome

#' Summarize a CIMTx_ATT_posterior object
#'
#' @param object a \code{CIMTx_ATT_posterior} object
#' obtained with \code{\link{ce_estimate}} function.
#' @param ... further arguments passed to or from other methods.
#'
#' @return a list with w-1 elements for ATT effect.
#' Each element of the list contains the estimation, standard error,
#' lower and upper 95\% CI for RD/RR/OR.
#' @importFrom stringr str_detect str_sub
#' @export
#'
#' @references
#' Hadley Wickham (2019).
#' \emph{stringr: Simple, Consistent Wrappers for Common String Operations}.
#' R package version 1.4.0.
#' URL:\url{https://CRAN.R-project.org/package=stringr}
#'
#' @examples
#' lp_w_all <-
#'   c(
#'     ".4*x1 + .1*x2  - .1*x4 + .1*x5", # w = 1
#'     ".2 * x1 + .2 * x2  - .2 * x4 - .3 * x5"
#'   ) # w = 2
#' nlp_w_all <-
#'   c(
#'     "-.5*x1*x4  - .1*x2*x5", # w = 1
#'     "-.3*x1*x4 + .2*x2*x5"
#'   ) # w = 2
#' lp_y_all <- rep(".2*x1 + .3*x2 - .1*x3 - .1*x4 - .2*x5", 3)
#' nlp_y_all <- rep(".7*x1*x1  - .1*x2*x3", 3)
#' X_all <- c(
#'   "rnorm(0, 0.5)", # x1
#'   "rbeta(2, .4)", # x2
#'   "runif(0, 0.5)", # x3
#'   "rweibull(1,2)", # x4
#'   "rbinom(1, .4)" # x5
#' )
#'
#' set.seed(111111)
#' data <- data_sim(
#'   sample_size = 300,
#'   n_trt = 3,
#'   x = X_all,
#'   lp_y = lp_y_all,
#'   nlp_y = nlp_y_all,
#'   align = FALSE,
#'   lp_w = lp_w_all,
#'   nlp_w = nlp_w_all,
#'   tau = c(-1.5, 0, 1.5),
#'   delta = c(0.5, 0.5),
#'   psi = 1
#' )
#' ce_estimate_ra_att_result <- ce_estimate(
#'   y = data$y, x = data$covariates,
#'   w = data$w, reference_trt = 1, ndpost = 10, method = "RA", estimand = "ATT"
#' )
#' summary(ce_estimate_ra_att_result)
summary.CIMTx_ATT_posterior <- function(object, ...) {
  object <- object[stringr::str_detect(names(object), "ATT")]
  reference_trt <-
    as.integer(stringr::str_sub(names(object)[1], 7, 7))
  trt_indicator_no_reference <-
    unique(as.integer(stringr::str_sub(names(object), 8, 8)))
  result_summary <- NULL
  for (j in seq_len(length(trt_indicator_no_reference))) {
    assign(
      paste0("ATT", reference_trt, trt_indicator_no_reference[j]),
      posterior_summary(object[[paste0("ATT_RD", reference_trt,
                                       trt_indicator_no_reference[j])]],
                        object[[paste0("ATT_RR", reference_trt,
                                       trt_indicator_no_reference[j])]],
                        object[[paste0("ATT_OR", reference_trt,
                                       trt_indicator_no_reference[j])]])
    )

    assign(paste0("ATT", reference_trt, trt_indicator_no_reference[j]),
           list(round(eval(parse(
             text = (
               paste0("ATT", reference_trt, trt_indicator_no_reference[j])
             )
           )), digits = 2)))
    assign(
      paste0("ATT", reference_trt, trt_indicator_no_reference[j]),
      stats::setNames(
        eval(parse(text = (
          paste0("ATT", reference_trt, trt_indicator_no_reference[j])
        ))),
        paste0("ATT", reference_trt, trt_indicator_no_reference[j])
      )
    )
    result_summary <-
      c(result_summary, (eval(parse(text = (
        paste0("ATT", reference_trt, trt_indicator_no_reference[j])
      )))))
  }
  return(result_summary)
}