feat: 优化性能

Author: zty012

crates/project_graph/src/stage.rs

@@ -3,29 +3,24 @@ pub mod context;
 pub mod elements;
 pub mod render_context;
 
+use crate::stage::{elements::ElementTrait, render_context::RenderContext};
 use camera::Camera;
 use context::StageContext;
-use egui::{Vec2, vec2};
-
-use crate::stage::{
-    elements::{Element, ElementTrait, entities::EntityTrait},
-    render_context::RenderContext,
-};
 
 /// egui 和画布之间的桥梁
 /// 负责坐标系转换、事件处理等
 pub struct Stage {
     pub camera: Camera,
     pub context: StageContext,
-    pub selection: Vec<String>,
+    pub selection: std::collections::HashSet<String>,
 }
 
 impl Stage {
     pub fn new() -> Self {
         Stage {
             camera: Camera::new(),
             context: StageContext::random(),
-            selection: Vec::new(),
+            selection: std::collections::HashSet::new(),
         }
     }
 
@@ -38,61 +33,46 @@ impl Stage {
             let screen_center = rect.center();
             let mut visible_count = 0;
 
+            // 获取视野范围（世界坐标）
+            let world_min = self.camera.screen_to_world(rect.min, screen_center);
+            let world_max = self.camera.screen_to_world(rect.max, screen_center);
+            let world_viewport = egui::Rect::from_two_pos(world_min, world_max);
+
             for element in self.context.elements().values() {
-                let entity = match element {
-                    Element::Entity(e) => e,
-                };
-                // 剔除
-                if !rect.contains(entity.position()) {
+                // 剔除：检查元素的包围盒是否与视野范围相交
+                if !world_viewport.intersects(element.world_rect().expand(20.0)) {
                     continue;
                 }
 
                 visible_count += 1;
 
-                let selected = self.selection.contains(&entity.id().to_string());
-
+                let selected = self.selection.contains(element.id());
                 let screen_pos = self
                     .camera
-                    .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::Background)
-                    .fixed_pos(screen_pos)
-                    .show(ui.ctx(), |ui| {
-                        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());
-                        }
-                    });
+                    .world_to_screen(element.world_rect().center(), screen_center);
+
+                // 在屏幕坐标系下进行局部渲染，不再使用 egui::Area
+                let mut rc = RenderContext {
+                    zoom: self.camera.zoom(),
+                    selected,
+                    clicked: false,
+                };
+
+                let element_size = element.world_rect().size() * self.camera.zoom();
+                let element_rect = egui::Rect::from_center_size(screen_pos, element_size);
+
+                // 显式限制裁剪区域，避免过度渲染
+                ui.scope_builder(egui::UiBuilder::new().max_rect(element_rect), |ui| {
+                    element.ui(ui, &mut rc);
+                });
+
+                if rc.clicked {
+                    log::info!("Entity {} clicked", element.id());
+                    if !ui.input(|i| i.modifiers.shift) {
+                        self.selection.clear();
+                    }
+                    self.selection.insert(element.id().to_string());
+                }
             }
 
             painter.text(
@@ -104,19 +84,38 @@ impl Stage {
             );
         });
 
-        let scroll_delta = ui.input(|i| i.smooth_scroll_delta);
-        if scroll_delta.y != 0.0 {
+        // 处理手势和滚动
+        let zoom_delta = ui.input(|i| i.zoom_delta());
+        if zoom_delta != 1.0 {
             if let Some(mouse_pos) = ui.input(|i| i.pointer.hover_pos()) {
-                let zoom_factor = (1.0 + scroll_delta.y * 0.01).clamp(0.9, 1.1);
                 let old_zoom = self.camera.zoom();
-                self.camera.zoom_by(zoom_factor);
+                self.camera.zoom_by(zoom_delta);
 
                 let offset = mouse_pos - rect.center();
                 let delta = offset * (self.camera.zoom() / old_zoom - 1.0);
                 self.camera.pan_by(delta);
             }
         }
 
+        let scroll_delta = ui.input(|i| i.smooth_scroll_delta);
+        if scroll_delta != egui::Vec2::ZERO {
+            if ui.input(|i| i.modifiers.command || i.modifiers.ctrl) {
+                // Ctrl + 滚动 = 缩放
+                if let Some(mouse_pos) = ui.input(|i| i.pointer.hover_pos()) {
+                    let zoom_factor = (1.0 + scroll_delta.y * 0.01).clamp(0.9, 1.1);
+                    let old_zoom = self.camera.zoom();
+                    self.camera.zoom_by(zoom_factor);
+
+                    let offset = mouse_pos - rect.center();
+                    let delta = offset * (self.camera.zoom() / old_zoom - 1.0);
+                    self.camera.pan_by(delta);
+                }
+            } else {
+                // 普通滚动 = 平移
+                self.camera.pan_by(-scroll_delta);
+            }
+        }
+
         // 中键拖拽平移
         if response.dragged_by(egui::PointerButton::Middle) {
             let drag_delta = ui.input(|i| i.pointer.delta());

crates/project_graph/src/stage/elements.rs

@@ -14,6 +14,8 @@ pub trait ElementTrait {
     fn id(&self) -> &str;
     /// 渲染函数
     fn ui(&self, ui: &mut egui::Ui, rc: &mut RenderContext);
+    /// 获取包围盒（世界坐标）
+    fn world_rect(&self) -> egui::Rect;
 }
 
 #[enum_dispatch(ElementTrait)]

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

@@ -32,22 +32,43 @@ impl ElementTrait for Text {
         &self.id
     }
     fn ui(&self, ui: &mut egui::Ui, rc: &mut RenderContext) {
-        let response = egui::Frame::new()
+        // 预计算，避免在 egui 生命周期内做重算
+        let zoom = rc.zoom;
+        let font_id = egui::FontId::proportional(16.0 * zoom);
+
+        // 使用更加紧凑的布局方式
+        let frame = egui::Frame::new()
             .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)
+            .corner_radius(16.0 * zoom)
+            .inner_margin(egui::Margin::symmetric(12, 10) * zoom)
+            .stroke(if rc.selected {
+                egui::Stroke::new(2.0 * zoom, egui::Color32::LIGHT_BLUE)
+            } else {
+                egui::Stroke::NONE
+            });
+
+        let response = frame
             .show(ui, |ui| {
+                // 极简渲染，避免 Label 的复杂逻辑
                 ui.add(
-                    egui::Label::new(egui::RichText::new(&self.val).size(16.0 * rc.zoom))
+                    egui::Label::new(egui::RichText::new(&self.val).font(font_id))
                         .wrap_mode(egui::TextWrapMode::Extend),
                 );
             })
-            .response
-            .interact(egui::Sense::click());
+            .response;
+
+        if response.interact(egui::Sense::click()).clicked() {
+            rc.clicked = true;
+        }
+    }
+
+    fn world_rect(&self) -> egui::Rect {
+        // 假设基础大小为 80x40
+        egui::Rect::from_center_size(self.position(), egui::vec2(80.0, 40.0))
     }
 }
 impl EntityTrait for Text {

crates/project_graph/src/stage/render_context.rs

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