R/module-select-geom-aes.R

In esquisse: Explore and Visualize Your Data Interactively

#' @importFrom shiny NS
#' @importFrom bslib nav_panel navset_pill
#' @importFrom htmltools tags tagList
select_geom_aes_ui <- function(id, n_geoms = 1, list_geoms = NULL) {

  if (is.null(list_geoms)) {
    list_geoms <- rep_len(list(geomIcons()), n_geoms)
  }

  if (!is_list(list_geoms)) {
    list_geoms <- rep_len(list(list_geoms), n_geoms)
  }

  ns <- NS(id)
  if (n_geoms == 1) {
    tags$div(
      class = "esquisse-geom-aes-main esquisse-geom-aes",
      tags$div(
        style = "padding: 3px 3px 0 3px; height: 122px;",
        dropInput(
          inputId = ns("geom_1"),
          choicesNames = list_geoms[[1]]$names,
          choicesValues = list_geoms[[1]]$values,
          dropWidth = "292px",
          width = "100%"
        )
      ),
      select_aes_ui(ns("aes_1"))
    )
  } else {
    navs_geom <- lapply(
      X = seq_len(n_geoms),
      FUN = function(i) {
        nav_panel(
          title = paste0("Geom #", i),
          value = paste0("geom", i),
          tags$div(
            class = "esquisse-geom-aes",
            tags$div(
              style = "padding: 3px 3px 0 3px; height: 122px;",
              dropInput(
                inputId = ns(paste0("geom_", i)),
                choicesNames = list_geoms[[i]]$names,
                choicesValues = list_geoms[[i]]$values,
                dropWidth = "292px",
                width = "100%"
              )
            ),
            select_aes_ui(ns(paste0("aes_", i)))
          )
        )
      }
    )
    tags$div(
      class = "esquisse-geom-aes-main",
      navset_pill(
        id = ns("navset_geoms"),
        !!!navs_geom
      )
    )
  }
}

#' @importFrom shiny reactive moduleServer reactiveValues observeEvent reactiveValuesToList
select_geom_aes_server <- function(id,
                                   n_geoms = 1,
                                   data_r = reactive(NULL),
                                   default_aes = c("fill", "color", "size", "group", "facet"),
                                   aesthetics_r = reactive(NULL)) {
  moduleServer(
    id,
    function(input, output, session) {

      rv <- reactiveValues()
      lapply(
        X = seq_len(n_geoms),
        FUN = function(i) {
          rv[[paste0("aes_", i)]] <- list()
          rv[[paste0("geom_", i)]] <- NA_character_
        }
      )

      lapply(
        X = seq_len(n_geoms),
        FUN = function(i) {

          aes_r <- select_aes_server(
            id = paste0("aes_", i),
            data_r = data_r,
            default_aes = if (i > 1) {
              grep("facet", x = default_aes, value = TRUE, invert = TRUE)
            } else {
              default_aes
            },
            input_aes = reactive({
              aesth <- aesthetics_r()
              if (i > 1) {
                aesth <- grep("facet", x = aesth, value = TRUE, invert = TRUE)
              }
              aesth
            })
          )

          observeEvent(aes_r(), {
            rv[[paste0("aes_", i)]] <- dropNulls(aes_r())
          })

          observeEvent(input[[paste0("geom_", i)]], {
            rv[[paste0("geom_", i)]] <- input[[paste0("geom_", i)]]
          })

        }
      )


      bindEvent(observe({
        aesthetics <- rv$aes_1
        data <- data_r()
        geoms <- potential_geoms(
          data = data,
          mapping = build_aes(
            data = data,
            .list = aesthetics
          )
        )

        if (inherits(data_r(), "sf")) {
          geoms <- c(geoms, "sf")
        }

        rv$possible <- c("auto", geoms)

      }), rv$aes_1, input$geom_1)

      observeEvent(rv$possible, {
        geoms <- geomIcons()$values
        geomposs <-  rv$possible
        updateDropInput(
          session = session,
          inputId = "geom_1",
          selected = setdiff(geomposs, "auto")[1],
          disabled = setdiff(geoms, geomposs)
        )
      })


      return(reactive({
        others <- reactiveValuesToList(rv)
        others$aes_1 <- NULL
        others$geom_1 <- NULL
        others$possible <- NULL
        # others[vapply(others, FUN = identical, "auto", FUN.VALUE = logical(1))] <- NULL
        # others[vapply(others, FUN = identical, "blank", FUN.VALUE = logical(1))] <- NULL
        others[grepl("geom_possible", names(others))] <- NULL
        result <- list(
          main = list(aes = rv$aes_1, geom = rv$geom_1),
          others = others
        )
        result$active <- input$navset_geoms
        return(result)
      }))
    }
  )
}