Fix print method for modfit class

Closes #1

NAMESPACE

@@ -1,12 +1,12 @@
 # Generated by roxygen2 (4.1.1): do not edit by hand
 
+S3method(print,modfit)
 export(check_cp)
 export(initialize_cp)
 export(merge_simfiles)
 export(mod_fit)
 export(netsim_hpc)
 export(netsim_par)
-export(print_modfit)
 export(qsub_master)
 export(save_cpdata)
 export(savesim)

R/mod_fit.R

@@ -57,6 +57,7 @@ mod_fit <- function(dir,
     odf$epithresh <- odf$epidiff < threshold
   }
 
+  class(odf) <- c("modfit", "data.frame")
   return(odf)
 }
 
@@ -72,22 +73,23 @@ mod_fit <- function(dir,
 #' @param print.gap Spacing method passed to \code{print} method.
 #' @param ... Additional arguments passed to \code{print} method.
 #'
+#' @method print modfit
 #' @export
 #'
-print_modfit <- function(x, out = "all", print.gap = 3, ...) {
+print.modfit <- function(x, out = "all", print.gap = 3, ...) {
 
   if (out == "all") {
-    print(x, print.gap = print.gap)
+    print.data.frame(x, print.gap = print.gap, ...)
   }
   if (out == "eql") {
     x <- x[x$thresh == TRUE, ]
     x <- x[order(x$reldiff), ]
-    print(x, print.gap = print.gap)
+    print.data.frame(x, print.gap = print.gap, ...)
   }
   if (out == "epi") {
     x <- x[x$thresh == TRUE & x$epithresh == TRUE, ]
     x <- x[order(x$epidiff), ]
-    print(x, print.gap = print.gap)
+    print.data.frame(x, print.gap = print.gap, ...)
   }
 
 }

man/print_modfit.Rd → man/print.modfit.Rd

@@ -1,10 +1,10 @@
 % Generated by roxygen2 (4.1.1): do not edit by hand
 % Please edit documentation in R/mod_fit.R
-\name{print_modfit}
-\alias{print_modfit}
+\name{print.modfit}
+\alias{print.modfit}
 \title{Printing Method for modfit Objects}
 \usage{
-print_modfit(x, out = "all", print.gap = 3, ...)
+\method{print}{modfit}(x, out = "all", print.gap = 3, ...)
 }
 \arguments{
 \item{x}{Output object from \code{\link{mod_fit}} function.}