I made a 3D movie with ggplot2 once – here’s how I did it
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.
Some time ago (last year actually ?) I had a blast developing a feature for ggforce
which had been on my mind for far to long than its limited utility warranted. The idea was to showcase the new facetting extension powers I’d added to ggplot2
by making a facetting function that created a stereoscopic pair of plots that would simulate 3D. To procrastinate and show off I made a little animated video with the feature and posted it on Twitter, promising I’d write about it someday. Now (again, one year later), I think the world is finally ready to see what went through my R console to make that little animation. While I’ve been very timely with this blog post, the feature is still not available on CRAN, so you’ll need to install the GitHub version of ggforce
to follow along.
devtools::install_github('thomasp85/ggforce')
Setup
The goal is to create a spinning hollow cube so we’ll need to define the cube somehow. This was way before ggraph
was published, and tidygraph
was not even a thought in my head, so while it may make sense to handle the cube as a network, I did it the hard way:
library(ggplot2) library(ggforce) # Define the corners of the cube cube <- matrix( c( 1, 1, 1, 1, -1, 1, -1, -1, 1, -1, 1, 1, 1, 1, -1, 1, -1, -1, -1, -1, -1, -1, 1, -1 ), ncol = 3, byrow = TRUE ) # Define each edge - practically making a network the hard way segments <- data.frame( ind = c(1, 2, 2, 3, 3, 4, 4, 1, 5, 6, 6, 7, 7, 8, 8, 5, 1, 5, 2, 6, 3, 7, 4, 8), group = rep(1:12, each = 2) )
Now that we have our data, we need to create some transformation functions. Currently, if plotted in ggplot2
it would just look like a square as, unsurprisingly, the third dimension would get lost. What we need is a projection of three dimension down to two. This is a bit hairy, and when I tried to achieve this back in the days by setting up a transformation matrix manually I failed miserably (if you are knowledgable in this and want to explain it to me - please reach out on twitter). In the end I took the shortcut and used the persp()
function, which, besides the side effect of plotting stuff in 3D, also returns the transformation matrix invisibly:
tr_mat <- persp(z = cube, theta = 0, phi = 0, xlim = c(-2, 2), ylim = c(-2, 2), zlim = c(-2,2))
Having a look at the transformation matrix I can safely say that I have no idea what’s going on:
tr_mat #> [,1] [,2] [,3] [,4] #> [1,] 0.5 0.000000e+00 0.000000e+00 0.000000e+00 #> [2,] 0.0 3.061617e-17 -5.000000e-01 5.000000e-01 #> [3,] 0.0 5.000000e-01 3.061617e-17 -3.061617e-17 #> [4,] 0.0 0.000000e+00 -2.732051e+00 3.732051e+00
But that doesn’t matter - the only thing required is that I know how to use it. The trans3d()
provides the means for taking in three dimensional points, and a transformation matrix, and outputting two dimensional points:
trans_cube <- trans3d(cube[,1], cube[,2], cube[,3], tr_mat) ggplot(as.data.frame(trans_cube)) + geom_point(aes(x, y))
With a bit of imagination we can see the 3D cube. Lets draw it with line segments as well - we add the added information of depth, which is simply the value in the dimension that gets dropped in the transformation.
to_grid <- function(mat, segments, tr_mat) { trans <- trans3d(mat[, 1], mat[, 2], mat[, 3], tr_mat) cube <- cbind(as.data.frame(trans), depth = -mat[, 2]) cbind(cube[segments$ind, ], segments) } cube_grid <- to_grid(cube, segments, tr_mat) ggplot(cube_grid) + geom_path(aes(x, y, group = group))
We can of course improve the illusion even more. Using geom_link2()
from ggforce
, we can add a gradient size to the lines, based on the depth of the endpoints (remember, these were added in the to_grid()
function).
ggplot(cube_grid) + geom_link2(aes(x, y, group = group, size = depth), lineend = 'round', n = 500) + scale_size(range = c(1.5, 3))
What can we do more? Well, we could decide to rotate it a bit. Let’s, make a function that rotates it around the vertical axis:
rot <- function(mat, angle) { mat_tmp <- mat mat_tmp[,1] <- mat[,1] * cos(angle) - mat[,2] * sin(angle) mat_tmp[,2] <- mat[,2] * cos(angle) + mat[,1] * sin(angle) mat_tmp } cube_grid <- to_grid(rot(cube, pi/4.5), segments, tr_mat) ggplot(cube_grid) + geom_link2(aes(x, y, group = group, size = depth), lineend = 'round', n = 500) + scale_size(range = c(1.5, 3))
We now have all the ingredients to make an animation of a spinning cube - we really only need to animate a 90 degree spin as it is selfrepeating. For added fiz we’ll improve the depth perception by simulating a bit of haze by greying parts more distant:
angles <- head(seq(0, pi/2, length.out = 51), 50) cubes <- lapply(angles, function(i) to_grid(rot(cube, i), segments, tr_mat)) for (d in cubes) { p <- ggplot(d) + geom_link2(aes(x, y, group = group, size = depth, color = depth), lineend = 'round', n = 500) + scale_size(range = c(1.5, 3)) + scale_color_gradient(low = 'grey70', high = 'black') + scale_x_continuous(limits = c(-0.4, 0.4)) + scale_y_continuous(limits = c(-0.4, 0.4)) + coord_fixed() + theme(legend.position = 'none') plot(p) }
That’s all fine and well, but this is just a regular and boring 2D video. Let us update it to the brave new world, where Avatar has taught us that 3D is not tacky: Enter facet_stereo()
.
for (d in cubes) { p <- ggplot(d) + geom_link2(aes(x, y, group = group, size = depth, color = depth, depth = depth), # This tells facet_stereo about depth lineend = 'round', n = 500) + scale_size(range = c(1.5, 3)) + scale_color_gradient(low = 'grey70', high = 'black') + scale_x_continuous(limits = c(-0.4, 0.4)) + scale_y_continuous(limits = c(-0.4, 0.4)) + coord_fixed() + theme(legend.position = 'none') + scale_depth(range = c(0, 0.15)) + # This scales depth perception facet_stereo(panel.size = 200) # This calculates the stereoscopic distortion plot(p) }
That is all it takes…
R-bloggers.com offers daily e-mail updates about R news and tutorials about learning R and many other topics. Click here if you're looking to post or find an R/data-science job.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.