From 631d722674588bfca4f833c3534872b0c3805472 Mon Sep 17 00:00:00 2001
From: Mansur Isaev <737dab2f169a@mail.ru>
Date: Mon, 2 Oct 2023 09:40:21 +0400
Subject: [PATCH] Add `set_slot_custom_icon` and `get_slot_custom_icon` to
GraphNode
---
doc/classes/GraphNode.xml | 30 ++++++++++++++++++++++++
scene/gui/graph_node.cpp | 48 +++++++++++++++++++++++++++++++++++++++
scene/gui/graph_node.h | 6 +++++
3 files changed, 84 insertions(+)
diff --git a/doc/classes/GraphNode.xml b/doc/classes/GraphNode.xml
index 9e1392567a2..5bb3a25158e 100644
--- a/doc/classes/GraphNode.xml
+++ b/doc/classes/GraphNode.xml
@@ -116,6 +116,20 @@
Returns the right (output) [Color] of the slot with the given [param slot_index].
+
+
+
+
+ Returns the left (input) custom [Texture2D] of the slot with the given [param slot_index].
+
+
+
+
+
+
+ Returns the right (output) custom [Texture2D] of the slot with the given [param slot_index].
+
+
@@ -195,6 +209,22 @@
Sets the [Color] of the right (output) side of the slot with the given [param slot_index] to [param color].
+
+
+
+
+
+ Sets the custom [Texture2D] of the left (input) side of the slot with the given [param slot_index] to [param custom_icon].
+
+
+
+
+
+
+
+ Sets the custom [Texture2D] of the right (output) side of the slot with the given [param slot_index] to [param custom_icon].
+
+
diff --git a/scene/gui/graph_node.cpp b/scene/gui/graph_node.cpp
index 3b1c1a153fe..1b960a9b62a 100644
--- a/scene/gui/graph_node.cpp
+++ b/scene/gui/graph_node.cpp
@@ -482,6 +482,27 @@ Color GraphNode::get_slot_color_left(int p_slot_index) const {
return slot_table[p_slot_index].color_left;
}
+void GraphNode::set_slot_custom_icon_left(int p_slot_index, const Ref &p_custom_icon) {
+ ERR_FAIL_COND_MSG(!slot_table.has(p_slot_index), vformat("Cannot set custom_port_icon_left for the slot with index '%d' because it hasn't been enabled.", p_slot_index));
+
+ if (slot_table[p_slot_index].custom_port_icon_left == p_custom_icon) {
+ return;
+ }
+
+ slot_table[p_slot_index].custom_port_icon_left = p_custom_icon;
+ queue_redraw();
+ port_pos_dirty = true;
+
+ emit_signal(SNAME("slot_updated"), p_slot_index);
+}
+
+Ref GraphNode::get_slot_custom_icon_left(int p_slot_index) const {
+ if (!slot_table.has(p_slot_index)) {
+ return Ref();
+ }
+ return slot_table[p_slot_index].custom_port_icon_left;
+}
+
bool GraphNode::is_slot_enabled_right(int p_slot_index) const {
if (!slot_table.has(p_slot_index)) {
return false;
@@ -545,6 +566,27 @@ Color GraphNode::get_slot_color_right(int p_slot_index) const {
return slot_table[p_slot_index].color_right;
}
+void GraphNode::set_slot_custom_icon_right(int p_slot_index, const Ref &p_custom_icon) {
+ ERR_FAIL_COND_MSG(!slot_table.has(p_slot_index), vformat("Cannot set custom_port_icon_right for the slot with index '%d' because it hasn't been enabled.", p_slot_index));
+
+ if (slot_table[p_slot_index].custom_port_icon_right == p_custom_icon) {
+ return;
+ }
+
+ slot_table[p_slot_index].custom_port_icon_right = p_custom_icon;
+ queue_redraw();
+ port_pos_dirty = true;
+
+ emit_signal(SNAME("slot_updated"), p_slot_index);
+}
+
+Ref GraphNode::get_slot_custom_icon_right(int p_slot_index) const {
+ if (!slot_table.has(p_slot_index)) {
+ return Ref();
+ }
+ return slot_table[p_slot_index].custom_port_icon_right;
+}
+
bool GraphNode::is_slot_draw_stylebox(int p_slot_index) const {
if (!slot_table.has(p_slot_index)) {
return false;
@@ -797,6 +839,9 @@ void GraphNode::_bind_methods() {
ClassDB::bind_method(D_METHOD("set_slot_color_left", "slot_index", "color"), &GraphNode::set_slot_color_left);
ClassDB::bind_method(D_METHOD("get_slot_color_left", "slot_index"), &GraphNode::get_slot_color_left);
+ ClassDB::bind_method(D_METHOD("set_slot_custom_icon_left", "slot_index", "custom_icon"), &GraphNode::set_slot_custom_icon_left);
+ ClassDB::bind_method(D_METHOD("get_slot_custom_icon_left", "slot_index"), &GraphNode::get_slot_custom_icon_left);
+
ClassDB::bind_method(D_METHOD("is_slot_enabled_right", "slot_index"), &GraphNode::is_slot_enabled_right);
ClassDB::bind_method(D_METHOD("set_slot_enabled_right", "slot_index", "enable"), &GraphNode::set_slot_enabled_right);
@@ -806,6 +851,9 @@ void GraphNode::_bind_methods() {
ClassDB::bind_method(D_METHOD("set_slot_color_right", "slot_index", "color"), &GraphNode::set_slot_color_right);
ClassDB::bind_method(D_METHOD("get_slot_color_right", "slot_index"), &GraphNode::get_slot_color_right);
+ ClassDB::bind_method(D_METHOD("set_slot_custom_icon_right", "slot_index", "custom_icon"), &GraphNode::set_slot_custom_icon_right);
+ ClassDB::bind_method(D_METHOD("get_slot_custom_icon_right", "slot_index"), &GraphNode::get_slot_custom_icon_right);
+
ClassDB::bind_method(D_METHOD("is_slot_draw_stylebox", "slot_index"), &GraphNode::is_slot_draw_stylebox);
ClassDB::bind_method(D_METHOD("set_slot_draw_stylebox", "slot_index", "enable"), &GraphNode::set_slot_draw_stylebox);
diff --git a/scene/gui/graph_node.h b/scene/gui/graph_node.h
index 04ca9e7cb4e..a0610b37fbe 100644
--- a/scene/gui/graph_node.h
+++ b/scene/gui/graph_node.h
@@ -129,6 +129,9 @@ public:
void set_slot_color_left(int p_slot_index, const Color &p_color);
Color get_slot_color_left(int p_slot_index) const;
+ void set_slot_custom_icon_left(int p_slot_index, const Ref &p_custom_icon);
+ Ref get_slot_custom_icon_left(int p_slot_index) const;
+
bool is_slot_enabled_right(int p_slot_index) const;
void set_slot_enabled_right(int p_slot_index, bool p_enable);
@@ -138,6 +141,9 @@ public:
void set_slot_color_right(int p_slot_index, const Color &p_color);
Color get_slot_color_right(int p_slot_index) const;
+ void set_slot_custom_icon_right(int p_slot_index, const Ref &p_custom_icon);
+ Ref get_slot_custom_icon_right(int p_slot_index) const;
+
bool is_slot_draw_stylebox(int p_slot_index) const;
void set_slot_draw_stylebox(int p_slot_index, bool p_enable);