extends GUINode

var _active : bool = false

const _screen : NationManagement.Screen = NationManagement.Screen.POPULATION

const _scene_name : String = "country_pops"

var _pop_screen_panel : Panel

var _province_listbox : GUIListBox
var _province_list_scroll_index : int = 0
var _province_list_types : Array[MenuSingleton.ProvinceListEntry]
var _province_list_indices : PackedInt32Array
var _province_list_panels : Array[Panel]
var _province_list_button_icons : Array[GFXSpriteTexture]
var _province_list_name_labels : Array[Label]
var _province_list_size_labels : Array[Label]
var _province_list_growth_icons : Array[GFXSpriteTexture]
var _province_list_colony_buttons : Array[Button]
var _province_list_national_focus_icons : Array[GFXSpriteTexture]
var _province_list_expand_icons : Array[GFXSpriteTexture]

var _pop_filter_buttons : Array[Button]
var _pop_filter_icons : Array[GFXSpriteTexture]
var _pop_filter_selected_icons : Array[GFXButtonStateTexture]
var _pop_filter_hover_icons : Array[GFXButtonStateTexture]

var _distribution_charts : Array[GFXPieChartTexture]
var _distribution_lists : Array[GUIListBox]

var _pop_list_scrollbar : GUIScrollbar
var _pop_list_scroll_index : int = 0

var _pop_list_rows : Array[Panel]
var _pop_list_size_labels : Array[Label]
var _pop_list_type_buttons : Array[Button]
var _pop_list_type_icons : Array[GFXSpriteTexture]
var _pop_list_producing_icons : Array[GFXSpriteTexture]
var _pop_list_culture_labels : Array[Label]
var _pop_list_religion_icons : Array[GFXSpriteTexture]
var _pop_list_location_labels : Array[Label]
var _pop_list_militancy_labels : Array[Label]
var _pop_list_consciousness_labels : Array[Label]
var _pop_list_ideology_charts : Array[GFXPieChartTexture]
var _pop_list_issues_charts : Array[GFXPieChartTexture]
var _pop_list_unemployment_progressbars : Array[TextureProgressBar]
var _pop_list_cash_labels : Array[Label]
var _pop_list_life_needs_progressbars : Array[TextureProgressBar]
var _pop_list_everyday_needs_progressbars : Array[TextureProgressBar]
var _pop_list_luxury_needs_progressbars : Array[TextureProgressBar]
var _pop_list_rebel_texture_rects : Array[TextureRect]
var _pop_list_rebel_icons : Array[GFXSpriteTexture]
var _pop_list_social_movement_texture_rects : Array[TextureRect]
var _pop_list_social_movement_icons : Array[GFXSpriteTexture]
var _pop_list_political_movement_texture_rects : Array[TextureRect]
var _pop_list_political_movement_icons : Array[GFXSpriteTexture]
var _pop_list_national_movement_texture_rects : Array[TextureRect]
var _pop_list_national_movement_flags : Array[GFXMaskedFlagTexture]
var _pop_list_size_change_icons : Array[GFXSpriteTexture]
var _pop_list_literacy_labels : Array[Label]

func _ready() -> void:
	GameSingleton.gamestate_updated.connect(_update_info)
	MenuSingleton.population_menu_province_list_changed.connect(_setup_province_list)
	MenuSingleton.population_menu_province_list_selected_changed.connect(_update_province_list)
	MenuSingleton.population_menu_pops_changed.connect(_update_pops)

	Events.NationManagementScreens.update_active_nation_management_screen.connect(_on_update_active_nation_management_screen)

	add_gui_element(_scene_name, "country_pop")

	var close_button : Button = get_button_from_nodepath(^"./country_pop/close_button")
	if close_button:
		close_button.pressed.connect(Events.NationManagementScreens.close_nation_management_screen.bind(_screen))

	_pop_screen_panel = get_panel_from_nodepath(^"./country_pop")

	# province list is set up via the population_menu_provinces_changed signal
	_setup_sort_buttons()
	_setup_pop_filter_buttons()
	_setup_distribution_windows()
	_setup_pop_list()

	_update_info()

func _generate_province_list_row(index : int, type : MenuSingleton.ProvinceListEntry) -> Error:
	while _province_list_types.size() <= index:
		_province_list_types.push_back(MenuSingleton.LIST_ENTRY_NONE)
		_province_list_indices.push_back(-1)
		_province_list_panels.push_back(null)
		_province_list_button_icons.push_back(null)
		_province_list_name_labels.push_back(null)
		_province_list_size_labels.push_back(null)
		_province_list_growth_icons.push_back(null)
		_province_list_colony_buttons.push_back(null)
		_province_list_national_focus_icons.push_back(null)
		_province_list_expand_icons.push_back(null)

	if _province_list_types[index] == type:
		return OK

	if _province_list_panels[index]:
		_province_listbox.remove_child(_province_list_panels[index])

	_province_list_types[index] = MenuSingleton.LIST_ENTRY_NONE
	_province_list_indices[index] = -1
	_province_list_panels[index] = null
	_province_list_button_icons[index] = null
	_province_list_name_labels[index] = null
	_province_list_size_labels[index] = null
	_province_list_growth_icons[index] = null
	_province_list_colony_buttons[index] = null
	_province_list_national_focus_icons[index] = null
	_province_list_expand_icons[index] = null

	if type == MenuSingleton.LIST_ENTRY_NONE:
		return OK

	const gui_element_names : Dictionary = {
		MenuSingleton.LIST_ENTRY_COUNTRY: "poplistitem_country",
		MenuSingleton.LIST_ENTRY_STATE: "poplistitem_state",
		MenuSingleton.LIST_ENTRY_PROVINCE: "poplistitem_province"
	}

	var entry_panel : Panel = GUINode.generate_gui_element(_scene_name, gui_element_names[type])

	if not entry_panel:
		return FAILED

	_province_list_types[index] = type

	_province_list_panels[index] = entry_panel

	var base_button : Button = GUINode.get_button_from_node(entry_panel.get_node(^"./poplistbutton"))
	if base_button:
		base_button.pressed.connect(
			func() -> void: MenuSingleton.population_menu_select_province_list_entry(_province_list_indices[index])
		)
		_province_list_button_icons[index] = GUINode.get_gfx_sprite_texture_from_node(base_button)

	_province_list_name_labels[index] = GUINode.get_label_from_node(entry_panel.get_node(^"./poplist_name"))

	_province_list_size_labels[index] = GUINode.get_label_from_node(entry_panel.get_node(^"./poplist_numpops"))

	_province_list_growth_icons[index] = GUINode.get_gfx_sprite_texture_from_node(entry_panel.get_node(^"./growth_indicator"))

	if type == MenuSingleton.LIST_ENTRY_STATE:
		_province_list_colony_buttons[index] = GUINode.get_button_from_node(entry_panel.get_node(^"./colonial_state_icon"))

		var national_focus_button : Button = GUINode.get_button_from_node(entry_panel.get_node(^"./state_focus"))
		if national_focus_button:
			# TODO - connect national focus button to national focus selection submenu
			_province_list_national_focus_icons[index] = GUINode.get_gfx_sprite_texture_from_node(national_focus_button)

		var expand_button : Button = GUINode.get_button_from_node(entry_panel.get_node(^"./expand"))
		if expand_button:
			expand_button.pressed.connect(
				func() -> void: MenuSingleton.population_menu_toggle_expanded(_province_list_indices[index])
			)
			_province_list_expand_icons[index] = GUINode.get_gfx_sprite_texture_from_node(expand_button)

	_province_listbox.add_child(entry_panel)
	_province_listbox.move_child(entry_panel, index)

	return OK

func _setup_province_list() -> void:
	if not  _province_listbox:
		_province_listbox = get_gui_listbox_from_nodepath(^"./country_pop/pop_province_list")

		if not _province_listbox:
			return
		_province_listbox.scroll_index_changed.connect(_update_province_list)

	if _province_list_panels.size() < 1 or not _province_list_panels[0]:
		if _generate_province_list_row(0, MenuSingleton.LIST_ENTRY_COUNTRY) != OK or _province_list_panels.size() < 1 or not _province_list_panels[0]:
			push_error("Failed to generate country row in population menu province list to determine row height!")
			return

	_province_listbox.set_fixed(MenuSingleton.get_population_menu_province_list_row_count(), _province_list_panels[0].size.y, false)

func _setup_sort_buttons() -> void:
	# button_path : NodePath, clear_text : bool, sort_key : GameSingleton.PopSortKey
	const sort_button_info : Array[Array] = [
		[^"./country_pop/sortby_size_button",         false, MenuSingleton.SORT_SIZE],
		[^"./country_pop/sortby_type_button",         false, MenuSingleton.SORT_TYPE],
		[^"./country_pop/sortby_nationality_button",  false, MenuSingleton.SORT_CULTURE],
		[^"./country_pop/sortby_religion_button",     false, MenuSingleton.SORT_RELIGION],
		[^"./country_pop/sortby_location_button",     false, MenuSingleton.SORT_LOCATION],
		[^"./country_pop/sortby_mil_button",          true,  MenuSingleton.SORT_MILITANCY],
		[^"./country_pop/sortby_con_button",          true,  MenuSingleton.SORT_CONSCIOUSNESS],
		[^"./country_pop/sortby_ideology_button",     true,  MenuSingleton.SORT_IDEOLOGY],
		[^"./country_pop/sortby_issues_button",       true,  MenuSingleton.SORT_ISSUES],
		[^"./country_pop/sortby_unemployment_button", true,  MenuSingleton.SORT_UNEMPLOYMENT],
		[^"./country_pop/sortby_cash_button",         true,  MenuSingleton.SORT_CASH],
		[^"./country_pop/sortby_subsistence_button",  true,  MenuSingleton.SORT_LIFE_NEEDS],
		[^"./country_pop/sortby_eve_button",          true,  MenuSingleton.SORT_EVERYDAY_NEEDS],
		[^"./country_pop/sortby_luxury_button",       true,  MenuSingleton.SORT_LUXURY_NEEDS],
		[^"./country_pop/sortby_revoltrisk_button",   true,  MenuSingleton.SORT_REBEL_FACTION],
		[^"./country_pop/sortby_change_button",       true,  MenuSingleton.SORT_SIZE_CHANGE],
		[^"./country_pop/sortby_literacy_button",     true,  MenuSingleton.SORT_LITERACY]
	]

	for button_info : Array in sort_button_info:
		var sort_button : Button = get_button_from_nodepath(button_info[0])
		if sort_button:
			if button_info[1]:
				sort_button.set_text("")
			sort_button.pressed.connect(MenuSingleton.population_menu_select_sort_key.bind(button_info[2]))

func _setup_pop_filter_buttons() -> void:
	if not _pop_screen_panel:
		push_error("Cannot set up pop filter buttons without pop screen to add them to")
		return

	var pop_filter_sprite_indices : PackedInt32Array = MenuSingleton.get_population_menu_pop_filter_setup_info()

	var pop_filter_start : Vector2 = GUINode.get_gui_position(_scene_name, "popfilter_start")
	var pop_filter_step : Vector2 = GUINode.get_gui_position(_scene_name, "popfilter_offset")

	for index : int in pop_filter_sprite_indices.size():
		var pop_filter_button : Button = GUINode.get_button_from_node(GUINode.generate_gui_element(_scene_name, "pop_filter_button"))
		var pop_filter_icon : GFXSpriteTexture = null
		var pop_filter_selected_icon : GFXButtonStateTexture = null
		var pop_filter_hover_icon : GFXButtonStateTexture = null

		if pop_filter_button:
			_pop_screen_panel.add_child(pop_filter_button)
			pop_filter_button.set_position(pop_filter_start + pop_filter_step * index)
			pop_filter_button.pressed.connect(MenuSingleton.population_menu_toggle_pop_filter.bind(index))
			pop_filter_icon = GUINode.get_gfx_sprite_texture_from_node(pop_filter_button)

			if pop_filter_icon:
				pop_filter_icon.set_icon_index(pop_filter_sprite_indices[index])
				pop_filter_selected_icon = pop_filter_icon.get_button_state_texture(GFXButtonStateTexture.SELECTED)
				pop_filter_hover_icon = pop_filter_icon.get_button_state_texture(GFXButtonStateTexture.HOVER)

		_pop_filter_buttons.push_back(pop_filter_button)
		_pop_filter_icons.push_back(pop_filter_icon)
		_pop_filter_selected_icons.push_back(pop_filter_selected_icon)
		_pop_filter_hover_icons.push_back(pop_filter_hover_icon)

	var select_all_button : Button = get_button_from_nodepath(^"./country_pop/popfilter_ALL")
	if select_all_button:
		select_all_button.pressed.connect(MenuSingleton.population_menu_select_all_pop_filters)

	var deselect_all_button : Button = get_button_from_nodepath(^"./country_pop/popfilter_DESELECT_ALL")
	if deselect_all_button:
		deselect_all_button.pressed.connect(MenuSingleton.population_menu_deselect_all_pop_filters)

func _setup_distribution_windows() -> void:
	if not _pop_screen_panel:
		push_error("Cannot set up distribution windows without pop screen to add them to")
		return

	const columns : int = 3

	var distribution_names : PackedStringArray = MenuSingleton.get_population_menu_distribution_setup_info()

	var distribution_start : Vector2 = GUINode.get_gui_position(_scene_name, "popdistribution_start")
	var distribution_step : Vector2 = GUINode.get_gui_position(_scene_name, "popdistribution_offset")

	for index : int in distribution_names.size():
		var distribution_panel : Panel = GUINode.generate_gui_element(_scene_name, "distribution_window")
		var distribution_chart : GFXPieChartTexture = null
		var distribution_list : GUIListBox = null

		if distribution_panel:
			_pop_screen_panel.add_child(distribution_panel)
			distribution_panel.set_position(distribution_start + distribution_step * Vector2(index % columns, index / columns))

			var name_label : Label = GUINode.get_label_from_node(distribution_panel.get_node(^"./item_name"))
			if name_label:
				name_label.text = distribution_names[index]

			distribution_chart = GUINode.get_gfx_pie_chart_texture_from_node(distribution_panel.get_node(^"./chart"))
			distribution_list = GUINode.get_gui_listbox_from_node(distribution_panel.get_node(^"./member_names"))

		_distribution_charts.push_back(distribution_chart)
		_distribution_lists.push_back(distribution_list)

func _setup_pop_list() -> void:
	_pop_list_scrollbar = get_gui_scrollbar_from_nodepath(^"./country_pop/external_scroll_slider")

	var pop_list_panel : Panel = get_panel_from_nodepath(^"./country_pop/pop_list")
	if not pop_list_panel:
		return

	if _pop_list_scrollbar:
		_pop_list_scrollbar.value_changed.connect(
			func (value : int) -> void:
				_pop_list_scroll_index = value
				_update_pop_list()
		)

		pop_list_panel.gui_input.connect(
			func (event : InputEvent) -> void:
				if event is InputEventMouseButton:
					if event.is_pressed():
						if event.get_button_index() == MOUSE_BUTTON_WHEEL_UP:
							_pop_list_scrollbar.decrement_value()
						elif event.get_button_index() == MOUSE_BUTTON_WHEEL_DOWN:
							_pop_list_scrollbar.increment_value()
		)

	var height : float = 0.0
	while height < pop_list_panel.size.y:
		var pop_row_panel : Panel = GUINode.generate_gui_element(_scene_name, "popinfomember_popview")
		if not pop_row_panel:
			break

		pop_list_panel.add_child(pop_row_panel)
		pop_row_panel.set_position(Vector2(0, height))
		height += pop_row_panel.size.y
		_pop_list_rows.push_back(pop_row_panel)

		_pop_list_size_labels.push_back(GUINode.get_label_from_node(pop_row_panel.get_node(^"./pop_size")))

		var pop_type_button : Button = GUINode.get_button_from_node(pop_row_panel.get_node(^"./pop_type"))
		# TODO - open pop details menu on pop type button press
		_pop_list_type_buttons.push_back(pop_type_button)

		_pop_list_type_icons.push_back(GUINode.get_gfx_sprite_texture_from_node(pop_type_button))

		_pop_list_producing_icons.push_back(GUINode.get_gfx_sprite_texture_from_node(pop_row_panel.get_node(^"./pop_producing_icon")))

		var culture_label : Label = GUINode.get_label_from_node(pop_row_panel.get_node(^"./pop_nation"))
		if culture_label:
			culture_label.set_text_overrun_behavior(TextServer.OVERRUN_TRIM_ELLIPSIS)
		_pop_list_culture_labels.push_back(culture_label)

		_pop_list_religion_icons.push_back(GUINode.get_gfx_sprite_texture_from_node(pop_row_panel.get_node(^"./pop_religion")))

		var location_label : Label = GUINode.get_label_from_node(pop_row_panel.get_node(^"./pop_location"))
		if location_label:
			location_label.set_text_overrun_behavior(TextServer.OVERRUN_TRIM_ELLIPSIS)
		_pop_list_location_labels.push_back(location_label)

		_pop_list_militancy_labels.push_back(GUINode.get_label_from_node(pop_row_panel.get_node(^"./pop_mil")))

		_pop_list_consciousness_labels.push_back(GUINode.get_label_from_node(pop_row_panel.get_node(^"./pop_con")))

		_pop_list_ideology_charts.push_back(GUINode.get_gfx_pie_chart_texture_from_node(pop_row_panel.get_node(^"./pop_ideology")))

		_pop_list_issues_charts.push_back(GUINode.get_gfx_pie_chart_texture_from_node(pop_row_panel.get_node(^"./pop_issues")))

		_pop_list_unemployment_progressbars.push_back(GUINode.get_progress_bar_from_node(pop_row_panel.get_node(^"./pop_unemployment_bar")))

		_pop_list_cash_labels.push_back(GUINode.get_label_from_node(pop_row_panel.get_node(^"./pop_cash")))

		var pop_list_life_needs_progressbar : TextureProgressBar = GUINode.get_progress_bar_from_node(pop_row_panel.get_node(^"./lifeneed_progress"))
		if pop_list_life_needs_progressbar:
			pop_list_life_needs_progressbar.position += Vector2(1, 0)
		_pop_list_life_needs_progressbars.push_back(pop_list_life_needs_progressbar)

		var pop_list_everyday_needs_progressbar : TextureProgressBar = GUINode.get_progress_bar_from_node(pop_row_panel.get_node(^"./eveneed_progress"))
		if pop_list_everyday_needs_progressbar:
			pop_list_everyday_needs_progressbar.position += Vector2(1, 0)
		_pop_list_everyday_needs_progressbars.push_back(pop_list_everyday_needs_progressbar)

		_pop_list_luxury_needs_progressbars.push_back(GUINode.get_progress_bar_from_node(pop_row_panel.get_node(^"./luxneed_progress")))

		var pop_list_rebel_texture_rect : TextureRect = GUINode.get_texture_rect_from_node(pop_row_panel.get_node(^"./pop_revolt"))
		_pop_list_rebel_texture_rects.push_back(pop_list_rebel_texture_rect)
		if pop_list_rebel_texture_rect:
			_pop_list_rebel_icons.push_back(GUINode.get_gfx_sprite_texture_from_node(pop_list_rebel_texture_rect))
		else:
			_pop_list_rebel_icons.push_back(null)

		var pop_list_social_movement_texture_rect : TextureRect = GUINode.get_texture_rect_from_node(pop_row_panel.get_node(^"./pop_movement_social"))
		_pop_list_social_movement_texture_rects.push_back(pop_list_social_movement_texture_rect)
		if pop_list_social_movement_texture_rect:
			_pop_list_social_movement_icons.push_back(GUINode.get_gfx_sprite_texture_from_node(pop_list_social_movement_texture_rect))
		else:
			_pop_list_social_movement_icons.push_back(null)

		var pop_list_political_movement_texture_rect : TextureRect = GUINode.get_texture_rect_from_node(pop_row_panel.get_node(^"./pop_movement_political"))
		_pop_list_political_movement_texture_rects.push_back(pop_list_political_movement_texture_rect)
		if pop_list_political_movement_texture_rect:
			_pop_list_political_movement_icons.push_back(GUINode.get_gfx_sprite_texture_from_node(pop_list_political_movement_texture_rect))
		else:
			_pop_list_political_movement_icons.push_back(null)

		var pop_list_national_movement_texture_rect : TextureRect = GUINode.get_texture_rect_from_node(pop_row_panel.get_node(^"./pop_movement_flag"))
		_pop_list_national_movement_texture_rects.push_back(pop_list_national_movement_texture_rect)
		if pop_list_national_movement_texture_rect:
			_pop_list_national_movement_flags.push_back(GUINode.get_gfx_masked_flag_texture_from_node(pop_list_national_movement_texture_rect))
		else:
			_pop_list_national_movement_flags.push_back(null)

		_pop_list_size_change_icons.push_back(GUINode.get_gfx_sprite_texture_from_node(pop_row_panel.get_node(^"./growth_indicator")))

		_pop_list_literacy_labels.push_back(GUINode.get_label_from_node(pop_row_panel.get_node(^"./pop_literacy")))

func _notification(what : int) -> void:
	match what:
		NOTIFICATION_TRANSLATION_CHANGED:
			_update_info()

func _on_update_active_nation_management_screen(active_screen : NationManagement.Screen) -> void:
	_active = active_screen == _screen
	_update_info()

func _update_info() -> void:
	if _active:
		# Province list
		_update_province_list()

		# Pop filter buttons, Distributions, Pop list
		_update_pops()

		show()
	else:
		hide()

func get_growth_icon_index(size_change : int) -> int:
	return 1 + int(size_change <= 0) + int(size_change < 0)

func _update_province_list(scroll_index : int = -1) -> void:
	if not _province_listbox:
		return

	if scroll_index >= 0:
		_province_listbox.set_scroll_index(scroll_index, false)

	_province_list_scroll_index = _province_listbox.get_scroll_index()

	var province_list_info_list : Array[Dictionary] = MenuSingleton.get_population_menu_province_list_rows(_province_list_scroll_index, _province_listbox.get_fixed_visible_items())

	for index : int in province_list_info_list.size():
		const type_key : StringName = &"type"
		const index_key : StringName = &"index"
		const name_key : StringName = &"name"
		const size_key : StringName = &"size"
		const change_key : StringName = &"change"
		const selected_key : StringName = &"selected"
		const expanded_key : StringName = &"expanded"
		const colony_key : StringName = &"colony"

		var province_list_info : Dictionary = province_list_info_list[index]

		var type : MenuSingleton.ProvinceListEntry = province_list_info[type_key]

		if _generate_province_list_row(index, type) != OK:
			continue

		if type == MenuSingleton.LIST_ENTRY_NONE or type != _province_list_types[index]:
			continue

		_province_list_indices[index] = province_list_info[index_key]

		if _province_list_button_icons[index]:
			_province_list_button_icons[index].set_icon_index(1 + int(province_list_info[selected_key]))

		if _province_list_name_labels[index]:
			_province_list_name_labels[index].set_text(
				GUINode.format_province_name(province_list_info[name_key]) if type == MenuSingleton.LIST_ENTRY_PROVINCE
				else province_list_info[name_key]
			)
			_province_list_name_labels[index].set_text_overrun_behavior(TextServer.OVERRUN_TRIM_ELLIPSIS)

		if _province_list_size_labels[index]:
			_province_list_size_labels[index].set_text(GUINode.int_to_string_suffixed(province_list_info[size_key]))

		if _province_list_growth_icons[index]:
			_province_list_growth_icons[index].set_icon_index(get_growth_icon_index(province_list_info[change_key]))

		if type == MenuSingleton.LIST_ENTRY_STATE:
			if _province_list_colony_buttons[index]:
				_province_list_colony_buttons[index].set_visible(province_list_info[colony_key])

			if _province_list_expand_icons[index]:
				_province_list_expand_icons[index].set_icon_index(1 + int(province_list_info[expanded_key]))

			# TODO - set _province_list_national_focus_icons[index]

	# Clear any excess rows
	for index : int in range(province_list_info_list.size(), _province_list_types.size()):
		_generate_province_list_row(index, MenuSingleton.LIST_ENTRY_NONE)

func _update_pops() -> void:
	_update_pop_filters()
	_update_distributions()
	_update_pop_list()

func _update_pop_filters() -> void:
	var pop_filter_info_list : Array[Dictionary] = MenuSingleton.get_population_menu_pop_filter_info()

	for index : int in pop_filter_info_list.size():
		const pop_filter_count_key : StringName = &"count"
		const pop_filter_change_key : StringName = &"change"
		const pop_filter_selected_key : StringName = &"selected"

		var pop_filter_info : Dictionary = pop_filter_info_list[index]

		var pop_filter_button : Button = _pop_filter_buttons[index]
		if not pop_filter_button:
			continue
		pop_filter_button.disabled = pop_filter_info[pop_filter_count_key] <= 0

		const normal_theme : StringName = &"normal"
		const hover_theme : StringName = &"hover"

		if pop_filter_info[pop_filter_selected_key] or pop_filter_button.disabled:
			pop_filter_button.get_theme_stylebox(normal_theme).set_texture(_pop_filter_icons[index])
			pop_filter_button.get_theme_stylebox(hover_theme).set_texture(_pop_filter_hover_icons[index])
		else:
			pop_filter_button.get_theme_stylebox(normal_theme).set_texture(_pop_filter_selected_icons[index])
			pop_filter_button.get_theme_stylebox(hover_theme).set_texture(_pop_filter_selected_icons[index])
		# TODO - size and promotion/demotion change tooltip

func _update_distributions():
	const slice_identifier_key : StringName = &"identifier"
	const slice_colour_key : StringName = &"colour"
	const slice_weight_key : StringName = &"weight"

	var distribution_info_list : Array[Array] = MenuSingleton.get_population_menu_distribution_info()

	for distribution_index : int in distribution_info_list.size():
		var distribution_info : Array[Dictionary] = distribution_info_list[distribution_index]

		if _distribution_charts[distribution_index]:
			_distribution_charts[distribution_index].set_slices_array(distribution_info)

		if _distribution_lists[distribution_index]:
			distribution_info.sort_custom(func(a : Dictionary, b : Dictionary) -> bool: return a[slice_weight_key] > b[slice_weight_key])

			var list : GUIListBox = _distribution_lists[distribution_index]

			list.clear_children(distribution_info.size())

			while list.get_child_count() < distribution_info.size():
				var child : Panel = GUINode.generate_gui_element(_scene_name, "pop_legend_item")
				if not child:
					break
				child.set_mouse_filter(Control.MOUSE_FILTER_IGNORE)
				list.add_child(child)

			for list_index in min(list.get_child_count(), distribution_info.size()):

				var child : Panel = list.get_child(list_index)

				var distribution_row : Dictionary = distribution_info[list_index]

				var colour_icon_rect : TextureRect = GUINode.get_texture_rect_from_node(child.get_node(^"./legend_color"))
				if colour_icon_rect:
					colour_icon_rect.set_modulate(distribution_row[slice_colour_key])

				var identifier_label : Label = GUINode.get_label_from_node(child.get_node(^"./legend_title"))
				if identifier_label:
					identifier_label.set_text_overrun_behavior(TextServer.OVERRUN_TRIM_ELLIPSIS)
					identifier_label.set_text(distribution_row[slice_identifier_key])

				var weight_label : Label = GUINode.get_label_from_node(child.get_node(^"./legend_value"))
				if weight_label:
					weight_label.set_text("%s%%" % GUINode.float_to_string_dp(distribution_row[slice_weight_key] * 100.0, 1))

func _update_pop_list() -> void:
	if _pop_list_scrollbar:
		var max_scroll_index : int = MenuSingleton.get_population_menu_pop_row_count() - _pop_list_rows.size()
		if max_scroll_index > 0:
			_pop_list_scrollbar.set_limits(0, max_scroll_index)
			_pop_list_scrollbar.show()
		else:
			_pop_list_scrollbar.set_limits(0, 0)
			_pop_list_scrollbar.hide()

	var pop_rows = MenuSingleton.get_population_menu_pop_rows(_pop_list_scroll_index, _pop_list_rows.size())

	for index : int in _pop_list_rows.size():
		if not _pop_list_rows[index]:
			continue
		if index < pop_rows.size():
			const pop_size_key : StringName = &"size"
			const pop_type_icon_key : StringName = &"pop_type_icon"
			const pop_culture_key : StringName = &"culture"
			const pop_religion_icon_key : StringName = &"religion_icon"
			const pop_location_key : StringName = &"location"
			const pop_militancy_key : StringName = &"militancy"
			const pop_consciousness_key : StringName = &"consciousness"
			const pop_ideology_key : StringName = &"ideology"
			const pop_issues_key : StringName = &"issues"
			const pop_unemployment_key : StringName = &"unemployment"
			const pop_cash_key : StringName = &"cash"
			const pop_life_needs_key : StringName = &"life_needs"
			const pop_everyday_needs_key : StringName = &"everyday_needs"
			const pop_luxury_needs_key : StringName = &"luxury_needs"
			const pop_rebel_icon_key : StringName = &"rebel_icon"
			const pop_size_change_key : StringName = &"size_change"
			const pop_literacy_key : StringName = &"literacy"

			var pop_row : Dictionary = pop_rows[index]

			if _pop_list_size_labels[index]:
				_pop_list_size_labels[index].set_text(GUINode.int_to_string_suffixed(pop_row[pop_size_key]))
			if _pop_list_type_icons[index]:
				_pop_list_type_icons[index].set_icon_index(pop_row[pop_type_icon_key])
			if _pop_list_culture_labels[index]:
				_pop_list_culture_labels[index].set_text(pop_row[pop_culture_key])
			if _pop_list_religion_icons[index]:
				_pop_list_religion_icons[index].set_icon_index(pop_row[pop_religion_icon_key])
			if _pop_list_location_labels[index]:
				_pop_list_location_labels[index].set_text(GUINode.format_province_name(pop_row.get(pop_location_key, "")))
			if _pop_list_militancy_labels[index]:
				_pop_list_militancy_labels[index].set_text(GUINode.float_to_string_dp(pop_row[pop_militancy_key], 2))
			if _pop_list_consciousness_labels[index]:
				_pop_list_consciousness_labels[index].set_text(GUINode.float_to_string_dp(pop_row[pop_consciousness_key], 2))
			if _pop_list_ideology_charts[index]:
				_pop_list_ideology_charts[index].set_slices_array(pop_row[pop_ideology_key])
			if _pop_list_issues_charts[index]:
				_pop_list_issues_charts[index].set_slices_array(pop_row[pop_issues_key])
			if _pop_list_unemployment_progressbars[index]:
				_pop_list_unemployment_progressbars[index].set_value_no_signal(pop_row[pop_unemployment_key])
			if _pop_list_cash_labels[index]:
				_pop_list_cash_labels[index].set_text(GUINode.float_to_string_dp(pop_row[pop_cash_key], 2))
			if _pop_list_life_needs_progressbars[index]:
				_pop_list_life_needs_progressbars[index].set_value_no_signal(pop_row[pop_life_needs_key])
			if _pop_list_everyday_needs_progressbars[index]:
				_pop_list_everyday_needs_progressbars[index].set_value_no_signal(pop_row[pop_everyday_needs_key])
			if _pop_list_luxury_needs_progressbars[index]:
				_pop_list_luxury_needs_progressbars[index].set_value_no_signal(pop_row[pop_luxury_needs_key])
			if _pop_list_rebel_texture_rects[index]:
				var rebel_icon : int = pop_row.get(pop_rebel_icon_key, 0)
				if rebel_icon > 0:
					if _pop_list_rebel_icons[index]:
						_pop_list_rebel_icons[index].set_icon_index(rebel_icon)
					_pop_list_rebel_texture_rects[index].show()
				else:
					_pop_list_rebel_texture_rects[index].hide()

			# TODO - handle social/political reform and country rebels
			if _pop_list_social_movement_texture_rects[index]:
				_pop_list_social_movement_texture_rects[index].hide()
			if _pop_list_political_movement_texture_rects[index]:
				_pop_list_political_movement_texture_rects[index].hide()
			if _pop_list_national_movement_texture_rects[index]:
				_pop_list_national_movement_texture_rects[index].hide()

			if _pop_list_size_change_icons[index]:
				_pop_list_size_change_icons[index].set_icon_index(get_growth_icon_index(pop_row[pop_size_change_key]))
			if _pop_list_literacy_labels[index]:
				_pop_list_literacy_labels[index].set_text("%s%%" % GUINode.float_to_string_dp(pop_row[pop_literacy_key], 2))

			_pop_list_rows[index].show()
		else:
			_pop_list_rows[index].hide()