R/BANOVA.floodlight.R

In BANOVA: Hierarchical Bayesian ANOVA Models

BANOVA.floodlight <-
  function(sol, var_numeric, var_factor, flood_values = list()){
    if(class(sol) %in% c('BANOVA', 'BANOVA.Normal', 'BANOVA.T', 'BANOVA.Poisson', 'BANOVA.Bern', 
                         'BANOVA.Bin', 'BANOVA.ordMultinomial', 'BANOVA.Multinomial', 'BANOVA.truncNormal')){
      
      if (sol$single_level){
        if (sol$model_name == 'BANOVA.Multinomial'){
          sol_tables <- floodlight.analysis(sol, 
                                            var_numeric, 
                                            var_factor, 
                                            sol$samples_l1_param, 
                                            sol$dMatrice$X_full[[1]], 
                                            sol$dMatrice$Z, 
                                            dataX= sol$dataX, 
                                            dataZ = sol$dataZ, 
                                            flood_values = flood_values)
        } else if(sol$model_name == 'BANOVA.multiNormal') {
          sol_tables <- list()
          for (i in 1:sol$num_depenent_variables){
            name <- sol$names_of_dependent_variables[i]
            title <- paste0("\nFloodlight analysis for ", name,"\n")
            sol_tables[[name]] <- floodlight.analysis(sol, 
                                                      var_numeric, 
                                                      var_factor, 
                                                      sol$samples_l1.list[[i]], 
                                                      sol$dMatrice$X, 
                                                      sol$dMatrice$Z, 
                                                      data = sol$data, 
                                                      flood_values = flood_values)
            
          }
        } else{
          sol_tables <- floodlight.analysis(sol, 
                                            var_numeric, 
                                            var_factor, 
                                            sol$samples_l1_param, 
                                            sol$dMatrice$X, 
                                            sol$dMatrice$Z, 
                                            data = sol$data, 
                                            flood_values = flood_values)
        }
      }else{
        if (sol$model_name == 'BANOVA.Multinomial'){
          sol_tables <- floodlight.analysis(sol, 
                                            var_numeric, 
                                            var_factor, 
                                            sol$samples_l2_param, 
                                            sol$dMatrice$X_full[[1]], 
                                            sol$dMatrice$Z, 
                                            dataX = sol$dataX, 
                                            dataZ = sol$dataZ, 
                                            flood_values = flood_values)
        } else if(sol$model_name == 'BANOVA.multiNormal') {
          sol_tables <- list()
          for (i in 1:sol$num_depenent_variables){
            name <- sol$names_of_dependent_variables[i]
            sol_tables[[name]] <- floodlight.analysis(sol, 
                                                      var_numeric, 
                                                      var_factor, 
                                                      sol$samples_l2.list[[i]], 
                                                      sol$dMatrice$X, 
                                                      sol$dMatrice$Z, 
                                                      data = sol$data, 
                                                      flood_values = flood_values)
          }
        }else{
          sol_tables <- floodlight.analysis(sol, 
                                            var_numeric, 
                                            var_factor, 
                                            sol$samples_l2_param, 
                                            sol$dMatrice$X, 
                                            sol$dMatrice$Z, 
                                            data = sol$data, 
                                            flood_values = flood_values)
        }
      }
      class(sol_tables) <- 'BANOVA.floodlight'
      return(sol_tables)
    }else{
      stop('Model is not recognized')
    }
  }