feat(ocr): 结构图过滤规则修正——Text区域同过图结构门控+形状约束

Author: Aparencia

app/src-tauri/src/structure_detect.rs

@@ -47,6 +47,12 @@ pub fn diagram_likeness(grid: &FrameGrid, x0: u32, y0: u32, x1: u32, y1: u32) ->
     if w < 2 || h < 2 {
         return 0.0;
     }
+    // 形状约束（实现校准）：标题细条/单行词条不可能是图结构——区域必须
+    // 是"块状"（高 ≥3 格、宽 ≥6 格）。真实流程图页密度高被判 Text，
+    // 行带可能是框线连成的大块；长标题是 1-2 格高的细条，借此拒绝。
+    if h < 3 || w < 6 {
+        return 0.0;
+    }
     // ① 长直线特征：行/列最长连续墨迹段 + 含长段行数
     let mut max_h_run = 0u32;
     let mut line_rows = 0u32;
@@ -169,9 +175,11 @@ pub fn pick_sharpest(candidates: &[(u64, FrameGrid)]) -> Option<usize> {
 
 /// 结构图候选过滤（纯函数）：版面区域 → 值得持久化的子集。
 ///
-/// @ai-context: table/formula/code 直收（is_structural）；Image 过
-///              diagram_likeness 阈值；text/unknown 跳过。输入输出均为
-///              网格坐标（编排层随后 regions_to_frame 换算帧坐标裁剪）。
+/// @ai-context: table/formula/code 直收（is_structural）；Image/Text 均过
+///              diagram_likeness 阈值（实现校准：真实流程图页密度高被判
+///              Text——长直线+形状约束天然拒纯文字/标题，故 Text 同门控）；
+///              unknown 跳过（归 V1.0 AI 补缝）。输入输出均为网格坐标
+///              （编排层随后 regions_to_frame 换算帧坐标裁剪）。
 pub fn filter_structure_regions(
     regions: &[LayoutRegion],
     grid: &FrameGrid,
@@ -180,11 +188,11 @@ pub fn filter_structure_regions(
         .iter()
         .filter(|r| match r.kind {
             RegionKind::Table | RegionKind::Formula | RegionKind::Code => true,
-            RegionKind::Image => {
+            RegionKind::Image | RegionKind::Text => {
                 diagram_likeness(grid, r.x, r.y, r.x + r.w - 1, r.y + r.h - 1)
                     >= DIAGRAM_LIKENESS_THRESHOLD
             }
-            RegionKind::Text | RegionKind::Unknown => false,
+            RegionKind::Unknown => false,
         })
         .cloned()
         .collect()

app/src-tauri/src/structure_detect_tests.rs

@@ -123,6 +123,19 @@ fn diagram_text_line_rejected() {
     assert!(score < crate::structure_detect::DIAGRAM_LIKENESS_THRESHOLD);
 }
 
+#[test]
+fn diagram_title_strip_rejected_by_shape() {
+    // Arrange：长标题细条（高 1-2 格、长词连续段占比高——形状约束拒绝）
+    let mut g = blank(32, 18, 255);
+    paint(&mut g, 2, 6, 25, 6); // 24 格长段、单行
+
+    // Act：区域 h=1（细条）
+    let score = diagram_likeness(&g, 2, 6, 25, 6);
+
+    // Assert：形状约束（h<3）→ 0 分——长标题不误收
+    assert_eq!(score, 0.0);
+}
+
 #[test]
 fn diagram_out_of_bounds_and_tiny_region_zero() {
     // Arrange：全屏空白 + 越界/过小区域
@@ -197,26 +210,34 @@ fn pick_sharpest_skips_zero_energy_frames() {
 
 #[test]
 fn filter_structure_kinds_direct_and_image_gated() {
-    // Arrange：流程图样网格 + 混合区域列表
+    // Arrange：流程图样网格 + 混合区域列表（含"被判 Text 的流程图区域"——
+    // 实现校准：真实流程图密度高被判 Text，同过图结构门控）
     let g = flowchart_like();
     let regions = vec![
         region(RegionKind::Table, 0, 0, 5, 3),
         region(RegionKind::Formula, 0, 4, 5, 3),
         region(RegionKind::Code, 0, 8, 5, 3),
-        region(RegionKind::Image, 4, 3, 20, 8), // 流程图区域（应过阈值）
+        region(RegionKind::Text, 4, 3, 20, 8), // 流程图区域被判 Text（应过门控）
+        region(RegionKind::Image, 4, 3, 20, 8), // 同区域 Image（应过门控）
         region(RegionKind::Image, 0, 12, 8, 4), // 空白区域（应拒）
-        region(RegionKind::Text, 26, 0, 5, 3),
+        region(RegionKind::Text, 26, 0, 5, 3), // 小文字区域（形状约束拒）
         region(RegionKind::Unknown, 26, 5, 5, 3),
     ];
 
     // Act
     let kept = filter_structure_regions(&regions, &g);
 
-    // Assert：结构三类直收 + Image 阈值门控；text/unknown 跳过
+    // Assert：结构三类直收 + Text/Image 过图结构门控；小文字/空白/unknown 跳过
     let kinds: Vec<RegionKind> = kept.iter().map(|r| r.kind).collect();
     assert_eq!(
         kinds,
-        vec![RegionKind::Table, RegionKind::Formula, RegionKind::Code, RegionKind::Image]
+        vec![
+            RegionKind::Table,
+            RegionKind::Formula,
+            RegionKind::Code,
+            RegionKind::Text,
+            RegionKind::Image
+        ]
     );
-    assert_eq!(kept.last().unwrap().x, 4); // 保留的是流程图区域而非空白区域
+    assert_eq!(kept[3].x, 4); // Text 流程图区域保留
 }