Factorio / calculator-ui

-- calculator.lua

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

-- ----------------------------------------------------------------
function show_think(player, enabled) 
	local root = get_gui_root(player)
	
	if enabled then
		root.calcui.calcui_table.calcui_table_col2.calcui_rant.sprite = "sprite_calcui_think"
		root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.style.height = 196
	else
		root.calcui.calcui_table.calcui_table_col2.calcui_rant.sprite = nil
		root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.style.height = 252
	end
end

-- ----------------------------------------------------------------
function show_rant(player, enabled) 
	local root = get_gui_root(player)
	
	if enabled then
		root.calcui.calcui_table.calcui_table_col2.calcui_rant.sprite = "sprite_calcui_rant"
		root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.style.height = 196
	else
		root.calcui.calcui_table.calcui_table_col2.calcui_rant.sprite = nil
		root.calcui.calcui_table.calcui_table_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()
	end
end

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

	if not root.calcui then
		local calcui = root.add({
			type = "frame", 
			name = "calcui", 
			style = "dialog_frame",
			direction = "vertical"
		})
		
		local flow = calcui.add({
			type = "flow", 
			name = "calcui_flow"
		})
		flow.style.horizontally_stretchable = "on"
		
		flow.add({
			type = "label", 
			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", 
			style = "calcui_textfield_style", 
			caption = "", 
			name = "calcui_display"
		})

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

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

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

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

		local row5 = col1.add({type="flow", name="calcui_col1_row5", direction="horizontal"})
		row5.add({type="button", style="calcui_button_style", caption="%",  name="calcui_button_PERC"})
		row5.add({type="button", style="calcui_button_style", caption="0",  name="calcui_button_0"})
		row5.add({type="button", style="calcui_button_style", caption=".",  name="calcui_button_DOT"})
		row5.add({type="button", style="calcui_button_style", caption="=",  name="calcui_button_EQU"})		


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

		local result = col2.add({
			type = "label", 
			caption = "= ", 
			name = "calcui_display_result"
		})
		result.style.font = "default-large"
		
		local rant = col2.add({
			type = "sprite",
			name = "calcui_rant"
		})
		
		col2.add({
			type = "line", 
			direction = "horizontal"
		})
		
		local scroll = col2.add({
			type = "scroll-pane",
			name = "calcui_scroll_pane"
		})
		scroll.style.height = 252
		
		local recents = scroll.add({
			type = "table", 
			caption = "", 
			name = "calcui_result_table", 
			column_count = "2"
		})
		recents.style.column_alignments[1] = "right"
		
		
		-- center the gui
		calcui.force_auto_center()
	end
end

-- ----------------------------------------------------------------
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

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

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

-- ----------------------------------------------------------------
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()
end

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

-- ----------------------------------------------------------------
function fix_equation(equation)
	local result = equation
	
	-- fix math library shortcuts
	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"
	}
	for key, val in pairs(math_lib) do
		result = result:gsub(key, val)
	end

	-- fix percentage
	result = result:gsub("(%%)", "/100")
	
	-- fix danish keyboard
	result = result:gsub(",", ".")
	result = result:gsub(";", ",")

	return result
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 = fix_equation(original_equation)
	
	-- 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_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_display_result.caption = "= " .. retval_show
		
		-- only write in recent table if actually a result
		if status then
			-- check last equation and only insert if not the same
			local item_size = #root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.calcui_result_table.children
			
			if item_size == 0 or root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.calcui_result_table.children[item_size-1].caption ~= original_equation then
				local recent_equation = root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.calcui_result_table.add({
					type = "label", 
					name = "calcui_recent_equation_" .. item_size, 
					caption = original_equation,
					tooltip = {"calculator-ui.recent_tooltip"}
				})
				local recent_result = root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.calcui_result_table.add({
					type = "label", 
					name = "calcui_recent_result_" .. item_size, 
					caption = root.calcui.calcui_table.calcui_table_col2.calcui_display_result.caption
				})
				root.calcui.calcui_table.calcui_table_col2.calcui_scroll_pane.scroll_to_bottom()
			end
		end
	end

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

-- ----------------------------------------------------------------
function display_addchar(player, char)
	local root = get_gui_root(player)
	root.calcui.calcui_table.calcui_table_col1.calcui_display.text = root.calcui.calcui_table.calcui_table_col1.calcui_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 recent_prefix = "calcui_recent_"
	local recent_prefix_len = string.len(recent_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)
	-- recent results
	elseif string.sub(event_name, 1, recent_prefix_len) == recent_prefix then
		if event.button == defines.mouse_button_type.left and 
		   event.shift == true then
			-- copy equation to display
			local root = get_gui_root(player)
			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
			focus_on_input(player)
		end
	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
		local player = game.players[event.player_index]
		local root = get_gui_root(player)
		root.calcui.location = event.element.location
	end
end