You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
337 lines
10 KiB
337 lines
10 KiB
@tool
|
|
extends Tree
|
|
|
|
enum TreeButtons {ADD_FOLDER, ADD_VARIABLE, DUPLICATE_FOLDER, DELETE, CHANGE_TYPE}
|
|
|
|
@onready var editor: DialogicEditor = find_parent("VariablesEditor")
|
|
|
|
|
|
#region INITIAL SETUP
|
|
|
|
func _ready() -> void:
|
|
set_column_title(0, "Name")
|
|
set_column_title(1, "")
|
|
set_column_title(2, "Default Value")
|
|
set_column_expand(1, false)
|
|
set_column_expand_ratio(2, 2)
|
|
set_column_title_alignment(0, 0)
|
|
set_column_title_alignment(2, 0)
|
|
|
|
%ChangeTypePopup.self_modulate = get_theme_color("dark_color_3", "Editor")
|
|
%ChangeTypePopup.theme.set_stylebox('pressed', 'Button', get_theme_stylebox("LaunchPadMovieMode", "EditorStyles"))
|
|
%ChangeTypePopup.theme.set_stylebox('hover', 'Button', get_theme_stylebox("LaunchPadMovieMode", "EditorStyles"))
|
|
for child in %ChangeTypePopup/HBox.get_children():
|
|
child.toggled.connect(_on_type_pressed.bind(child.get_index()+1))
|
|
child.icon = get_theme_icon(["String", "float", "int", "bool"][child.get_index()], "EditorIcons")
|
|
|
|
#endregion
|
|
|
|
|
|
#region POPULATING THE TREE
|
|
|
|
func load_info(dict:Dictionary, parent:TreeItem = null, is_new:=false) -> void:
|
|
if parent == null:
|
|
clear()
|
|
parent = add_folder_item("VAR", null)
|
|
|
|
var sorted_keys := dict.keys()
|
|
sorted_keys.sort()
|
|
for key in sorted_keys:
|
|
if typeof(dict[key]) != TYPE_DICTIONARY:
|
|
var item := add_variable_item(key, dict[key], parent)
|
|
if is_new:
|
|
item.set_meta("new", true)
|
|
|
|
for key in sorted_keys:
|
|
if typeof(dict[key]) == TYPE_DICTIONARY:
|
|
var folder := add_folder_item(key, parent)
|
|
if is_new:
|
|
folder.set_meta("new", true)
|
|
load_info(dict[key], folder, is_new)
|
|
|
|
|
|
|
|
func add_variable_item(name:String, value:Variant, parent:TreeItem) -> TreeItem:
|
|
var item := create_item(parent)
|
|
item.set_meta("type", "VARIABLE")
|
|
|
|
item.set_text(0, name)
|
|
item.set_editable(0, true)
|
|
item.set_metadata(0, name)
|
|
item.set_icon(0, load(DialogicUtil.get_module_path('Variable').path_join("variable.svg")))
|
|
var folder_color: Color = parent.get_meta('color', Color.DARK_GOLDENROD)
|
|
item.set_custom_bg_color(0, folder_color.lerp(get_theme_color("background", "Editor"), 0.8))
|
|
|
|
item.add_button(1, get_theme_icon("String", "EditorIcons"), TreeButtons.CHANGE_TYPE)
|
|
adjust_variable_type(item, DialogicUtil.get_variable_value_type(value), value)
|
|
item.set_editable(2, true)
|
|
item.add_button(2, get_theme_icon("Remove", "EditorIcons"), TreeButtons.DELETE)
|
|
|
|
item.set_meta('prev_path', get_item_path(item))
|
|
return item
|
|
|
|
|
|
func add_folder_item(name:String, parent:TreeItem) -> TreeItem:
|
|
var item := create_item(parent)
|
|
item.set_icon(0, get_theme_icon("Folder", "EditorIcons"))
|
|
item.set_text(0, name)
|
|
item.set_editable(0, item != get_root())
|
|
|
|
var folder_color: Color
|
|
if parent == null:
|
|
folder_color = Color(0.33000001311302, 0.15179999172688, 0.15179999172688)
|
|
else:
|
|
folder_color = parent.get_meta('color')
|
|
folder_color.h = wrap(folder_color.h+0.15*(item.get_index()+1), 0, 1)
|
|
item.set_custom_bg_color(0, folder_color)
|
|
item.set_custom_bg_color(1, folder_color)
|
|
item.set_custom_bg_color(2, folder_color)
|
|
item.set_meta('color', folder_color)
|
|
item.add_button(2, load(self.get_script().get_path().get_base_dir().get_base_dir() + "/add-variable.svg"), TreeButtons.ADD_VARIABLE)
|
|
item.add_button(2, load("res://addons/dialogic/Editor/Images/Pieces/add-folder.svg"), TreeButtons.ADD_FOLDER)
|
|
item.add_button(2, get_theme_icon("Duplicate", "EditorIcons"), TreeButtons.DUPLICATE_FOLDER, item == get_root())
|
|
item.add_button(2, get_theme_icon("Remove", "EditorIcons"), TreeButtons.DELETE, item == get_root())
|
|
item.set_meta("type", "FOLDER")
|
|
return item
|
|
|
|
|
|
#endregion
|
|
|
|
|
|
#region EDITING THE TREE
|
|
|
|
func set_variable_item_type(item:TreeItem, type:int) -> void:
|
|
item.set_meta('value_type', type)
|
|
item.set_button(1, 0, get_theme_icon(["Variant", "String", "float", "int", "bool"][type], "EditorIcons"))
|
|
|
|
|
|
func get_variable_item_default(item:TreeItem) -> Variant:
|
|
match int(item.get_meta('value_type', DialogicUtil.VarTypes.STRING)):
|
|
DialogicUtil.VarTypes.STRING:
|
|
return item.get_text(2)
|
|
DialogicUtil.VarTypes.FLOAT:
|
|
return item.get_range(2)
|
|
DialogicUtil.VarTypes.INT:
|
|
return int(item.get_range(2))
|
|
DialogicUtil.VarTypes.BOOL:
|
|
return item.is_checked(2)
|
|
return ""
|
|
|
|
|
|
func _on_button_clicked(item: TreeItem, column: int, id: int, mouse_button_index: int) -> void:
|
|
match id:
|
|
TreeButtons.ADD_FOLDER:
|
|
var new_item := add_folder_item("Folder", item)
|
|
new_item.select(0)
|
|
new_item.set_meta("new", true)
|
|
await get_tree().process_frame
|
|
edit_selected()
|
|
TreeButtons.ADD_VARIABLE:
|
|
var new_item := add_variable_item("Var", "", item)
|
|
new_item.select(0)
|
|
new_item.set_meta("new", true)
|
|
await get_tree().process_frame
|
|
edit_selected()
|
|
TreeButtons.DELETE:
|
|
item.free()
|
|
TreeButtons.DUPLICATE_FOLDER:
|
|
load_info({item.get_text(0)+"(copy)":get_info(item)}, item.get_parent(), true)
|
|
TreeButtons.CHANGE_TYPE:
|
|
%ChangeTypePopup.show()
|
|
%ChangeTypePopup.set_meta('item', item)
|
|
%ChangeTypePopup.position = get_local_mouse_position()+Vector2(-%ChangeTypePopup.size.x/2, 10)
|
|
for child in %ChangeTypePopup/HBox.get_children():
|
|
child.set_pressed_no_signal(false)
|
|
%ChangeTypePopup/HBox.get_child(int(item.get_meta('value_type', DialogicUtil.VarTypes.STRING)-1)).set_pressed_no_signal(true)
|
|
|
|
|
|
func _on_type_pressed(pressed:bool, type:int) -> void:
|
|
%ChangeTypePopup.hide()
|
|
var item := %ChangeTypePopup.get_meta('item')
|
|
adjust_variable_type(item, type, item.get_metadata(2))
|
|
|
|
|
|
func _on_item_edited() -> void:
|
|
var item := get_edited()
|
|
match item.get_meta('type'):
|
|
"VARIABLE":
|
|
match get_edited_column():
|
|
0:
|
|
if item.get_text(0).is_empty():
|
|
item.set_text(0, item.get_metadata(0))
|
|
|
|
else:
|
|
if item.get_text(0) != item.get_metadata(0):
|
|
item.set_metadata(0, item.get_text(0))
|
|
report_name_changes(item)
|
|
|
|
2:
|
|
item.set_metadata(2, get_variable_item_default(item))
|
|
"FOLDER":
|
|
report_name_changes(item)
|
|
|
|
|
|
func adjust_variable_type(item:TreeItem, type:int, prev_value:Variant) -> void:
|
|
set_variable_item_type(item, type)
|
|
match type:
|
|
DialogicUtil.VarTypes.STRING:
|
|
item.set_metadata(2, str(prev_value))
|
|
item.set_cell_mode(2, TreeItem.CELL_MODE_STRING)
|
|
item.set_text(2, str(prev_value))
|
|
DialogicUtil.VarTypes.FLOAT:
|
|
item.set_metadata(2, float(prev_value))
|
|
item.set_cell_mode(2, TreeItem.CELL_MODE_RANGE)
|
|
item.set_range_config(2, -9999, 9999, 0.001, false)
|
|
item.set_range(2, float(prev_value))
|
|
DialogicUtil.VarTypes.INT:
|
|
item.set_metadata(2, int(prev_value))
|
|
item.set_cell_mode(2, TreeItem.CELL_MODE_RANGE)
|
|
item.set_range_config(2, -9999, 9999, 1, false)
|
|
item.set_range(2, int(prev_value))
|
|
DialogicUtil.VarTypes.BOOL:
|
|
item.set_metadata(2, prev_value and true)
|
|
item.set_cell_mode(2, TreeItem.CELL_MODE_CHECK)
|
|
item.set_checked(2, prev_value and true)
|
|
|
|
|
|
func _input(event):
|
|
if !%ChangeTypePopup.visible:
|
|
return
|
|
if event is InputEventMouseButton and event.pressed:
|
|
if not %ChangeTypePopup.get_global_rect().has_point(get_global_mouse_position()):
|
|
%ChangeTypePopup.hide()
|
|
|
|
#endregion
|
|
|
|
|
|
func filter(filter_term:String, item:TreeItem = null) -> bool:
|
|
if item == null:
|
|
item = get_root()
|
|
|
|
var any := false
|
|
for child in item.get_children():
|
|
match child.get_meta('type'):
|
|
"VARIABLE":
|
|
child.visible = filter_term.is_empty() or filter_term.to_lower() in child.get_text(0).to_lower()
|
|
|
|
"FOLDER":
|
|
child.visible = filter(filter_term, child)
|
|
if child.visible:
|
|
any = true
|
|
return any
|
|
|
|
|
|
## Parses the tree and returns a dictionary representing it.
|
|
func get_info(item:TreeItem = null) -> Dictionary:
|
|
if item == null:
|
|
item = get_root()
|
|
if item == null:
|
|
return {}
|
|
|
|
var dict := {}
|
|
|
|
for child in item.get_children():
|
|
match child.get_meta('type'):
|
|
"VARIABLE":
|
|
dict[child.get_text(0)] = child.get_metadata(2)
|
|
"FOLDER":
|
|
dict[child.get_text(0)] = get_info(child)
|
|
|
|
return dict
|
|
|
|
|
|
#region DRAG AND DROP
|
|
################################################################################
|
|
|
|
func _get_drag_data(position:Vector2) -> Variant:
|
|
drop_mode_flags = DROP_MODE_INBETWEEN
|
|
var preview := Label.new()
|
|
preview.text = " "+get_selected().get_text(0)
|
|
preview.add_theme_stylebox_override('normal', get_theme_stylebox("Background", "EditorStyles"))
|
|
set_drag_preview(preview)
|
|
|
|
return get_selected()
|
|
|
|
|
|
func _can_drop_data(position:Vector2, data:Variant) -> bool:
|
|
return data is TreeItem
|
|
|
|
|
|
func _drop_data(position:Vector2, item:Variant) -> void:
|
|
var to_item := get_item_at_position(position)
|
|
|
|
if !to_item:
|
|
return
|
|
|
|
var drop_section := get_drop_section_at_position(position)
|
|
var parent: TreeItem = null
|
|
if (drop_section == 1 and to_item.get_meta('type') == "FOLDER") or to_item == get_root():
|
|
parent = to_item
|
|
else:
|
|
parent = to_item.get_parent()
|
|
|
|
## Test for inheritance-recursion
|
|
var test_item:= to_item
|
|
while true:
|
|
if test_item == item:
|
|
return
|
|
test_item = test_item.get_parent()
|
|
if test_item == get_root():
|
|
break
|
|
|
|
var new_item: TreeItem = null
|
|
match item.get_meta('type'):
|
|
"VARIABLE":
|
|
new_item = add_variable_item(item.get_text(0), item.get_metadata(2), parent)
|
|
new_item.set_meta('prev_path', get_item_path(item))
|
|
if item.get_meta("new", false):
|
|
new_item.set_meta("new", true)
|
|
"FOLDER":
|
|
new_item = add_folder_item(item.get_text(0), parent)
|
|
load_info(get_info(item), new_item)
|
|
if item.get_meta("new", false):
|
|
new_item.set_meta("new", true)
|
|
|
|
# If this was dropped on a variable (or the root node)
|
|
if to_item != parent:
|
|
if drop_section == -1:
|
|
new_item.move_before(to_item)
|
|
else:
|
|
new_item.move_after(to_item)
|
|
|
|
report_name_changes(new_item)
|
|
|
|
item.free()
|
|
|
|
#endregion
|
|
|
|
|
|
#region NAME CHANGES
|
|
################################################################################
|
|
|
|
func report_name_changes(item:TreeItem) -> void:
|
|
|
|
match item.get_meta('type'):
|
|
"VARIABLE":
|
|
if item.get_meta("new", false):
|
|
return
|
|
var new_path := get_item_path(item)
|
|
editor.variable_renamed(item.get_meta('prev_path'), new_path)
|
|
item.set_meta('prev_path', new_path)
|
|
"FOLDER":
|
|
for child in item.get_children():
|
|
report_name_changes(child)
|
|
|
|
|
|
func get_item_path(item:TreeItem) -> String:
|
|
if item.get_meta('type') == "VARIABLE":
|
|
var path := item.get_text(0)
|
|
while item.get_parent() != get_root():
|
|
item = item.get_parent()
|
|
path = item.get_text(0)+"."+path
|
|
return path
|
|
return ""
|
|
|
|
|
|
#endregion
|