R/add_undefined.R

In barefootbiology/heyexr: Import, Transform and Visualize Optical Coherence Tomography Volumes in R

#' Add a column specifying if the values in the segmentation are undefined
#'
#' Adds a column `is_defined` which indicates if a particular A-scan could be
#' reliably segmented by the Iowa Reference Algorithms. NOTE: This function
#' assumes there is a column named
#' "value", but this column  name is likely to change in future iterations of
#' the software.
#'
#' @export
#' @importFrom magrittr %>%
#' @importFrom dplyr full_join mutate if_else
#' @importFrom rlang .data
add_undefined <- function(data, undefined) {
  na_value <- as.numeric(NA)

  if(is.integer(data$value)) {
    na_value <- as.integer(NA)
  }

  data %>%
    full_join(undefined) %>%
    mutate(is_defined = if_else(is.na(.data$is_defined),
                                TRUE, .data$is_defined)) %>%
    mutate(value_defined =
             if_else(.data$is_defined,
                     .data$value,
                     na_value))
}