R/ancestor_abbrvs.R

In CKMRpop: Forward-in-Time Simulation and Tallying of Pairwise Relationships

#' Return a string of ancestor abbreviations in the order of the ancestor vectors
#'
#' Use this to put axis labels on plots, etc.
#' @param L desired length of the output
#' @keywords internal
#' @export
#' @examples
#' ancestor_abbrvs(15)
ancestor_abbrvs <- function(L) {

  if(L <= 3) {
    return(c("s", "p", "m")[1:L])
  }
  # determine how many generations to go (and go a bit over, possibly)
  g <- ceiling(log(L + 1, base = 2))

  alist <- list(
    "s",
    c("p", "m")
  )

  for(i in 3:g) {
    alist[[i]] <- paste0(rep(alist[[i-1]], each = 2), c("p", "m") )
  }

  unlist(alist)[1:L]
}