Add method to coerse SimInf_events to a data.frame

stewid · Feb 5, 2020 · ba95edd · ba95edd
1 parent 68a2499
commit ba95edd
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -1,5 +1,6 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method(as.data.frame,SimInf_events)
 export("gdata<-")
 export("punchcard<-")
 export("select_matrix<-")

diff --git a/R/SimInf_events.R b/R/SimInf_events.R
@@ -369,6 +369,16 @@ setAs(from = "SimInf_events",
       }
 )
 
+##' Coerce to data frame
+##'
+##' @method as.data.frame SimInf_events
+##'
+##' @inheritParams base::as.data.frame
+##' @export
+as.data.frame.SimInf_events <- function(x, ...) {
+    as(x, "data.frame")
+}
+
 ##' Plot scheduled events
 ##'
 ##' @param x the time points of the events.

diff --git a/man/as.data.frame.SimInf_events.Rd b/man/as.data.frame.SimInf_events.Rd