#
# NotebookPlots.R
#
# Copyright (C) 2009-16 by RStudio, Inc.
#
# Unless you have received this program directly from RStudio pursuant
# to the terms of a commercial license agreement with RStudio, then
# this program is licensed to you under the terms of version 3 of the
# GNU Affero General Public License. This program is distributed WITHOUT
# ANY EXPRESS OR IMPLIED WARRANTY, INCLUDING THOSE OF NON-INFRINGEMENT,
# MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Please refer to the
# AGPL (http://www.gnu.org/licenses/agpl-3.0.txt) for more details.
#
#

# creates the notebook graphics device 
.rs.addFunction("createNotebookGraphicsDevice", function(filename,
  height, width, units, pixelRatio, extraArgs)
{
  # form the arguments to the graphics device creator
  require(grDevices, quietly = TRUE)
  args <- list(
    filename = filename,
    width    = width * pixelRatio,
    height   = height * pixelRatio, 
    units    = units,
    res      = 96 * pixelRatio)
  
  if (nchar(extraArgs) > 0)
  {
    # trim leading comma from extra args if present
    if (identical(substr(extraArgs, 1, 1), ","))
      extraArgs <- substring(extraArgs, 2)

    # parse extra args and merge with existing args
    extraList <- NULL
    tryCatch({
      extraList <- eval(parse(text = paste("list(", extraArgs, ")")))
    }, error = function(e) {
      # failure to parse implies we won't merge below
    })

    if (is.list(extraList))
      args <- c(args, extraList)
  }

  # create the device
  do.call(what = png, args = args)
})

.rs.addFunction("setNotebookGraphicsOption", function(filename,
  height, width, units, pixelRatio, extraArgs)
{
  options(device = function() {
    .rs.createNotebookGraphicsDevice(filename, height, width, units, 
                                     pixelRatio, extraArgs)
    dev.control(displaylist = "enable")
    .rs.setNotebookMargins()
  })
})

.rs.addFunction("saveNotebookGraphics", function(plot, filename)
{
   save(plot, file = filename)
})

.rs.addFunction("setNotebookMargins", function() {
  #           bot  left top  right
  par(mar = c(5.1, 4.1, 2.1, 2.1))
})

.rs.addFunction("replayNotebookPlots", function(width, height, pixelRatio, tempFile, extraArgs) {
  # open stdin (for consuming snapshots from parent process)
  stdin <- file("stdin")

  require(grDevices, quietly = TRUE)

  snapshots <- readLines(stdin, warn = FALSE)
  lapply(snapshots, function(snapshot) {

    # ignore empty lines
    if (nchar(tools::file_ext(snapshot)) < 1)
      return(invisible(NULL))

    # create the PNG device on which we'll regenerate the plot -- it's somewhat
    # wasteful to use a separate device per plot, but the alternative is
    # doing a lot of state management and post-processing of the files
    # output from the device
    output <- paste(tools::file_path_sans_ext(snapshot), "resized.png",
                    sep = ".")

    height <- if (height <= 0) width / 1.618 else height

    .rs.createNotebookGraphicsDevice(output, height, width, 
                                     "px", pixelRatio, extraArgs)

    # actually replay the plot onto the device
    tryCatch({
      .rs.restoreGraphics(snapshot)
    }, error = function(e) {
      # nothing reasonable we can do here; we'll just omit this file from the
      # output
    })

    # close the device; has the side effect of committing the plot to disk
    dev.off()

    # if the plot file was written out, emit to standard out for the caller to 
    # consume
    if (file.exists(output)) {
      if (tempFile)
      {
        chunksPath <- dirname(snapshot)
        chunksTempPath <- file.path(chunksPath, "temp")

        if (!dir.exists(chunksTempPath))
          dir.create(chunksTempPath)

        chunkBaseName <- basename(tools::file_path_sans_ext(snapshot))

        final <- file.path(
          chunksTempPath,
          paste(chunkBaseName, "png", sep = ".")
        )
      }
      else
      {
        final <- paste(tools::file_path_sans_ext(snapshot), "png", sep = ".")
      }

      # remove the old copy of the plot if it existed
      if (file.exists(final))
        unlink(final)

      file.copy(output, final)

      if (file.exists(output))
        unlink(output)

      if (file.exists(final)) {
        cat(final, "\n")
      }
    }
  })

  invisible(NULL)
})
