@@ -45,6 +45,7 @@ export(StatHilight)

 export(add_colorbar)

 export(aes)

 export(annotation_image)

+export(arrow)

 export(as.polytomy)

 export(collapse)

 export(decimal2Date)

@@ -128,6 +129,7 @@ export(theme_dendrogram)

 export(theme_inset)

 export(theme_tree)

 export(theme_tree2)

+export(unit)

 export(viewClade)

 export(xlim)

 export(xlim_expand)

@@ -210,6 +212,7 @@ importFrom(ggplot2,xlim)

 importFrom(ggplot2,ylab)

 importFrom(ggplot2,ylim)

 importFrom(graphics,identify)

+importFrom(grid,arrow)

 importFrom(grid,convertX)

 importFrom(grid,convertY)

 importFrom(grid,dataViewport)

@@ -20,7 +20,8 @@ geom_tree <- function(mapping=NULL, data=NULL, layout="rectangular", multiPhylo=

 stat_tree <- function(mapping=NULL, data=NULL, geom="segment", position="identity",

                       layout="rectangular", multiPhylo=FALSE, lineend="round", MAX_COUNT=5,

-					  ..., show.legend=NA, inherit.aes=TRUE, na.rm=TRUE, check.param=TRUE) {

+                      ..., arrow=NULL, rootnode=TRUE, show.legend=NA, inherit.aes=TRUE,

+                      na.rm=TRUE, check.param=TRUE) {

     default_aes <- aes_(x=~x, y=~y,node=~node, parent=~parent)

     if (multiPhylo) {

@@ -33,6 +34,10 @@ stat_tree <- function(mapping=NULL, data=NULL, geom="segment", position="identit

         mapping <- modifyList(default_aes, mapping)

     }

+    if (!is.null(arrow)) {

+        rootnode <- FALSE

+    }

+

     if (layout %in% c("rectangular", "fan", "circular")) {

         list(layer(data=data,

                    mapping=mapping,

@@ -44,6 +49,8 @@ stat_tree <- function(mapping=NULL, data=NULL, geom="segment", position="identit

                    params=list(layout = layout,

                                lineend = lineend,

                                na.rm = na.rm,

+                               arrow = arrow,

+                               rootnode = rootnode,

                                ...),

                    check.aes = FALSE

                    ),

@@ -57,6 +64,8 @@ stat_tree <- function(mapping=NULL, data=NULL, geom="segment", position="identit

                    params=list(layout = layout,

                                lineend = lineend,

                                na.rm = na.rm,

+                               arrow = arrow,

+                               rootnode = rootnode,

                                ...),

                    check.aes = FALSE

                    )

@@ -72,6 +81,8 @@ stat_tree <- function(mapping=NULL, data=NULL, geom="segment", position="identit

               params=list(layout = layout,

                           lineend = lineend,

                           na.rm = na.rm,

+                          arrow = arrow,

+                          rootnode = rootnode,

                           ...),

               check.aes = FALSE

               )

@@ -151,7 +162,7 @@ StatTreeVertical <- ggproto("StatTreeVertical", Stat,

                                     df$yend <- y

                                     if (!rootnode) {

-                                        df <- dplyr::filter(df, .data$node != tidytree:::rootnode.tbl_tree(df)$node)

+                                        df <- dplyr::filter(df, .data$node != rootnode.tbl_tree(df)$node)

                                     }

                                     if (continuous && !is.null(df$colour ))

@@ -190,7 +201,7 @@ StatTree <- ggproto("StatTree", Stat,

                             df$yend <- y

                             if (!rootnode) {

-                                df <- dplyr::filter(df, .data$node != tidytree:::rootnode.tbl_tree(df)$node)

+                                df <- dplyr::filter(df, .data$node != rootnode.tbl_tree(df)$node)

                             }

                             if (continuous && !is.null(df$colour)) {

@@ -46,7 +46,9 @@ ggtree <- function(tr,

                    ...) {

     # Check if layout string is valid.

-    layout %<>% match.arg(c("rectangular", "slanted", "fan", "circular", "radial", "unrooted", "equal_angle", "daylight"))

+    layout %<>% match.arg(c("rectangular", "slanted", "fan", "circular",

+                            "radial", "unrooted", "equal_angle", "daylight"))

+

     if (layout == "unrooted") {

         layout <- "daylight"

         message('"daylight" method was used as default layout for unrooted tree.')

@@ -70,4 +70,10 @@ ggplot2::geom_label

 ##' @export

 ggplot2::geom_point

+##' @importFrom grid arrow

+##' @export

+grid::arrow

+##' @importFrom grid unit

+##' @export

+grid::unit

@@ -1,2 +1,3 @@

 nodeid.tbl_tree <- utils::getFromNamespace("nodeid.tbl_tree", "tidytree")

+rootnode.tbl_tree <- utils::getFromNamespace("rootnode.tbl_tree", "tidytree")

 offspring.tbl_tree <- utils::getFromNamespace("offspring.tbl_tree", "tidytree")

@@ -21,6 +21,8 @@

 \alias{geom_text}

 \alias{geom_label}

 \alias{geom_point}

+\alias{arrow}

+\alias{unit}

 \title{Objects exported from other packages}

 \keyword{internal}

 \description{

@@ -34,6 +36,8 @@ below to see their documentation.

   \item{ggplot2}{\code{\link[ggplot2]{fortify}}, \code{\link[ggplot2]{ggplot}}, \code{\link[ggplot2]{xlim}}, \code{\link[ggplot2]{theme}}, \code{\link[ggplot2]{ggsave}}, \code{\link[ggplot2]{aes}}, \code{\link[ggplot2]{geom_text}}, \code{\link[ggplot2]{geom_label}}, \code{\link[ggplot2]{geom_point}}}

+  \item{grid}{\code{\link[grid]{arrow}}, \code{\link[grid]{unit}}}

+

   \item{magrittr}{\code{\link[magrittr]{\%>\%}}}

   \item{tidytree}{\code{\link[tidytree]{groupOTU}}, \code{\link[tidytree]{groupClade}}, \code{\link[tidytree]{nodeid}}, \code{\link[tidytree]{nodelab}}, \code{\link[tidytree]{MRCA}}}