Merge pull request #3 from ThePat02/feature-editor-ui

Add new editor UI for creating FDG Nodes

Author: ThePat02

README.md

@@ -4,6 +4,7 @@ is a simple implementation of a [Force-Directed Graph](https://en.wikipedia.org/
 #### Features
 - 2D implementation of a Force-Directed Graph
 - Flexible and easy to use Graph Nodes that can be extended and customized
+- Editor UI for adding new nodes to your graph
 - Real-time graph simulation in the editor (optional)
 
 #### Supported Versions

addons/force_directed_graph/plugin.gd

@@ -2,6 +2,9 @@
 extends EditorPlugin
 
 
+var _graph_ui = preload("res://addons/force_directed_graph/ui/graph_ui.tscn")
+
+
 # Icons
 const icon_force_directed_graph = preload("res://addons/force_directed_graph/icons/ForceDirectedGraph.svg")
 const icon_fdg_node = preload("res://addons/force_directed_graph/icons/FDGNode.svg")
@@ -19,9 +22,40 @@ func _enter_tree():
 	add_custom_type("FDGNode", "Node2D", fdg_node, icon_fdg_node)
 	add_custom_type("FDGSpring", "Node2D", fdg_spring, icon_fdg_spring)
 
+	# Add UI
+	_graph_ui = _graph_ui.instantiate()
+	add_control_to_container(EditorPlugin.CONTAINER_CANVAS_EDITOR_SIDE_LEFT, _graph_ui)
+	_graph_ui.visible = false
+
+	# Connect editor signals 
+	get_editor_interface().get_selection().selection_changed.connect(_on_selection_changed)
+
 
 func _exit_tree():
 	# Remove custom types
 	remove_custom_type("ForceDirectedGraph")
 	remove_custom_type("FDGNode")
 	remove_custom_type("FDGSpring")
+
+	# Remove UI
+	remove_control_from_container(EditorPlugin.CONTAINER_CANVAS_EDITOR_SIDE_LEFT, _graph_ui)
+
+
+func _on_selection_changed() -> void:
+
+	# Get current selection
+	var selection = get_editor_interface().get_selection().get_selected_nodes()
+
+	if selection.size() == 0:
+		_graph_ui.visible = false
+		return
+	
+	
+	_graph_ui.set_current_selection(selection[0])
+
+	for node in selection:
+		if node is ForceDirectedGraph or node is FDGNode or node is FDGSpring:
+			_graph_ui.visible = true
+			return
+
+	_graph_ui.visible = false

addons/force_directed_graph/ui/graph_ui.gd

@@ -0,0 +1,37 @@
+@tool
+extends MarginContainer
+
+
+var current_selection
+
+
+func _ready():
+	# Connect buttons
+	%ButtonGraph.connect("pressed", _on_button_pressed.bind(ForceDirectedGraph, "ForceDirectedGraph"))
+	%ButtonNode.connect("pressed", _on_button_pressed.bind(FDGNode, "FDGNode"))
+	%ButtonSpring.connect("pressed", _on_button_pressed.bind(FDGSpring, "FDGSpring"))
+
+
+func set_current_selection(selection):
+	current_selection = selection
+
+
+func _on_button_pressed(type, name: String):
+	var new_node = type.new()
+
+	# Check if name already exists
+	var already_exists = false
+	var count = 0
+	for child in current_selection.get_children():
+		if child.name.begins_with(name):
+			count += 1
+			already_exists = true
+
+	if not already_exists:
+		new_node.name = name
+	else:
+		new_node.name = name + str(count + 1)
+
+
+	current_selection.add_child(new_node)
+	new_node.set_owner(current_selection.get_tree().edited_scene_root)

addons/force_directed_graph/ui/graph_ui.tscn

@@ -0,0 +1,64 @@
+[gd_scene load_steps=11 format=3 uid="uid://csom57ywvasf5"]
+
+[ext_resource type="Texture2D" uid="uid://3i6spt2g10rr" path="res://addons/force_directed_graph/icons/ForceDirectedGraph.svg" id="1_p4yg8"]
+[ext_resource type="Script" path="res://addons/force_directed_graph/ui/graph_ui.gd" id="1_t3d8u"]
+[ext_resource type="Texture2D" uid="uid://dvsr10wvptj0j" path="res://addons/force_directed_graph/icons/FDGNode.svg" id="2_mtvv5"]
+[ext_resource type="Texture2D" uid="uid://dydh7pbp8qiqm" path="res://addons/force_directed_graph/icons/FDGSpring.svg" id="3_7jlyv"]
+
+[sub_resource type="InputEventKey" id="InputEventKey_gf45o"]
+device = -1
+alt_pressed = true
+keycode = 71
+
+[sub_resource type="Shortcut" id="Shortcut_m2piv"]
+events = [SubResource("InputEventKey_gf45o")]
+
+[sub_resource type="InputEventKey" id="InputEventKey_odhcc"]
+device = -1
+alt_pressed = true
+keycode = 78
+
+[sub_resource type="Shortcut" id="Shortcut_tdywb"]
+events = [SubResource("InputEventKey_odhcc")]
+
+[sub_resource type="InputEventKey" id="InputEventKey_1tft4"]
+device = -1
+alt_pressed = true
+keycode = 83
+
+[sub_resource type="Shortcut" id="Shortcut_6inb6"]
+events = [SubResource("InputEventKey_1tft4")]
+
+[node name="GraphUI" type="MarginContainer"]
+custom_minimum_size = Vector2(100, 0)
+anchors_preset = 9
+anchor_bottom = 1.0
+grow_vertical = 2
+theme_override_constants/margin_left = 5
+theme_override_constants/margin_top = 5
+theme_override_constants/margin_bottom = 5
+script = ExtResource("1_t3d8u")
+
+[node name="VBoxContainer" type="VBoxContainer" parent="."]
+layout_mode = 2
+
+[node name="ButtonGraph" type="Button" parent="VBoxContainer"]
+unique_name_in_owner = true
+layout_mode = 2
+shortcut = SubResource("Shortcut_m2piv")
+text = "Graph"
+icon = ExtResource("1_p4yg8")
+
+[node name="ButtonNode" type="Button" parent="VBoxContainer"]
+unique_name_in_owner = true
+layout_mode = 2
+shortcut = SubResource("Shortcut_tdywb")
+text = "Node"
+icon = ExtResource("2_mtvv5")
+
+[node name="ButtonSpring" type="Button" parent="VBoxContainer"]
+unique_name_in_owner = true
+layout_mode = 2
+shortcut = SubResource("Shortcut_6inb6")
+text = "Spring"
+icon = ExtResource("3_7jlyv")