rtichoke: rtichoke

library(dplyr)
library(tidyr)
library(magrittr)

# TODO Fix frame order

plot_decision_combined_curve <- function(rtichoke_decision_combined_curve_list){
  

  interactive_marker <- list(
    size = 12,
    line = list(
      width = 3,
      color = I("black")
    )
  )
  
  if ( ! (rtichoke_decision_combined_curve_list$perf_dat_type %in% 
          c("several models", "several populations")) ) {
    
    interactive_marker$color <- "#f6e3be"
    
  }
  
  combined_decision <- rtichoke_decision_combined_curve_list$performance_data_ready_for_curve$conventional |> 
    dplyr::inner_join(
      rtichoke_decision_combined_curve_list$performance_data_ready_for_curve$`interventions avoided`,
      by = c("reference_group", "stratified_by"),
      suffix = c("_conventional", "_interventions_avoided")
    )
  
  size_height <- switch(is.null(rtichoke_decision_combined_curve_list$size) +1, 
                        1.25*rtichoke_decision_combined_curve_list$size + 50, NULL)
  

  conventional_decision_curve <- plotly::plot_ly(
    height = size_height,
    width = rtichoke_decision_combined_curve_list$size
  ) |>
    plotly::add_lines(
      data = rtichoke_decision_combined_curve_list$reference_data$conventional,
      x = ~x,
      y = ~y,
      text = ~text,
      line = list(
        dash = "dot"
      ),
      hoverinfo = "text",
      color = ~reference_group,
      colors = unlist(rtichoke_decision_combined_curve_list$group_colors_vec)
    ) |>
    plotly::add_trace(
      data = combined_decision,
      x = ~x_conventional,
      y = ~ y_conventional,
      text =~ text_conventional,
      hoverinfo = "text",
      type = 'scatter',
      mode = 'lines+markers',
      line = list(dash = 'solid'),
      color =~ reference_group,
      colors = unlist(rtichoke_decision_combined_curve_list$group_colors_vec)
    ) |>
    plotly::add_markers(
      data = combined_decision,
      x = ~x_conventional,
      y = ~ y_conventional,
      text =~ text_conventional,
      hoverinfo = "text",
      type = 'scatter',
      color =~ reference_group,
      colors = unlist(rtichoke_decision_combined_curve_list$group_colors_vec),
      frame =~ stratified_by,
      marker = interactive_marker
    ) |>
    plotly::layout(
      xaxis = list(showgrid = FALSE, fixedrange = TRUE,
                   range = rtichoke_decision_combined_curve_list$axes_ranges$conventional$xaxis, 
                   title = rtichoke_decision_combined_curve_list$axes_labels$conventional$xaxis),
      yaxis = list(showgrid = FALSE, fixedrange = TRUE,
                   range = rtichoke_decision_combined_curve_list$axes_ranges$conventional$yaxis)
    )
  
  print(names(combined_decision))
  
  interventions_avoided_curve <- plotly::plot_ly(
    height = size_height,
    width = rtichoke_decision_combined_curve_list$size
  ) |>
    plotly::add_trace(
      x = ~x_interventions_avoided ,
      y = ~y_interventions_avoided,
      data = combined_decision,
      type = 'scatter',
      mode = 'lines+markers',
      line = list(dash = 'solid'),
      color = ~reference_group,
      colors = unlist(rtichoke_decision_combined_curve_list$group_colors_vec),
      text = ~text_interventions_avoided,
      hoverinfo = "text"
    ) |>
    plotly::add_markers(
      data = combined_decision,
      x = ~x_interventions_avoided,
      y = ~y_interventions_avoided,
      frame =~ stratified_by,
      color = ~reference_group,
      colors = unlist(rtichoke_decision_combined_curve_list$group_colors_vec),
      marker = interactive_marker,
      text = ~text_interventions_avoided,
      hoverinfo = "text"
    ) |>
    plotly::layout(
      xaxis = list(showgrid = FALSE, fixedrange = TRUE,
                   range = rtichoke_decision_combined_curve_list$axes_ranges$`interventions avoided`$xaxis),
      yaxis = list(showgrid = FALSE, fixedrange = TRUE,
                   range = rtichoke_decision_combined_curve_list$axes_ranges$`interventions avoided`$yaxis)
    )
  
  interventions_avoided_annotation <- list(
    text = "Interventions Avoided (per 100)",
    font = list(
      size = 18,
      color = "black"),
    xref = "paper",
    yref = "paper",
    yanchor = "bottom",
    xanchor = "center",
    align = "center",
    x = 0.5,
    y = 1,
    showarrow = FALSE
  )
  
  conventional_decision_annotation <- list(
    text = "Net Benefit",
    font = list(
      size = 18,
      color = "black"),
    xref = "paper",
    yref = "paper",
    yanchor = "bottom",
    xanchor = "center",
    align = "center",
    x = 0.5,
    y = 1,
    showarrow = FALSE
  )
  
  plotly::subplot(
    interventions_avoided_curve |> plotly::layout(annotations = interventions_avoided_annotation), 
    conventional_decision_curve |> plotly::layout(annotations = conventional_decision_annotation),
    nrows = 2,
    shareX = TRUE,
    shareY = FALSE, 
    heights = c(0.5, 0.5)
  ) |>
    plotly::config(displayModeBar = FALSE) |>  
    plotly::animation_button(visible=FALSE) |>
    plotly::animation_slider(
      currentvalue = list(
        prefix = rtichoke_decision_combined_curve_list$animation_slider_prefix,
        font = list(color = "black"),
        xanchor = "left"
      ),
      pad = list(t = 50)
    ) |> 
    plotly::layout(
      showlegend = FALSE
    )
  

}




check_equality_between_decision_curve_lists <- function(
    decision_conventional_list, 
    decision_interventions_avoided_list) {
  
  stopifnot(
    
    all.equal(
      decision_conventional_list[c("size", "animation_slider_prefix",
                                   "perf_dat_type", "group_colors_vec")],
      decision_interventions_avoided_list[c("size", "animation_slider_prefix",
                                            "perf_dat_type", "group_colors_vec")]
    )
    
  )
  
}

unify_decision_curve_lists_for_combined_decision_curve_list <- function(
    rtichoke_decision_curve_lists) {
  
  check_equality_between_decision_curve_lists(
    rtichoke_decision_curve_lists$conventional,  
    rtichoke_decision_curve_lists$`interventions avoided`
  )
  
  
  rtichoke_decision_combined_curve_list$size <- rtichoke_decision_curve_lists$conventional$size
  
  rtichoke_decision_combined_curve_list <- rtichoke_decision_curve_lists$conventional[
    c("animation_slider_prefix", "perf_dat_type", "group_colors_vec")]
  
  rtichoke_decision_combined_curve_list$reference_data <- list(
    "conventional" = rtichoke_decision_curve_lists$conventional$reference_data
  )
  
  rtichoke_decision_combined_curve_list$axes_labels <- list(
    "conventional" = rtichoke_decision_curve_lists$conventional$axes_labels,
    "interventions avoided" = rtichoke_decision_curve_lists$`interventions avoided`$axes_labels
  )
  
  rtichoke_decision_combined_curve_list$performance_data_ready_for_curve <- list(
    "conventional" = rtichoke_decision_curve_lists$conventional$performance_data_ready_for_curve,
    "interventions avoided" = rtichoke_decision_curve_lists$`interventions avoided`$performance_data_ready_for_curve
  )
  
  rtichoke_decision_combined_curve_list$axes_ranges <- list(
    "conventional" = rtichoke_decision_curve_lists$conventional$axes_ranges,
    "interventions avoided" = rtichoke_decision_curve_lists$`interventions avoided`$axes_ranges
  )
  
  rtichoke_decision_combined_curve_list
  
}


create_rtichoke_combined_decision_curve_list <- function(
    performance_data,
    curve, 
    min_p_threshold = 0.01, 
    max_p_threshold = 0.99, 
    size = NULL, 
    color_values = c(
                             "#1b9e77", "#d95f02",
                             "#7570b3", "#e7298a",
                             "#07004D", "#E6AB02",
                             "#FE5F55", "#54494B",
                             "#006E90", "#BC96E6",
                             "#52050A", "#1F271B",
                             "#BE7C4D", "#63768D",
                             "#08A045", "#320A28",
                             "#82FF9E", "#2176FF",
                             "#D1603D", "#585123")) {
  
  rtichoke_decision_curve_lists <- c("decision", "interventions avoided") |>
    purrr::map(~create_rtichoke_curve_list(performance_data , 
                                           curve = .x, 
                                           min_p_threshold = min_p_threshold, 
                                           max_p_threshold = max_p_threshold, 
                                           size = size, 
                                           color_values = color_values)) |> 
    purrr::set_names("conventional", "interventions avoided")
                             
  
  rtichoke_decision_curve_lists |> 
    unify_decision_curve_lists_for_combined_decision_curve_list() |> 
    plot_decision_combined_curve()                         
  
}

create_rtichoke_combined_decision_curve_list(one_pop_one_model)



create_rtichoke_combined_decision_curve_list(multiple_populations)


rtichoke_decision_curve_lists <- c("decision", "interventions avoided") |>
  purrr::map(~create_rtichoke_curve_list(multiple_models , .x, 
                                         min_p_threshold = 0.01, 
                                         max_p_threshold = 0.99)) |> 
  purrr::set_names("conventional", "interventions avoided") |> 
  unify_decision_curve_lists_for_combined_decision_curve_list() |> 
  plot_decision_combined_curve()

rtichoke_decision_curve_lists


rtichoke_decision_curve_lists |> 
  unify_decision_curve_lists_for_combined_decision_curve_list() |> 
  # plot_decision_combined_curve()

  rtichoke_decision_curve_lists$conventional
  

rtichoke_decision_curve_lists |> 
  unify_decision_curve_lists_for_combined_decision_curve_list() |> 
  names()
  
rtichoke_decision_curve_lists |> 
  unify_decision_curve_lists_for_combined_decision_curve_list() |> 
  # plot_decision_combined_curve() |> 
  jsonlite::toJSON(auto_unbox = TRUE) |> 
  write("C:/Users/CRI_user/Documents/rtichoke_curve_decision_combined_json_array.json")


names(rtichoke_decision_curve_lists$conventional)

rtichoke_decision_curve_lists

plot_decision_combined_curve(rtichoke_decision_combined_curve_list)