Fix for Issue #120

R/plotLoadings.R

@@ -176,6 +176,8 @@ plotLoadings.mixo_pls <-
         if (length(block) == 1 & !is.null(name.var))
             name.var = list(name.var = name.var)
 
+        contrib.df <- list()
+
         for (i in 1 : length(block))
         {
             res = get.loadings.ndisplay(object = object, comp = comp, block = block[i], name.var = name.var[[i]], name.var.complete = name.var.complete, ndisplay = ndisplay)
@@ -204,8 +206,12 @@ plotLoadings.mixo_pls <-
             } else if (length(block) > 1 & !missing(subtitle)) {
                 title(paste(subtitle[i]), line=0, cex.main = size.subtitle)
             }
+
+            contrib.df <- c(contrib.df, list(df))
         }
 
+        names(contrib.df) <- block
+
         if (length(block) > 1 & !is.null(title))
             title(title, outer=TRUE, line = -2, cex.main = size.title)
 
@@ -215,7 +221,7 @@ plotLoadings.mixo_pls <-
         par(mar = omar) #reset mar
 
         # return the contribution matrix
-        return(invisible(df))
+        return(invisible(contrib.df))
     }
 
 #' @rdname plotLoadings
@@ -391,6 +397,9 @@ plotLoadings.mixo_plsda <-
             if (length(block) == 1 & !is.null(name.var))
                 name.var = list(name.var = name.var)
 
+
+            contrib.df <- list()
+
             for (i in 1 : length(block))
             {
                 res = get.loadings.ndisplay(object = object, comp = comp, block = block[i], name.var = name.var[[i]], name.var.complete = name.var.complete, ndisplay = ndisplay)
@@ -457,9 +466,14 @@ plotLoadings.mixo_plsda <-
                                title = paste(legend.title),
                                cex = size.legend)
                     }
+
+                    contrib.df <- c(contrib.df, list(df))
                 } # end if plot
             }
 
+            names(contrib.df) <- block
+
+
             if(plot) # overall title and reset par if needed
             {
                 # legend
@@ -473,7 +487,7 @@ plotLoadings.mixo_plsda <-
             }
 
             # return the contribution matrix
-            return(invisible(df))
+            return(invisible(contrib.df)) # df
         }# end contrib missing
     }
 
@@ -1253,6 +1267,7 @@ get.contrib.df <- function(Y,
         method.group[[k]] = tapply(X[, k], Y, method, na.rm=TRUE) #method is either mean or median
         # determine which group has the highest mean/median
         which.contrib[k, 1:nlevels(Y)] = (method.group[[k]]) == get(contrib)((method.group[[k]]), na.rm=TRUE) # contrib is either min or max
+
     }
 
     # we also add an output column indicating the group that is max