Plot the iRBC % ratio for each scenario

plot-scenarios.R

@@ -6,6 +6,7 @@ script_name <- "plot-scenarios.R"
 prepare_data <- function(input_file) {
   read.csv(input_file, header = TRUE) |>
     infection_flows_and_loss() |>
+    calculate_irbc_ratio() |>
     mutate(sim_no = factor(scenario)) |>
     pivot_longer(! c("time", "sim_no")) |>
     mutate(days = time / 24)
@@ -57,8 +58,13 @@ make_plots <- function(df) {
       xlab("Time (days)") +
       scale_y_log10() +
       facet_wrap(~ name),
+    ratio = ggplot(df |> filter(name == c("iRBC_Ratio")),
+                   aes(days, value, colour = sim_no)) +
+      geom_line() +
+      xlab("Time (days)") +
+      scale_y_log10(),
     macro = ggplot(df |> filter(name == "M_t"),
-                aes(days, value, colour = sim_no)) +
+                   aes(days, value, colour = sim_no)) +
       geom_line() +
       xlab("Time (days)") +
       ylab(expression(M(t))) +
@@ -77,12 +83,6 @@ make_plots <- function(df) {
 
 
 main <- function(args) {
-  suppressPackageStartupMessages(library(dplyr))
-  suppressPackageStartupMessages(library(tidyr))
-  suppressPackageStartupMessages(library(ggplot2))
-
-  devtools::load_all(".", export_all = FALSE)
-
   if (length(args) != 1) {
     cat("\n")
     cat("USAGE:", script_name, "results.csv\n")
@@ -96,6 +96,12 @@ main <- function(args) {
     return(2)
   }
 
+  suppressPackageStartupMessages(library(dplyr))
+  suppressPackageStartupMessages(library(tidyr))
+  suppressPackageStartupMessages(library(ggplot2))
+
+  devtools::load_all(".", export_all = FALSE)
+
   plot_dir <- dirname(input_file)
   plot_prefix <- tools::file_path_sans_ext(basename(input_file))
   df <- prepare_data(input_file)