89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
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())
|