R/stopsignal-preprocessor.R

In BrainVR/nudz-stopsignal:

#' Title
#'
#' @param df_experiment
#'
#' @return
#' @export
#'
#' @examples
preprocess_experiment <- function(df_experiment){
  if(!("trialcode" %in% colnames(df_experiment))){
    df_experiment$trialcode <- ifelse(df_experiment$values.signal == 1,
                                      "sstrial", "nstrial")
  }
  if(!("blockcode" %in% colnames(df_experiment))){
    warning("BLOCKCODE not found. Assuming ALL TRIALS ARE TEST!")
    df_experiment$blockcode <- "testblock"
  }
  dat <- df_experiment %>%
    mutate(blocknumber = as.character(expressions.blocknumber),
           correct_signal = ifelse((values.response > 0 & values.signal == 0) |
                                     (values.response == 0 & values.signal ==1 ), "correct", "error"),
           correct_key = ifelse(values.response == values.stimulus, "correct", "error"),
           correct_key = ifelse(values.response == 0, NA, correct_key),
           b_correct_response = ((correct_signal  == "correct") & (correct_key == "correct")) |
             (values.signal == 1 & correct_signal == "correct"),
           correct_response = ifelse(b_correct_response, "correct", "error"),
           trialcode = recode(trialcode, sstrial = "stop", nstrial="nostop"),
           stopsignalstart = expressions.ssrt) %>%
    select(-starts_with("expressions"))
  dat[(dat$trialcode == "stop" & dat$correct_response == "correct"), "values.rt"] <- NA_real_

  # The any of is there as some datasets do NOT have these columns
  dat <- select(dat, -any_of(c("values.ns_ntotal", "values.ss_ntotal", "values.ssd")))
  return(dat)
}

#' Title
#'
#' @param df_experiment
#'
#' @return
#' @export
#'
#' @examples
extract_settings <- function(df_experiment){
  settings_fields <- c("build", "computer.platform", "date", "time", "subject", "group")
  settings <- sapply(settings_fields, function(x){return(dat[[x]][1])}, simplify = TRUE)
  dat <- select(df_experiment, -settings_fields)
  return(list(settings = settings, data = dat))
}