feat: 支持选择节点

Author: zty012

crates/project_graph/src/stage.rs

@@ -5,6 +5,7 @@ pub mod render_context;
 
 use camera::Camera;
 use context::StageContext;
+use egui::{Vec2, vec2};
 
 use crate::stage::{
     elements::{Element, ElementTrait, entities::EntityTrait},
@@ -16,13 +17,15 @@ use crate::stage::{
 pub struct Stage {
     pub camera: Camera,
     pub context: StageContext,
+    pub selection: Vec<String>,
 }
 
 impl Stage {
     pub fn new() -> Self {
         Stage {
             camera: Camera::new(),
             context: StageContext::random(),
+            selection: Vec::new(),
         }
     }
 
@@ -46,20 +49,49 @@ impl Stage {
 
                 visible_count += 1;
 
+                let selected = self.selection.contains(&entity.id().to_string());
+
                 let screen_pos = self
                     .camera
-                    .world_to_screen(entity.position(), screen_center);
+                    .world_to_screen(entity.position(), screen_center)
+                    - if selected {
+                        vec2(4.0, 4.0) * self.camera.zoom()
+                    } else {
+                        Vec2::ZERO
+                    };
 
                 egui::Area::new(ui.make_persistent_id(entity.id()))
                     .order(egui::Order::Foreground)
                     .fixed_pos(screen_pos)
                     .show(ui.ctx(), |ui| {
-                        entity.ui(
-                            ui,
-                            &RenderContext {
-                                zoom: self.camera.zoom(),
-                            },
-                        );
+                        let response = egui::Frame::new()
+                            .inner_margin(4.0 * self.camera.zoom())
+                            .corner_radius(24.0 * self.camera.zoom())
+                            .stroke(if selected {
+                                egui::Stroke::new(
+                                    4.0 * self.camera.zoom(),
+                                    egui::Color32::LIGHT_BLUE,
+                                )
+                            } else {
+                                egui::Stroke::NONE
+                            })
+                            .show(ui, |ui| {
+                                entity.ui(
+                                    ui,
+                                    &mut RenderContext {
+                                        zoom: self.camera.zoom(),
+                                        selected: selected,
+                                    },
+                                );
+                            })
+                            .response
+                            .interact(egui::Sense::click());
+
+                        if response.clicked() {
+                            log::info!("Entity {} clicked", entity.id());
+                            self.selection.clear();
+                            self.selection.push(entity.id().to_string());
+                        }
                     });
             }
 

crates/project_graph/src/stage/elements.rs

@@ -13,7 +13,7 @@ pub trait ElementTrait {
     /// 原则上应该是一个 nanoid
     fn id(&self) -> &str;
     /// 渲染函数
-    fn ui(&self, ui: &mut egui::Ui, rc: &RenderContext);
+    fn ui(&self, ui: &mut egui::Ui, rc: &mut RenderContext);
 }
 
 #[enum_dispatch(ElementTrait)]

crates/project_graph/src/stage/elements/entities/text.rs

@@ -31,9 +31,13 @@ impl ElementTrait for Text {
     fn id(&self) -> &str {
         &self.id
     }
-    fn ui(&self, ui: &mut egui::Ui, rc: &RenderContext) {
+    fn ui(&self, ui: &mut egui::Ui, rc: &mut RenderContext) {
         let response = egui::Frame::new()
-            .fill(egui::Color32::from_rgba_unmultiplied(255, 255, 255, 25))
+            .fill(if rc.selected {
+                egui::Color32::from_rgba_unmultiplied(150, 150, 255, 75)
+            } else {
+                egui::Color32::from_rgba_unmultiplied(255, 255, 255, 25)
+            })
             .corner_radius(16.0 * rc.zoom)
             .inner_margin(egui::Margin::symmetric(12, 10) * rc.zoom)
             .show(ui, |ui| {
@@ -42,7 +46,8 @@ impl ElementTrait for Text {
                         .wrap_mode(egui::TextWrapMode::Extend),
                 );
             })
-            .response;
+            .response
+            .interact(egui::Sense::click());
     }
 }
 impl EntityTrait for Text {

crates/project_graph/src/stage/render_context.rs

@@ -1,5 +1,6 @@
 pub struct RenderContext {
     pub zoom: f32,
+    pub selected: bool,
 }
 
 impl RenderContext {}