add dplyr helpers and methods for complete

DESCRIPTION

@@ -6,7 +6,7 @@ Date: 2018-02-08
 Author:  Simon Grund [aut,cre], Alexander Robitzsch [aut], Oliver Luedtke [aut]
 Maintainer: Simon Grund <grund@ipn.uni-kiel.de>
 BugReports: https://github.com/simongrund1/mitml/issues
-Imports: pan, jomo, haven, grDevices, graphics, stats, utils
+Imports: pan, jomo, haven, grDevices, graphics, stats, utils, dplyr
 Suggests: mice, miceadds, Amelia, lme4, nlme, geepack, survival, knitr, rmarkdown
 LazyData: true
 LazyLoad: true

R/complete_method.R

@@ -0,0 +1,34 @@
+
+n_imputations.mitml <- function(imps) {
+  imps$iter$m
+}
+
+n_imputations.mitml.list <- function(imps) {
+  length(imps)
+}
+
+n_imputations.mids <- function(imps) {
+  imps$m
+}
+n_imputations.amelia <- function(imps) {
+  length(imps$imputations)
+}
+n_imputations = function(imps) { UseMethod("n_imputations") }
+
+complete.mitml.list <- function(imps, action = "long") {
+  if (action == "long") {
+    dplyr::bind_rows(imps, .id = "m")
+  } else if (is.numeric(action)) {
+    imps[[action]]
+  }
+}
+
+complete.amelia <- function(imps, action = "long") {
+  if (action == "long") {
+    dplyr::bind_rows(imps$imputations, .id = "m")
+  } else if (is.numeric(action)) {
+    imps$imputations[[action]]
+  }
+}
+complete.mids <- mice::complete
+complete <- function(...) { UseMethod("complete") }

R/dplyr_helpers.R

@@ -0,0 +1,56 @@
+
+mutate.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::mutate, ...)
+  class(r) = class(.data)
+  r
+}
+
+mutate_at.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::mutate_at, ...)
+  class(r) = class(.data)
+  r
+}
+
+mutate_if.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::mutate_if, ...)
+  class(r) = class(.data)
+  r
+}
+
+tbl_df.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::tbl_df, ...)
+  class(r) = class(.data)
+  r
+}
+
+select.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::select, ...)
+  class(r) = class(.data)
+  r
+}
+
+arrange.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::arrange, ...)
+  class(r) = class(.data)
+  r
+}
+
+group_by.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::group_by, ...)
+  class(r) = class(.data)
+  r
+}
+
+
+ungroup.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::ungroup, ...)
+  class(r) = class(.data)
+  r
+}
+
+summarise.mitml.list <- function(.data, ...) {
+  r <- lapply(.data, FUN = dplyr::summarise, ...)
+  class(r) = class(.data)
+  r
+}
+

R/mitmlComplete.R

@@ -82,3 +82,5 @@ mitmlComplete <- function(x, print="all", force.list=FALSE){
   x
 
 }
+
+complete.mitml <- mitmlComplete