如何在ggplot中找到geom_curve控制点

时间:2018-07-18 16:45:06

标签: r ggplot2

我想知道如何在ggplot2中找到 geom_curve 控制点吗?例如

p <- ggplot(mtcars, aes(wt, mpg)) + 
geom_curve(aes(x = 2.62, y = 21.0, xend = 3.57, yend = 15.0),curvature = -0.2, data = df) + 
geom_point()

b <- ggplot_build(p)
b$data[[1]]
p$layers[[1]]$geom_params

b$data[[1]]给出起点终点点,而
p$layers[[1]]$geom_params给出曲线信息(角度,曲率,... )。
但是,如何找到控制点,以便可以复制图形?

1 个答案:

答案 0 :(得分:0)

library(ggplot)
library(grid)
library(stringr)

df <- data.frame(x = 1:3, y = 1:3)
df2 <- data.frame(x = c(1,3), y = c(1,3),
              xend = c(2,2), yend = c(2,2))
g <- ggplot(df, aes(x, y)) +
  geom_point() +
  geom_curve(aes(x = x ,y = y,
                 xend = xend, yend = yend),
             data = df2,
             color = c("red", "blue"))
g

getCurve_controlPoints <- function(ggplotObject) {
  len_layers <- length(ggplotObject$layers)
  layerNames <- lapply(seq_len(len_layers),
                       function(j) {
                         className <- class(ggplotObject$layers[[j]]$geom)
                         className[-which(className %in% c("ggproto"  ,"gg", "Geom"))] 
                       })
  curveLayerId <- which(sapply(layerNames,
                               function(layerName){
                                 "GeomCurve" %in% layerName
                               }) == TRUE
  )
  gg_build <- ggplot_build(ggplotObject)
  # you can also add yes or no in your code
  # answer <- utils::menu(c("y", "n"), title="Do you want to draw the ggplot?")
  grid.draw(ggplotObject)
  grid.force()
  gridList <- grid.ls(print = FALSE)
  gridList.name <- gridList$name
  xspline.name <- gridList.name[which(str_detect(gridList.name, "curve") == TRUE)]
  xspline.len <- length(xspline.name)

  controlPoints <- lapply(seq_len(length(curveLayerId)),
                          function (j) {
                            # curve data
                            curve_data <- gg_build$data[[curveLayerId[j]]]
                            # avoid duplicated rows
                            curve_data <- curve_data[!duplicated(curve_data), ]
                            n <- dim(curve_data)[1]
                            # here we go! But wait a second, it seems like the starting and ending position do not match
                            xsplinePoints <- xsplinePoints(grid.get("xspline", grep=TRUE))  
                            # mapping data to our coordinates
                            control_data <- lapply(seq_len(n),
                                                   function(i){
                                                     if (n == 1) {
                                                       xy <- lapply(xsplinePoints,
                                                                    function(coord){
                                                                      as.numeric(coord) 
                                                                    })
                                                     } else {
                                                       xy <- lapply(xsplinePoints[[i]],
                                                                    function(coord){
                                                                      as.numeric(coord)
                                                                    })
                                                     }
                                                     x.start <- curve_data[i, ]$x
                                                     x.end <- curve_data[i, ]$xend
                                                     y.start <- curve_data[i, ]$y
                                                     y.end <- curve_data[i, ]$yend
                                                     # mapping to ggplot coordinates
                                                     xy_x.diff <- xy$x[length(xy$x)] - xy$x[1] 
                                                     xy_y.diff <- xy$y[length(xy$y)] - xy$y[1]
                                                     # maybe there is a better way?
                                                    if(xy_x.diff == 0){
                                                       xy_x.diff <- 1e-16
                                                     }
                                                     if(xy_y.diff == 0){
                                                        xy_y.diff <- 1e-16
                                                     }
                                                     x <- (x.end - x.start) / (xy_x.diff) * (xy$x - xy$x[1]) + x.start
                                                     y <- (y.end - y.start) / (xy_y.diff) * (xy$y - xy$y[1]) + y.start
                                                     list(x = x, y = y)
                                                   })
                            # grid remove
                            grid.remove(xspline.name[j], redraw = FALSE)
                            control_data
                          })

  controlPoints
}
controlPoints <- getCurve_controlPoints(g)
# check the points
plot(controlPoints[[1]][[1]]$x, controlPoints[[1]][[1]]$y, 
     xlim = c(1,3), 
     ylim = c(1,3),
     xlab = "x", 
     ylab = "y", 
     pch = 19)
points(controlPoints[[1]][[2]]$x, controlPoints[[1]][[2]]$y, pch = 19)

我认为到目前为止,效果很好,我使用的ggplot版本为 3.0.0 。如果您使用的版本低于 2.2.1 ,则可能会发生错误。

Paul Murrell教授提出了这个想法。捕获geom_curve控制点的最简单方法也许是?缺点是grob s(ggplot对象)必须首先绘制,因为这些点仅在绘制时生成。