InterfacePreview.lua

--- Original: Divvy's Preview for Balatro - Interface.lua
--
-- The user interface components that display simulation results.

-- Append node for preview text to the HUD:
local orig_hud = create_UIBox_HUD
function create_UIBox_HUD()
   local contents = orig_hud()
   
   local hand_text_area = FN.PRE.find_hand_text_area(contents)
   if not hand_text_area then return contents end
   table.insert(hand_text_area.nodes[1].nodes, FN.PRE.get_preview_container())

   --[[local dollars_node_wrap = {n=G.UIT.C, config={id = "fn_pre_dollars_wrap", align = "cm"}, nodes={}}
   if G.SETTINGS.FN.preview_dollars then table.insert(dollars_node_wrap.nodes, FN.PRE.get_dollars_node()) end
   table.insert(contents.nodes[1].nodes[1].nodes[5].nodes[2].nodes[3].nodes[1].nodes[1].nodes[1].nodes, dollars_node_wrap) --]]

   return contents
end

function G.FUNCS.calculate_score_button()
   FN.PRE.start_new_coroutine()
end

function FN.PRE.get_preview_container()
   return {n=G.UIT.R, config={id = "fn_preview_container", align = "cm"}, nodes={
      {n=G.UIT.C, config={align = "cm"}, nodes={
         {n=G.UIT.R, config={id = "fn_pre_score_wrap", align = "cm", padding = 0.1}, nodes={
            FN.PRE.get_score_node()
         }},
         {n=G.UIT.R, config={id = "fn_calculate_score_button_wrap", align = "cm", padding = 0.1}, nodes={
            FN.PRE.get_calculate_score_button()
         }}
      }}
   }}
end

function FN.PRE.get_calculate_score_button()
   return {n=G.UIT.C, config={id = "calculate_score_button", button = "calculate_score_button", align = "cm", minh = 0.42, padding = 0.05, r = 0.02, colour = G.C.RED, hover = true, shadow = true}, nodes={
      {n=G.UIT.R, config={align = "cm"}, nodes={
         {n=G.UIT.T, config={text = "  Calculate Score  ", colour = G.C.UI.TEXT_LIGHT, shadow = true, scale = 0.36}}
      }}
   }}
end


function FN.PRE.get_score_node()
   local text_scale = nil
   if true then text_scale = 0.5
   else text_scale = 0.75 end

   return {n = G.UIT.C, config = {id = "fn_pre_score", align = "cm"}, nodes={
              {n=G.UIT.O, config={id = "fn_pre_l", func = "fn_pre_score_UI_set", object = DynaText({string = {{ref_table = FN.PRE.text.score, ref_value = "l"}}, colours = {G.C.UI.TEXT_LIGHT}, shadow = true, float = true, scale = text_scale})}},
              {n=G.UIT.O, config={id = "fn_pre_r", func = "fn_pre_score_UI_set", object = DynaText({string = {{ref_table = FN.PRE.text.score, ref_value = "r"}}, colours = {G.C.UI.TEXT_LIGHT}, shadow = true, float = true, scale = text_scale})}},
   }}
end

function FN.PRE.find_hand_text_area(node)
   if node.config and node.config.id == "hand_text_area" then
      return node
   end
   if node.nodes then
      for _, child in ipairs(node.nodes) do
         local found = FN.PRE.find_hand_text_area(child, id)
         if found then return found end
      end
   end
   return nil
end

--[[function FN.PRE.get_dollars_node()
   local top_color = FN.PRE.get_dollar_colour(0)
   local bot_color = top_color
   if FN.PRE.data ~= nil then
      top_color = FN.PRE.get_dollar_colour(FN.PRE.data.dollars.max)
      bot_color = FN.PRE.get_dollar_colour(FN.PRE.data.dollars.min)
   else
   end
   return {n=G.UIT.C, config={id = "fn_pre_dollars", align = "cm"}, nodes={
       {n=G.UIT.R, config={align = "cm"}, nodes={
           {n=G.UIT.O, config={id = "fn_pre_dollars_top", func = "fn_pre_dollars_UI_set", object = DynaText({string = {{ref_table = FN.PRE.text.dollars, ref_value = "top"}}, colours = {top_color}, shadow = true, spacing = 2, bump = true, scale = 0.5})}}
       }},
       {n=G.UIT.R, config={minh = 0.05}, nodes={}},
       {n=G.UIT.R, config={align = "cm"}, nodes={
           {n=G.UIT.O, config={id = "fn_pre_dollars_bot", func = "fn_pre_dollars_UI_set", object = DynaText({string = {{ref_table = FN.PRE.text.dollars, ref_value = "bot"}}, colours = {bot_color}, shadow = true, spacing = 2, bump = true, scale = 0.5})}},
       }}
   }}
end--]]

--
-- SETTINGS:
--

function FN.get_preview_settings_page()
   local function preview_score_toggle_callback(e)
      if not G.HUD then return end

      if G.SETTINGS.FN.preview_score then
         -- Preview was just enabled, so add preview node:
         G.HUD:add_child(FN.PRE.get_score_node(), G.HUD:get_UIE_by_ID("fn_pre_score_wrap"))
         FN.PRE.data = FN.PRE.simulate()
      else
         -- Preview was just disabled, so remove preview node:
         G.HUD:get_UIE_by_ID("fn_pre_score").parent:remove()
      end
      G.HUD:recalculate()
   end

   local function preview_dollars_toggle_callback(_)
      if not G.HUD then return end

      if G.SETTINGS.FN.preview_dollars then
         -- Preview was just enabled, so add preview node:
         G.HUD:add_child(FN.PRE.get_dollars_node(), G.HUD:get_UIE_by_ID("fn_pre_dollars_wrap"))
         FN.PRE.data = FN.PRE.simulate()
      else
         -- Preview was just disabled, so remove preview node:
         G.HUD:get_UIE_by_ID("fn_pre_dollars").parent:remove()
      end
      G.HUD:recalculate()
   end

   local function face_down_toggle_callback(_)
      if not G.HUD then return end

      FN.PRE.data = FN.PRE.simulate()
      G.HUD:recalculate()
   end

   return
      {n=G.UIT.ROOT, config={align = "cm", padding = 0.05, colour = G.C.CLEAR}, nodes={
          create_toggle({id = "score_toggle",
                         label = "Enable Score Preview",
                         ref_table = G.SETTINGS.FN,
                         ref_value = "preview_score",
                         callback = preview_score_toggle_callback}),
          create_toggle({id = "dollars_toggle",
                         label = "Enable Money Preview",
                         ref_table = G.SETTINGS.FN,
                         ref_value = "preview_dollars",
                         callback = preview_dollars_toggle_callback}),
          create_toggle({label = "Hide Preview if Any Card is Face-Down",
                         ref_table = G.SETTINGS.FN,
                         ref_value = "hide_face_down",
                         callback = face_down_toggle_callback})
      }}
end