elias/egui_node_graph
1
0
Fork 0
mirror of https://github.com/eliasstepanik/egui_node_graph.git synced 2026-01-11 05:48:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Snap Connections and Ports

Browse Source
This commit is contained in:
KOKI 2022-08-01 16:54:00 +09:00
parent 007c3e5480
commit b97a567eb7
1 changed files with 42 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
egui_node_graph/src/editor_ui.rs
View File

@ -8,6 +8,7 @@ use egui::epaint::{CubicBezierShape, RectShape};
use egui::*;
pub type PortLocations = std::collections::HashMap<AnyParameterId, Pos2>;
const DISTANCE_TO_CONNECT: f32 = 10.0;
/// Nodes communicate certain events to the parent graph when drawn. There is
/// one special `User` variant which can be used by users as the return value
@ -176,8 +177,37 @@ where
let connection_color = port_type.data_type_color(&self.user_state);
let start_pos = port_locations[locator];
let (src_pos, dst_pos) = match locator {
AnyParameterId::Output(_) => (start_pos, cursor_pos),
AnyParameterId::Input(_) => (cursor_pos, start_pos),
AnyParameterId::Output(_) => {
// Find a port to connect to
let port = self.graph.inputs.iter().find_map(|(input_id, _)| {
let port_pos = port_locations[&AnyParameterId::Input(input_id)];
if port_pos.distance(cursor_pos) < DISTANCE_TO_CONNECT {
Some(port_pos)
} else {
None
}
});
if let Some(port_pos) = port {
(start_pos, port_pos)
} else {
(start_pos, cursor_pos)
}
}
AnyParameterId::Input(_) => {
let port = self.graph.outputs.iter().find_map(|(output_id, _)| {
let port_pos = port_locations[&AnyParameterId::Output(output_id)];
if port_pos.distance(cursor_pos) < DISTANCE_TO_CONNECT {
Some(port_pos)
} else {
None
}
});
if let Some(port_pos) = port {
(port_pos, start_pos)
} else {
(cursor_pos, start_pos)
}
}
};
draw_connection(ui.painter(), src_pos, dst_pos, connection_color);
}
@ -464,7 +494,15 @@ where
};
let resp = ui.allocate_rect(port_rect, sense);
let port_color = if resp.hovered() {
// Check if the distance between the port and the mouse is the distance to connect
let close_enough = if let Some(pointer_pos) = ui.ctx().pointer_hover_pos() {
port_rect.center().distance(pointer_pos) < DISTANCE_TO_CONNECT
} else {
false
};
let port_color = if close_enough {
Color32::WHITE
} else {
port_type.data_type_color(user_state)
@ -491,7 +529,7 @@ where
if origin_node != node_id {
// Don't allow self-loops
if graph.any_param_type(origin_param).unwrap() == port_type
&& resp.hovered()
&& close_enough
&& ui.input().pointer.any_released()
{
match (param_id, origin_param) {