first commit

Jonathan Chang · Jonathan Chang · commit 37559b6b25ac · 2011-07-25T16:05:55.000-07:00
diff --git a/Rpipe.R b/Rpipe.R
@@ -0,0 +1,74 @@
+#!Rscript
+
+################################
+# Wrappers for common operations
+################################
+
+Rpipe.group <- function(d, key, expr) {
+  expr <- substitute(expr)
+  key <- eval(substitute(key), d)
+  do.call(rbind, 
+          by(d, key, function(x) Rpipe.eval(x, expr)))
+}
+
+Rpipe.aggregate <- function(d, key, ...) {
+  key.name <- deparse(substitute(key))
+  key <- eval(substitute(key), d)
+  expr <- substitute(list(...))
+  do.call(rbind,
+          by(cbind(d, .key = key), key, function(x) {
+             do.call(data.frame,
+                     c(structure(list(x[1, ".key"]),
+                                 names = key.name),
+                            eval(expr, x)))
+          }))
+}
+
+Rpipe.sort <- function(d, ..., decreasing = FALSE) {
+  expr <- substitute(list(...))
+  d[do.call("order", c(eval(expr, d), decreasing = decreasing)),]
+}
+
+################################
+# Core operations
+################################
+
+Rpipe.prefix <- function(e1) {
+  UseMethod("Rpipe.prefix")
+}
+
+Rpipe.prefix.default <- function(e1) {
+  c("Rpipe.", "")
+}
+
+Rpipe.eval <- function(e1, expr) {
+  if (as.character(expr[1]) == "(") {
+    do.call(`%|%`, list(e1 = e1, e2 = expr[[2]]))
+  } else if (as.character(expr[1]) == "%|%") {
+    left <- do.call(`%|%`, list(e1 = e1, e2 = expr[[2]]))
+    do.call(`%|%`, list(e1 = left, e2 = expr[[3]]))
+  } else if (as.character(expr[1]) == "%&&%") {
+    structure(list(e1 = do.call(`%|%`, list(e1 = e1, e2 = expr[[2]])),
+                   e2 = do.call(`%|%`, list(e1 = e1, e2 = expr[[3]]))),
+              class = "Rpipe.tuple")
+  } else {
+    prefixes <- Rpipe.prefix(e1)
+    func.names <- paste(prefixes, expr[1], sep = "")
+    lookups <- mget(func.names, 
+                    envir = as.environment(-1),
+                    ifnotfound = list(NULL),
+                    inherits = TRUE)
+    func.names <- lookups[-which(sapply(lookups, is.null))]
+    if (length(func.names) == 0) {
+      stop(paste("Could not find pipe function", expr[1]))
+    }
+    do.call(names(func.names)[1],
+            c(list(e1), as.list(expr[-1])))
+  }
+}
+
+`%|%` <- function(e1, e2) {
+  expr <- substitute(e2)
+  Rpipe.eval(e1, expr)
+}
+
diff --git a/Rpipe.Rd b/Rpipe.Rd
@@ -0,0 +1,39 @@
+\name{Rpipe}
+\alias{Rpipe}
+\title{
+  Manipulate a data frame through a series of pipe operations.
+}
+\description{
+  
+}
+\usage{
+df %|% operation()
+}
+\arguments{
+  \item{df}{
+    A data frame to operate on.  Note that this can be the output of another \code{Rpipe} stage.}
+  \item{e2}{
+%%     ~~Describe \code{e2} here~~
+}
+}
+\details{
+%%  ~~ If necessary, more details than the description above ~~
+}
+\value{
+%%  ~Describe the value returned
+%%  If it is a LIST, use
+%%  \item{comp1 }{Description of 'comp1'}
+%%  \item{comp2 }{Description of 'comp2'}
+%% ...
+}
+\author{
+  Jonathan Chang <jonchang@fb.com>
+}
+
+\examples{
+iris %|%
+  subset(Sepal.Width > 3.0) %|%
+    sort(Sepal.Length) %|%
+      group(Species, head()) %|%
+        aggregate(Species, Mean.Petal.Width = mean(Petal.Width))
+}
diff --git a/Rpipe.jjplot.R b/Rpipe.jjplot.R
@@ -0,0 +1,49 @@
+################################
+# Initial Rpipe interface for jjplot
+################################
+
+  ## Need to do checks on e1 = jjplot.state?
+  ## both jjplot.stat.foo and jjplot.geom.foo
+
+Rpipe.prefix.jjplot.state <- function(e1) {
+  c("jjplot.stat",
+    "jjplog.geom",
+    "Rpipe",
+    "")
+}
+
+jjplot.data <- function(data, ...) {
+  structure(list(data = transform(data, ...)),
+            class = "jjplot.state")
+}
+
+jjplot.range <- function(state, ...) {
+  list(x = range(state$data$x),
+       y = range(state$data$y))
+}
+
+jjplot.geom <- function(plot.function, 
+                        range.function = jjplot.range) {
+  ## Add extra info here!
+  function(state, ...) {
+    structure(list(plot.function = plot.function,
+                   range = range.function(state, ...),
+                   state = state),
+              class = "jjplot.goem")
+  }
+}
+
+## Logic for scales:
+## Two aspects
+##  1.) ticks
+##  2.) range
+##  range should be union of numeric ranges
+##  ticks depend on type
+##   numeric - pretty
+##   factor - at ticks
+##  cast numeric -> factor
+##  check if factors equal.
+
+jjplot.data(iris) %|% 
+  aggregate(Species, Mean.Sepal.Width = mean(Sepal.Width)) %|%
+  transform(x = Species, y = Mean.Sepal.Width)
