refactor: 将Painter渲染改为Ui渲染

Author: zty012

crates/project_graph/src/stage.rs

@@ -40,11 +40,17 @@ impl Stage {
                     .camera
                     .world_to_screen(entity.position(), screen_center);
 
-                entity.render(&mut RenderContext {
-                    painter: &painter,
-                    position: screen_pos,
-                    scale: self.camera.zoom(),
-                });
+                egui::Area::new(ui.make_persistent_id(entity.id()))
+                    .order(egui::Order::Foreground)
+                    .fixed_pos(screen_pos)
+                    .show(ui.ctx(), |ui| {
+                        entity.ui(
+                            ui,
+                            &mut RenderContext {
+                                zoom: self.camera.zoom(),
+                            },
+                        );
+                    });
             }
 
             painter.text(

crates/project_graph/src/stage/render_context.rs

@@ -1,42 +1,5 @@
-use egui::{Painter, Pos2, Stroke, vec2};
-
-pub struct RenderContext<'a> {
-    pub painter: &'a Painter,
-    pub position: Pos2,
-    pub scale: f32,
+pub struct RenderContext {
+    pub zoom: f32,
 }
 
-impl<'a> RenderContext<'a> {
-    pub fn local_to_screen(&self, local_pos: Pos2) -> Pos2 {
-        Pos2 {
-            x: self.position.x + local_pos.x * self.scale,
-            y: self.position.y + local_pos.y * self.scale,
-        }
-    }
-
-    pub fn rect(&self, local_rect: egui::Rect, color: egui::Color32) {
-        let screen_rect = egui::Rect {
-            min: self.local_to_screen(local_rect.min),
-            max: self.local_to_screen(local_rect.max),
-        };
-        self.painter.rect_stroke(
-            screen_rect,
-            8.0 * self.scale,
-            Stroke {
-                color,
-                width: self.scale,
-            },
-            egui::StrokeKind::Middle,
-        );
-    }
-
-    pub fn text(&self, position: Pos2, text: &str) {
-        self.painter.text(
-            self.local_to_screen(position) - vec2(0.0, 14.0 * 0.15) * self.scale,
-            egui::Align2::LEFT_TOP,
-            text,
-            egui::FontId::proportional(14.0 * self.scale),
-            egui::Color32::WHITE,
-        );
-    }
-}
+impl RenderContext {}

crates/project_graph/src/stage/structs.rs

@@ -8,7 +8,7 @@ use crate::{stage::render_context::RenderContext, utils::kdl::KdlPos2};
 pub trait EntityTrait {
     fn id(&self) -> &str;
     fn position(&self) -> Pos2;
-    fn render(&self, rc: &mut RenderContext);
+    fn ui(&self, ui: &mut egui::Ui, rc: &mut RenderContext);
 }
 
 #[derive(knus::Decode, Debug, Clone)]
@@ -39,12 +39,12 @@ impl EntityTrait for Text {
     fn position(&self) -> Pos2 {
         self.pos.into()
     }
-    fn render(&self, rc: &mut RenderContext) {
+    fn ui(&self, ui: &mut egui::Ui, rc: &mut RenderContext) {
         let padding = 8.0;
 
         let text_width = *self.text_width.get_or_init(|| {
             // 这里调用你的 get_text_width 或者直接计算
-            rc.painter
+            ui.painter()
                 .layout_no_wrap(
                     self.val.clone(),
                     FontId::proportional(14.0),
@@ -54,17 +54,10 @@ impl EntityTrait for Text {
                 .x
         });
 
-        rc.rect(
-            Rect {
-                min: Pos2::ZERO,
-                max: Pos2 {
-                    x: text_width + padding * 2.0,
-                    y: 14.0 + padding * 2.0,
-                },
-            },
-            Color32::WHITE,
+        ui.add(
+            egui::Label::new(egui::RichText::new(&self.val).size(14.0 * rc.zoom))
+                .wrap_mode(egui::TextWrapMode::Extend),
         );
-        rc.text(pos2(padding, padding), &self.val);
     }
 }