R/datatable.R

In PRL-PRG/dynalyzer: Dynamic Analysis Data Analysis Pipeline

#' @export
write_data_table <- function(data_table,
                             filepath,
                             truncate = FALSE,
                             binary = FALSE,
                             compression_level = 0) {

    compression_level <- as.integer(compression_level)

    invisible(.Call(C_write_data_table,
                    data_table,
                    filepath,
                    truncate,
                    binary,
                    compression_level))
}

#' @export
read_data_table <- function(filepath_without_ext,
                            binary = FALSE,
                            compression_level = 0) {

    binary = as.logical(binary)

    compression_level <- as.integer(compression_level)

    ext <- data_table_extension(binary, compression_level)

    filepath <- paste0(filepath_without_ext, ".", ext)

    if (!binary & compression_level == 0) {
        read.table(filepath,
                   header = TRUE,
                   sep = "\x1f",
                   comment.char = "",
                   stringsAsFactors = FALSE)
    }
    else {
        .Call(C_read_data_table,
              filepath,
              binary,
              compression_level)
    }
}


#' @export
data_table_extension <- function(binary = FALSE,
                                 compression_level = 0) {

    ext <- if (binary) "bin" else "csv"

    if (compression_level == 0) ext else paste0(ext, ".zst")
}