fixed typo

skchronicles · Jul 5, 2020 · cefc232 · cefc232
1 parent 005624c
commit cefc232
Showing 1 changed file with 25 additions and 4 deletions.
diff --git a/R/HeatmapList-legends.R b/R/HeatmapList-legends.R
@@ -405,21 +405,42 @@ calc_ht_w = function(ht_list, include_right = FALSE) {
 }
 
 calc_before_h = function(ht_list) {
-    sum(component_height(ht_list@ht_list[[ which_first_ht(ht_list) ]], 1:4))
+    i = which_last_ht(ht_list)
+    if(i > 1) {
+        unit(0, "mm")
+    } else {
+        sum(component_height(ht_list@ht_list[[ which_last_ht(ht_list) ]], 6:9))
+    }
 }
 
 calc_after_h = function(ht_list) {
-    sum(component_height(ht_list@ht_list[[ which_last_ht(ht_list) ]], 6:9))
+    i = which_last_ht(ht_list)
+    if(i < length(ht_list@ht_list)) {
+        unit(0, "mm")
+    } else {
+        sum(component_height(ht_list@ht_list[[ which_last_ht(ht_list) ]], 6:9))
+    }
 }
 
 calc_before_w = function(ht_list) {
-    sum(component_width(ht_list@ht_list[[ which_first_ht(ht_list) ]], 1:4))
+    i = which_last_ht(ht_list)
+    if(i > 1) {
+        unit(0, "mm")
+    } else {
+        sum(component_width(ht_list@ht_list[[ which_first_ht(ht_list) ]], 1:4))
+    }
 }
 
 calc_after_w = function(ht_list) {
-    sum(component_width(ht_list@ht_list[[ which_last_ht(ht_list) ]], 6:9))
+    i = which_last_ht(ht_list)
+    if(i < length(ht_list@ht_list)) {
+        unit(0, "mm")
+    } else {
+        sum(component_width(ht_list@ht_list[[ which_last_ht(ht_list) ]], 6:9))
+    }
 }
 
+
 guess_align_legend = function(ht_list, 
     heatmap_legend_size, annotation_legend_size,
     heatmap_legend_side, annotation_legend_side,