feat: 完善 Face 一级编辑与隔离计算

2026-08-05 15:08:16 +08:00
parent a76282d7dd
commit ed1faf51d8
28 changed files with 3740 additions and 876 deletions
@@ -0,0 +1,88 @@
+from __future__ import annotations
+
+from pathlib import Path
+import sys
+
+PROJECT_ROOT = Path(__file__).resolve().parent.parent
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
+from step_editor.model import StepModel
+from step_editor.window_state import WindowStateMixin, _feature_dimension_keys
+
+
+MODEL_PATH = PROJECT_ROOT / "assets" / "models" / "geom_extract.step"
+FACE_ID = 594
+BASE_FACE_KEYS = (
+    "area",
+    "local_face_width",
+    "local_face_height",
+    "face_center_position",
+    "face_target_normal_position",
+)
+
+
+class _Probe(WindowStateMixin):
+    pass
+
+
+def _feature_rows(model: StepModel, face_id: int) -> tuple[str, ...]:
+    info = model.quick_face_info(face_id)
+    probe = object.__new__(_Probe)
+    probe.model = model
+    probe.operation_in_progress = False
+    probe.scan_in_progress = False
+    probe.load_in_progress = False
+    probe.selected_face_id = face_id
+    probe.selected_edge_id = None
+    probe.selected_kind = "feature"
+    probe.selected_part_id = int(info["part_id"])
+    probe.selected_solid_id = int(info["solid_id"])
+    probe.feature_detection_level = "current-only"
+    probe.manual_bottom_face_id = None
+    probe.manual_slot_pair_face_id = None
+    context = probe._feature_context_info(face_id)
+    specs, _used = probe._editable_property_specs(context)
+    rows = probe._feature_context_property_specs(specs, context)
+    return tuple(str(row.get("key")) for row in rows if row.get("parameter_role") == "dimension")
+
+
+def _assert_contains(keys: tuple[str, ...], expected: tuple[str, ...], label: str) -> None:
+    missing = [key for key in expected if key not in keys]
+    if missing:
+        raise AssertionError(f"{label} missing {missing}, got {keys}")
+
+
+def main() -> int:
+    model = StepModel.load(MODEL_PATH)
+    if FACE_ID >= len(model.faces):
+        raise AssertionError(f"test model has no Face {FACE_ID}")
+
+    quick_info = model.quick_face_info(FACE_ID)
+    if quick_info.get("surface") != "plane":
+        raise AssertionError(f"Face {FACE_ID} should be planar in the baseline model: {quick_info}")
+
+    before_cached_rows = _feature_rows(model, FACE_ID)
+    _assert_contains(before_cached_rows, BASE_FACE_KEYS, "Face 594 before full feature cache")
+
+    full_info = model.feature_info(FACE_ID)
+    if full_info.get("shell_region_status") == "candidate":
+        dimension_keys = _feature_dimension_keys(full_info)
+        _assert_contains(dimension_keys, BASE_FACE_KEYS, "shell candidate feature dimensions")
+        if "shell_thickness_estimate" not in dimension_keys:
+            raise AssertionError(f"shell candidate should keep shell thickness as an extra dimension: {dimension_keys}")
+
+    after_cached_rows = _feature_rows(model, FACE_ID)
+    _assert_contains(after_cached_rows, BASE_FACE_KEYS, "Face 594 after full feature cache")
+    if before_cached_rows != after_cached_rows:
+        raise AssertionError(
+            "Face 594 current-only feature rows changed after full recognition cache: "
+            f"before={before_cached_rows}, after={after_cached_rows}"
+        )
+
+    print("Face feature parameter consistency ok")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())