improve print.data.table

rpkgs · May 6, 2023 · e4a405f · e4a405f
1 parent 817cfe1
commit e4a405f
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 15 deletions.
diff --git a/R/print.R b/R/print.R
@@ -0,0 +1,22 @@
+# #' @importFrom pillar dim_desc 
+
+#' print.data.table
+#' 
+#' @examples 
+#' d = data.table(1:100)
+#' options(datatable.print.nrow = 100)
+#' print(d)
+#' @export
+print.data.table <- function(d, n = NULL, ..., maxrows = 1e6) {
+  if (is.null(n)) n = getOption("datatable.print.nrow") %||% 10
+
+  # cat(sprintf("[data.table]: %s \n", dim_desc(d)))
+  if (nrow(d) <= maxrows) {
+    cat("[data.table]: \n")
+
+    class(d) <- c("tbl", "data.table", "data.frame")
+    print(d, n = n, ...)
+  } else {
+    data.table:::print.data.table(d)
+  }
+}
diff --git a/R/reexports.R b/R/reexports.R
@@ -89,21 +89,6 @@ data.table::fread
 #' @export
 data.table::fwrite
 
-
-# #' @importFrom pillar dim_desc 
-#' @export
-print.data.table <- function(d, n = 10, ..., maxrows = 1e6) {
-  # cat(sprintf("[data.table]: %s \n", dim_desc(d)))
-  if (nrow(d) <= maxrows) {
-    cat("[data.table]: \n")
-
-    class(d) <- c("tbl", "data.table", "data.frame")
-    print(d, n = n, ...)
-  } else {
-    data.table:::print.data.table(d)
-  }
-}
-
 #' @export
 data.table::as.data.table
 

diff --git a/man/print.data.table.Rd b/man/print.data.table.Rd