R/divime_run_all.R

In gobbios/avutils: various utilities for DiViMe tools and audio processing

#' run all DiViMe modules
#'
#' @param audio_loc character, path to the audio files
#' @param divime_loc character, path to the DiViMe directory with a VM
#' @param vmstart logical, perform a check whether the VM is running and if not start it up (by default \code{TRUE}). Turning this off, will speed up the function a little bit, but requires that you are sure that the VM is indeed running in \code{divime_loc}.
#' @param vmshutdown logical, should the VM shut down after the operations are done (by default \code{TRUE})
#' @param messages logical, should the file names of each processed file be printed
#' @param overwrite logical, should output files be overwritten if they already exist (default is \code{FALSE})
#' @return a list of data.frames with the locations of the created rttm files and some diagnostics
#' @export

divime_run_all <- function(audio_loc,
                           divime_loc,
                           vmstart = TRUE,
                           vmshutdown = TRUE,
                           messages = TRUE,
                           overwrite = FALSE) {
  # audio_loc = "~/Desktop/test_audio/smallfiles"
  # audio_loc = "~/Desktop/error-producing files"
  # divime_loc = "/Volumes/Data/VM2/ooo/DiViMe"
  # vmstart = T; vmshutdown = F; messages = TRUE; overwrite = FALSE
  logres <- list()
  totalruns <- 15


  cat("###### SAD noisemes ###########\n")
  temp1 <- divime_sad(audio_loc = audio_loc,
                      divime_loc = divime_loc,
                      module = "noisemes",
                      vmstart = vmstart,
                      vmshutdown = FALSE,
                      messages = messages,
                      overwrite = overwrite)
  logres$sad_noisemes <- temp1
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### SAD opensmile ###########\n")
  temp2 <- divime_sad(audio_loc = audio_loc,
                      divime_loc = divime_loc,
                      module = "opensmile",
                      vmstart = FALSE,
                      vmshutdown = FALSE,
                      messages = messages,
                      overwrite = overwrite)
  logres$sad_opensmile <- temp2
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### SAD tocombo ###########\n")
  temp3 <- divime_sad(audio_loc = audio_loc,
                      divime_loc = divime_loc,
                      module = "tocombo",
                      vmstart = FALSE,
                      vmshutdown = FALSE,
                      messages = messages,
                      overwrite = overwrite)
  logres$sad_tocombo <- temp3
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### diartk noisemes ###########\n")
  temp4 <- divime_diarization(audio_loc = audio_loc,
                              divime_loc = divime_loc,
                              speech_annos = "noisemes",
                              vmstart = FALSE,
                              vmshutdown = FALSE,
                              messages = messages,
                              overwrite = overwrite)
  logres$diartk_noisemes <- temp4
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### diartk opensmile ###########\n")
  temp5 <- divime_diarization(audio_loc = audio_loc,
                              divime_loc = divime_loc,
                              speech_annos = "opensmile",
                              vmstart = FALSE,
                              vmshutdown = FALSE,
                              messages = messages,
                              overwrite = overwrite)
  logres$diartk_opensmile <- temp5
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### diartk tocombo ###########\n")
  temp6 <- divime_diarization(audio_loc = audio_loc,
                              divime_loc = divime_loc,
                              speech_annos = "tocombo",
                              vmstart = FALSE,
                              vmshutdown = FALSE,
                              messages = messages,
                              overwrite = overwrite)
  logres$diartk_tocombo <- temp6
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### yunitate ###########\n")
  temp7 <- divime_talkertype(audio_loc = audio_loc,
                             divime_loc = divime_loc,
                             vmstart = FALSE,
                             vmshutdown = FALSE,
                             marvinator = FALSE,
                             messages = messages,
                             overwrite = overwrite)
  logres$yuni <- temp7
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### yunitate (marvinator) ###########\n")
  temp8 <- divime_talkertype(audio_loc = audio_loc,
                             divime_loc = divime_loc,
                             vmstart = FALSE,
                             vmshutdown = FALSE,
                             marvinator = TRUE,
                             messages = messages,
                             overwrite = overwrite)
  logres$yuni_marv <- temp8
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### vcm (marvinator) ###########\n")
  temp9 <- divime_classify_vox(audio_loc = audio_loc,
                               divime_loc = divime_loc,
                               vmstart = FALSE,
                               vmshutdown = FALSE,
                               marvinator = TRUE,
                               messages = messages,
                               overwrite = overwrite)
  logres$vcm_marv <- temp9
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### vcm ###########\n")
  temp10 <- divime_classify_vox(audio_loc = audio_loc,
                               divime_loc = divime_loc,
                               vmstart = FALSE,
                               vmshutdown = FALSE,
                               marvinator = FALSE,
                               messages = messages,
                               overwrite = overwrite)
  logres$vcm <- temp10
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### wce (noisemes) ###########\n")
  temp11 <- divime_wordcount(audio_loc = audio_loc,
                             divime_loc = divime_loc,
                             vmstart = FALSE,
                             vmshutdown = FALSE,
                             speech_annos = "noisemes",
                             messages = messages,
                             overwrite = overwrite)
  logres$wce_nois <- temp11
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### wce (opensmile) ###########\n")
  temp12 <- divime_wordcount(audio_loc = audio_loc,
                             divime_loc = divime_loc,
                             vmstart = FALSE,
                             vmshutdown = FALSE,
                             speech_annos = "opensmile",
                             messages = messages,
                             overwrite = overwrite)
  logres$wce_open <- temp12
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### wce (tocombo) ###########\n")
  temp13 <- divime_wordcount(audio_loc = audio_loc,
                             divime_loc = divime_loc,
                             vmstart = FALSE,
                             vmshutdown = FALSE,
                             speech_annos = "tocombo",
                             messages = messages,
                             overwrite = overwrite)
  logres$wce_toco <- temp13
  cat(length(logres), "out of", totalruns, "done\n")

  cat("###### wce (marvinator) ###########\n")
  temp14 <- divime_wordcount(audio_loc = audio_loc,
                             divime_loc = divime_loc,
                             vmstart = FALSE,
                             vmshutdown = FALSE,
                             speech_annos = "yunitator_english",
                             messages = messages,
                             overwrite = overwrite)
  logres$wce_marv <- temp14
  cat(length(logres), "out of", totalruns, "done\n")



  cat("###### full noisemes ###########\n")
  temp15 <- divime_fullnoisemes(audio_loc = audio_loc,
                               divime_loc = divime_loc,
                               vmstart = FALSE,
                               vmshutdown = vmshutdown,
                               messages = messages,
                               overwrite = overwrite)
  logres$fullnoisemes <- temp15
  cat(length(logres), "out of", totalruns, "done\n")

  logres
}