mrwaterPlots: mrwater plots

plotmap3 <- function(data,file=NULL,title="World map",legend_range=NULL,legendname="Cell share",lowcol="grey95",midcol="orange",highcol="darkred",midpoint=0.5,facet_grid="Year~Data1",nrow=NULL,ncol=NULL,scale=2,breaks=T,labs=T,borders=TRUE,MAgPIE_regions=F,axis_text_col="black",legend_discrete=FALSE,legend_breaks=NULL,show_percent=FALSE,sea=TRUE,land_colour="white",legend_height=2,legend_width=NULL,text_size=12,legend_position="right",facet_style="default",plot_height=10,plot_width=20) {
  #require("ggplot2", quietly = TRUE)
  #require("RColorBrewer", quietly = TRUE)
  wrld_simpl_df <- NULL
  data("world", envir = environment(), package="luplot")
  if (MAgPIE_regions) {
    facet_grid <- NULL
    map <- ggplot(wrld_simpl_df,aes_(~long, ~lat)) +
      geom_polygon(aes_(group=~group, fill=~magpie)) +
      #  scale_fill_manual("MAgPIE\nregions",values=brewer.pal(11,"Paired")[2:11],na.value="white")
      scale_fill_manual("MAgPIE\nregion",values=c("purple3", "red3", "hotpink", "cyan3", "goldenrod1", "gray44","#8C5400FF","darkorange2","royalblue3","green4"),na.value="white")
    #      scale_fill_manual("MAgPIE\nregions",values=c(nice_colors(style="contrast_area",saturation=1)[1:10]),na.value="white")
    #      scale_fill_brewer(palette="Set3",na.value="white")
  } else {
    if (!is.list(data)) {
      temp <- data
      data <- list()
      data[["default"]] <- temp
    }
    if (is.null(legend_range)) {
      midpoint <- max(unlist(lapply(data,max,na.rm=T)))*midpoint
    } else {
      data <- lapply(data, function(x) {
        x[which(x < legend_range[1])] <- legend_range[1]
        x[which(x > legend_range[2])] <- legend_range[2]
        return(x)
      })
    }
    if (any(unlist(lapply(data,function(x) return(is.null(attr(x,"coordinates"))))))) {
      data <- lapply(data, function(x) {
        attr(x,"coordinates") <- getCoordinates(degree = T)
        return(x)
      })
      warning("Missing coordinates in attributes for at least one MAgPIE object. Added coordinates in default MAgPIE cell order.")
    }
    data <- as.ggplot(data,asDate=F)
    if(legend_discrete) {
      if(is.null(legend_breaks)) {
        data$Breaks<-as.character(data$Value)
        #replace NA's with a value that is not contained in data yet
        if(any(is.na(data$Breaks))){
          data$Breaks[is.na(data$Breaks)]<-"No data"
        }
        data$Breaks<-as.factor(data$Breaks)
        legend_labels<-levels(data$Breaks)
      } else {
        tmp<-as.vector(data$Value)
        tmp[]<-length(legend_breaks)+1
        legend_labels<-rep("",length(legend_breaks)+1)
        legend_labels[length(legend_breaks)+1]<-paste(">",legend_breaks[length(legend_breaks)])
        for(i in length(legend_breaks):2){
          tmp[which(as.vector(data$Value)<=legend_breaks[i])]<-i
          legend_labels[i]<-paste(legend_breaks[i-1],"-",legend_breaks[i])
        }
        tmp[which(as.vector(data$Value)<legend_breaks[1])]<-1
        tmp[which(is.na(as.vector(data$Value)))]<-NA
        legend_labels[1]<-paste("<",legend_breaks[1])
        legend_labels<-legend_labels[as.numeric(rev(levels(as.factor(tmp))))]
        tmpchar<-as.character(tmp)
        tmpchar[is.na(tmpchar)]<-"No data"
        levels<-rev(levels(as.factor(tmp)))
        if("No data" %in% tmpchar){
          legend_labels<-c(legend_labels,"No data")
          levels<-c(levels,"No data")
        }
        data$Breaks <- factor(tmpchar, levels = levels)
      }

      if("No data" %in% legend_labels){
        colours <- c(colorRampPalette(c(highcol,midcol,lowcol))( length(legend_labels)-1 ),"grey")
      } else {
        colours <- colorRampPalette(c(highcol,midcol,lowcol))( length(legend_labels))
      }

      if(show_percent){
        tmp<-table(data$Breaks)
        if("No data" %in% names(tmp)) tmp<-tmp[-which(names(tmp)=="No data")]
        percent<-round(tmp/sum(tmp)*100,1)
        percent<-paste("(",percent,"%)",sep="")
        legend_labels[which(legend_labels!="No data")]<-paste(legend_labels[which(legend_labels!="No data")],percent,sep="     ")
      }

    }
    if (!is.null(legend_breaks)) {
      labels <- c(bquote(""<=.(head(legend_breaks,1))),legend_breaks[2:(length(legend_breaks)-1)],bquote("">=.(tail(legend_breaks,1))))
    } else {
      legend_breaks <- waiver()
      labels <- waiver()
    }
    map <- ggplot(data,aes_(~x,~y)) + geom_polygon(data=wrld_simpl_df, aes_(~long,~lat, group=~group, fill=~hole), fill=land_colour)
    if (is.null(data$Breaks)) {
      if (!is.null(midcol)) map <- map + geom_raster(aes_(fill=~Value)) + scale_fill_gradient2(name=legendname,low=lowcol,mid=midcol,high=highcol,midpoint=midpoint,limits=legend_range,breaks=legend_breaks,labels=labels,na.value="grey")
      else map <- map + geom_raster(aes_(fill=~Value)) + scale_fill_gradient(name=legendname,low=lowcol,high=highcol,limits=legend_range,breaks=legend_breaks,labels=labels,na.value="grey")
    } else map <- map + geom_raster(aes_(fill=~Breaks)) + scale_fill_manual(name=legendname,values=colours,labels=legend_labels,na.value="yellow")
    if(!is.null(legend_height)) map <- map + theme(legend.key.height = unit(legend_height, "cm"))
    if(!is.null(legend_width)) map <- map + theme(legend.key.width = unit(legend_width, "cm"))
  }
  map <- map +
    #coord_cartesian(xlim = c(-180, 180), ylim = c(-58, 86)) +

    coord_map(projection = "mercator", xlim = c(-180, 180), ylim = c(-58, 86), clip = "on")

    theme(aspect.ratio = 0.5) +
    ggtitle(title)
  if (sea) map <- map + theme(panel.background = element_rect(fill = "lightsteelblue2"))
  else map <- map + theme(panel.background = element_rect(fill="white", colour="black")) + theme(panel.grid.major=element_line(colour="grey80"),panel.grid.minor=element_line(colour="grey90"))
  if (!is.null(facet_grid)) {
    if (substr(facet_grid,1,1) == "~") map <- map + facet_wrap(facet_grid, nrow = nrow, ncol = ncol)
    else map <- map + facet_grid(facet_grid)
  }
  if(breaks) map <- map + scale_x_continuous(breaks=c(-90, 0, 90)) + scale_y_continuous(breaks=c(-66,-38,-23, 0, 23,38, 66))
  else map <- map + scale_x_continuous(breaks=NULL) + scale_y_continuous(breaks=NULL)
  if(labs) map <- map + labs(x="Longitude",y="Latitude")
  else map <- map + labs(y=NULL,x=NULL)
  if(borders) map <- map + geom_path(data=wrld_simpl_df, aes_(~long, ~lat, group=~group, fill=NULL), color="grey10", size=0.1)
  if (!is.null(axis_text_col)) map <- map + theme(axis.text = element_text(colour=axis_text_col),axis.ticks = element_line(colour=axis_text_col))
  map <- map + theme(panel.grid.minor=element_line(colour="white"),plot.title=element_text(size=text_size+4,face="bold",vjust=1.5), legend.position=legend_position, legend.title=element_text(size=text_size,face="bold"), legend.text=element_text(size=text_size), axis.title.y=element_text(angle=90,size=text_size,vjust=0.3), axis.text.y=element_text(size=text_size-2), axis.title.x=element_text(size=text_size,vjust=-0.3), axis.text.x=element_text(size=text_size-2, vjust=0.5))
  if (facet_style == "default") map <- map + theme(strip.text.x=element_text(size=text_size-1), strip.text.y=element_text(size=text_size-1))
  else if (facet_style == "paper") map <- map + theme(strip.text.x=element_text(size=text_size,face="bold"),strip.text.y=element_text(size=text_size,face="bold")) + theme(strip.background = element_blank())
  if (legend_position %in% c("top","bottom")) map <- map + guides(fill = guide_colorbar(title.position = "top")) + theme(legend.box.just = "left")
  if(!is.null(file)) {
    ggsave(file,map,scale=scale,limitsize=FALSE,units = "cm",height = plot_height,width = plot_width)
  } else {
    return(map)
  }
}