feat: add MVP version of line and barplot

Author: vedhav

NAMESPACE

@@ -23,6 +23,8 @@ export(tm_g_scatterplot)
 export(tm_g_scatterplotmatrix)
 export(tm_missing_data)
 export(tm_outliers)
+export(tm_p_bargraph)
+export(tm_p_lineplot)
 export(tm_p_scatterplot)
 export(tm_p_spiderplot)
 export(tm_p_swimlane)

R/tm_p_bargraph.R

@@ -0,0 +1,80 @@
+#' @export
+tm_p_bargraph <- function(label = "Bar Plot",
+                          plot_dataname,
+                          y_var,
+                          color_var,
+                          count_var,
+                          bar_colors = NULL) {
+  module(
+    label = label,
+    ui = ui_p_bargraph,
+    server = srv_p_bargraph,
+    ui_args = list(),
+    server_args = list(
+      plot_dataname = plot_dataname,
+      y_var = y_var,
+      color_var = color_var,
+      count_var = count_var,
+      bar_colors = bar_colors
+    )
+  )
+}
+
+ui_p_bargraph <- function(id) {
+  ns <- NS(id)
+  bslib::page_fluid(
+    bslib::card(
+      full_screen = TRUE,
+      tags$div(
+        # trigger_tooltips_deps(),
+        plotly::plotlyOutput(ns("plot"), height = "100%")
+      )
+    )
+  )
+}
+
+srv_p_bargraph <- function(id, data, plot_dataname, y_var, color_var, count_var, bar_colors) {
+  moduleServer(id, function(input, output, session) {
+    plotly_q <- reactive({
+      df <- data()[[plot_dataname]]
+      df[[color_var]] <- as.character(df[[color_var]])
+
+      plot_data <- df %>%
+        group_by(!!as.name(y_var), !!as.name(color_var)) %>%
+        summarize(count = n_distinct(!!as.name(count_var)), .groups = "drop")
+
+      event_type_order <- plot_data %>%
+        group_by(!!as.name(y_var)) %>%
+        summarize(total = sum(count)) %>%
+        arrange(total) %>%
+        pull(!!as.name(y_var))
+
+      plot_data[[y_var]] <- factor(plot_data[[y_var]], levels = event_type_order)
+
+      p <- plot_ly(
+        data = plot_data,
+        y = as.formula(paste0("~", y_var)),
+        x = ~count,
+        color = as.formula(paste0("~", color_var)),
+        colors = bar_colors,
+        type = "bar",
+        orientation = "h"
+      ) %>%
+        layout(
+          barmode = "stack",
+          xaxis = list(title = "Count"),
+          yaxis = list(title = "Adverse Event Type"),
+          legend = list(title = list(text = "AE Type"))
+        )
+
+      p
+    })
+
+
+    output$plot <- plotly::renderPlotly({
+      p <- plotly_q()
+      plotly::event_register(p, "plotly_selected")
+      p
+    })
+  })
+}

R/tm_p_lineplot.R

@@ -0,0 +1,91 @@
+#' @export
+tm_p_lineplot <- function(label = "Line Plot",
+                          plot_dataname,
+                          x_var,
+                          y_var,
+                          transformators = list()) {
+  module(
+    label = label,
+    ui = ui_p_lineplot,
+    server = srv_p_lineplot,
+    ui_args = list(),
+    server_args = list(
+      plot_dataname = plot_dataname,
+      x_var = x_var,
+      y_var = y_var
+    ),
+    transformators = transformators
+  )
+}
+
+ui_p_lineplot <- function(id) {
+  ns <- NS(id)
+  bslib::page_fluid(
+    tags$div(
+      # trigger_tooltips_deps(),
+      plotly::plotlyOutput(ns("plot"), height = "100%")
+    )
+  )
+}
+
+srv_p_lineplot <- function(id, data, plot_dataname, x_var, y_var) {
+  moduleServer(id, function(input, output, session) {
+    plotly_q <- reactive({
+      df <- data()[[plot_dataname]]
+
+      validate(need(nrow(df) > 0, "No data after applying filters."))
+
+      # TODO: implement the high/low lines with annotations
+      y_low_last <- if ("si_low" %in% names(df)) utils::tail(stats::na.omit(df[["si_low"]]), 1) else NA
+      y_high_last <- if ("si_high" %in% names(df)) utils::tail(stats::na.omit(df[["si_high"]]), 1) else NA
+
+      p <- plotly::plot_ly(data = df, x = df[[x_var]]) |>
+        plotly::add_trace(
+          y = df[[y_var]],
+          mode = "lines+markers", type = "scatter", name = "Lab Result",
+          line = list(color = "green"),
+          marker = list(color = "green"),
+          showlegend = FALSE
+        ) |>
+        # plotly::add_trace(
+        #   y = df[["si_low"]],
+        #   mode = "lines",
+        #   line = list(color = "red", dash = "dash"),
+        #   showlegend = FALSE
+        # ) |>
+        # plotly::add_annotations(
+        #   x = max(df[[x_var]], na.rm = TRUE),
+        #   y = y_low_last,
+        #   yshift = 15,
+        #   text = "Original LLN",
+        #   showarrow = FALSE
+        # ) |>
+        # plotly::add_trace(
+        #   y = df[["si_high"]],
+        #   mode = "lines",
+        #   line = list(color = "red", dash = "solid"),
+        #   showlegend = FALSE
+        # ) |>
+        # plotly::add_annotations(
+        #   x = max(df[[x_var]], na.rm = TRUE),
+        #   y = y_high_last,
+        #   yshift = -15,
+        #   text = "Original ULN",
+        #   showarrow = FALSE
+        # ) |>
+        plotly::layout(
+          xaxis = list(title = "Study Day of Sample Collection", zeroline = FALSE),
+          yaxis = list(title = "Original Result")
+        )
+
+      p
+    })
+
+
+    output$plot <- plotly::renderPlotly({
+      p <- plotly_q()
+      plotly::event_register(p, "plotly_selected")
+      p
+    })
+  })
+}