print line numbers

stan-dev · May 10, 2024 · f178a44 · f178a44
1 parent c3d0168
commit f178a44
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/R/model.R b/R/model.R
@@ -302,7 +302,16 @@ CmdStanModel <- R6::R6Class(
       if (length(private$stan_code_) == 0) {
         stop("'$print()' cannot be used because the 'CmdStanModel' was not created with a Stan file.", call. = FALSE)
       }
-      cat(self$code(), sep = "\n")
+      lines <- self$code()
+      line_num_indent <- nchar(as.character(length(lines)))
+      line_nums <- vapply(seq_along(lines), function(y) {
+        paste0(
+          rep(" ", line_num_indent - nchar(as.character(y))), y, collapse = ""
+        )
+      }, character(1))
+      cat(
+        paste(paste(line_nums, lines, sep = ": "), collapse = "\n"), sep = "\n"
+      )
       invisible(self)
     },
     stan_file = function() {