fixed optional error visualization for butterfly plot

hadexversum · Oct 3, 2024 · 500008d · 500008d
1 parent eb9a10d
commit 500008d
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/R/plot_butterfly.R b/R/plot_butterfly.R
@@ -100,12 +100,9 @@ plot_butterfly <- function(uptake_dat,
     ))
   ) else geom_point()
 
-  chosen_uncertainty_geom <- switch (
-    uncertainty_type,
-    ribbon = geom_ribbon(alpha = 0.5, size = 0, linetype = "blank"),
-    bars = geom_errorbar(width = 0.25, alpha = 0.5),
-    `bars + line` = geom_errorbar(width = 0.25, alpha = 0.5) + geom_line()
-  )
+  chosen_uncertainty_geom <- if(uncertainty_type == "ribbon") {
+    geom_ribbon(alpha = 0.5, size = 0, linetype = "blank")
+  } else { geom_errorbar(width = 0.25, alpha = 0.5) }
 
   butterfly_plot <- ggplot(
     plot_dat, 
@@ -125,6 +122,8 @@ plot_butterfly <- function(uptake_dat,
          y = y_label) +
     theme(legend.position = "bottom")
 
+  if(uncertainty_type == "bars + line") butterfly_plot <- butterfly_plot + geom_line()
+
   return(HaDeXify(butterfly_plot))
 
 }