Improve usage of StringNames in GraphEdit

This commit is contained in:
kobewi 2022-08-12 13:06:08 +02:00
parent 6f1d4fd887
commit 5490c22ffb
3 changed files with 34 additions and 65 deletions

View file

@ -291,7 +291,7 @@
</description> </description>
</signal> </signal>
<signal name="connection_drag_started"> <signal name="connection_drag_started">
<param index="0" name="from_node" type="String" /> <param index="0" name="from_node" type="StringName" />
<param index="1" name="from_port" type="int" /> <param index="1" name="from_port" type="int" />
<param index="2" name="is_output" type="bool" /> <param index="2" name="is_output" type="bool" />
<description> <description>

View file

@ -623,7 +623,7 @@ void GraphEdit::_top_layer_input(const Ref<InputEvent> &p_ev) {
//check disconnect //check disconnect
for (const Connection &E : connections) { for (const Connection &E : connections) {
if (E.from == gn->get_name() && E.from_port == j) { if (E.from == gn->get_name() && E.from_port == j) {
Node *to = get_node(String(E.to)); Node *to = get_node(NodePath(E.to));
if (Object::cast_to<GraphNode>(to)) { if (Object::cast_to<GraphNode>(to)) {
connecting_from = E.to; connecting_from = E.to;
connecting_index = E.to_port; connecting_index = E.to_port;
@ -637,7 +637,7 @@ void GraphEdit::_top_layer_input(const Ref<InputEvent> &p_ev) {
just_disconnected = true; just_disconnected = true;
emit_signal(SNAME("disconnection_request"), E.from, E.from_port, E.to, E.to_port); emit_signal(SNAME("disconnection_request"), E.from, E.from_port, E.to, E.to_port);
to = get_node(String(connecting_from)); //maybe it was erased to = get_node(NodePath(connecting_from)); // Maybe it was erased.
if (Object::cast_to<GraphNode>(to)) { if (Object::cast_to<GraphNode>(to)) {
connecting = true; connecting = true;
emit_signal(SNAME("connection_drag_started"), connecting_from, connecting_index, false); emit_signal(SNAME("connection_drag_started"), connecting_from, connecting_index, false);
@ -673,10 +673,10 @@ void GraphEdit::_top_layer_input(const Ref<InputEvent> &p_ev) {
if (is_in_input_hotzone(gn, j, click_pos, port_size)) { if (is_in_input_hotzone(gn, j, click_pos, port_size)) {
if (right_disconnects || valid_right_disconnect_types.has(gn->get_connection_input_type(j))) { if (right_disconnects || valid_right_disconnect_types.has(gn->get_connection_input_type(j))) {
//check disconnect // Check disconnect.
for (const Connection &E : connections) { for (const Connection &E : connections) {
if (E.to == gn->get_name() && E.to_port == j) { if (E.to == gn->get_name() && E.to_port == j) {
Node *fr = get_node(String(E.from)); Node *fr = get_node(NodePath(E.from));
if (Object::cast_to<GraphNode>(fr)) { if (Object::cast_to<GraphNode>(fr)) {
connecting_from = E.from; connecting_from = E.from;
connecting_index = E.from_port; connecting_index = E.from_port;
@ -689,7 +689,7 @@ void GraphEdit::_top_layer_input(const Ref<InputEvent> &p_ev) {
if (connecting_type >= 0) { if (connecting_type >= 0) {
emit_signal(SNAME("disconnection_request"), E.from, E.from_port, E.to, E.to_port); emit_signal(SNAME("disconnection_request"), E.from, E.from_port, E.to, E.to_port);
fr = get_node(String(connecting_from)); //maybe it was erased fr = get_node(NodePath(connecting_from)); // Maybe it was erased.
if (Object::cast_to<GraphNode>(fr)) { if (Object::cast_to<GraphNode>(fr)) {
connecting = true; connecting = true;
emit_signal(SNAME("connection_drag_started"), connecting_from, connecting_index, true); emit_signal(SNAME("connection_drag_started"), connecting_from, connecting_index, true);
@ -780,26 +780,16 @@ void GraphEdit::_top_layer_input(const Ref<InputEvent> &p_ev) {
if (mb.is_valid() && mb->get_button_index() == MouseButton::LEFT && !mb->is_pressed()) { if (mb.is_valid() && mb->get_button_index() == MouseButton::LEFT && !mb->is_pressed()) {
if (connecting_valid) { if (connecting_valid) {
if (connecting && connecting_target) { if (connecting && connecting_target) {
String from = connecting_from; if (connecting_out) {
int from_port = connecting_index; emit_signal(SNAME("connection_request"), connecting_from, connecting_index, connecting_target_to, connecting_target_index);
String to = connecting_target_to;
int to_port = connecting_target_index;
if (!connecting_out) {
SWAP(from, to);
SWAP(from_port, to_port);
}
emit_signal(SNAME("connection_request"), from, from_port, to, to_port);
} else if (!just_disconnected) {
String from = connecting_from;
int from_port = connecting_index;
Vector2 ofs = mb->get_position();
if (!connecting_out) {
emit_signal(SNAME("connection_from_empty"), from, from_port, ofs);
} else { } else {
emit_signal(SNAME("connection_to_empty"), from, from_port, ofs); emit_signal(SNAME("connection_request"), connecting_target_to, connecting_target_index, connecting_from, connecting_index);
}
} else if (!just_disconnected) {
if (connecting_out) {
emit_signal(SNAME("connection_to_empty"), connecting_from, connecting_index, mb->get_position());
} else {
emit_signal(SNAME("connection_from_empty"), connecting_from, connecting_index, mb->get_position());
} }
} }
} }
@ -935,17 +925,12 @@ void GraphEdit::_draw_connection_line(CanvasItem *p_where, const Vector2 &p_from
void GraphEdit::_connections_layer_draw() { void GraphEdit::_connections_layer_draw() {
Color activity_color = get_theme_color(SNAME("activity")); Color activity_color = get_theme_color(SNAME("activity"));
//draw connections // Draw connections.
List<List<Connection>::Element *> to_erase; List<List<Connection>::Element *> to_erase;
for (List<Connection>::Element *E = connections.front(); E; E = E->next()) { for (List<Connection>::Element *E = connections.front(); E; E = E->next()) {
NodePath fromnp(E->get().from); const Connection &c = E->get();
Node *from = get_node(fromnp);
if (!from) {
to_erase.push_back(E);
continue;
}
Node *from = get_node(NodePath(c.from));
GraphNode *gfrom = Object::cast_to<GraphNode>(from); GraphNode *gfrom = Object::cast_to<GraphNode>(from);
if (!gfrom) { if (!gfrom) {
@ -953,13 +938,7 @@ void GraphEdit::_connections_layer_draw() {
continue; continue;
} }
NodePath tonp(E->get().to); Node *to = get_node(NodePath(c.to));
Node *to = get_node(tonp);
if (!to) {
to_erase.push_back(E);
continue;
}
GraphNode *gto = Object::cast_to<GraphNode>(to); GraphNode *gto = Object::cast_to<GraphNode>(to);
if (!gto) { if (!gto) {
@ -967,21 +946,20 @@ void GraphEdit::_connections_layer_draw() {
continue; continue;
} }
Vector2 frompos = gfrom->get_connection_output_position(E->get().from_port) + gfrom->get_position_offset() * zoom; Vector2 frompos = gfrom->get_connection_output_position(c.from_port) + gfrom->get_position_offset() * zoom;
Color color = gfrom->get_connection_output_color(E->get().from_port); Color color = gfrom->get_connection_output_color(c.from_port);
Vector2 topos = gto->get_connection_input_position(E->get().to_port) + gto->get_position_offset() * zoom; Vector2 topos = gto->get_connection_input_position(c.to_port) + gto->get_position_offset() * zoom;
Color tocolor = gto->get_connection_input_color(E->get().to_port); Color tocolor = gto->get_connection_input_color(c.to_port);
if (E->get().activity > 0) { if (c.activity > 0) {
color = color.lerp(activity_color, E->get().activity); color = color.lerp(activity_color, c.activity);
tocolor = tocolor.lerp(activity_color, E->get().activity); tocolor = tocolor.lerp(activity_color, c.activity);
} }
_draw_connection_line(connections_layer, frompos, topos, color, tocolor, lines_thickness, zoom); _draw_connection_line(connections_layer, frompos, topos, color, tocolor, lines_thickness, zoom);
} }
while (to_erase.size()) { for (List<Connection>::Element *&E : to_erase) {
connections.erase(to_erase.front()->get()); connections.erase(E);
to_erase.pop_front();
} }
} }
@ -989,7 +967,7 @@ void GraphEdit::_top_layer_draw() {
_update_scroll(); _update_scroll();
if (connecting) { if (connecting) {
Node *fromn = get_node(connecting_from); Node *fromn = get_node(NodePath(connecting_from));
ERR_FAIL_COND(!fromn); ERR_FAIL_COND(!fromn);
GraphNode *from = Object::cast_to<GraphNode>(fromn); GraphNode *from = Object::cast_to<GraphNode>(fromn);
ERR_FAIL_COND(!from); ERR_FAIL_COND(!from);
@ -1087,22 +1065,13 @@ void GraphEdit::_minimap_draw() {
// Draw node connections. // Draw node connections.
Color activity_color = get_theme_color(SNAME("activity")); Color activity_color = get_theme_color(SNAME("activity"));
for (const Connection &E : connections) { for (const Connection &E : connections) {
NodePath fromnp(E.from); Node *from = get_node(NodePath(E.from));
Node *from = get_node(fromnp);
if (!from) {
continue;
}
GraphNode *gfrom = Object::cast_to<GraphNode>(from); GraphNode *gfrom = Object::cast_to<GraphNode>(from);
if (!gfrom) { if (!gfrom) {
continue; continue;
} }
NodePath tonp(E.to); Node *to = get_node(NodePath(E.to));
Node *to = get_node(tonp);
if (!to) {
continue;
}
GraphNode *gto = Object::cast_to<GraphNode>(to); GraphNode *gto = Object::cast_to<GraphNode>(to);
if (!gto) { if (!gto) {
continue; continue;
@ -2428,7 +2397,7 @@ void GraphEdit::_bind_methods() {
ADD_SIGNAL(MethodInfo("begin_node_move")); ADD_SIGNAL(MethodInfo("begin_node_move"));
ADD_SIGNAL(MethodInfo("end_node_move")); ADD_SIGNAL(MethodInfo("end_node_move"));
ADD_SIGNAL(MethodInfo("scroll_offset_changed", PropertyInfo(Variant::VECTOR2, "offset"))); ADD_SIGNAL(MethodInfo("scroll_offset_changed", PropertyInfo(Variant::VECTOR2, "offset")));
ADD_SIGNAL(MethodInfo("connection_drag_started", PropertyInfo(Variant::STRING, "from_node"), PropertyInfo(Variant::INT, "from_port"), PropertyInfo(Variant::BOOL, "is_output"))); ADD_SIGNAL(MethodInfo("connection_drag_started", PropertyInfo(Variant::STRING_NAME, "from_node"), PropertyInfo(Variant::INT, "from_port"), PropertyInfo(Variant::BOOL, "is_output")));
ADD_SIGNAL(MethodInfo("connection_drag_ended")); ADD_SIGNAL(MethodInfo("connection_drag_ended"));
BIND_ENUM_CONSTANT(SCROLL_ZOOMS); BIND_ENUM_CONSTANT(SCROLL_ZOOMS);

View file

@ -136,14 +136,14 @@ private:
bool arrange_nodes_button_hidden = false; bool arrange_nodes_button_hidden = false;
bool connecting = false; bool connecting = false;
String connecting_from; StringName connecting_from;
bool connecting_out = false; bool connecting_out = false;
int connecting_index = 0; int connecting_index = 0;
int connecting_type = 0; int connecting_type = 0;
Color connecting_color; Color connecting_color;
bool connecting_target = false; bool connecting_target = false;
Vector2 connecting_to; Vector2 connecting_to;
String connecting_target_to; StringName connecting_target_to;
int connecting_target_index = 0; int connecting_target_index = 0;
bool just_disconnected = false; bool just_disconnected = false;
bool connecting_valid = false; bool connecting_valid = false;