Make extract_samples a method of coevfit

DESCRIPTION

@@ -16,7 +16,7 @@ URL: https://github.com/ScottClaessens/coevolve
 BugReports: https://github.com/ScottClaessens/coevolve/issues
 Encoding: UTF-8
 Roxygen: list(markdown = TRUE)
-RoxygenNote: 7.2.3
+RoxygenNote: 7.3.1
 Suggests: 
     knitr,
     rmarkdown,

NAMESPACE

@@ -1,11 +1,11 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method(extract_samples,coevfit)
 S3method(print,coevfit)
 S3method(print,coevsummary)
 S3method(summary,coevfit)
 export(coev_calculate_delta_theta)
 export(coev_calculate_theta)
-export(coev_extract_samples)
 export(coev_fit)
 export(coev_make_stancode)
 export(coev_make_standata)
@@ -14,4 +14,5 @@ export(coev_plot_flowfield)
 export(coev_plot_predictive_check)
 export(coev_plot_selection_gradient)
 export(coev_simulate_coevolution)
+export(extract_samples)
 importFrom(rlang,.data)

R/extract_samples.R

@@ -0,0 +1,19 @@
+#' Extract samples (draws) from a fitted \code{coevfit} object
+#'
+#' @param object An object of class \code{coevfit}
+#'
+#' @return Samples in 'rethinking' style list format
+#' @export
+extract_samples <- function(object) {
+    UseMethod("extract_samples")
+}
+
+#' @export
+extract_samples.coevfit <- function(object) {
+  vars <- object$fit$metadata()$stan_variables
+  draws <- posterior::as_draws_rvars(object$fit$draws())
+
+  return(lapply(vars, \(var_name){  
+    posterior::draws_of(draws[[var_name]], with_chains = FALSE)
+  }) |> setNames(vars))
+}