R/rmixweibull.R

In GaryBAYLOR/mixR: Finite Mixture Modeling for Raw and Binned Data

#' Generating Random Data From A Weibull Mixture Model
#'
#' The function \code{rmixweibull} generates random data from a normal Weibull model.
#'
#' The number of random data from each component \eqn{n_0} (a vector) is generated from a multinomial
#' distribution Multinom\eqn{(n, pi)}. Then the random data from each component is generated with
#' the sample sized specified in \eqn{n_0} and parameters of Weibull distributions specified in
#' \code{mu} and \code{sd}.
#'
#' @param n a positive integer specifying the number of observations we want to generate from the mixture model
#' @param pi a numeric vector for the proportion of each component
#' @param mu a numeric vector for the mean of each component
#' @param sd a numeric vector for the standard deviation of each component
#'
#' @return The function \code{rmixweibull} returns a numeric vector of random data from the specified Weibull mixture model.
#'
#' @seealso \code{\link{rmixnormal}}, \code{\link{rmixgamma}}, \code{\link{rmixlnorm}}
#'
#' @examples
#' x <- rmixweibull(1000, c(0.4, 0.6), c(2, 5), c(1, 0.5))
#' hist(x, breaks = 40)
#'
#' @export
rmixweibull <- function(n, pi, mu, sd) {
  check_inputs_for_rmix(n, pi, mu, sd)
  if(any(mu <= 0)) stop("'mu' should be positive!")

  n0 <- rmultinom(1, n, pi)[, 1]
  pars <- to_k_lambda_weibull(mu, sd)
  k <- pars[[1]]
  lambda <- pars[[2]]
  res <- numeric(0)
  for(i in 1:length(pi)) {
    tmp <- rweibull(n0[i], k[i], lambda[i])
    res <- c(res, tmp)
  }
  res
}