256 lines
11 KiB
Python
256 lines
11 KiB
Python
from __future__ import annotations
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
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.info_panel import InfoPanelMixin
|
||
from step_editor.ui_helpers import INFO_LABELS
|
||
from step_editor.window_state import WindowStateMixin
|
||
|
||
|
||
class _SelectionProbe(InfoPanelMixin, WindowStateMixin):
|
||
pass
|
||
|
||
|
||
class _LogicalFaceModel:
|
||
def __init__(self, logical_id: int) -> None:
|
||
self.logical_id = logical_id
|
||
|
||
def face_region_logical_id(self, _face_id: int) -> int:
|
||
return self.logical_id
|
||
|
||
def face_logical_id(self, _face_id: int) -> int:
|
||
return self.logical_id
|
||
|
||
|
||
class _FeatureContextModel(_LogicalFaceModel):
|
||
faces = [object()] * 800
|
||
face_part_ids = [1] * 800
|
||
face_solid_ids = [0] * 800
|
||
|
||
def quick_face_info(self, face_id: int) -> dict[str, object]:
|
||
return {
|
||
"kind": "face",
|
||
"face_id": face_id,
|
||
"topological_face_id": face_id,
|
||
"logical_face_id": self.logical_id,
|
||
"surface": "plane",
|
||
"part_id": 1,
|
||
"solid_id": 0,
|
||
"area": 100.0,
|
||
"area_center": (5.0, 5.0, 0.0),
|
||
"local_face_width": 10.0,
|
||
"local_face_height": 10.0,
|
||
"feature_source_face_id": face_id,
|
||
"feature_highlight_face_ids": (face_id,),
|
||
}
|
||
|
||
def cached_feature_info(self, _face_id: int) -> dict[str, object] | None:
|
||
return None
|
||
|
||
def face_first_level_topology(self, face_id: int) -> dict[str, object]:
|
||
return {
|
||
"topology_relation_depth": 1,
|
||
"topology_relation_model": "STEP/B-Rep shared-edge first-level",
|
||
"topology_relation_scope": "selected same-domain region + direct shared-edge adjacent Faces",
|
||
"topology_relation_boundary": "shared-edge",
|
||
"topology_ignored_relation_depths": ("second-level", "third-level", "deeper"),
|
||
"topology_ignored_relation_note": "当前阶段只传播一级关系；二级、三级拓扑暂不自动递归编辑。",
|
||
"same_domain_face_ids": (face_id,),
|
||
"same_domain_face_count": 1,
|
||
"same_domain_region_kind": "single-face",
|
||
"first_level_boundary_edge_ids": (1, 2, 3, 4),
|
||
"first_level_boundary_edge_count": 4,
|
||
"first_level_boundary_vertex_count": 4,
|
||
"first_level_adjacent_face_ids": (10, 11, 12, 13),
|
||
"first_level_adjacent_face_count": 4,
|
||
"first_level_face_ids": (face_id, 10, 11, 12, 13),
|
||
"first_level_face_count": 5,
|
||
"first_level_topology_note": "已识别当前 Face 区域 1 个 Face、边界 Edge 4 条、边界 Vertex 4 个、共享边一级相邻 Face 4 个。",
|
||
}
|
||
|
||
def face_first_level_facts(self, face_id: int, scope: str = "auto") -> dict[str, object]:
|
||
return {
|
||
"first_level_fact_model": "STEP/B-Rep first-level fact graph",
|
||
"first_level_fact_status": "ready",
|
||
"first_level_fact_relation_depth": 1,
|
||
"first_level_fact_scope": scope,
|
||
"first_level_fact_relation_boundary": "shared-edge",
|
||
"first_level_fact_subject_face_ids": (face_id,),
|
||
"first_level_fact_subject_face_count": 1,
|
||
"first_level_fact_boundary_edge_count": 4,
|
||
"first_level_fact_boundary_vertex_count": 4,
|
||
"first_level_fact_adjacent_face_ids": (10, 11, 12, 13),
|
||
"first_level_fact_adjacent_face_count": 4,
|
||
"first_level_fact_included_face_count": 5,
|
||
"first_level_fact_ignored_relation_depths": ("second-level", "third-level", "deeper"),
|
||
"first_level_fact_summary": "一级事实=当前 Face、边界和共享边相邻 Face。",
|
||
}
|
||
|
||
def _recognition_summary_fields(self, _info: dict[str, object]) -> dict[str, object]:
|
||
return {"recognition_summary": "识别摘要包含一级事实。"}
|
||
|
||
|
||
class _FeatureRefreshProbe(WindowStateMixin):
|
||
def __init__(self) -> None:
|
||
self.model = _FeatureContextModel(594)
|
||
self.selected_kind = "feature"
|
||
self.selected_face_id = 637
|
||
self.selected_edge_id = None
|
||
self.selected_part_id = 1
|
||
self.selected_solid_id = 0
|
||
self.selected_pick_position = (1.0, 2.0, 3.0)
|
||
self.feature_detection_level = "current-only"
|
||
self.set_info_payload: dict[str, object] | None = None
|
||
self.highlighted_face_ids: tuple[int, ...] = ()
|
||
self.synced_id: tuple[str, int] | None = None
|
||
|
||
def set_info(self, info: dict[str, object]) -> None:
|
||
self.set_info_payload = dict(info)
|
||
self.current_info_values = dict(info)
|
||
|
||
def _highlight_faces(self, face_ids: list[int] | tuple[int, ...] | None = None, **_kwargs) -> None:
|
||
self.highlighted_face_ids = tuple(int(item) for item in (face_ids or ()))
|
||
|
||
def _update_selected_object_title(self) -> None:
|
||
return
|
||
|
||
def _sync_id_picker(self, kind: str, target_id: int) -> None:
|
||
self.synced_id = (kind, int(target_id))
|
||
|
||
|
||
def _probe(kind: str, selected_face_id: int, info: dict[str, object]) -> _SelectionProbe:
|
||
probe = object.__new__(_SelectionProbe)
|
||
probe.model = None
|
||
probe.selected_kind = kind
|
||
probe.selected_part_id = None
|
||
probe.selected_solid_id = None
|
||
probe.selected_face_id = selected_face_id
|
||
probe.selected_edge_id = None
|
||
probe.current_info_values = dict(info)
|
||
return probe
|
||
|
||
|
||
def _assert(condition: bool, message: str) -> None:
|
||
if not condition:
|
||
raise AssertionError(message)
|
||
|
||
|
||
def main() -> int:
|
||
for key in ("selection_title", "selection_display_id", "selection_topological_face_id"):
|
||
_assert(key in INFO_LABELS, f"{key} should have a readable UI label")
|
||
_assert(INFO_LABELS.get("face_id") == "当前拓扑 Face ID", "raw face_id should be labelled as topological")
|
||
|
||
info = {
|
||
"feature_type": "可拉伸/切除平面候选",
|
||
"selection_display_id": 594,
|
||
"selection_topological_face_id": 637,
|
||
"face_region_logical_id": 594,
|
||
"logical_face_id": 594,
|
||
}
|
||
|
||
feature_probe = _probe("feature", 637, info)
|
||
_assert(
|
||
feature_probe._selected_id_text() == "face 594",
|
||
f"feature copy ID should prefer logical Face ID, got {feature_probe._selected_id_text()!r}",
|
||
)
|
||
feature_title = feature_probe._selected_object_title_suffix()
|
||
_assert(
|
||
"Face 594" in feature_title and "拓扑 637" in feature_title,
|
||
f"feature title should show logical and topological IDs: {feature_title}",
|
||
)
|
||
_assert("候选" not in feature_title, f"feature title should use a user-facing label, got: {feature_title}")
|
||
_assert("可拉伸/切除平面" in feature_title, f"feature title should keep the editable feature name: {feature_title}")
|
||
|
||
face_probe = _probe("face", 637, info)
|
||
_assert(
|
||
face_probe._selected_id_text() == "face 594",
|
||
f"Face copy ID should prefer logical Face ID, got {face_probe._selected_id_text()!r}",
|
||
)
|
||
face_title = face_probe._selected_object_title_suffix()
|
||
_assert(
|
||
"Face 594" in face_title and "拓扑 637" in face_title,
|
||
f"Face title should show logical and topological IDs: {face_title}",
|
||
)
|
||
_assert("候选" not in face_title, f"Face title should not expose candidate wording: {face_title}")
|
||
|
||
fallback_probe = _probe("feature", 637, {"face_region_logical_id": 594})
|
||
_assert(fallback_probe._selected_id_text() == "face 594", "logical Face ID fallback should be copied")
|
||
|
||
selection_only_probe = _probe("feature", 637, {"selection_display_id": 594})
|
||
_assert(
|
||
selection_only_probe._selected_id_text() == "face 594",
|
||
"selection_display_id should be enough for copied Face ID",
|
||
)
|
||
selection_only_title = selection_only_probe._selected_object_title_suffix()
|
||
_assert(
|
||
"Face 594" in selection_only_title and "拓扑 637" in selection_only_title,
|
||
f"selection_display_id should be enough for feature title: {selection_only_title}",
|
||
)
|
||
|
||
face_selection_only_probe = _probe("face", 637, {"selection_display_id": 594})
|
||
face_selection_only_title = face_selection_only_probe._selected_object_title_suffix()
|
||
_assert(
|
||
"Face 594" in face_selection_only_title and "拓扑 637" in face_selection_only_title,
|
||
f"selection_display_id should be enough for Face title: {face_selection_only_title}",
|
||
)
|
||
|
||
raw_probe = _probe("feature", 637, {})
|
||
_assert(raw_probe._selected_id_text() == "face 637", "topological Face ID fallback should still work")
|
||
|
||
helper_probe = _probe("feature", 637, {})
|
||
helper_probe.model = _LogicalFaceModel(594)
|
||
fields = helper_probe._selection_identity_fields(637, "特征来源 Face")
|
||
_assert(fields.get("selection_display_id") == 594, f"helper should expose logical Face ID: {fields}")
|
||
_assert(
|
||
fields.get("selection_topological_face_id") == 637,
|
||
f"helper should keep current topological Face ID: {fields}",
|
||
)
|
||
_assert(
|
||
fields.get("selection_title") == "特征来源 Face 594（当前拓扑 Face 637）",
|
||
f"helper title should explain both IDs: {fields}",
|
||
)
|
||
|
||
refresh_probe = _FeatureRefreshProbe()
|
||
context = refresh_probe._feature_context_info(637)
|
||
_assert(context.get("selection_display_id") == 594, f"feature context should include logical ID: {context}")
|
||
_assert(
|
||
context.get("selection_topological_face_id") == 637,
|
||
f"feature context should include topological ID: {context}",
|
||
)
|
||
refresh_probe._on_feature_detection_level_changed()
|
||
payload = refresh_probe.set_info_payload or {}
|
||
_assert(payload.get("selection_display_id") == 594, f"detection refresh should call set_info with identity: {payload}")
|
||
_assert(payload.get("pick_position") == (1.0, 2.0, 3.0), f"detection refresh should preserve pick: {payload}")
|
||
_assert(refresh_probe.highlighted_face_ids == (637,), f"detection refresh should rehighlight source Face")
|
||
|
||
source_probe = _FeatureRefreshProbe()
|
||
source_probe.selected_face_id = 100
|
||
source_probe.selected_pick_position = (9.0, 9.0, 9.0)
|
||
source_probe._activate_property_source_feature(
|
||
{
|
||
"source_face_id": 637,
|
||
"source_feature_info": {
|
||
"feature_type": "可拉伸/切除平面候选",
|
||
"part_id": 1,
|
||
"solid_id": 0,
|
||
"feature_highlight_face_ids": (637,),
|
||
},
|
||
}
|
||
)
|
||
source_payload = source_probe.set_info_payload or {}
|
||
_assert(source_probe.synced_id == ("Feature", 594), f"associated jump should sync logical ID: {source_probe.synced_id}")
|
||
_assert(source_payload.get("selection_display_id") == 594, f"associated jump should keep logical ID: {source_payload}")
|
||
_assert("pick_position" not in source_payload, f"associated jump should not reuse stale pick position: {source_payload}")
|
||
|
||
print("selection identity UI ok")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|