R/load_csv_data.R

In titaniumtroop/rweightlifting: Analyze Weightlifting Progress and Create Custom Weightlifting Programs

#' @importFrom rlang .data

#' @title Load weightlifting logs
#' @description Loads weightlifting logs in CSV format into a data frame
#'
#' @param datadir A directory containing weightlifting logs in CSV format. The expected format is \code{date, exercise, variant, set1weight, set1reps, ..., setNweight, setNreps}
#' @param files A list of files containing weightlifting logs in CSV format. Not currently implemented.
#' @param header Whether the CSV file contains a header. Passed to read.csv.
#' @return A data frame containing a weightlifting log with one set of an exercise per row. The program name listed in the data frame will correspond to the name of the CSV file from which the data was read.
#'
#' @export

load_csv_data <- function(files = NA, datadir = NA, header = TRUE) {

  # if (is.na(datadir)) {
  #   stop("You must enter a directory for your weightlifting files.")
  # }

  out.file <- data.frame()

  if (! is.na(files)) {
    # Load data from files passed to function (interactive Rshiny)
    # files = files
  } else if (! is.na(datadir)) {
    # Load data from datadir
    files <- dir(datadir, pattern = ".csv")
    files <- paste0(datadir, "/", files)
  } else {
    stop("You must enter either a set of files or a directory path.")
  }

  for (i in 1:length(files)) {

    file <- utils::read.csv(
      paste(files[i], sep=""),
      header = header,
      sep = ",",
      stringsAsFactors = FALSE,
      strip.white = TRUE
    )

    file$program <- sub(".+\\/(.+?)[.csv]*$", "\\1", files[i]) # set program to name of file
    file <- file[, c("program", setdiff(names(file), c("program")))] # Move program to first column

    if (requireNamespace("tidyr", quietly = TRUE) &
        requireNamespace("dplyr", quietly = TRUE) &
        requireNamespace("readr", quietly = TRUE)) {
      col.classes <- sapply(file, class)
      numeric.cols <- names(col.classes[col.classes == "numeric" | col.classes == "integer"])
      logical.cols <- names(col.classes[col.classes == "logical"])

      if (all(is.na(file[, logical.cols]))) {
        file <- dplyr::select(file, -logical.cols)
      }

      file <- tidyr::gather(file, "key", "value", numeric.cols, na.rm = TRUE)
      file <- dplyr::mutate(file, set = readr::parse_number(.data$key))
      file <- dplyr::mutate(file, key = ifelse(grepl("rep", .data$key), "reps", "weight"))
      file <- tidyr::spread(file, .data$key, .data$value)
      out.file <- dplyr::bind_rows(out.file, file)
    }

    else {
      requireNamespace("tidyr", quietly = F)
      requireNamespace("dplyr", quietly = F)
      requireNamespace("readr", quietly = F)
      set.cols <- grep("set", names(file), ignore.case = TRUE, value = TRUE)
      other.cols <- names(file[, ! names(file) %in% set.cols])

      # Get max set number
      set.num <- max(as.numeric(sub("set[_ ]*(\\d+).+", "\\1", set.cols)), na.rm = TRUE)

      # Loop through sets and add to dataframe
      for (j in seq(1:set.num)) {
        this.set <- grep(paste("set[_ ]*", j, "", sep = ""), set.cols, ignore.case = TRUE, value = TRUE)[1:2]
        temp <- file[ , c(other.cols, this.set)]

        if (grepl("reps", this.set[1], ignore.case = TRUE)) {
          names(temp) <- c(other.cols, "reps","weight")
        } else {
          names(temp) <- c(other.cols, "weight","reps")
        }

        temp$set <- j
        out.file <- rbind(out.file, temp)
      }
    }
  }
  # Return merged dataframe, removing NA values
  out.file$date <- as.Date(out.file$date)
  out.file[! is.na(out.file$weight) & ! is.na(out.file$reps), ]
}


#' Checks to see if a data frame is a valid weightlifting log. Looks for names.
#' @export
#'
#' @param weightlifting.log A data frame containing at least the following elements: \code{program, date, exercise, variant, reps,  weight}
#' @return Boolean

is_valid_weightlifting_log <- function(weightlifting.log = NULL) {

  if (is.null(weightlifting.log)) {
    stop("Please provide a valid weightlifting log.")
  }

  if (! all(c("program", "date", "exercise", "equipment", "variant", "reps", "weight") %in% names(weightlifting.log))) {
    stop("Please provide a weightlifting log that includes program, date, exercise, equipment, variant, reps, and weight.")
  }

  return(TRUE)
}