scratchpad/aux_distributions.R

In meghapsimatrix/wildmeta: Cluster Wild Bootstrapping for Meta-Analysis

return_wts <- function(auxiliary_dist, cluster_var) {

  if(auxiliary_dist == "Rademacher"){

    wts <- sample(c(-1, 1),
                  size = length(cluster_var),
                  replace = TRUE,
                  prob = rep(1/2, 2))

  } else if(auxiliary_dist == "Mammen"){

   wts <- sample(c(- (sqrt(5) - 1)/2,
                   (sqrt(5) + 1)/2),
                 size = length(cluster_var),
                 replace = TRUE,
                 prob = c((sqrt(5) + 1) /(2 * sqrt(5)), (sqrt(5) - 1) /(2 * sqrt(5)) ))

  } else if(auxiliary_dist == "Webb six"){

   wts <- sample(c(-sqrt(3/2),
                   -sqrt(2/2),
                   -sqrt(1/2),
                   sqrt(1/2),
                   sqrt(2/2),
                   sqrt(3/2)),
                 size = length(cluster_var),
                 replace = TRUE,
                 prob = rep(1/6, 6))

  } else if(auxiliary_dist = "uniform"){

    wts <- runif(n = length(cluster_var),
                 min = -sqrt(3),
                 max = sqrt(3))

  } else if(auxiliary_dist = "standard normal"){

    wts <- rnorm(n = length(cluster_var))

  }

  return(wts)

}