R/mplot.list.R

In itsaquestion/MyPlot:

#' mplot
#'
#' @param x
#'
#' @return
#' @export
#'
#' @examples
mplot = function(x,...){
  UseMethod("mplot",x)
}



#' mplot.list
#'
#' @param plots
#' @param use.one.x
#' @param theme
#'
#' @return
#' @export
#'
#' @examples
mplot.list = function(plots, use.one.x = F) {
	# mplot的核心部分

  numPlots = length(plots)

  tmp.plots = NULL

  for(i in 1:numPlots){

    theme = plots[[i]]$theme

    if(use.one.x){
      axisTheme = theme + theme(axis.title.x = element_blank(),
                                axis.text.x = element_blank())
    }else{
      axisTheme = theme
    }

    top.margin = 0.4
    right.margin = 0.1
    bottom.margin = 0.1
    left.margin = 0.1

    #if(plots[[1]]$theme$legend.position == "right"){
      #right.margin = 0.1
    #}
    #if(plots[[1]]$theme$legend.position == "bottom"){
      #bottom.margin = -0.5
    #}

    top.theme = axisTheme +
      theme(plot.margin = unit(c(top.margin, right.margin, bottom.margin, left.margin), "cm"))

    mid.theme = axisTheme +
      theme(plot.margin = unit(c(0, right.margin, bottom.margin, left.margin), "cm"))

    bottom.theme = theme + theme(plot.margin = unit(c(0, right.margin, 0.1, left.margin), "cm"))

    if(numPlots > 1){
      if(i == 1){
        # top plot
        gp = ggplotGrob(plots[[i]] + top.theme)

      }else if( i!= numPlots ){
        # middles
        gp = ggplotGrob(plots[[i]] + mid.theme)

      }else{
        # bottom one
        gp = ggplotGrob(plots[[i]] + bottom.theme)
      }
    }else{
      gp = ggplotGrob(plots[[i]] + theme)
    }


    if( i == 1){
      tmp.plots =gp
    }else{
      tmp.plots= rbind(tmp.plots,gp,size="last")
    }
  }

  grid::grid.newpage()
  grid::grid.draw(tmp.plots)
}