R/EM.plot.R

In franciscorichter/dmea: Diversification and Macro-Evolution Analysis

EM.plot <- function(M,dim=FALSE){
  if(dim){
    EM.plot2(M)
  }
  else{
    EM.plot1(M)
  }
}

EM.plot1 <-function(M){

grid.newpage()

# two plots
p1 <- ggplot(M, aes(it, ltt)) + geom_point(colour = "blue") + theme_bw()
p2 <- ggplot(M, aes(it, EM)) + geom_line(colour = "red") + theme_bw() %+replace%
  theme(panel.background = element_rect(fill = NA))

# extract gtable
g1 <- ggplot_gtable(ggplot_build(p1))
g2 <- ggplot_gtable(ggplot_build(p2))

# overlap the panel of 2nd plot on that of 1st plot
pp <- c(subset(g1$layout, name == "panel", se = t:r))
g <- gtable_add_grob(g1, g2$grobs[[which(g2$layout$name == "panel")]], pp$t,
                     pp$l, pp$b, pp$l)

# axis tweaks
ia <- which(g2$layout$name == "axis-l")
ga <- g2$grobs[[ia]]
ax <- ga$children[[2]]
ax$widths <- rev(ax$widths)
ax$grobs <- rev(ax$grobs)
ax$grobs[[1]]$x <- ax$grobs[[1]]$x - unit(1, "npc") + unit(0.15, "cm")
g <- gtable_add_cols(g, g2$widths[g2$layout[ia, ]$l], length(g$widths) - 1)
g <- gtable_add_grob(g, ax, pp$t, length(g$widths) - 1, pp$b)

# draw it
grid.draw(g)
}


EM.plot2 <-function(M){

  grid.newpage()

  # two plots
  col = rep("blue",length(M$ltt))
  col[M$sd == max(M$sd)] = 'green'
  p1 <- ggplot(M, aes(it, ltt, size=sd)) + geom_point(colour = col) + theme_bw()
  p2 <- ggplot(M, aes(it, EM)) + geom_line(colour = "red") + theme_bw() %+replace%
    theme(panel.background = element_rect(fill = NA))

  # extract gtable
  g1 <- ggplot_gtable(ggplot_build(p1))
  g2 <- ggplot_gtable(ggplot_build(p2))

  # overlap the panel of 2nd plot on that of 1st plot
  pp <- c(subset(g1$layout, name == "panel", se = t:r))
  g <- gtable_add_grob(g1, g2$grobs[[which(g2$layout$name == "panel")]], pp$t,
                       pp$l, pp$b, pp$l)

  # axis tweaks
  ia <- which(g2$layout$name == "axis-l")
  ga <- g2$grobs[[ia]]
  ax <- ga$children[[2]]
  ax$widths <- rev(ax$widths)
  ax$grobs <- rev(ax$grobs)
  ax$grobs[[1]]$x <- ax$grobs[[1]]$x - unit(1, "npc") + unit(0.15, "cm")
  g <- gtable_add_cols(g, g2$widths[g2$layout[ia, ]$l], length(g$widths) - 1)
  g <- gtable_add_grob(g, ax, pp$t, length(g$widths) - 1, pp$b)

  # draw it
  grid.draw(g)
}