Copy extras over from plotly.js, if necessary

Author: cpsievert

R/plotly.R

@@ -192,6 +192,9 @@ layout <- function(p = last_plot(), ...,
 #' @param displayModeBar display the modebar (T, F, or 'hover')
 #' @param displaylogo add the plotly logo on the end of the modebar
 #' @param plot3dPixelRatio increase the pixel ratio for 3D plot images
+#' @param mathjax whether to include mathjax. If \code{TRUE}, you must have
+#' a local copy of plotly.js and set the environment variable plotly_jsdir to 
+#' its location
 #' @author Carson Sievert
 #' @export
 
@@ -201,7 +204,7 @@ config <- function(p = last_plot(), staticPlot = F, workspace = F, editable = F,
                    autosizable = F, fillFrame = F, scrollZoom = F,
                    doubleClick = 'reset+autosize', showTips = F, showLink = T, 
                    sendData = T, linkText = 'Edit chart', displayModeBar = 'hover',
-                   displaylogo = T, plot3dPixelRatio = 2) {
+                   displaylogo = T, plot3dPixelRatio = 2, mathjax = FALSE) {
   conf <- list(
     staticPlot = staticPlot,
     workspace = workspace,
@@ -216,7 +219,8 @@ config <- function(p = last_plot(), staticPlot = F, workspace = F, editable = F,
     linkText = linkText,
     displayModeBar = displayModeBar,
     displaylogo = displaylogo,
-    plot3dPixelRatio = plot3dPixelRatio
+    plot3dPixelRatio = plot3dPixelRatio,
+    mathjax = mathjax
   )
   p <- last_plot(p)
   p$config <- c(p$config, conf)

R/print.R

@@ -40,6 +40,8 @@ as.widget <- function(x, ...) {
   )
   # customize the JSON serializer (for htmlwidgets)
   attr(p, 'TOJSON_FUNC') <- to_JSON
+  # get plotlyjs assets, if necessary
+  deps <- get_assets(p)
   htmlwidgets::createWidget(
     name = "plotly",
     x = p,
@@ -49,6 +51,7 @@ as.widget <- function(x, ...) {
       padding = 5, 
       browser.fill = TRUE
     ),
+    dependencies = deps,
     ...
   )
 }

R/utils.R

@@ -229,3 +229,62 @@ cat_profile <- function(key, value, path = "~") {
   message("Adding plotly_", key, " environment variable to ", r_profile)
   cat(snippet, file = r_profile, append = TRUE)
 }
+
+get_assets <- function(p) {
+  types <- unlist(lapply(p$data, "[[", "type"))
+  deps <- list(NULL)
+  # if there are any geo trace(s), add the geo assets dependency
+  if (any(grepl("geo|choropleth", types))) {
+    deps[[1]] <- plotly_dist("geo")
+  }
+  if (isTRUE(p$config$mathjax)) {
+    deps[[2]] <- plotly_dist("mathjax")
+  }
+  deps
+}
+
+# helper function for including large plotlyjs assets
+plotly_dist <- function(extra = c("mathjax", "geo")) {
+  path <- Sys.getenv("plotly_jsdir", unset = NA)
+  if (extra[1] == "mathjax") {
+    if (is.na(path)) {
+      stop("MathJax support requires a local clone of the plotly.js repo \n",
+           "https://github.com/plotly/plotly.js \n",
+           "Once you have plotly.js locally, set the plotly_jsdir \n",
+           "environment variable with the path to plotly.js")
+    } else {
+      mj <- file.path(path, "dist", "extras", "mathjax", "MathJax.js")
+      if (!file.exists(mj)) stop("Couldn't locate MathJax.js")
+      # parse the version
+      mathjax <- readLines(mj)
+      pat <- 'MathJax.fileversion="[0-9]+.[0-9]+.[0-9]+'
+      ver <- regmatches(mathjax, regexpr(pat, mathjax))
+      ver <- sub('"', '', strsplit(ver, "=")[[1]][2])
+      htmltools::htmlDependency(
+        name = "mathjax", 
+        version = ver, 
+        src = dirname(mj),
+        script = basename(mj)
+      )
+    }
+  }
+  if (extra[1] == "geo") {
+    if (is.na(path)) {
+      warning("Rendering 'geo' traces without an internet connection \n",
+              "requires a local clone of the plotly.js repo \n",
+              "https://github.com/plotly/plotly.js \n",
+              "Once you have plotly.js locally, set the plotly_jsdir \n",
+              "environment variable with the path to plotly.js")
+      return(NULL)
+    } else {
+      geo <- file.path(path, "dist", "plotly-geo-assets.js")
+      if (!file.exists(geo)) stop("Couldn't locate plotly-geo-assets.js")
+      htmltools::htmlDependency(
+        name = "plotly-geo-assets",
+        src = dirname(geo),
+        version = "1.0",
+        script = basename(geo)
+      )
+    }
+  }
+}