R/fusionPlot.R

In alexym1/fusionChartsR: Embedding FusionCharts in R

#' Create new charts
#'
#' Main function to make interactive charts.
#' Check all charts at \url{https://www.fusioncharts.com/charts}
#'
#'
#' @import htmlwidgets
#' @importFrom jsonlite toJSON
#' @importFrom grDevices boxplot.stats
#' @importFrom stats na.omit
#'
#' @param data Default dataset to use
#' @param x,y character name of variable
#' @param type type of chart
#' @param numberSuffix Specify the suffix for all the Y-axis values on the chart
#'
#' @details A 2x2 confusion matrix can be displayed using `type = "confusionMatrix"`.
#' 
#' @export
fusionPlot <- function(data, x, y, type = "column2d", numberSuffix = NULL) {
  
  # Include unit tests
  if(type %in% c("confusionMatrix", "heatmap")){
    stopifnot(is.matrix(data))
  } else {
    stopifnot(is.data.frame(data))
    stopifnot(is.character(x))
    stopifnot(is.character(y))
  }
  
  stopifnot(is.character(type))
  stopifnot(is.character(numberSuffix) | is.null(numberSuffix))
  

  # Include key license
  license <- Sys.getenv("LICENSE_FUSIONCHARTS")
  
  # Main arguments
  category <- NULL
  dataset <- NULL
  mapbycategory <- "0"
  colorrange <- NULL
  columns <- NULL
  rows <- NULL
  
  if(type == "boxandwhisker2d"){
    
    xaxis <- factor(data[,x])
    df <- list(category = data.frame(label = as.character(levels(xaxis))))
    category <- toJSON(x = df, pretty = TRUE)
    
    n <- unique(levels(xaxis))
    df.list <- lapply(1:length(n), function(i){
      yaxis <- na.omit(data[data[,x] == n[i],y])
      stats <- boxplot.stats(yaxis)
      if(length(stats$out) >= 1){
        list(
          value = toString(yaxis[! yaxis %in% unique(stats$out)]),
          outliers = toString(unique(stats$out))
        )
      } else {
        list(
          value = toString(yaxis)
        )
      }
    })
    
    newlist <- list(seriesname = y, data = df.list)
    dataset <- toJSON(x = newlist, pretty = TRUE, auto_unbox = TRUE)
    
  } else if(type %in% c("confusionMatrix", "heatmap")){
    
    column01 <- list(
      column = data.frame(
        id = paste0("GROUP", colnames(data)),
        label = colnames(data)
      )
    )
    
    row01 <- list(
      row = data.frame(
        id = paste0("group", rownames(data)),
        label = rownames(data)
      )
    )
    
    columnid <- lapply(1:length(column01$column$id), function(x){
      rep(column01$column$id[x], times = ncol(data))
    })
    
    if(type == "confusionMatrix"){
      
      mapbycategory <- "1"
      
      color <- list(
        gradient = c("0", "0"),
        color = data.frame(
          code = c("#5E72E4", "#FFFFFF"),
          minvalue = c("0", "0"),
          maxvalue = c("Infinity", "Infinity"),
          label = c("Good", "Bad")
        )
      )
      
      data01 <- list(
        data = data.frame(
          rowid = rep(row01$row$id, times = nrow(data)),
          columnid = do.call(c, columnid),
          displayvalue = as.character(as.vector(data)),
          colorrangelabel = c("Good", "Bad", "Bad", "Good")
        )
      )
      
    } else {
    
      color <- list(
        gradient = "1",
        startlabel = "Negative",
        endlabel = "Positive",
        color = data.frame(
          code = c("#FF595E", "#FFFFFF", "#5E72E3"),
          maxvalue = c("-1", "0", "1")
        )
      )
      
      data01 <- list(
        data = data.frame(
          rowid = rep(row01$row$id, times = nrow(data)),
          columnid = do.call(c, columnid),
          value = as.character(as.vector(data))
        )
      )
      
    }
    
    type <- "heatmap"
    colorrange <- toJSON(x = color, pretty = TRUE, auto_unbox = TRUE)
    columns <- toJSON(x = column01, pretty = TRUE)
    rows <- toJSON(x = row01, pretty = TRUE)
    dataset <- toJSON(x = data01, pretty = TRUE, auto_unbox = TRUE)
    
  }  else if(type == "scatter"){
    
    df <- data.frame(x = data[,x], y = data[,y])
    dataset <- toJSON(x = list(data = df), pretty = TRUE, auto_unbox = TRUE)
    
  } else {
    
    new.data <- data.frame(label = factor(data[,x]), value = data[,y])
    data <- toJSON(x = new.data, pretty = TRUE)
    
  }
  


#' @examples
#' library(fusionchartsR)
#'
#' # Single
#' df <- data.frame(label = c("Venezuela", "Saudi", "Canada", "Russia"), value = c(290, 260,180, 115))
#' df %>%
#' fusionPlot(x = "label", y = "value", type = "pie2d") %>%
#' fusionTheme(theme = "fusion")
#' 

  # forward options using x
  x <- list(
    key_license = license,
    data = data,
    categories = category,
    dataset = dataset,
    type = type,
    numberSuffix = numberSuffix,
    columns = columns,
    rows = rows,
    colorrange = colorrange,
    mapbycategory = mapbycategory
    )

  # create widget
  widgets <- htmlwidgets::createWidget(
    name = 'fusionPlot',
    x = x,
    package = 'fusionchartsR'
  )

  widgets %>%
  fusionCaption() %>%
    fusionSubcaption() %>%
    fusionBackground() %>%
    fusionCanvas() %>%
    fusionAxis() %>%
    fusionCustomAxis() %>%
    fusionLegend() %>%
    fusionCustomLegend() %>%
    fusionPalette() %>%
    fusionAnchors() %>%
    fusionTrendline() %>%
    fusionCustomBoxplot() %>%
    fusionDiv() %>%
    fusionTooltip() %>%
    fusionLogo() %>%
    fusionTheme()

}


#' Shiny bindings for fusionPlot
#'
#' Output and render functions for using fusionPlot within Shiny
#' applications and interactive Rmd documents.
#'
#' @param outputId output variable to read from
#' @param width,height Must be a valid CSS unit (like \code{'100\%'},
#'   \code{'400px'}, \code{'auto'}) or a number, which will be coerced to a
#'   string and have \code{'px'} appended.
#' @param expr An expression that generates a fusionPlot
#' @param env The environment in which to evaluate \code{expr}.
#' @param quoted Is \code{expr} a quoted expression (with \code{quote()})? This
#'   is useful if you want to save an expression in a variable.
#'
#' @name fusionPlotOutput
#' @aliases renderfusionPlot
#'
#' @export
fusionPlotOutput <- function(outputId, width = '100%', height = '400px'){
  htmlwidgets::shinyWidgetOutput(outputId, 'fusionPlot', width, height, package = 'fusionchartsR')
}

#' @rdname fusionPlotOutput
#' @export
renderfusionPlot <- function(expr, env = parent.frame(), quoted = FALSE) {
  if (!quoted) { expr <- substitute(expr) } # force quoted
  htmlwidgets::shinyRenderWidget(expr, fusionPlotOutput, env, quoted = TRUE)
}