read_log.R

#' Read log file
#' 
#' @param path log file path
#' @return Parsed tibble of log file in \code{path}
#' @export
#' @examples 
#' rcg <- read_log("data/20220405162804-HELIOS_base_3-vs-enemy_2.rcg")
#' rcl <- read_log("data/20220405162804-HELIOS_base_3-vs-enemy_2.rcl")
read_log <- function(path) {
  ext <- path |> path() |> path_ext()
  
  if (ext == "rcg") {
    return(read_rcg(path))
  }
  if (ext == "rcl") {
    return(read_rcl(path))
  }
  
  stop("Not supported format: ", path)
}

#' @rdname read_log
#' @export 
read_rcg <- function(path) {
  rcg <- path |>
    read_file() |>
    parse_json(simplifyVector = TRUE, flatten = TRUE) |>
    as_tibble() |>
    filter(type == "show") |>
    select(time, stime, players, ball.x, ball.y, ball.vx, ball.vy) |>
    unnest(players) |>
    select(time:capacity, ball.x:ball.vy) |>
    rename(step = time) |>
    rename_with(str_replace, pattern = "\\.", replacement = "_")
  
  return(rcg)
}

#' @rdname read_log
#' @export
read_rcl <- function(path) {
  rcl <- path |>
    read_lines() |>
    as_tibble() |>
    mutate(
      step     = value |> str_extract("\\d+") |> as.numeric(),
      agent    = value |> str_extract("\\w+_([0-9]{1,2}|Coach)(?!\\))"),
      team     = agent |> str_remove("_([0-9]{1,2}|Coach)"),
      unum     = agent |> str_extract("([0-9]{1,2}|Coach)$"),
      commands = value |>
        str_extract("\\(.+\\)$") |>
        map(~ .x |>
              str_split("\\(|\\)", simplify = TRUE) |>
              str_trim() |>
              discard(~ .x == "")),
    ) |>
    unnest(commands) |>
    mutate(
      commands = commands |> str_split("\\ ", n = 2),
      command  = commands |> map_chr(1),
      args     = commands |> map(~ .x[-1]),
    ) |>
    select(
      step,
      team,
      unum,
      command,
      args
    )
  
  return(rcl)
}