From 9b4924c73ebb12e4ede4fd50be379c9e3baec65e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E5=BE=B7=E5=AE=87?= <kvtodev@outlook.com>
Date: Wed, 9 Sep 2026 14:54:37 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85=E5=88=86=E7=A6=BB=E6=89=B9?=
 =?UTF-8?q?=E6=AC=A1=E9=87=8D=E5=8F=A0=E7=AD=9B=E9=80=89=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=20=E9=AA=8C=E8=AF=81=20DFS=20=E9=A1=BA=E5=BA=8F=E4=B8=8B?=
 =?UTF-8?q?=E8=B4=AA=E5=BF=83=E8=B7=B3=E8=BF=87=E9=87=8D=E5=8F=A0=E5=8F=B6?=
 =?UTF-8?q?=E8=8A=82=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/ui_editor/commands/separation/mod.rs  | 62 +++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs
index 83a984c2e..179dfdd6f 100644
--- a/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs
+++ b/apps/ai-game-creator-shell/src-tauri/src/ui_editor/commands/separation/mod.rs
@@ -157,4 +157,66 @@ mod tests {
         assert_eq!(state.bound[0].node_id, id);
         assert_eq!(state.trees[0].root.children.len(), 1);
     }
+
+    #[test]
+    fn batch_selection_greedily_skips_overlapping_leaves() {
+        let a = SeparationNode {
+            id: NodeId::new("a").unwrap(),
+            global_pos_x_px: 0,
+            global_pos_y_px: 0,
+            width_px: 10,
+            height_px: 10,
+            note: SeparationNote::default(),
+            children: vec![],
+            rework_count: 0,
+        };
+        let b = SeparationNode {
+            id: NodeId::new("b").unwrap(),
+            global_pos_x_px: 5,
+            global_pos_y_px: 5,
+            width_px: 10,
+            height_px: 10,
+            note: SeparationNote::default(),
+            children: vec![],
+            rework_count: 0,
+        };
+        let c = SeparationNode {
+            id: NodeId::new("c").unwrap(),
+            global_pos_x_px: 20,
+            global_pos_y_px: 0,
+            width_px: 5,
+            height_px: 5,
+            note: SeparationNote::default(),
+            children: vec![],
+            rework_count: 0,
+        };
+        let tree = SeparationTree {
+            src_ui_design: UIDesignImageId::new("page").unwrap(),
+            root: SeparationNode {
+                id: NodeId::new("root").unwrap(),
+                global_pos_x_px: 0,
+                global_pos_y_px: 0,
+                width_px: 100,
+                height_px: 100,
+                note: SeparationNote::default(),
+                children: vec![a, b, c],
+                rework_count: 0,
+            },
+            root_extractable: false,
+        };
+        let state = SeparationState {
+            schema_version: SEPARATION_STATE_SCHEMA_VERSION.to_string(),
+            trees: vec![tree.clone()],
+            bound: vec![],
+            problematic_nodes: vec![],
+        };
+        let batch = next_leaf_batch(&state, &tree);
+        assert_eq!(
+            batch
+                .iter()
+                .map(|node| node.id.as_str())
+                .collect::<Vec<_>>(),
+            vec!["a", "c"]
+        );
+    }
 }