R/qweibull_mix.R

In matheuscastro43/finiteMix: Finite Mixture Models

qweibull_mix <- function(p, pi, shape, scale, lower.tail = TRUE, log.p = FALSE){
  if(length(p) == 1){
    g <- length(pi)
    pi = pi/sum(pi)
    if(min(c(pi, shape, scale)) > 0 && length(shape) == g &&
       length(scale) == g){
      if(log.p){
        p = exp(p)
      }
      if(!lower.tail){
        p = 1 - p
      }
      if(p < 0 || p > 1){
        warning("The probability must be in (0, 1) interval.")
        return(NaN)
      }
      else{
        if(p == 0){
          return(0)
        }
        else{
          if(p == 1){
            return(Inf)
          }
          else{
            U = aux = 100
            h = function(q){
              pweibull_mix(q, pi, shape, scale) - p
            }
            while(aux >= 0.9*U){
              U = 2*U
              aux = uniroot(h, lower = 0, upper = U)$root
            }
            return(aux)
          }
        }
      }
    }
    else stop("The parametric space must be respected.")
    
  }else{
    j = function(p){qweibull_mix(p, pi, shape, scale, lower.tail, log.p)}
    return(sapply(p, j))
  }
}