inst/shiny/server/get/getStep7Results.R

In ahnjedid/dataprep-test: ONDRI Data Prep ShinyApp

# getStep7Results.R
#
# Purpose: Return the imputed data data frame after step #7 is complete.
#
# Date: 2020-04-15
#
# ==============================================================================

getStep7Results <- function(input, rvStep6Results, rvVarDataTypes){
  results <- rvStep6Results()
  types <- rvVarDataTypes()
  
  transDataDF <- results$transDataDF
  asIsDF <- results$asIsDF
  
  imputedDataDF <- NULL
  
  if (input$selectImputeOpt7 == "yes"){
    
    # All continuous/numeric variables.
    if (length(types$catVars) == 0 && length(types$ordVars) == 0){
      if (sum(is.na(transDataDF)) > 0){
        # OLD CODE before manual input of number of components through slider.
        # imputeComponentMax <- ncol(transDataDF) - 1
        # ncpTest <- missMDA::estim_ncpPCA(transDataDF, 
        #                                  ncp.max = imputeComponentMax)
        # output <- missMDA::imputePCA(transDataDF, ncp = ncpTest$ncp)
        # imputedDataDF <- as.data.frame(output$completeObs)
        imputedDataDF <- as.data.frame(
          missMDA::imputePCA(transDataDF, ncp = input$numComponents7, 
                             scale = TRUE)$completeObs
        )
      }
      else{
        imputedDataDF <- transDataDF
      }
    }
    # Mix of categorical, ordinal, and continuous/numeric variables.
    else{
      if (sum(is.na(transDataDF)) > 0){
        imputedDataDF <- as.data.frame(
          missMDA::imputeCA(transDataDF, ncp = input$numComponents7),
          stringsAsFactors = FALSE
        )
      }
      else{
        imputedDataDF <- transDataDF
      }
    }
    
  }
  # Skip imputation step. Leave NA's as they are.
  else if (input$selectImputeOpt7 == "no"){
    imputedDataDF <- transDataDF
  }
  
  # LAST STEP: Bind back ONDRI standard columns and AS IS variables to rest
  # of data.
  DF_PREPPED_DATA <- cbind(asIsDF, imputedDataDF) %>% 
    rownames_to_column(var = "SUBJECT")
  
  return (DF_PREPPED_DATA)
}

# [END]