Issue 80: Default generic for epidist method

NAMESPACE

@@ -1,6 +1,6 @@
 # Generated by roxygen2: do not edit by hand
 
-S3method(epidist,epidist_ltcad)
+S3method(epidist,default)
 S3method(epidist_family,epidist_ltcad)
 S3method(epidist_formula,epidist_ltcad)
 S3method(epidist_prepare,default)

R/ltcad.R

@@ -74,27 +74,27 @@
 epidist_stancode.epidist_ltcad <- function(data,
                                            family = epidist_family(data)) {
   stanvars_version <- epidist_version_stanvar()
 
   stanvars_functions <- brms::stanvar(
     block = "functions", scode = epidist_stan_chunk("functions.stan")
   )
 
   family_name <- gsub("latent_", "", family$name)
 
   stanvars_functions[[1]]$scode <- gsub(
     "family", family_name, stanvars_functions[[1]]$scode
   )
 
   stanvars_data <- brms::stanvar(
     block = "data",
     scode = "int wN;",
     x = nrow(data[woverlap > 0]),
     name = "wN"
   ) +
     brms::stanvar(
       block = "data",
       scode = "array[N - wN] int noverlap;",
       x = data[woverlap == 0][, row_id],
       name = "noverlap"
     ) +
     brms::stanvar(
@@ -103,7 +103,7 @@
       x = data[woverlap > 0][, row_id],
       name = "woverlap"
     )
 
   stanvars_parameters <- brms::stanvar(
     block = "parameters", scode = epidist_stan_chunk("parameters.stan")
   )
@@ -119,23 +119,3 @@
   stanvars_all <- stanvars_version + stanvars_functions + stanvars_data +
     stanvars_parameters + stanvars_tparameters + stanvars_priors
 }
-
-#' @method epidist epidist_ltcad
-#' @family ltcad
-#' @export
-epidist.epidist_ltcad <- function(data, formula = epidist_formula(data),
-                                  family = epidist_family(data),
-                                  priors = epidist_priors(data),
-                                  stancode = epidist_stancode(data),
-                                  fn = brms::brm,
-                                  ...) {
-
-  fit <- fn(
-    formula = formula, family = family, stanvars = stancode,
-    backend = "cmdstanr", data = data, ...
-  )
-
-  class(fit) <- c(class(fit), "epidist_fit")
-
-  return(fit)
-}

R/methods.R

@@ -81,3 +81,26 @@ epidist_stancode <- function(data, ...) {
 epidist <- function(data, formula, family, priors, custom_stancode, fn, ...) {
   UseMethod("epidist")
 }
+
+#' Default method used for interface using `brms`
+#'
+#' @inheritParams epidist
+#' @rdname epidist
+#' @method epidist default
+#' @family methods
+#' @export
+epidist.default <- function(data, formula = epidist_formula(data),
+                            family = epidist_family(data),
+                            priors = epidist_priors(data),
+                            stancode = epidist_stancode(data), fn = brms::brm,
+                            ...) {
+
+  fit <- fn(
+    formula = formula, family = family, stanvars = stancode,
+    backend = "cmdstanr", data = data, ...
+  )
+
+  class(fit) <- c(class(fit), "epidist_fit")
+
+  return(fit)
+}