R/1.3.extractComponents.R

In Mmcsd: Modeling Complex Longitudinal Data in a Quick and Easy Way

extractComponents <- function(modelFrame, componentName) {
  componentValue <- do.call(stats::model.extract, list(frame = modelFrame$data, component = componentName))

  if (is.null(componentValue)) {
    if (componentName %in% c("weights", "stratum")) {
      componentValue <- tibble::tibble(value = rep(1, dim(modelFrame$data)[1]))
    } else if (componentName == "cluster") {
      componentValue <- tibble::tibble(value = seq(dim(modelFrame$data)[1]))
    }
  } else {
    componentValue <- tibble::as_tibble(componentValue)
    if (componentName == "waves") {
      if (typeof(componentValue$value) == "character") componentValue$value <- as.numeric(factor(componentValue$value))
    }
  }

  return(componentValue)
}