R/myToDataFrame_fns.R

In n8thangreen/ltbiScreenLite: ltbiScreenLite

#' my_ToDataFrameTable
#'
#' This is the same as the same named function in \code{data.tree}
#' except it is not filtered by leaf.
#'
#' @param x
#' @param ...
#' @param pruneFun
#'
#' @return
#' @export
#' @seealso \link[data.tree]{ToDataFrameTable}
#'
#' @examples
my_ToDataFrameTable <- function(x,
                                ...,
                                pruneFun = NULL) {

  df <- as.data.frame(x,
                      row.names = NULL,
                      optional = FALSE,
                      ...,
                      filterFun = NULL,
                      pruneFun = pruneFun,
                      inheritFromAncestors = TRUE)
  df[, -1]
}


#' my_ToDataFrameTypeCol
#'
#' This is the same as the same named function in \code{data.tree}
#' except it is not filtered by leaf.
#'
#' @param x
#' @param ...
#' @param type
#' @param prefix
#' @param pruneFun
#'
#' @return
#' @export
#' @seealso \link[data.tree]{ToDataFrameTypeCol}
#'
#' @examples
my_ToDataFrameTypeCol <- function(x,
                                  ...,
                                  type = "level",
                                  prefix = type,
                                  pruneFun = NULL){

  cols <-
    c(x$Get(type, filterFun = isNotLeaf),
      x$Get(type)) %>%
    unique() #%>%
    # as.character()

  pathArgs <- data.tree:::GetPathArgV(cols, type)

  if (!is.null(prefix))
    cols <- paste0(prefix, "_", cols)

  names(pathArgs) <- cols

  do.call(my_ToDataFrameTable, c(x, pathArgs, ...))
}