R/collect_samples.R

In j-g-b/BTF: What the Package Does (One Line, Title Case)

#'
#'
#'
collect_samples <- function(save_dir, last = NULL){
  #
  require(plyr)
  require(magrittr)
  #
  sample_list <- list()
  #
  dir_files <- dir(save_dir) %>%
    magrittr::extract(., order(as.numeric(gsub(".*_", "", gsub("[.]csv", "", .)))))
  if(!is.null(last)){
    file_nums <- as.numeric(gsub(".*_", "", gsub("[.]csv", "", dir_files)))
    max_iter <- max(file_nums)
    min_num <- min(sort(unique(file_nums), decreasing = T)[1:min(length(unique(file_nums)), last)])
    dir_files <- dir_files[file_nums > min_num]
  }
  #
  if(any(grepl("^U_.*[.]csv", dir_files))){
    sample_list[["U"]] <- dir_files %>%
      magrittr::extract(., grepl("^U_.*[.]csv", .)) %>%
      paste0(save_dir, "/", .) %>%
      plyr::aaply(1, function(fn){

        read.csv(fn, header = F) %>%
          as.matrix()

      })
  }
  if(any(grepl("^V_.*[.]csv", dir_files))){
    sample_list[["V"]] <- dir_files %>%
      magrittr::extract(., grepl("^V_.*[.]csv", .)) %>%
      paste0(save_dir, "/", .) %>%
      plyr::aaply(1, function(fn){

        read.csv(fn, header = F) %>%
          as.matrix()

      })
  }
  if(any(grepl("^R.*_.*[.]csv", dir_files))){
    sample_list[["R"]] <- dir_files %>%
      magrittr::extract(., grepl("^R.*_.*[.]csv", .)) %>%
      data.frame() %>%
      magrittr::set_colnames("fname") %>%
      dplyr::mutate(rel_num = as.numeric(gsub("_.*", "", gsub("R", "", fname)))) %>%
      plyr::daply(.(rel_num), function(d){
        plyr::aaply(as.character(d[["fname"]]), 1, function(fn){
          read.csv(paste0(save_dir, "/", fn), header = F) %>% as.matrix()
        })
      }) %>%
      aperm(c(2, 1, 3, 4))
  }
  if(any(grepl("^SS_.*[.]csv", dir_files))){
    sample_list[["SS"]] <- dir_files %>%
      magrittr::extract(., grepl("^SS_.*[.]csv", .)) %>%
      paste0(save_dir, "/", .) %>%
      plyr::aaply(1, function(fn){

        read.csv(fn, header = F) %>%
          as.matrix() %>%
          c()

      })
  }
  if(any(grepl("^Mu_.*[.]csv", dir_files))){
    sample_list[["Mu"]] <- dir_files %>%
      magrittr::extract(., grepl("^Mu_.*[.]csv", .)) %>%
      paste0(save_dir, "/", .) %>%
      plyr::aaply(1, function(fn){

        read.csv(fn, header = F) %>%
          as.matrix() %>%
          c()

      })
  }
  return(sample_list)
}