Factorio / calculator-ui

-- calculator.lua

-- ----------------------------------------------------------------
local function get_gui_root(player)
	return player.gui.screen
end

local nilaus_think = { "calcui_nilaus_really" }
local nilaus_rant = { "calcui_nilaus_ugghhhh", "utility/cannot_build" }

-- ----------------------------------------------------------------
local function play_sfx(player, sfx)
	if settings.get_player_settings(player)["calcui-sfx"].value then
		player.play_sound{
			path = sfx,
			volume_modifier = 1.0
		}
	end
end

-- ----------------------------------------------------------------
local function show_think(player, enabled)
	local root = get_gui_root(player)
	local col2 = root.calcui.calcui_table.calcui_table_col2

	if enabled then
		if settings.get_player_settings(player)["calcui-nilaus-mode"].value then
			col2.calcui_rant.sprite = "sprite_calcui_think"
			col2.calcui_scroll_pane.style.height = 196
			play_sfx(player, nilaus_think[math.random(1, #nilaus_think)])
		else
			play_sfx(player, "calcui_home_improvement")
		end
	else
		col2.calcui_rant.sprite = nil
		col2.calcui_scroll_pane.style.height = 252
	end
end

-- ----------------------------------------------------------------
local function show_rant(player, enabled)
	local root = get_gui_root(player)
	local col2 = root.calcui.calcui_table.calcui_table_col2

	if enabled then
		if settings.get_player_settings(player)["calcui-nilaus-mode"].value then
			col2.calcui_rant.sprite = "sprite_calcui_rant"
			col2.calcui_scroll_pane.style.height = 196
			play_sfx(player, nilaus_rant[math.random(1, #nilaus_rant)])
		else
			play_sfx(player, "utility/cannot_build")
		end
	else
		col2.calcui_rant.sprite = nil
		col2.calcui_scroll_pane.style.height = 252
	end
end

-- ----------------------------------------------------------------
local function destroy_calculator(player)
	local root = get_gui_root(player)
	if root.calcui then
		root.calcui.destroy()
		global.recent_results[player.index] = {}
	end
end

-- ----------------------------------------------------------------
local function fix_oob_ui(player)
	if global.gui_position[player.index].x < 0 then
		global.gui_position[player.index].x = 0
	end
	if global.gui_position[player.index].y < 0 then
		global.gui_position[player.index].y = 0
	end

	-- TODO fixed box size, because there is no API call for that
	local width = 255
	local height = 350

	if global.gui_position[player.index].x + width > player.display_resolution.width then
		global.gui_position[player.index].x = player.display_resolution.width - width
	end
	if global.gui_position[player.index].y + height > player.display_resolution.height then
		global.gui_position[player.index].y = player.display_resolution.height - height
	end
end

-- ----------------------------------------------------------------
function show_calculator(player)
	local root = get_gui_root(player)

	if not global.recent_results then
		global.recent_results = {}
	end
	if not global.recent_results[player.index] then
		global.recent_results[player.index] = {}
	end

	if not root.calcui then
		local calcui = root.add({
			type = "frame",
			name = "calcui",
			direction = "vertical"
		})

		local flow = calcui.add({
			type = "flow",
			name = "calcui_flow"
		})
		flow.style.horizontally_stretchable = "on"

		flow.add({
			type = "label",
			name = "calcui_title",
			caption = {"calculator-ui.title"},
			style = "frame_title"
		}).drag_target = calcui

		local widget = flow.add({
			type = "empty-widget",
			style = "draggable_space_header",
			name = "calcui_drag"
		})
		widget.drag_target = calcui
		widget.style.horizontally_stretchable = "on"
		widget.style.minimal_width = 24
		widget.style.natural_height = 24

		flow.add({
			type = "sprite-button",
			sprite = "utility/close_white",
			style = "frame_action_button",
			name = "calcui_close"
		})

		local table = calcui.add({
			type = "table",
			name = "calcui_table",
			column_count = "2",
			vertical_centering = "false"
		})

		local col1 = table.add({
			type = "flow",
			name = "calcui_table_col1",
			direction = "vertical"
		})

		local display = col1.add({
			type = "textfield",
			caption = "",
			name = "calcui_display"
		})
		display.style.width = 212

		local row1 = col1.add({type="flow", name="calcui_col1_row1", direction="horizontal"})
		row1.add({type="sprite-button", style="calcui_button_style_light", caption="CE", name="calcui_button_CE"}).sprite = "sprite_calcui_light"		-- CE = Clear Entry (just this line)
		row1.add({type="sprite-button", style="calcui_button_style_light", caption="C",  name="calcui_button_C"}).sprite = "sprite_calcui_light"		-- C  = Clear (all, past results as well)
		row1.add({type="sprite-button", style="calcui_button_style_light", caption="",   name="calcui_button_BS"}).sprite = "sprite_calcui_backspace"
		row1.add({type="sprite-button", style="calcui_button_style_light", caption="/",  name="calcui_button_DIV"}).sprite = "sprite_calcui_light"

		local row2 = col1.add({type="flow", name="calcui_col1_row2", direction="horizontal"})
		row2.add({type="sprite-button", style="calcui_button_style_dark",  caption="7",  name="calcui_button_7"}).sprite = "sprite_calcui_dark"
		row2.add({type="sprite-button", style="calcui_button_style_dark",  caption="8",  name="calcui_button_8"}).sprite = "sprite_calcui_dark"
		row2.add({type="sprite-button", style="calcui_button_style_dark",  caption="9",  name="calcui_button_9"}).sprite = "sprite_calcui_dark"
		row2.add({type="sprite-button", style="calcui_button_style_light", caption="*",  name="calcui_button_MUL"}).sprite = "sprite_calcui_light"

		local row3 = col1.add({type="flow", name="calcui_col1_row3", direction="horizontal"})
		row3.add({type="sprite-button", style="calcui_button_style_dark",  caption="4",  name="calcui_button_4"}).sprite = "sprite_calcui_dark"
		row3.add({type="sprite-button", style="calcui_button_style_dark",  caption="5",  name="calcui_button_5"}).sprite = "sprite_calcui_dark"
		row3.add({type="sprite-button", style="calcui_button_style_dark",  caption="6",  name="calcui_button_6"}).sprite = "sprite_calcui_dark"
		row3.add({type="sprite-button", style="calcui_button_style_light", caption="-",  name="calcui_button_SUB"}).sprite = "sprite_calcui_light"

		local row4 = col1.add({type="flow", name="calcui_col1_row4", direction="horizontal"})
		row4.add({type="sprite-button", style="calcui_button_style_dark",  caption="1",  name="calcui_button_1"}).sprite = "sprite_calcui_dark"
		row4.add({type="sprite-button", style="calcui_button_style_dark",  caption="2",  name="calcui_button_2"}).sprite = "sprite_calcui_dark"
		row4.add({type="sprite-button", style="calcui_button_style_dark",  caption="3",  name="calcui_button_3"}).sprite = "sprite_calcui_dark"
		row4.add({type="sprite-button", style="calcui_button_style_light", caption="+",  name="calcui_button_ADD"}).sprite = "sprite_calcui_light"

		local row5 = col1.add({type="flow", name="calcui_col1_row5", direction="horizontal"})
		row5.add({type="sprite-button", style="calcui_button_style_light", caption="%",  name="calcui_button_PERC"}).sprite = "sprite_calcui_light"
		row5.add({type="sprite-button", style="calcui_button_style_dark",  caption="0",  name="calcui_button_0"}).sprite = "sprite_calcui_dark"
		row5.add({type="sprite-button", style="calcui_button_style_dark",  caption=".",  name="calcui_button_DOT"}).sprite = "sprite_calcui_dark"
		row5.add({type="sprite-button", style="calcui_button_style_red",   caption="=",  name="calcui_button_EQU"}).sprite = "sprite_calcui_red"

		local col2 = table.add({
			type = "flow",
			name = "calcui_table_col2",
			direction = "vertical"
		})

		local result = col2.add({
			type = "flow",
			name = "calcui_result",
			direction = "horizontal"
		})

		result.add({
			type = "label",
			caption = "=",
			name = "calcui_display_sign"
		}).style.font = "default-large"

		result.add({
			type = "label",
			caption = "",
			name = "calcui_copy_display_result"
		}).style.font = "default-large"

		col2.add({
			type = "sprite",
			name = "calcui_rant"
		})

		col2.add({
			type = "line",
			name = "calcui_line",
			direction = "horizontal"
		})

		local scroll = col2.add({
			type = "scroll-pane",
			name = "calcui_scroll_pane"
		})
		scroll.style.height = 252

		scroll.add({
			type = "table",
			caption = "",
			name = "calcui_result_table",
			column_count = "3"
		}).style.column_alignments[1] = "right"


		-- use last saved location or center the gui
		if not global.gui_position then
			global.gui_position = {}
		end
		if global.gui_position[player.index] then
			-- fix weird saved positions (out of reach)
			fix_oob_ui(player)

			calcui.location = global.gui_position[player.index]
		else
			calcui.force_auto_center()
		end

		-- focus on display
		display.focus()
	end
end

-- ----------------------------------------------------------------
local function hide_calculator(player)
	destroy_calculator(player)
end

-- ----------------------------------------------------------------
function toggle_calculator(player)
	local root = get_gui_root(player)
	if root and root.calcui then
		hide_calculator(player)
	else
		show_calculator(player)
	end
end

-- ----------------------------------------------------------------
function focus_on_input(player)
	local root = get_gui_root(player)
	root.calcui.calcui_table.calcui_table_col1.calcui_display.focus()
end

-- ----------------------------------------------------------------
local function clear_equation(player)
	local root = get_gui_root(player)
	root.calcui.calcui_table.calcui_table_col1.calcui_display.text = ""
end

-- ----------------------------------------------------------------
local function process_ce_key(player, button)
	local root = get_gui_root(player)
	clear_equation(player)
	local result = root.calcui.calcui_table.calcui_table_col2.calcui_result.calcui_copy_display_result
	result.caption = ""
	result.tooltip = ""
end

-- ----------------------------------------------------------------
local function process_c_key(player, button)
	local root = get_gui_root(player)
	process_ce_key(player, button)
	root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.calcui_result_table.clear()
	global.recent_results[player.index] = {}
end

-- ----------------------------------------------------------------
local function process_backspace_key(player, button)
	local root = get_gui_root(player)
	local display = root.calcui.calcui_table.calcui_table_col1.calcui_display
	display.text = string.sub(display.text, 1, -2)
end

-- ----------------------------------------------------------------
local function draw_recent_table(player)
	local root = get_gui_root(player)

	local recent = root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane

	-- drop old table
	recent.calcui_result_table.clear()

	for i, result in ipairs(global.recent_results[player.index]) do
		recent.calcui_result_table.add({
			type = "label",
			name = "calcui_copy_equation_" .. i,
			caption = result["equation"],
			tooltip = {"calculator-ui.recent_tooltip"}
		})
		recent.calcui_result_table.add({
			type = "label",
			name = "calcui_sign_" .. i,
			caption = "="
		})
		recent.calcui_result_table.add({
			type = "label",
			name = "calcui_copy_result_" .. i,
			caption = result["result"]
		})
	end
	recent.scroll_to_top()
end

-- ----------------------------------------------------------------
local math_lib = {
	["abs"]   = "math.abs",
	["acos"]  = "math.acos",
	["asin"]  = "math.asin",
	["atan"]  = "math.atan",
	["atan2"] = "math.atan2",
	["ceil"]  = "math.ceil",
	["floor"] = "math.floor",
	["cos"]   = "math.cos",
	["cosh"]  = "math.cosh",
	["sin"]   = "math.sin",
	["sinh"]  = "math.sinh",
	["tan"]   = "math.tan",
	["tanh"]  = "math.tanh",
	["deg"]   = "math.deg",
	["rad"]   = "math.rad",
	["exp"]   = "math.exp",
	["log"]   = "math.log",
	["log10"] = "math.log10",
	["min"]   = "math.min",
	["max"]   = "math.max",
	["modf"]  = "math.modf",
	["fmod"]  = "math.fmod",
	["frexp"] = "math.frexp",
	["ldexp"] = "math.ldexp",
	["sqrt"]  = "math.sqrt",
	["huge"]  = "math.huge",
	["pi"]    = "math.pi",
	["pow"]   = "math.pow"
}
local function fix_equation(equation, root)
	local result = equation
	local prev_result = root.calcui.calcui_table.calcui_table_col2.calcui_result.calcui_copy_display_result.tooltip


	-- 1. visible part
	-- if equation does not start with a number or char, use the previous result (if available) and put it in front
	if not (string.match(result:sub(1, 1), "[^%w%(%)]") == nil) and prev_result then
		result = prev_result .. result
	end

	-- remove "math."
	result = result:gsub("math.", "")
	local new_equation = result


	-- 2. invisible part
	-- fix math library shortcuts
	for key, val in pairs(math_lib) do
		result = result:gsub(key, val)
	end

	-- fix percentage
	-- complex equations like "20+10%" = 22, before it was 20.1 -- big thanks to GWulf
	result = result:gsub("(%d+)(.)(%d+)%%", function (base, sign, perc)
		if sign == "+" then
			return base .. "*1." .. perc
		elseif sign == "-" then
			return base .. "*(1-0." ..perc .. ")"
		elseif sign == "*" then
			return "(" .. base .. "/100)*" .. perc
		elseif sign == "/" then
			return "(" .. base .. "*100)/" .. perc
		end
	end)
	-- still the simple equation needs some work: 1% = 0.01
	result = result:gsub("(%%)", "/100")

	-- fix danish keyboard
	result = result:gsub(",", ".")
	result = result:gsub(";", ",")


	return result, new_equation
end

-- ----------------------------------------------------------------
function process_equal_key(player, button)
	local root = get_gui_root(player)
	local original_equation = root.calcui.calcui_table.calcui_table_col1.calcui_display.text;

	equation, original_equation = fix_equation(original_equation, root)

	-- just testing
	--root.calcui.calcui_table.calcui_table_col1.calcui_display.text = equation

	if not (equation == nil or equation == "") then
		local status, retval = pcall(function()
			return load("return " .. equation)()
		end)
		root.calcui.calcui_table.calcui_table_col2.calcui_result.calcui_copy_display_result.tooltip = retval
		if not (retval == math.huge or retval ~= retval) then
			status, retval_show = pcall(function()
				local result = string.format("%0." .. settings.get_player_settings(player)["calcui-decimal-places"].value .. "f", retval)
				if result:len() > tostring(retval):len() then
					result = retval
				end
				return result
			end)
		else
			status = false
		end
		if retval_show == nil or retval_show == "" then
			status = false
		end
		if not status then
			retval_show = "NaN"
			show_rant(player, true)
		else
			if retval <= 0 then
				show_think(player, true)
			else
				show_rant(player, false)
			end
		end
		root.calcui.calcui_table.calcui_table_col2.calcui_result.calcui_copy_display_result.caption = retval_show

		-- only write in recent table if actually a result
		if status then
			-- check first equation and only insert if not the same
			if #global.recent_results[player.index] == 0 or global.recent_results[player.index][1]["equation"] ~= original_equation then
				table.insert(global.recent_results[player.index], 1, {
					equation = original_equation,
					result = retval_show
				})
			end
			draw_recent_table(player)
		end
	end

	if settings.get_player_settings(player)["calcui-clear-on-calc"].value then
		clear_equation(player)
	end
end

-- ----------------------------------------------------------------
local function display_addchar(player, char)
	local root = get_gui_root(player)
	local display = root.calcui.calcui_table.calcui_table_col1.calcui_display
	display.text = display.text .. char
	show_rant(player, false)
end

-- ----------------------------------------------------------------
local button_dispatch = {
	["CE"]  = process_ce_key,
	["C"]   = process_c_key,
	["BS"]  = process_backspace_key,
	--
	["EQU"] = process_equal_key
}
local button_addchar = {
	["DIV"]  = "/",
	--
	["7"]    = "7",
	["8"]    = "8",
	["9"]    = "9",
	["MUL"]  = "*",
	--
	["4"]    = "4",
	["5"]    = "5",
	["6"]    = "6",
	["SUB"]  = "-",
	--
	["1"]    = "1",
	["2"]    = "2",
	["3"]    = "3",
	["ADD"]  = "+",
	--
	["PERC"] = "%",
	["0"]    = "0",
	["DOT"]  = "."
}

function handle_calcui_click(event, player)
	debug_print("handle_calcui_click()")
	local event_name = event.element.name
	local button_prefix = "calcui_button_"
	local button_prefix_len = string.len(button_prefix)

	local copy_prefix = "calcui_copy_"
	local copy_prefix_len = string.len(copy_prefix);

	-- calculator buttons
	if string.sub(event_name, 1, button_prefix_len) == button_prefix then
		show_rant(player, false)

		button = string.sub(event_name, button_prefix_len + 1 )
		debug_print("handle_calcui_click button " .. button)
		local dispatch_func = button_dispatch[button]
		if dispatch_func then
			dispatch_func(player, button)
		end

		local addchar = button_addchar[button]
		if addchar then
			display_addchar(player, addchar)
		end
	-- close button
	elseif event_name == "calcui_close" then
		hide_calculator(player)
	-- copy results
	elseif string.sub(event_name, 1, copy_prefix_len) == copy_prefix then
		if event.button == defines.mouse_button_type.left and
		   event.shift == true then
			-- copy equation or result to display
			local root = get_gui_root(player)
			if event_name == "calcui_copy_display_result" then
				root.calcui.calcui_table.calcui_table_col1.calcui_display.text = root.calcui.calcui_table.calcui_table_col2.calcui_result.calcui_copy_display_result.caption
			else
				root.calcui.calcui_table.calcui_table_col1.calcui_display.text = root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.calcui_result_table[event_name].caption
			end
		end
		focus_on_input(player)
	-- if else focus on focus on display
	else
		focus_on_input(player)
	end
end

-- ----------------------------------------------------------------
function calcui_on_gui_text_changed(event)
	if event.element.name == "calcui_display" then
		local player = game.players[event.player_index]
		local root = get_gui_root(player)
		if string.find(root.calcui.calcui_table.calcui_table_col1.calcui_display.text, "=") then
			root.calcui.calcui_table.calcui_table_col1.calcui_display.text = root.calcui.calcui_table.calcui_table_col1.calcui_display.text:gsub("=", "")
			process_equal_key(player)
		end
	end
end

-- ----------------------------------------------------------------
function calcui_on_gui_location_changed(event)
	if event.element.name == "calcui" then
		if not global.gui_position then
			global.gui_position = {}
		end
		global.gui_position[event.player_index] = event.element.location
	end
end