scrap-code/ggtransform.r

In garrettgman/ggplyr: Explore large data with binned modelling

ggtransform <- function(ggobject, mapping = NULL, ...) {
	UseMethod("ggtransform")
}

ggtransform.proto <- function(ggobject, mapping = NULL, ...) {

	ggobject <- layer_clone(ggobject)
	ggobject$mapping <- combine_aes(mapping, ggobject$mapping)
	ggobject$geom_params <- c(ggobject$geom_params, list(...))
	ggobject
}

ggtransform.ggplot <- function(ggobject, mapping = NULL, ...) {

	ggobject <- ggplot2:::plot_clone(ggobject)
	ggobject$mapping <- combine_aes(mapping, ggobject$mapping)
	ggobject$layers <- lapply(ggobject$layers, ggtransform, 
		mapping = mapping, ...)
	labels <- ggobject$options$labels
	ggobject$options$labels <- combine_labels(lapply(mapping, deparse), 
		labels) 
	ggobject
	
}
	
#' @export
setGeneric("ggtransform")

#' @export
setMethod("ggtransform", signature(ggobject = "glyphs"), 
          function(ggobject, mapping, ...){
            ggplot <- ggtransform(ggobject@.Data, mapping, ...)
            new("glyphs", ggplot)
          }
)

#' @export
setMethod("ggtransform", signature(ggobject = "glayer"), 
          function(ggobject, mapping, ...){
            layer <- ggtransform(ggobject@layer, mapping, ...)
            new("glayer", layer = layer)
          }
)