R/qexp_mix.R

In matheuscastro43/finiteMix: Finite Mixture Models

qexp_mix <- function(p, pi, rate, lower.tail = TRUE, log.p = FALSE){
  if(length(p) == 1){
    g <- length(pi)
    pi = pi/sum(pi)
    if(min(c(pi, rate)) > 0 && length(rate) == 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){
              pexp_mix(q, pi, rate) - p
            }
            while(aux >= 0.9*U){
              U = 2*U
              aux = uniroot(h, lower = 0, upper = U)$root
            }
            return(aux)
            if(aux > 0){return(aux)
            }else{return(0)}
          }
        }
      }
    }
    else stop("The parametric space must be respected.")
  }else{
    j = function(p){qexp_mix(p, pi, rate, lower.tail, log.p)}
    return(sapply(p, j))
  }
}