From 3c7ad13f98d0216970ffa90aebce28b86a9f91a0 Mon Sep 17 00:00:00 2001
From: "Brenton M. Wiernik" <bwiernik@users.noreply.github.com>
Date: Fri, 7 Jul 2023 13:37:43 -0400
Subject: [PATCH] Filter prior density layer to include only shown parameters.

---
 R/plot.estimate_density.R | 4 ++++
 R/plot.p_direction.R      | 4 ++++
 R/plot.p_significance.R   | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/R/plot.estimate_density.R b/R/plot.estimate_density.R
index 04bf29a94..4c1eee6cb 100644
--- a/R/plot.estimate_density.R
+++ b/R/plot.estimate_density.R
@@ -131,6 +131,9 @@ plot.see_estimate_density <- function(x,
     n_columns <- NULL
   }
 
+  # get parameter names for filtering
+  params <- unique(x$y)
+
   # get labels
   labels <- .clean_parameter_names(x$Parameter, grid = !is.null(n_columns))
 
@@ -152,6 +155,7 @@ plot.see_estimate_density <- function(x,
       p <- p +
         .add_prior_layer_ridgeline(
           model,
+          parameter = params,
           show_intercept = show_intercept,
           priors_alpha = priors_alpha,
           show_ridge_line = FALSE
diff --git a/R/plot.p_direction.R b/R/plot.p_direction.R
index 0e6cf9102..d8e5d9d88 100644
--- a/R/plot.p_direction.R
+++ b/R/plot.p_direction.R
@@ -180,6 +180,9 @@ plot.see_p_direction <- function(x,
     n_columns <- NULL
   }
 
+  # get parameter names for filtering
+  params <- unique(x$y)
+
   # get labels
   labels <- .clean_parameter_names(x$y, grid = !is.null(n_columns))
 
@@ -203,6 +206,7 @@ plot.see_p_direction <- function(x,
   if (priors) {
     p <- p + .add_prior_layer_ridgeline(
       model,
+      parameter = params,
       show_intercept = show_intercept,
       priors_alpha = priors_alpha
     )
diff --git a/R/plot.p_significance.R b/R/plot.p_significance.R
index c1d631d43..d47a8423a 100644
--- a/R/plot.p_significance.R
+++ b/R/plot.p_significance.R
@@ -192,6 +192,9 @@ plot.see_p_significance <- function(x,
     n_columns <- NULL
   }
 
+  # get parameter names for filtering
+  params <- unique(x$y)
+
   # get labels
   labels <- .clean_parameter_names(x$y, grid = !is.null(n_columns))
 
@@ -216,6 +219,7 @@ plot.see_p_significance <- function(x,
     p <- p +
       .add_prior_layer_ridgeline(
         model,
+        parameter = params,
         show_intercept = show_intercept,
         priors_alpha = priors_alpha
       ) +