fix for divRates to add fossilization rates/ times to processed output.

revbayes · Jan 5, 2024 · 5048b69 · 5048b69
1 parent 1cdaecf
commit 5048b69
Showing 1 changed file with 28 additions and 10 deletions.
diff --git a/R/utils.R b/R/utils.R
@@ -412,17 +412,35 @@
         summary = summary
       ))
   plotdata <- do.call(rbind, res)
-  plotdata$item <- factor(
-    plotdata$item,
-    levels = c(
-      "speciation rate",
-      "extinction rate",
-      "speciation time",
-      "extinction time",
-      "net-diversification rate",
-      "relative-extinction rate"
+
+  if (sum(grepl("fossil", plotdata$item)) == 0) {
+    plotdata$item <- factor(
+      plotdata$item,
+      levels = c(
+        "speciation rate",
+        "extinction rate",
+        "speciation time",
+        "extinction time",
+        "net-diversification rate",
+        "relative-extinction rate"
+      )
     )
-  )
+  } else {
+    plotdata$item <- factor(
+      plotdata$item,
+      levels = c(
+        "speciation rate",
+        "extinction rate",
+        "fossilization rate",
+        "speciation time",
+        "extinction time",
+        "fossilization time",
+        "net-diversification rate",
+        "relative-extinction rate"
+      )
+    )
+  }
+
   return(plotdata)
 }