R/spectrace_import_battery.R

In steffenhartmeyer/spectrace: Functions for importing and processing raw Spectrace output

#' Import raw spectrace battery data
#'
#' This function imports the raw battery data of the spectrace sensors. It assumes
#' that the data is stored in CSV format.
#'
#' @param batteryFile Path to the file containing the battery data.
#' @param tz Time zone to be used for datetime conversion. See supported time zones
#'    by calling \code{\link[base]{OlsonNames}}.
#' @param serial_number Serial number of spectrace device. Defaults to NULL.
#'
#' @return Data frame.
#' @export
#'
#' @examples
spectrace_import_battery <- function(batteryFile, tz, serial_number = NA) {
  # Get file type (CSV or TSV)
  if (endsWith(batteryFile, ".csv")) {
    sep <- ","
  } else if (endsWith(batteryFile, ".tsv")) {
    sep <- "\t"
  } else {
    stop("Unsupported file format! Must be CSV or TSV.")
  }

  # Get header
  header <- readr::read_delim(
    batteryFile,
    col_names = FALSE,
    col_types = readr::cols(.default = "c"),
    n_max = 3,
    delim = sep,
    skip_empty_rows = FALSE
  )

  # Check type of file (including header or not)
  if (header$X1[1] == "SERIAL") {
    serial_number <- header$X2[1]
    batData <- utils::read.delim(
      batteryFile,
      skip = 5,
      header = FALSE,
      sep = sep
    ) %>% dplyr::select(c(1:4))
  }
  # Version 3 file
  else if (header$X1[1] == "Raw Spectrace Data") {
    serial_number <- header$X2[3]
    batData <- utils::read.delim(
      batteryFile,
      skip = 6,
      header = FALSE,
      sep = sep
    ) %>% dplyr::select(c(1:4))
  } else {
    # Check whether serial number available
    if (is.na(serial_number)) {
      warning("No serial number specified!")
    }
    batData <- utils::read.delim(
      batteryFile,
      header = FALSE,
      sep = sep
    ) %>% dplyr::select(c(1:4))
  }

  col_names <- c("unix", "battery_voltage", "battery_percent", "battery_isCharging")
  names(batData) <- col_names

  batData <-
    batData %>%
    dplyr::mutate(
      datetime = lubridate::as_datetime(unix, tz = tz),
      serial = serial_number
    ) %>%
    dplyr::relocate(serial, datetime)

  # Return
  return(batData)
}