feature #90: colorize interactive epoch plots

shawntz · Nov 25, 2024 · 6ad98de · 6ad98de
1 parent d0c90de
commit 6ad98de
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/R/bidsify.R b/R/bidsify.R
@@ -293,6 +293,8 @@ bidsify <- function(eyeris, save_all = TRUE, epochs_list = NULL,
               y_units <- "(a.u.)"
             }
 
+            colors <- c("black", rainbow(length(pupil_steps) - 1))
+
             y_label <- paste("pupil size", y_units)
 
             file_out <- file.path(epochs_out, paste0(group, "_", pstep, ".png"))
@@ -305,8 +307,9 @@ bidsify <- function(eyeris, save_all = TRUE, epochs_list = NULL,
               pointsize = 6
             )
             plot(group_df$timebin, group_df[[pupil_steps[pstep]]],
-              type = "l", xlab = "time (s)",
-              ylab = y_label, main = paste0(group, "\n", pupil_steps[pstep])
+              type = "l", xlab = "time (s)", ylab = y_label,
+              col = colors[pstep],
+              main = paste0(group, "\n", pupil_steps[pstep])
             )
             dev.off()
           }