R/power.4m.R

Defines functions power.4m

Documented in power.4m

#' Budget and/or sample size, power, MDES calculation for
#' four-level MRTs detecting main effects
#'
#' @description This function can calculate required budget for desired power,
#'     power or minimum detectable effect size (MDES) under fixed budget
#'     for four-level multisite randomized trials (MRTs).
#'     It also can perform conventional power analyses
#'     (e.g., required sample size, power, and MDES calculation).
#'
#' @inheritParams power.4
#'
#' @param expr Returned objects from function \code{\link{od.4m}}; default is NULL;
#'     if \code{expr} is specified, parameter values of \code{icc2}, \code{icc3}, \code{icc4},
#'     \code{r12}, \code{r22}, \code{r32}, \code{r42m},
#'     \code{c1}, \code{c2}, \code{c3}, \code{c4},
#'     \code{c1t}, \code{c2t}, \code{c3t}, \code{p}, \code{n}, \code{J}, and \code{K}
#'     used or solved in function \code{\link{od.4m}} will
#'     be passed to current function;
#'     only the values of \code{p}, \code{n}, \code{J}, and/or \code{K} that specified or solved in
#'     function \code{\link{od.4m}} can be overwritten
#'     if \code{constraint} is specified.
#' @param constraint The constrained values of \code{p}, \code{n}, \code{J},
#'     and/or \code{K} in list format to overwrite
#'     those from \code{expr}; default value is NULL.
#' @param omega The standardized variance of site-specific treatment effect.
#' @param r42m The proportion of variance of site-specific treatment effect explained by covariates.
#' @param c4 The cost of sampling one level-4 unit.
#' @param p The proportion of level-3 units to be assigned to treatment.
#' @param q The number of covariates at level 4.
#' @param mlim The range for searching the root of budget (\code{m}) numerically,
#'     default is the costs sampling \code{Llim} level-4 units
#'     or c(4 * Lcost, 1e+10 * Lcost) with Lcost =
#'     ((1 - p) * (c1 * n * J * K + c2 * J * K + c3 * K) +
#'     p * (c1t * n * J * K + c2t * J * K + c3t * K) + c4.
#'
#' @return Required budget (and/or required level-4 sample size), statistical power, or MDES
#'     depending on the specification of parameters.
#'     The function also returns the function name, design type,
#'     and parameters used in the calculation.
#'
#' @export power.4m
#'
#' @examples
#' # Unconstrained optimal design #---------
#'   myod1 <- od.4m(icc2 = 0.2, icc3 = 0.1, icc4 = 0.05, omega = 0.02,
#'               r12 = 0.5, r22 = 0.5, r32 = 0.5, r42m = 0.5,
#'               c1 = 1, c2 = 5, c3 = 25,
#'               c1t = 1, c2t = 50, c3t = 250, c4 = 500,
#'               varlim = c(0, 0.005))
#'   myod1$out # n = 8.3, J = 3.2, K = 4.9, p = 0.36
#'
#' # ------- Power analyses by default considering costs and budget -------
#' # Required budget and sample size
#'   mym.1 <- power.4m(expr = myod1, d = 0.2, q = 1, power = 0.8)
#'   mym.1$out  # m = 30201, L = 20.6
#'   # mym.1$par  # parameters and their values used for the function
#' # Or, equivalently, specify every argument in the function
#'   mym.1 <- power.4m(d = 0.2, power = 0.8, q = 1,
#'                  icc2 = 0.2, icc3 = 0.1, icc4 = 0.05, omega = 0.02,
#'                  r12 = 0.5, r22 = 0.5, r32 = 0.5, r42m = 0.5,
#'                  c1 = 1, c2 = 5, c3 = 25,
#'                  c1t = 1, c2t = 50, c3t = 250, c4 = 500,
#'                  n = 8, J = 3, K = 5, p = 0.36)
#' # Required budget and sample size with constrained p
#'   mym.2 <- power.4m(expr = myod1, d = 0.2, q = 1, power = 0.8,
#'                  constraint = list(p = 0.5))
#'   mym.2$out  # m = 33183, L = 19.3
#' # Required budget and sample size with constrained p and n
#'   mym.3 <- power.4m(expr = myod1, d = 0.2, q = 1, power = 0.8,
#'                  constraint = list(p = 0.5, n = 20))
#'   mym.3$out  # m = 34262, L = 18.0
#'
#' # Power calculation
#'   mypower <- power.4m(expr = myod1, q = 1, d = 0.2, m = 30201)
#'   mypower$out  # power = 0.80
#' # Power calculation under constrained p (p = 0.5)
#'   mypower.1 <- power.4m(expr = myod1, q = 1, d = 0.2, m = 30201,
#'                  constraint = list(p = 0.5))
#'   mypower.1$out  # power = 0.76
#'
#' # MDES calculation
#'   mymdes <- power.4m(expr = myod1, q = 1, power = 0.80, m = 30201)
#'   mymdes$out  # d = 0.20
#'
#'
#' # ------- Conventional power analyses with cost.model = FALSE-------
#' # Required sample size
#'   myL <- power.4m(cost.model = FALSE, expr = myod1, d = 0.2, q = 1, power = 0.8)
#'   myL$out  # L = 20.6
#'   # myL$par  # parameters and their values used for the function
#' # Or, equivalently, specify every argument in the function
#'   myL <- power.4m(cost.model = FALSE, d = 0.2, power = 0.8, q = 1,
#'                   icc2 = 0.2, icc3 = 0.1, icc4 = 0.05, omega = 0.02,
#'                  r12 = 0.5, r22 = 0.5, r32 = 0.5, r42m = 0.5,
#'                  c1 = 1, c2 = 5, c3 = 25,
#'                  c1t = 1, c2t = 50, c3t = 250, c4 = 500,
#'                  n = 8, J = 3, K = 5, p = 0.36)
#'
#' # Power calculation
#'   mypower1 <- power.4m(cost.model = FALSE, expr = myod1, L = 20.6, d = 0.2, q = 1)
#'   mypower1$out  # power = 0.80
#'
#' # MDES calculation
#'   mymdes1 <- power.4m(cost.model = FALSE, expr = myod1, L = 20.6, power = 0.8, q = 1)
#'   mymdes1$out  # d = 0.20
#'
power.4m <- function(cost.model = TRUE, expr = NULL, constraint = NULL,
                    sig.level = 0.05, two.tailed = TRUE,
                    d = NULL, power = NULL, m = NULL,
                    n = NULL, J = NULL, K = NULL, L = NULL, p = NULL,
                    icc2 = NULL, icc3 = NULL, icc4 = NULL,
                    r12 = NULL, r22 = NULL, r32 = NULL, r42m = NULL, q = NULL,
                    c1 = NULL, c2 = NULL, c3 = NULL, c4 = NULL,
                    c1t = NULL, c2t = NULL, c3t = NULL, omega = NULL,
                    dlim = NULL, powerlim = NULL, Llim = NULL, mlim = NULL,
                    rounded = TRUE) {
  funName <- "power.4m"
  designType <- "four-level MRTs"
  if (cost.model == TRUE) {
    if (sum(sapply(list(m, d, power), is.null)) != 1)
      stop("exactly one of 'm', 'd', and 'power' must be NULL
           when cost.model is TRUE")
    if (!is.null(L))
      stop("'L' must be NULL when cost.model is TRUE")
  } else {
    if (sum(sapply(list(L, d, power), is.null)) != 1)
      stop("exactly one of 'L', 'd', and 'power' must be NULL
           when cost.model is FALSE")
    if (!is.null(m))
      stop("'m' must be NULL when cost.model is FALSE")
  }
  if (!is.null(expr)) {
    if (expr$funName != "od.4m") {
      stop("'expr' can only be NULL or
           the return from the function of 'od.4m'")
    } else {
      if (sum(sapply(list(icc2, icc3, icc4, r12, r22, r32, r42m, c1, c2, c3, c4,
                          c1t, c2t, c3t, omega, n, J, K, p), function(x) {!is.null(x)})) >= 1)
        stop("parameters of 'icc2', 'icc3', 'icc4', 'r12', 'r22', 'r32', 'r42m',
             'c1', 'c2', 'c3', 'c4', 'c1t', 'c2t', 'c3t', 'omega', 'n', 'J', 'K', and 'p'
             have been specified in expr of 'od.4m'")
      icc2 <- expr$par$icc2
      icc3 <- expr$par$icc3
      icc4 <- expr$par$icc4
      r12 <- expr$par$r12
      r22 <- expr$par$r22
      r32 <- expr$par$r32
      r42m <- expr$par$r42m
      c1 <- expr$par$c1
      c2 <- expr$par$c2
      c3 <- expr$par$c3
      c4 <- expr$par$c4
      c1t <- expr$par$c1t
      c2t <- expr$par$c2t
      c3t <- expr$par$c3t
      omega <- expr$par$omega
      if (rounded == TRUE) {
        n <- round(expr$out$n, 0)
        J <- round(expr$out$J, 0)
        K <- round(expr$out$K, 0)
        p <- round(expr$out$p, 2)
      } else {
        n <- expr$out$n
        J <- expr$out$J
        K <- expr$out$K
        p <- expr$out$p
      }
    }
  } else {
    if (!is.null(constraint))
      stop("'constraint' must be NULL when 'expr' is NULL")
  }
  NumberCheck <- function(x) {!is.null(x) && !is.numeric(x)}
  if (!is.null(constraint) && !is.list(constraint))
    stop("'constraint' must be in list format
         (e.g., constraint = list(p = 0.5, n = 20))")
  if (length(constraint) > 4)
    stop("'constraint' must be limited to 'n', 'J', 'K', and/or 'p'")
  if (!is.null(constraint$n)) {
    if(NumberCheck(constraint$n) || constraint$n <= 0)
      stop("constrained 'n' must be numeric with n > 0")
    n <- constraint$n
  }
  if (!is.null(constraint$J)) {
    if(NumberCheck(constraint$J) || constraint$J <= 0)
      stop("constrained 'J' must be numeric with J > 0")
    J <- constraint$J
  }
  if (!is.null(constraint$K)) {
    if(NumberCheck(constraint$K) || constraint$K <= 0)
      stop("constrained 'K' must be numeric with K > 0")
    K <- constraint$K
  }
  if (!is.null(constraint$p)) {
    if(NumberCheck(constraint$p) ||
       any (0 >= constraint$p | constraint$p >= 1))
      stop("constrained 'p' must be numeric in (0, 1)")
    p <- constraint$p
  }
  if (sum(sapply(list(icc2, icc3, icc4, p, power, sig.level), function(x) {
    NumberCheck(x) || any(0 > x | x >= 1)
  })) >= 1) stop("'icc2', 'icc3', 'icc4', 'p', 'power', and 'sig.level'
                 must be numeric in (0, 1)")
  if (sum(sapply(list(r12, r22, r32, r42m), function(x) {
    NumberCheck(x) || any(0 > x | x >= 1)
  })) >= 1) stop("'r12', 'r22', 'r32', and 'r42m' must be numeric in [0, 1)")
  if (cost.model == TRUE){
    if (sum(sapply(list(c1, c2, c3, c4, c1t, c2t, c3t), function(x) {
      NumberCheck(x) || x < 0})) >= 1)
      stop("'c1', 'c2', 'c3', 'c4', 'c1t', 'c2t', 'c3t' must be numeric in [0, Inf)")
    if (NumberCheck(m))
      stop("'m' must be numeric in [0, Inf)")
  }
  if (NumberCheck(q) | q < 0)
    stop("'q' must be numeric with q >= 0")
  if (NumberCheck(n) || n <= 0)
    stop("'n' must be numeric with n > 0")
  if (NumberCheck(J) || J <= 0)
    stop("'J' must be numeric with J > 0")
  if (NumberCheck(K) || K <= 0)
    stop("'K' must be numeric with K > 0")
  if (NumberCheck(d) || any(0 > d | d > 5))
    stop("'d' must be numeric in [0, 5],
         please transfer negative effect size to positive one if needed")
  par <- list(cost.model = cost.model,
              sig.level = sig.level,
              two.tailed = two.tailed,
              d = d, icc2 = icc2, icc3 = icc3, icc4 = icc4,
              r12 = r12, r22 = r22, r32 = r32, r42m = r42m,
              c1 = c1, c2 = c2, c3 = c3, c4 = c4,
              c1t = c1t, c2t = c2t, c3t = c3t, omega = omega,
              n = n, J = J, K = K, L = L, p = p,
              q = q, m = m, power = power)
  tside <- ifelse(two.tailed == TRUE, 2, 1)
  if (cost.model == TRUE) {
    Lcost <- ((1 - p) * (c1 * n * J * K + c2 * J * K + c3 * K) +
                p * (c1t * n * J * K + c2t * J * K + c3t * K) + c4)
    if (two.tailed == TRUE) {
      pwr.expr <- quote({
        L <- m / ((1 - p) * K * (c1 * n * J + c2 * J + c3) +
                    p * K * (c1t * n * J + c2t * J + c3t) + c4);
        lambda <- d * sqrt((p * (1 - p) * n * J * K * m / ((1 - p) * K * (c1 * n * J + c2 * J + c3) +
                                          p * K * (c1t * n * J + c2t * J + c3t) + c4)) /
          (p * (1 - p) * n * J * K * omega * (1 - r42m) + n * J * icc3 * (1 - r32) +
                 n * icc2 * (1 - r22)  +
                 (1 - icc2 - icc3 - icc4) * (1 - r12)));
        1 - pt(qt(1 - sig.level / tside, df = (L - q - 1)),
               df = (L - q - 1), lambda) +
          pt(qt(sig.level / tside, df = (L - q -1)),
             df = (L - q -1), lambda)
      })
    } else {
      pwr.expr <- quote({
        L <- m / ((1 - p) * K * (c1 * n * J + c2 * J + c3) +
                    p * K * (c1t * n * J + c2t * J + c3t) + c4);
        lambda <- d * sqrt((p * (1 - p) * n * J * K * m / ((1 - p) * K * (c1 * n * J + c2 * J + c3) +
                                                            p * K * (c1t * n * J + c2t * J + c3t) + c4)) /
          (p * (1 - p) * n * J * K * omega * (1 - r42m) + n * J * icc3 * (1 - r32) +
                 n * icc2 * (1 - r22)  +
                 (1 - icc2 - icc3 - icc4) * (1 - r12)));
        1 - pt(qt(1 - sig.level / tside, df = (L - q -1)),
               df = (L - q -1), lambda)
      })
    }
  } else {
    if (two.tailed == TRUE) {
      pwr.expr <- quote({
        lambda <- d * sqrt(p * (1 - p) * n * J * K *L) /
          sqrt(p * (1 - p) * n * J * K * omega * (1 - r42m) + n * J * icc3 * (1 - r32) +
                 n * icc2 * (1 - r22)  +
                 (1 - icc2 - icc3 - icc4) * (1 - r12));
        1 - pt(qt(1 - sig.level / tside, df = (L - q -1)),
               df = (L - q -1), lambda) +
          pt(qt(sig.level / tside, df = (L - q -1)),
             df = (L - q -1), lambda)
      })
    } else {
      pwr.expr <- quote({
        lambda <- d * sqrt(p * (1 - p) * n * J * K *L) /
          sqrt(p * (1 - p) * n * J * K * omega * (1 - r42m) + n * J * icc3 * (1 - r32) +
                 n * icc2 * (1 - r22)  +
                 (1 - icc2 - icc3 - icc4) * (1 - r12));
        1 - pt(qt(1 - sig.level / tside, df = (L - q -1)),
               df = (L - q -1), lambda)
      })
    }
  }
  limFun <- function(x, y) {
    if (!is.null(x) && length(x) == 2 && is.numeric(x)) {x} else {y}
  }
  Llim <- limFun(x = Llim, y = c(4, 1e+10))
  powerlim <- limFun(x = powerlim, y = c(1e-10, 1 - 1e-10))
  dlim <- limFun(x = dlim, y = c(0, 5))
  if(cost.model == TRUE) {
    mlim <- limFun(x = mlim, y = c(Llim[1] * Lcost, Llim[2] * Lcost))
    if (is.null(power)) {
      out <- list(power = eval(pwr.expr))
    } else if (is.null(m)) {
      out <- list(m = stats::uniroot(function(m)
        eval(pwr.expr) - power, mlim)$root)
      out <- c(out, list(L = out$m / ((1 - p) * (c1 * n * J * K + c2 * J * K + c3 * K) +
                                        p * (c1t * n * J * K + c2t * J * K + c3t * K) + c4)))
    } else if (is.null(d)) {
      out <- list(d = stats::uniroot(function(d)
        eval(pwr.expr) - power, dlim)$root)
    }
  } else {
    if (is.null(power)) {
      out <- list(power = eval(pwr.expr))
    } else if (is.null(L)) {
      out <- list(L = stats::uniroot(function(L)
        eval(pwr.expr) - power, Llim)$root)
    } else if (is.null(d)) {
      out <- list(d = stats::uniroot(function(d)
        eval(pwr.expr) - power, dlim)$root)
    }
  }
  power.out <- list(funName = funName,
                    designType = designType,
                    par = par, out = out)
  return(power.out)
}

Try the odr package in your browser

Any scripts or data that you put into this service are public.

odr documentation built on Aug. 8, 2023, 5:13 p.m.