format a standard phylo object for use in RevGadgets plotting functions

convertTreeFormat.R

# randomly sample a tree for this example
t <- rtree(10)
# check that the example tree is a phylo object
class(t) == "phylo"
# convert to RevGadgets plotting format 
t_td <- convertTreeFormat(t) # [note, run the function code below before executing this line of code]


# The function:
# `tree` is the argument for your phylo object.
# `format` should be either 1 or 2, where 1 
# products a list of treedata objects and 2
# produces a list of lists of treedata objects.
# The appropriate format will depend on which 
# downstream plotting functions you wish to use. 
convertTreeFormat <- function(tree, format = 2){
  td <- list(tidytree::as.treedata(tree))
  if (format == 1){
    return(td)
  } else if (format == 2){
    return(list(td))
  } else {
    simpleError("Format arg should be either 1 or 2")
  }
}