Add initial stab at position_jitterdodge

Author: kevinushey

DESCRIPTION

@@ -143,6 +143,7 @@ Collate:
     'position-fill.r'
     'position-identity.r'
     'position-jitter.r'
+    'position-jitterdodge.R'
     'position-stack.r'
     'quick-plot.r'
     'save.r'

NAMESPACE

@@ -286,7 +286,9 @@ export(plotmatrix)
 export(position_dodge)
 export(position_fill)
 export(position_identity)
+export(position_jd)
 export(position_jitter)
+export(position_jitterdodge)
 export(position_stack)
 export(qplot)
 export(quickplot)

R/position-jitterdodge.R

@@ -0,0 +1,56 @@
+#' Jitter-dodge points to align them with a boxplot including fill aesthetic
+#'
+#' @family position adjustments
+#' @param width degree of jitter in x direction. Defaults to 40\% of the
+#'   resolution of the data.
+#' @param height degree of jitter in y direction. Defaults to 40\% of the
+#'   resolution of the data
+#' @export
+#' @examples
+#' dsub <- diamonds[ sample(1:nrow(diamonds), 1000), ]
+#' ggplot(dsub, aes(x=cut, y=carat, fill=clarity)) +
+#'   geom_boxplot(outlier.size=0) +
+#'   geom_point( pch=21, position=position_jitterdodge() )
+position_jitterdodge <- function (width = NULL, height = NULL) {
+  PositionJitterDodge$new(width = width, height = height)
+}
+
+#' @rdname position_jitterdodge
+#' @export
+position_jd <- position_jitterdodge
+
+PositionJitterDodge <- proto(Position, {
+  objname <- "jitterdodge"
+
+  adjust <- function(., data) {
+
+    if (empty(data)) return(data.frame())
+    check_required_aesthetics(c("x", "y", "fill"), names(data), "position_jitterdodge")
+
+    ## Workaround to avoid this warning:
+    ## ymax not defined: adjusting position using y instead
+    if (!("ymax" %in% names(data))) {
+      data$ymax <- data$y
+    }
+
+    ## Adjust the x transformation based on the number of 'fill' variables
+    nfill <- length( levels(data$fill) )
+
+    if (is.null(.$width)) .$width <- resolution(data$x, zero = FALSE) * 0.4
+    if (is.null(.$height)) .$height <- resolution(data$y, zero = FALSE) * 0.4
+
+    trans_x <- NULL
+    trans_y <- NULL
+    if(.$width > 0) {
+      trans_x <- function(x) jitter(x, amount = .$width / (nfill + 2))
+    }
+    if(.$height > 0) {
+      trans_y <- function(x) jitter(x, amount = .$height)
+    }
+
+    ## dodge, then jitter
+    data <- collide(data, 0.75, .$my_name(), pos_dodge, check.width = FALSE)
+    transform_position(data, trans_x, trans_y)
+  }
+
+})

man/position_dodge.Rd

@@ -42,6 +42,8 @@ p + geom_errorbar(aes(ymin = y-1, ymax = y+1, width = 0.2),
 \seealso{
 Other position adjustments: \code{\link{position_fill}};
   \code{\link{position_identity}};
+  \code{\link{position_jd}},
+  \code{\link{position_jitterdodge}};
   \code{\link{position_jitter}};
   \code{\link{position_stack}}
 }

man/position_fill.Rd

@@ -37,6 +37,8 @@ See \code{\link{geom_bar}} and \code{\link{geom_area}} for
 
 Other position adjustments: \code{\link{position_dodge}};
   \code{\link{position_identity}};
+  \code{\link{position_jd}},
+  \code{\link{position_jitterdodge}};
   \code{\link{position_jitter}};
   \code{\link{position_stack}}
 }

man/position_identity.Rd

@@ -17,7 +17,8 @@ Don't adjust position
 }
 \seealso{
 Other position adjustments: \code{\link{position_dodge}};
-  \code{\link{position_fill}};
+  \code{\link{position_fill}}; \code{\link{position_jd}},
+  \code{\link{position_jitterdodge}};
   \code{\link{position_jitter}};
   \code{\link{position_stack}}
 }

man/position_jitter.Rd

@@ -36,6 +36,8 @@ qplot(class, hwy, data = mpg, geom = c("boxplot", "jitter"))
 Other position adjustments: \code{\link{position_dodge}};
   \code{\link{position_fill}};
   \code{\link{position_identity}};
+  \code{\link{position_jd}},
+  \code{\link{position_jitterdodge}};
   \code{\link{position_stack}}
 }
 

man/position_jitterdodge.Rd

@@ -0,0 +1,34 @@
+% Generated by roxygen2 (4.0.0): do not edit by hand
+\name{position_jitterdodge}
+\alias{position_jd}
+\alias{position_jitterdodge}
+\title{Jitter-dodge points to align them with a boxplot including fill aesthetic}
+\usage{
+position_jitterdodge(width = NULL, height = NULL)
+
+position_jd(width = NULL, height = NULL)
+}
+\arguments{
+  \item{width}{degree of jitter in x direction. Defaults to
+  40\% of the resolution of the data.}
+
+  \item{height}{degree of jitter in y direction. Defaults
+  to 40\% of the resolution of the data}
+}
+\description{
+Jitter-dodge points to align them with a boxplot including fill aesthetic
+}
+\examples{
+dsub <- diamonds[ sample(1:nrow(diamonds), 1000), ]
+ggplot(dsub, aes(x=cut, y=carat, fill=clarity)) +
+  geom_boxplot(outlier.size=0) +
+  geom_point( pch=21, position=position_jitterdodge() )
+}
+\seealso{
+Other position adjustments: \code{\link{position_dodge}};
+  \code{\link{position_fill}};
+  \code{\link{position_identity}};
+  \code{\link{position_jitter}};
+  \code{\link{position_stack}}
+}
+

man/position_stack.Rd

@@ -45,6 +45,8 @@ qplot(Time, Value, data = data.set, colour = Type, geom = "line",
 Other position adjustments: \code{\link{position_dodge}};
   \code{\link{position_fill}};
   \code{\link{position_identity}};
+  \code{\link{position_jd}},
+  \code{\link{position_jitterdodge}};
   \code{\link{position_jitter}}
 }