100 lines
4.4 KiB
Python
100 lines
4.4 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
import sys
|
|
import tempfile
|
|
|
|
from OCC.Core.BRepPrimAPI import BRepPrimAPI_MakeBox
|
|
|
|
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.step_io import _write_step
|
|
|
|
|
|
DEFAULT_MODEL = PROJECT_ROOT / "assets" / "models" / "cube_10mm.step"
|
|
|
|
|
|
def _first_plane_face(model: StepModel) -> int:
|
|
for face_id in range(len(model.faces)):
|
|
info = model.face_info(face_id)
|
|
if info.get("surface") == "plane":
|
|
return face_id
|
|
raise SystemExit("no plane Face found")
|
|
|
|
|
|
def _first_shell_face(model: StepModel, thickness: float, tolerance: float) -> int:
|
|
candidates: list[tuple[int, int]] = []
|
|
for face_id in range(len(model.faces)):
|
|
info = model.feature_info(face_id)
|
|
if info.get("surface") != "plane":
|
|
continue
|
|
if info.get("shell_region_status") != "candidate":
|
|
continue
|
|
current = float(info.get("shell_thickness_estimate") or 0.0)
|
|
if abs(current - thickness) > tolerance:
|
|
continue
|
|
confidence_rank = {"high": 0, "medium": 1, "low": 2}.get(str(info.get("shell_confidence")), 3)
|
|
candidates.append((confidence_rank, face_id))
|
|
if not candidates:
|
|
raise SystemExit(f"no shell thickness candidate near {thickness:g}")
|
|
candidates.sort()
|
|
return candidates[0][1]
|
|
|
|
|
|
def _assert_noop_blocked(label: str, plan: dict[str, object]) -> None:
|
|
status = str(plan.get("status", ""))
|
|
message = str(plan.get("message", ""))
|
|
if status != "blocked":
|
|
raise SystemExit(f"{label} no-op plan was not blocked: status={status}, message={message}")
|
|
if not any(fragment in message for fragment in ("不需要修改", "无需移动", "为 0", "几乎相同")):
|
|
raise SystemExit(f"{label} no-op blocker did not explain unchanged target: {message}")
|
|
|
|
|
|
def _run_cube_face_noops() -> None:
|
|
model = StepModel.load(DEFAULT_MODEL)
|
|
face_id = _first_plane_face(model)
|
|
info = model.face_info(face_id)
|
|
center = info.get("area_center") or info.get("bbox_center")
|
|
if not isinstance(center, tuple) or len(center) != 3:
|
|
raise SystemExit("plane Face center is missing")
|
|
area = float(info.get("area") or 0.0)
|
|
width = float(info.get("local_face_width") or 0.0)
|
|
height = float(info.get("local_face_height") or 0.0)
|
|
|
|
_assert_noop_blocked("Face push/pull distance", model.push_pull_plan(face_id, 0.0))
|
|
_assert_noop_blocked("Face offset current face", model.face_plane_offset_local_plan(face_id, 0.0))
|
|
_assert_noop_blocked("Face offset owning feature", model.face_plane_offset_owning_translation_plan(face_id, 0.0))
|
|
_assert_noop_blocked("Face center current face", model.face_center_local_move_plan(face_id, center))
|
|
_assert_noop_blocked("Face center owning feature", model.face_center_owning_translation_plan(face_id, center))
|
|
_assert_noop_blocked("Face area current face", model.face_area_local_resize_plan(face_id, area))
|
|
_assert_noop_blocked("Face area owning feature", model.face_area_scale_plan(face_id, area))
|
|
_assert_noop_blocked("Face width current face", model.face_size_local_resize_plan(face_id, width, "width"))
|
|
_assert_noop_blocked("Face width owning feature", model.face_size_owning_scale_plan(face_id, width, "width"))
|
|
_assert_noop_blocked("Face height current face", model.face_size_local_resize_plan(face_id, height, "height"))
|
|
_assert_noop_blocked("Face height owning feature", model.face_size_owning_scale_plan(face_id, height, "height"))
|
|
|
|
|
|
def _run_shell_noops() -> None:
|
|
with tempfile.TemporaryDirectory(prefix="geom_param_face_noop_shell_") as temp_dir:
|
|
model_path = Path(temp_dir) / "plate.step"
|
|
_write_step(BRepPrimAPI_MakeBox(30.0, 20.0, 2.0).Shape(), model_path)
|
|
model = StepModel.load(model_path)
|
|
face_id = _first_shell_face(model, 2.0, 2e-4)
|
|
current = float(model.feature_info(face_id).get("shell_thickness_estimate") or 0.0)
|
|
_assert_noop_blocked("thin wall current face", model.shell_thickness_plan(face_id, current))
|
|
_assert_noop_blocked("thin wall owning feature", model.shell_thickness_owning_scale_plan(face_id, current))
|
|
|
|
|
|
def main() -> int:
|
|
_run_cube_face_noops()
|
|
_run_shell_noops()
|
|
print("Face no-op guards ok")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|