Use tinytable for print_md.compare_parameters()

easystats · Mar 13, 2024 · 160833a · 160833a
1 parent 66dc88b
commit 160833a
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 10 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,7 +1,7 @@
 Type: Package
 Package: parameters
 Title: Processing of Model Parameters
-Version: 0.21.5.6
+Version: 0.21.5.7
 Authors@R:
     c(person(given = "Daniel",
              family = "Lüdecke",

diff --git a/R/format.R b/R/format.R
@@ -253,6 +253,7 @@ format.compare_parameters <- function(x,
                                       zap_small = FALSE,
                                       format = NULL,
                                       groups = NULL,
+                                      engine = NULL,
                                       ...) {
   m_class <- attributes(x)$model_class
   x$Method <- NULL
@@ -393,7 +394,7 @@ format.compare_parameters <- function(x,
   # group parameters - this function find those parameters that should be
   # grouped, reorders parameters into groups and indents lines that belong
   # to one group, adding a header for each group
-  if (!is.null(groups)) {
+  if (!is.null(groups) && !identical(engine, "tt")) {
     out <- .parameter_groups(out, groups)
   }
   indent_groups <- attributes(x)$indent_groups

diff --git a/R/print_md.R b/R/print_md.R
@@ -136,6 +136,7 @@ print_md.compare_parameters <- function(x,
                                         ci_brackets = c("(", ")"),
                                         zap_small = FALSE,
                                         groups = NULL,
+                                        engine = "tt",
                                         ...) {
   # check if user supplied digits attributes
   if (missing(digits)) {
@@ -153,6 +154,9 @@ print_md.compare_parameters <- function(x,
     select <- attributes(x)$output_style
   }
 
+  # markdown engine?
+  engine <- match.arg(engine, c("tt", "default"))
+
   formatted_table <- format(
     x,
     select = select,
@@ -164,16 +168,24 @@ print_md.compare_parameters <- function(x,
     ci_brackets = ci_brackets,
     format = "markdown",
     zap_small = zap_small,
-    groups = groups
+    groups = groups,
+    engine = engine
   )
 
-  insight::export_table(
-    formatted_table,
-    format = "markdown",
-    caption = caption,
-    subtitle = subtitle,
-    footer = footer
-  )
+  if (identical(engine, "tt")) {
+    insight::check_if_installed("tinytable", minimum_version = "0.1.0")
+    gsub("(.*) \\((.*)\\)$", "\\2", colnames(out))
+    gsub("(.*) \\((.*)\\)$", "\\1", colnames(out))
+    lapply(groups, function(i) match(i, out$Parameter))
+  } else {
+    insight::export_table(
+      formatted_table,
+      format = "markdown",
+      caption = caption,
+      subtitle = subtitle,
+      footer = footer
+    )
+  }
 }