R/get_transformed.R

In trafo: Estimation, Comparison and Selection of Transformations

#' Function that bundles the return of a trafo object
#' 
#' @keywords internal


get_transformed <- function(trafo, ans, y, lambda, custom_func, 
                            custom_func_std, 
                            custom_family) {
  
  if (trafo == "boxcoxshift") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- box_cox_shift(y = y, lambda = lambda)$y
    ans$zt <- box_cox_shift_std(y = y, lambda = lambda)
    
    # Save transformation family
    ans$family <- "Box-Cox shift"
  } else if (trafo == "boxcox") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- box_cox(y = y, lambda = lambda)
    ans$zt <- box_cox_std(y = y, lambda = lambda)
    
    # Save transformation family
    ans$family <- "Box-Cox"
  } else if (trafo == "bickeldoksum") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- Bick_dok(y = y, lambda = lambda)
    ans$zt <- Bick_dok_std(y = y, lambda = lambda)
    
    # Save transformation family 
    ans$family <- "Bickel-Doksum"
    
  } else if (trafo == "manly") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- Manly(y = y, lambda = lambda)
    ans$zt <- Manly_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- "Manly"
  } else if (trafo == "modulus") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- modul(y = y, lambda = lambda)
    ans$zt <- modul_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- "Modulus"
  } else if (trafo == "dual") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- Dual(y = y, lambda = lambda)
    ans$zt <- Dual_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- "Dual"
  } else if (trafo == "yeojohnson") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- Yeo_john(y = y, lambda = lambda)
    ans$zt <- Yeo_john_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- "Yeo-Johnson"
  } else if (trafo == "logshiftopt") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- log_shift_opt(y = y, lambda = lambda)
    ans$zt <- log_shift_opt_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- "Log shift opt"
  } else if (trafo == "sqrtshift") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- sqrt_shift(y = y, lambda = lambda)
    ans$zt <- sqrt_shift_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- "Square root shift"
  } else if (trafo == "gpower") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- gPower(y = y, lambda = lambda)
    ans$zt <- gPower_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- "Gpower"
  } else if (trafo == "custom") {
    # Get vector of transformed and standardized transformed variable
    ans$yt <- custom_func(y = y, lambda = lambda)
    ans$zt <- custom_func_std(y = y, lambda = lambda)
    
    # Save transformation family and method
    ans$family <- custom_family
  } 
  
  return(ans)
}