From 13703b9ab598fed0fa66eaedcff9e8b3900ba038 Mon Sep 17 00:00:00 2001
From: Daniel <mail@danielluedecke.de>
Date: Fri, 9 Aug 2024 23:49:05 +0200
Subject: [PATCH] plot.check_dag plots multiple adjustment sets

---
 DESCRIPTION        |  2 +-
 R/plot.check_dag.R | 17 +++++++++++++++++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/DESCRIPTION b/DESCRIPTION
index fea9afb6a..f7ee79635 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -1,7 +1,7 @@
 Type: Package
 Package: see
 Title: Model Visualisation Toolbox for 'easystats' and 'ggplot2'
-Version: 0.8.5.4
+Version: 0.8.5.5
 Authors@R:
     c(person(given = "Daniel",
              family = "Lüdecke",
diff --git a/R/plot.check_dag.R b/R/plot.check_dag.R
index bf6e8a5d7..747cf6fd1 100644
--- a/R/plot.check_dag.R
+++ b/R/plot.check_dag.R
@@ -65,6 +65,14 @@ plot.see_check_dag <- function(x,
   p1 <- p2 <- suppressWarnings(ggdag::dag_adjustment_sets(x))
   adjusted_for <- attributes(x)$adjusted
 
+  # if we have multiple sets, we only need one for the current model
+  if (!is.null(p1$data$set) && insight::n_unique(p1$data$set) > 1) {
+    p1$data <- p1$data[p1$data$set == p1$data$set[1], ]
+    # rename set-variables
+    p2$data$set <- gsub("\\{(.*)\\}", "\\1", p2$data$set)
+    p2$data$set <- paste0("Adjusted for ", p2$data$set)
+  }
+
   # for current plot, we need to update the "adjusted" column
   p1$data$adjusted <- "unadjusted"
   if (!is.null(adjusted_for)) {
@@ -142,6 +150,15 @@ plot.see_check_dag <- function(x,
     common_layers +
     ggplot2::ggtitle("Required model")
 
+  # if we have multiple sets, we want to facet the second plot by sets
+  if (!is.null(p2$data$set) && insight::n_unique(p2$data$set) > 1) {
+    plot2 <- plot2 + ggplot2::facet_wrap(
+      ~set,
+      scales = "free",
+      ncol = ceiling(sqrt(insight::n_unique(p2$data$set)))
+    )
+  }
+
   if (which == "all") {
     # fix legends - remove the legend that has fewer items, so all items
     # in the legend are shown for the integrated plot