R/panel.R

In d-callan/plot.data: Create `data.table`s of Ready to Plot Data

panelOddsRatio <- function(data, col, group, panel = NULL) {
  names(data)[names(data) == col] <- 'y'
  names(data)[names(data) == group] <- 'x'
  col <- 'y'
  group <- 'x'

  if (is.null(panel)) {
    dt <- oddsRatio(data)
  } else {
    dt.list <- split(data, list(data[[panel]]))
    dt.list <- lapply(dt.list, oddsRatio)
    dt <- purrr::reduce(dt.list, rbind)
    dt$panel <- names(dt.list)
  }

  return(dt)
}

panelRelativeRisk <- function(data, col, group, panel = NULL) {
  names(data)[names(data) == col] <- 'y'
  names(data)[names(data) == group] <- 'x'
  col <- 'y'
  group <- 'x'

  if (is.null(panel)) {
    dt <- relativeRisk(data)
  } else {
    dt.list <- split(data, list(data[[panel]]))
    dt.list <- lapply(dt.list, relativeRisk)
    dt <- purrr::reduce(dt.list, rbind)
    dt$panel <- names(dt.list)
  }

  return(dt)
}

panelBothRatios <- function(data, col, group, panel = NULL) {
  names(data)[names(data) == col] <- 'y'
  names(data)[names(data) == group] <- 'x'
  col <- 'y'
  group <- 'x'
  mergeByCols <- c(panel, 'p.value', 'x', 'y')

  if (is.null(panel)) {
    dt <- bothRatios(data)
  } else {
    dt.list <- split(data, list(data[[panel]]))
    dt.list <- lapply(dt.list, bothRatios)
    dt <- purrr::reduce(dt.list, rbind)
    dt$panel <- names(dt.list)
  }

  return(dt)
}

panelChiSq <- function(data, col, group, panel = NULL) {
  names(data)[names(data) == col] <- 'y'
  names(data)[names(data) == group] <- 'x'
  col <- 'y'
  group <- 'x'

  if (is.null(panel)) {
    dt <- chiSq(data)
  } else {
    dt.list <- split(data, list(data[[panel]]))
    dt.list <- lapply(dt.list, chiSq)
    dt <- purrr::reduce(dt.list, rbind)
    dt$panel <- names(dt.list)
  }

  return(dt)
}