speedup parsing parameters

Author: rok-cesnovar

R/read_csv.R

@@ -209,7 +209,7 @@ read_cmdstan_csv <- function(files,
         fread_cmd <- paste0(grep_path, " -v '^#' ", output_file)
       } else {
         fread_cmd <- paste0("grep -v '^#' ", output_file)
-      }      
+      }
       suppressWarnings(
       draws <- data.table::fread(
           cmd = fread_cmd,
@@ -412,21 +412,9 @@ read_csv_metadata <- function(csv_file) {
     if (!startsWith(line, "#") && is.null(csv_file_info[["model_params"]])) {
       # if no # at the start of line, the line is the CSV header
       all_names <- strsplit(line, ",")[[1]]
-      csv_file_info[["sampler_diagnostics"]] <- c()
-      csv_file_info[["model_params"]] <- c()
-      for (x in all_names) {
-        if (all(csv_file_info$algorithm != "fixed_param")) {
-          if (endsWith(x, "__") && !(x %in% c("lp__", "log_p__", "log_g__"))) {
-            csv_file_info[["sampler_diagnostics"]] <- c(csv_file_info[["sampler_diagnostics"]], x)
-          } else {
-            csv_file_info[["model_params"]] <- c(csv_file_info[["model_params"]], x)
-          }
-        } else {
-          if (!endsWith(x, "__")) {
-            csv_file_info[["model_params"]] <- c(csv_file_info[["model_params"]], x)
-          }
-        }
-      }
+      csv_file_info[["sampler_diagnostics"]] <- all_names[endsWith(all_names, "__")]
+      csv_file_info[["sampler_diagnostics"]] <- csv_file_info[["sampler_diagnostics"]][!(csv_file_info[["sampler_diagnostics"]] %in% c("lp__", "log_p__", "log_g__"))]
+      csv_file_info[["model_params"]] <- all_names[!(all_names %in% csv_file_info[["sampler_diagnostics"]])]
     } else {
       parse_key_val <- TRUE
       if (regexpr("# Diagonal elements of inverse mass matrix:", line, perl = TRUE) > 0