871070c440
Phase 2 of the decoupling refactor (behavior-preserving move): - runtime_base.py: RuntimeExecutionError, FeatureExecutionError, ExtentVector - session.py: GeometryAdapter protocol + ExecutionSession - extents.py: end-condition planning (_extent_vectors family) - pattern_transform.py: translate/mirror/rotate replay parameter algebra - runtime.py: keeps executors + registry + entry points; re-exports all moved names (incl. test-referenced privates) for import stability No behavior change; verified against baseline (zero new failures).
1664 lines
85 KiB
Python
1664 lines
85 KiB
Python
"""Session-based CDSL execution with atomic executor registry.
|
||
|
||
The runtime was split into focused modules (behavior-preserving move):
|
||
|
||
- ``runtime_base``: shared error types and the ``ExtentVector`` value.
|
||
- ``session``: ``ExecutionSession`` and the ``GeometryAdapter`` protocol.
|
||
- ``extents``: end-condition planning.
|
||
- ``pattern_transform``: translate/mirror/rotate parameter algebra for replay.
|
||
|
||
This module keeps the executor registry, the per-atomic executors, and the
|
||
``analyze_cdsl`` / ``rebuild_cdsl`` entry points. The moved names are
|
||
re-exported so every historical ``cdsl_engine.runtime`` import keeps working.
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from copy import deepcopy
|
||
import math
|
||
from pathlib import Path
|
||
from typing import Any, Callable, Protocol
|
||
|
||
from .build123d_adapter import Build123dGeometryAdapter # noqa: F401 (historical re-export)
|
||
from .capabilities import CapabilityAnalyzer, pattern_transform_blocker, sketch_ids_required_by_contract
|
||
from .extents import (
|
||
_extent_reference,
|
||
_extent_vectors,
|
||
_extent_vectors_from_normal,
|
||
_normal_from_sketch,
|
||
_side_extent_vectors,
|
||
_targeted_extent_vector,
|
||
)
|
||
from .pattern_transform import (
|
||
_box_circular_is_exact,
|
||
_coordinate_axis_direction,
|
||
_mirrored_node,
|
||
_mirrored_sketch,
|
||
_normal_is_coordinate_axis,
|
||
_owner_plane_frame,
|
||
_pattern_operation_node,
|
||
_reflect_point,
|
||
_rotated_node,
|
||
_rotated_point,
|
||
_rotated_sketch,
|
||
_rotated_vector,
|
||
_transformed_loft_profiles,
|
||
_translated_node,
|
||
_translated_sketch,
|
||
)
|
||
from .runtime_base import ExtentVector, FeatureExecutionError, RuntimeExecutionError
|
||
from .session import ExecutionSession, GeometryAdapter
|
||
from .sketch_solver import CORE_SHAPE_GENERATORS, resolve_required_sketches
|
||
from .specs import (
|
||
AxisSpec, BendSpec, GearSpec, HoleSpec, PlaneSpec, RackSpec, ThreadSpec, Vector3,
|
||
pattern_instance_member_id, transform_copy_member_id,
|
||
vector_add, vector_cross, vector_dot, vector_scale, vector_subtract, vector_unit,
|
||
)
|
||
from .topology import (
|
||
CapabilityResult, FeaturePlanNode, FeatureResult, RuntimeDiagnostic,
|
||
SelectorResolution, TopologyDelta, TopologyDeltaRelation, TopologyRecord, TopologyRegistry,
|
||
)
|
||
|
||
|
||
ALL_ATOMIC_IDS = frozenset({
|
||
"extrude_add_blind", "extrude_add_blind_with_hole", "extrude_add_two_sided", "extrude_cut_blind", "extrude_cut_two_sided", "extrude_surface",
|
||
"extrude_cut_through", "extrude_from_face", "loft_add", "loft_add_with_cap_face", "sweep_add",
|
||
"revolve_add", "revolve_cut", "revolve_surface", "hole_blind", "hole_countersink",
|
||
"hole_counterbore", "sphere_add", "box_add", "cylinder_add",
|
||
"reference_plane", "reference_axis",
|
||
"hole_wizard", "fillet", "chamfer", "shell", "pattern_linear", "pattern_mirror",
|
||
"pattern_circular", "boolean_bodies", "transform_bodies", "delete_bodies",
|
||
"thread_add", "thread_cut",
|
||
"bend_add",
|
||
"gear_add", "rack_add",
|
||
})
|
||
|
||
|
||
class AtomicExecutor(Protocol):
|
||
atomic_id: str
|
||
|
||
def preflight(self, node: FeaturePlanNode, session: "ExecutionSession") -> CapabilityResult: ...
|
||
def execute(self, node: FeaturePlanNode, session: "ExecutionSession") -> FeatureResult: ...
|
||
|
||
|
||
def _revolve_axis(node: FeaturePlanNode, session: ExecutionSession) -> AxisSpec:
|
||
raw_axis = node.params.get("axis") or {}
|
||
if raw_axis.get("origin_mm") is not None and raw_axis.get("direction") is not None:
|
||
return AxisSpec.from_mapping(raw_axis)
|
||
selector = raw_axis.get("selector") if isinstance(raw_axis, dict) else None
|
||
if not isinstance(selector, dict):
|
||
selector = next((item for item in node.selectors if item.get("kind") == "axis"), None)
|
||
if not isinstance(selector, dict):
|
||
raise FeatureExecutionError(
|
||
"missing_revolve_axis",
|
||
"Revolve requires an explicit axis or an owner-qualified reference-axis selector",
|
||
)
|
||
resolution = session.resolve(selector)
|
||
if resolution.status != "resolved" or resolution.record is None:
|
||
raise ValueError(resolution.diagnostic.message if resolution.diagnostic else "revolve axis was not resolved")
|
||
if not isinstance(resolution.record.value, AxisSpec):
|
||
raise FeatureExecutionError(
|
||
"unsupported_revolve_axis", "The resolved context is not an axis", actual_kind=resolution.record.kind,
|
||
)
|
||
return resolution.record.value
|
||
|
||
|
||
def _validate_revolve_axis_in_sketch_plane(axis: AxisSpec, sketch: dict[str, Any]) -> None:
|
||
"""Defend direct CDSL execution from an out-of-plane revolve axis."""
|
||
plane = PlaneSpec.from_mapping(sketch.get("workplane") or {})
|
||
direction_normal_dot = abs(vector_dot(axis.direction, plane.normal))
|
||
if direction_normal_dot > 1e-7:
|
||
raise ValueError(
|
||
"REVOLVE_AXIS_NOT_IN_SKETCH_PLANE: params.axis.direction must be parallel to "
|
||
f"sketch.workplane; abs(dot(axis_direction, plane_normal))={direction_normal_dot:.3g}"
|
||
)
|
||
origin_plane_offset = abs(vector_dot(vector_subtract(axis.origin_mm, plane.origin_mm), plane.normal))
|
||
if origin_plane_offset > 1e-6:
|
||
raise ValueError(
|
||
"REVOLVE_AXIS_NOT_IN_SKETCH_PLANE: params.axis.origin_mm must lie in "
|
||
f"sketch.workplane; plane_offset_mm={origin_plane_offset:.3g}"
|
||
)
|
||
|
||
|
||
def _cut_explicit_body_members(session: ExecutionSession, tool: Any) -> dict[str, Any]:
|
||
"""Apply a cut to each independently owned body without erasing ownership.
|
||
|
||
A CADFS NEW body stays independently addressable even when a later REMOVE
|
||
feature affects several active bodies. Cutting the aggregate first loses
|
||
that identity, so this path uses the equivalent per-member set difference
|
||
and drops only members that the tool removes completely.
|
||
"""
|
||
members: dict[str, Any] = {}
|
||
for feature_id, body in session.body_members.items():
|
||
result = session.adapter.cut(body, tool)
|
||
if abs(float(result.volume)) > 1e-12:
|
||
members[feature_id] = result
|
||
return members
|
||
|
||
|
||
def _extruded_tool(
|
||
node: FeaturePlanNode,
|
||
faces: list[Any],
|
||
profile_normal: Vector3,
|
||
session: ExecutionSession,
|
||
) -> tuple[Any, TopologyDelta | None]:
|
||
"""Build one extrude tool, retaining caps only from one exact builder result."""
|
||
extents = _extent_vectors_from_normal(node, faces, profile_normal, session)
|
||
draft = node.params.get("draft")
|
||
taper_deg = 0.0
|
||
if isinstance(draft, dict):
|
||
taper_deg = float(draft["angle_deg"])
|
||
if not bool(draft["pull_direction"]):
|
||
taper_deg = -taper_deg
|
||
topology_delta: TopologyDelta | None = None
|
||
solids: list[Any] = []
|
||
for face in faces:
|
||
for extent in extents:
|
||
if draft is not None:
|
||
if len(faces) == 1 and len(extents) == 1:
|
||
solid, topology_delta = session.adapter.extrude_taper_with_topology_delta(
|
||
face, extent.vector, taper_deg,
|
||
)
|
||
solids.append(solid)
|
||
else:
|
||
solids.append(session.adapter.extrude_taper(face, extent.vector, taper_deg))
|
||
elif extent.trim_to is None and len(faces) == 1 and len(extents) == 1:
|
||
solid, topology_delta = session.adapter.extrude_with_topology_delta(face, extent.vector)
|
||
solids.append(solid)
|
||
elif extent.trim_to is None:
|
||
solids.append(session.adapter.extrude(face, extent.vector))
|
||
else:
|
||
solids.append(session.adapter.extrude_trimmed(face, extent.trim_to, extent.vector))
|
||
tool = None
|
||
for solid in solids:
|
||
tool = session.adapter.fuse(tool, solid)
|
||
if tool is None:
|
||
raise ValueError("extrude produced no solid")
|
||
return tool, topology_delta
|
||
|
||
|
||
def _apply_primary_tool(
|
||
node: FeaturePlanNode,
|
||
session: ExecutionSession,
|
||
tool: Any,
|
||
*,
|
||
cutting: bool,
|
||
topology_delta: TopologyDelta | None = None,
|
||
) -> FeatureResult:
|
||
"""Apply a profile-derived tool while preserving only final-snapshot topology evidence."""
|
||
if cutting:
|
||
if session.body is None:
|
||
raise ValueError("cut feature has no body")
|
||
members = _cut_explicit_body_members(session, tool)
|
||
if not members:
|
||
session.clear_body()
|
||
return session.result(node)
|
||
body = session.adapter.cut(session.body, tool)
|
||
topology_delta = None
|
||
elif node.params.get("result_mode") == "new_body":
|
||
body = session.adapter.combine(session.body, tool)
|
||
members = {**session.body_members, node.feature_id: tool}
|
||
else:
|
||
body = session.adapter.fuse(session.body, tool)
|
||
members = {node.feature_id: body}
|
||
# A fuse rebuilds subshape identity. Builder evidence belongs only to
|
||
# an unchanged standalone/new-body prism snapshot.
|
||
if session.body is not None:
|
||
topology_delta = None
|
||
session.register_body(
|
||
node.feature_id, body, replay_node=node, body_members=members, topology_delta=topology_delta,
|
||
)
|
||
return session.result(node)
|
||
|
||
|
||
def _shape_from_primary(node: FeaturePlanNode, session: ExecutionSession, *, sketch: dict[str, Any] | None = None) -> FeatureResult:
|
||
# 主形状特征(拉伸 / 旋转)的统一入口:由草图生成实体并与当前主体做布尔合并或切除。
|
||
|
||
# 1. 取草图:优先使用外部传入的 sketch_override(阵列/镜像等重放场景),
|
||
# 否则按 sketch_id 从会话草图表中取原始草图。
|
||
selected_sketch = sketch or session.sketches.get(str(node.sketch_id))
|
||
if selected_sketch is None:
|
||
raise ValueError("primary feature has no resolved sketch")
|
||
# 2. 从草图解析闭合轮廓区域(faces),没有闭合区域就无法生成实体。
|
||
faces = session.adapter.faces_for_sketch(selected_sketch)
|
||
if not faces:
|
||
raise ValueError("sketch does not create a closed profile region")
|
||
if node.atomic_id == "extrude_add_blind_with_hole":
|
||
resolved = [session.resolve(selector) for selector in node.selectors]
|
||
failed = next((item for item in resolved if item.status != "resolved"), None)
|
||
if failed or len(resolved) != 1 or resolved[0].record is None or resolved[0].record.kind != "face":
|
||
raise ValueError(failed.diagnostic.message if failed and failed.diagnostic else "profile hole selector is unresolved")
|
||
if len(faces) != 1:
|
||
raise ValueError("profile hole extrusion requires exactly one outer sketch region")
|
||
faces = [session.adapter.face_with_holes(faces[0], [resolved[0].record.value])]
|
||
topology_delta: TopologyDelta | None = None
|
||
# 3. 按特征类型生成子实体:
|
||
if node.atomic_id.startswith("extrude_"):
|
||
# 拉伸:先按终止条件(盲孔/贯穿/至面/双侧等)求出位移向量,
|
||
# 再对每个面沿每个向量做拉伸,得到实体列表。up_to_surface 在
|
||
# profile 与目标面非均匀相交时(extent.trim_to 非空)改用裁剪
|
||
# 拉伸:穿透后与目标面求交,只保留可达部分(issue #5)。
|
||
tool, topology_delta = _extruded_tool(
|
||
node, faces, _normal_from_sketch(selected_sketch), session,
|
||
)
|
||
else:
|
||
# 旋转:解析旋转轴并校验旋转角,然后绕轴旋转每个面得到实体列表。
|
||
axis = _revolve_axis(node, session)
|
||
_validate_revolve_axis_in_sketch_plane(axis, selected_sketch)
|
||
angle = float(node.params.get("angle_deg") or 0.0)
|
||
if angle <= 0:
|
||
raise ValueError("revolve requires angle_deg > 0")
|
||
# reverse=true 表示绕轴反向扫掠(SolidWorks 旋转方向反转):取负
|
||
# 旋转角,与 extrude 的 reverse(_extent_vectors 反转拉伸方向)同一
|
||
# 语义。profile_schema.json 已声明 revolve.* optional_params 含
|
||
# reverse,cdsl_schema.json revolveParams 也已允许,这里补齐 runtime
|
||
# 侧实现,使三方合同一致。
|
||
if bool(node.params.get("reverse")):
|
||
angle = -angle
|
||
tool = None
|
||
for solid in (session.adapter.revolve(face, angle, axis) for face in faces):
|
||
tool = session.adapter.fuse(tool, solid)
|
||
if tool is None:
|
||
raise ValueError("revolve produced no solid")
|
||
return _apply_primary_tool(
|
||
node, session, tool, cutting="cut" in node.atomic_id, topology_delta=topology_delta,
|
||
)
|
||
|
||
|
||
def _execute_extrude_from_face(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
resolved = [session.resolve(selector) for selector in node.selectors]
|
||
failed = next((item for item in resolved if item.status != "resolved"), None)
|
||
if failed or len(resolved) != 1 or resolved[0].record is None or resolved[0].record.kind != "face":
|
||
raise ValueError(failed.diagnostic.message if failed and failed.diagnostic else "derived profile face is unresolved")
|
||
face = resolved[0].record.value
|
||
tool, topology_delta = _extruded_tool(node, [face], session.adapter.face_normal(face), session)
|
||
return _apply_primary_tool(
|
||
node, session, tool, cutting=node.params.get("operation") == "cut", topology_delta=topology_delta,
|
||
)
|
||
|
||
|
||
def _execute_revolve_surface(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# Surface revolve 的 profile 是单一闭合 wire。它只生成独立 shell,不能参与
|
||
# 当前实体 body 的 fuse/cut,也不能把其结果误报为新的实体 body。
|
||
sketch = session.sketches.get(str(node.sketch_id))
|
||
if sketch is None:
|
||
raise ValueError("surface revolve has no resolved sketch")
|
||
faces = session.adapter.faces_for_sketch(sketch)
|
||
if len(faces) != 1 or faces[0].inner_wires():
|
||
raise ValueError("surface revolve requires exactly one closed profile without holes")
|
||
axis = _revolve_axis(node, session)
|
||
_validate_revolve_axis_in_sketch_plane(axis, sketch)
|
||
angle = float(node.params.get("angle_deg") or 0.0)
|
||
if angle <= 0:
|
||
raise ValueError("surface revolve requires angle_deg > 0")
|
||
if bool(node.params.get("reverse")):
|
||
angle = -angle
|
||
surface_id = session.register_surface(
|
||
node.feature_id,
|
||
session.adapter.revolve_surface(faces[0].outer_wire(), angle, axis),
|
||
)
|
||
return session.result(node, include_body=False, surface_id=surface_id)
|
||
|
||
|
||
def _execute_extrude_surface(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# surfaceEntities 的曲面拉伸沿用实体特征已 lower 的距离,但始终独立登记为
|
||
# shell。它既不改变 active solid,也不以曲面参与实体 fuse/cut。
|
||
sketch = session.sketches.get(str(node.sketch_id))
|
||
if sketch is None:
|
||
raise ValueError("surface extrude has no resolved sketch")
|
||
direction = vector_unit(_normal_from_sketch(sketch), field_name="sketch normal")
|
||
if bool(node.params.get("reverse")):
|
||
direction = vector_scale(direction, -1)
|
||
distance = float(node.params.get("distance_mm") or 0.0)
|
||
if distance <= 0:
|
||
raise ValueError("surface extrude requires distance_mm > 0")
|
||
wires = session.adapter.surface_wires_for_sketch(sketch)
|
||
surface = session.adapter.extrude_surface(wires, vector_scale(direction, distance))
|
||
reverse_distance = float(node.params.get("reverse_distance_mm") or 0.0)
|
||
if reverse_distance > 0:
|
||
opposite = session.adapter.extrude_surface(wires, vector_scale(direction, -reverse_distance))
|
||
surface = session.adapter.combine_surfaces(surface, opposite)
|
||
surface_id = session.register_surface(node.feature_id, surface)
|
||
return session.result(node, include_body=False, surface_id=surface_id)
|
||
|
||
|
||
def _combine_members(session: ExecutionSession, members: dict[str, Any]) -> Any:
|
||
body = None
|
||
for member in members.values():
|
||
body = session.adapter.combine(body, member)
|
||
if body is None:
|
||
raise ValueError("booleanBodies produced no result bodies")
|
||
return body
|
||
|
||
|
||
def _execute_boolean_bodies(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# booleanBodies 总是作用于 source feature 的明确 body 输出,不能回退为
|
||
# 当前聚合 body。这样相邻独立实体不会意外成为工具或目标。
|
||
params = node.params
|
||
target_ids = _member_sources(
|
||
node, session, "target_feature_ids", pattern_instance_parameter="target_pattern_instance_refs",
|
||
)
|
||
tool_ids = _member_sources(
|
||
node, session, "tool_feature_ids", pattern_instance_parameter="tool_pattern_instance_refs",
|
||
)
|
||
targets = {feature_id: session.body_members[feature_id] for feature_id in target_ids}
|
||
tools = {feature_id: session.body_members[feature_id] for feature_id in tool_ids}
|
||
target = _combine_members(session, targets)
|
||
tool = _combine_members(session, tools)
|
||
operation = str(params.get("operation") or "")
|
||
topology_delta: TopologyDelta | None = None
|
||
if operation == "union":
|
||
result, topology_delta = session.adapter.fuse_with_topology_delta(target, tool)
|
||
elif operation == "subtract":
|
||
result, topology_delta = session.adapter.cut_with_topology_delta(target, tool)
|
||
elif operation == "intersect":
|
||
result, topology_delta = session.adapter.intersect_with_topology_delta(target, tool)
|
||
else:
|
||
raise ValueError(f"unsupported booleanBodies operation {operation!r}")
|
||
members = {
|
||
feature_id: body
|
||
for feature_id, body in session.body_members.items()
|
||
if feature_id not in set(target_ids + tool_ids)
|
||
}
|
||
members[node.feature_id] = result
|
||
if bool(params.get("keep_tools")):
|
||
members.update(tools)
|
||
session.register_body(
|
||
node.feature_id, _combine_members(session, members), body_members=members, topology_delta=topology_delta,
|
||
)
|
||
return session.result(node)
|
||
|
||
|
||
def _pattern_instance_sources(
|
||
node: FeaturePlanNode,
|
||
session: ExecutionSession,
|
||
parameter: str = "pattern_instance_refs",
|
||
) -> list[str]:
|
||
"""Resolve CDSL pattern-instance refs to their internal body-member keys."""
|
||
resolved: list[str] = []
|
||
for reference in node.params.get(parameter) or ():
|
||
if not isinstance(reference, dict):
|
||
raise ValueError("pattern instance reference must be an object")
|
||
pattern_id = str(reference.get("pattern_feature_id") or "")
|
||
source_id = str(reference.get("source_feature_id") or "")
|
||
instance = reference.get("instance_index")
|
||
if not pattern_id or not source_id or not isinstance(instance, int):
|
||
raise ValueError("pattern instance reference is incomplete")
|
||
pattern = session.nodes.get(pattern_id)
|
||
if pattern is None or pattern.atomic_id not in {"pattern_circular", "pattern_mirror"}:
|
||
raise ValueError(f"pattern instance owner is unavailable: {pattern_id}")
|
||
params = pattern.params
|
||
if source_id not in {str(value) for value in params.get("source_feature_ids") or ()}:
|
||
raise ValueError("pattern instance source is not selected by its pattern")
|
||
count = int(params.get("pattern_count") or 0)
|
||
excluded = {int(value) for value in params.get("excluded_instance_indices") or ()}
|
||
if (
|
||
pattern.atomic_id == "pattern_mirror" and instance != 1
|
||
) or (
|
||
pattern.atomic_id == "pattern_circular" and (instance < 1 or instance >= count or instance in excluded)
|
||
):
|
||
raise ValueError("pattern instance is outside the pattern's surviving instances")
|
||
member_id = pattern_instance_member_id(pattern_id, source_id, instance)
|
||
if member_id not in session.body_members:
|
||
raise ValueError(f"pattern instance body is unavailable: {pattern_id}/{source_id}/{instance}")
|
||
if member_id not in resolved:
|
||
resolved.append(member_id)
|
||
return resolved
|
||
|
||
|
||
def _transform_copy_sources(node: FeaturePlanNode, session: ExecutionSession) -> list[str]:
|
||
"""Resolve source-qualified outputs of preceding multi-body COPY transforms."""
|
||
resolved: list[str] = []
|
||
for reference in node.params.get("transform_copy_refs") or ():
|
||
if not isinstance(reference, dict):
|
||
raise ValueError("transform COPY reference must be an object")
|
||
transform_id = str(reference.get("transform_feature_id") or "")
|
||
source_id = str(reference.get("source_feature_id") or "")
|
||
if not transform_id or not source_id:
|
||
raise ValueError("transform COPY reference is incomplete")
|
||
transform = session.nodes.get(transform_id)
|
||
params = transform.params if transform is not None else {}
|
||
sources = params.get("source_feature_ids") or []
|
||
if (
|
||
transform is None
|
||
or transform.atomic_id != "transform_bodies"
|
||
or not bool(params.get("make_copy"))
|
||
or not isinstance(sources, list)
|
||
or len(sources) < 2
|
||
or source_id not in {str(value) for value in sources}
|
||
):
|
||
raise ValueError(f"transform COPY owner/source is unavailable: {transform_id}/{source_id}")
|
||
member_id = transform_copy_member_id(transform_id, source_id)
|
||
if member_id not in session.body_members:
|
||
raise ValueError(f"transform COPY body is unavailable: {transform_id}/{source_id}")
|
||
if member_id not in resolved:
|
||
resolved.append(member_id)
|
||
return resolved
|
||
|
||
|
||
def _member_sources(
|
||
node: FeaturePlanNode,
|
||
session: ExecutionSession,
|
||
parameter: str,
|
||
*,
|
||
pattern_instance_parameter: str | None = None,
|
||
allow_transform_copies: bool = False,
|
||
) -> list[str]:
|
||
source_ids = [str(value) for value in node.params.get(parameter) or []]
|
||
if pattern_instance_parameter is not None:
|
||
source_ids.extend(_pattern_instance_sources(node, session, pattern_instance_parameter))
|
||
if allow_transform_copies:
|
||
source_ids.extend(_transform_copy_sources(node, session))
|
||
if not source_ids:
|
||
raise ValueError(f"{node.atomic_id} requires explicit {parameter}")
|
||
missing = [feature_id for feature_id in source_ids if feature_id not in session.body_members]
|
||
if missing:
|
||
raise ValueError(f"{node.atomic_id} source bodies are unavailable: " + ", ".join(missing))
|
||
return source_ids
|
||
|
||
|
||
def _execute_transform_bodies(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# FeatureScript transform targets explicit bodies. Do not move the
|
||
# aggregate session body, because it may include unrelated members.
|
||
source_ids = _member_sources(
|
||
node, session, "source_feature_ids", pattern_instance_parameter="pattern_instance_refs", allow_transform_copies=True,
|
||
)
|
||
make_copy = bool(node.params.get("make_copy"))
|
||
direct_sources = node.params.get("source_feature_ids") or []
|
||
if make_copy and isinstance(direct_sources, list) and len(direct_sources) > 1:
|
||
# The aggregate is only an export compound. Each source transform has
|
||
# its own B-rep builder and is the only output a later COPY query may
|
||
# select. Do not attach an aggregate topology delta to source members.
|
||
members = dict(session.body_members)
|
||
members.update({
|
||
transform_copy_member_id(node.feature_id, source_id): session.adapter.transform(
|
||
session.body_members[source_id], dict(node.params.get("transform") or {}),
|
||
)
|
||
for source_id in source_ids
|
||
})
|
||
session.register_body(
|
||
node.feature_id, _combine_members(session, members), body_members=members,
|
||
)
|
||
return session.result(node)
|
||
source = _combine_members(session, {feature_id: session.body_members[feature_id] for feature_id in source_ids})
|
||
transformed, topology_delta = session.adapter.transform_with_topology_delta(
|
||
source, dict(node.params.get("transform") or {}),
|
||
)
|
||
members = dict(session.body_members)
|
||
if not make_copy:
|
||
for feature_id in source_ids:
|
||
members.pop(feature_id)
|
||
members[node.feature_id] = transformed
|
||
session.register_body(
|
||
node.feature_id, _combine_members(session, members), body_members=members, topology_delta=topology_delta,
|
||
)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_delete_bodies(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# Deletion is a body-graph operation, never a Boolean subtraction. A
|
||
# selected member can be disjoint or overlap another independent body.
|
||
source_ids = _member_sources(node, session, "target_feature_ids")
|
||
members = {feature_id: body for feature_id, body in session.body_members.items() if feature_id not in set(source_ids)}
|
||
if members:
|
||
session.register_body(node.feature_id, _combine_members(session, members), body_members=members)
|
||
else:
|
||
session.clear_body()
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_loft_add(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 放样截面不占用 feature.sketch_id;按有序 profile_sketch_ids 取已解析
|
||
# 草图,并由 adapter 统一校验单闭环、无内环等内核输入约束。
|
||
profile_ids = node.params.get("profile_sketch_ids") or []
|
||
profiles: list[dict[str, Any]] = []
|
||
for sketch_id in profile_ids:
|
||
sketch = session.sketches.get(str(sketch_id))
|
||
if sketch is None:
|
||
raise ValueError(f"loft profile sketch {sketch_id!r} is not resolved")
|
||
profiles.append(sketch)
|
||
solid, topology_delta = session.adapter.loft_with_topology_delta(profiles)
|
||
body = session.adapter.fuse(session.body, solid)
|
||
# Fusing a loft into an existing body replaces its subshapes through a
|
||
# different builder. Only an initial direct loft can expose this builder's
|
||
# cap evidence for the final B-rep snapshot.
|
||
if session.body is not None:
|
||
topology_delta = None
|
||
session.register_body(node.feature_id, body, replay_node=node, topology_delta=topology_delta)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_loft_add_with_cap_face(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
resolved = [session.resolve(selector) for selector in node.selectors]
|
||
failed = next((item for item in resolved if item.status != "resolved"), None)
|
||
if failed or len(resolved) != 1 or resolved[0].record is None or resolved[0].record.kind != "face":
|
||
raise ValueError(failed.diagnostic.message if failed and failed.diagnostic else "cap-face loft selector is unresolved")
|
||
profile_ids = node.params.get("profile_sketch_ids") or []
|
||
profiles: list[dict[str, Any]] = []
|
||
for sketch_id in profile_ids:
|
||
sketch = session.sketches.get(str(sketch_id))
|
||
if sketch is None:
|
||
raise ValueError(f"loft profile sketch {sketch_id!r} is not resolved")
|
||
profiles.append(sketch)
|
||
solid = session.adapter.loft_with_cap_face(resolved[0].record.value, profiles)
|
||
session.register_body(node.feature_id, session.adapter.fuse(session.body, solid), replay_node=node)
|
||
return session.result(node)
|
||
|
||
|
||
def _sweep_path(node: FeaturePlanNode, session: ExecutionSession) -> Any:
|
||
# 路径是 self-contained CDSL 数据,避免重放时依赖临时草图或 source id。
|
||
path = node.params.get("path") or {}
|
||
if not isinstance(path, dict):
|
||
raise ValueError("sweep path must be an object")
|
||
plane = PlaneSpec.from_mapping(path.get("workplane") or {})
|
||
segment = path.get("segment") or {}
|
||
if not isinstance(segment, dict):
|
||
raise ValueError("sweep path segment must be an object")
|
||
kind = str(segment.get("type") or "")
|
||
if kind == "line":
|
||
local_points = [segment.get("start"), segment.get("end")]
|
||
elif kind == "bspline":
|
||
local_points = segment.get("points") or []
|
||
else:
|
||
raise ValueError(f"unsupported sweep path segment {kind!r}")
|
||
if len(local_points) < 2 or any(not isinstance(point, list) or len(point) != 2 for point in local_points):
|
||
raise ValueError("sweep path requires two-dimensional points")
|
||
|
||
def point(value: list[float]) -> Vector3:
|
||
return vector_add(
|
||
plane.origin_mm,
|
||
vector_add(vector_scale(plane.x_dir, float(value[0])), vector_scale(plane.y_dir, float(value[1]))),
|
||
)
|
||
|
||
def tangent(value: Any) -> Vector3 | None:
|
||
if value is None:
|
||
return None
|
||
if not isinstance(value, list) or len(value) != 2:
|
||
raise ValueError("sweep path tangent must contain two coordinates")
|
||
return vector_add(vector_scale(plane.x_dir, float(value[0])), vector_scale(plane.y_dir, float(value[1])))
|
||
|
||
return session.adapter.sweep_path(
|
||
[point(value) for value in local_points],
|
||
start_tangent=tangent(segment.get("start_tangent")),
|
||
end_tangent=tangent(segment.get("end_tangent")),
|
||
parameters=[float(value) for value in segment.get("parameters") or []] or None,
|
||
)
|
||
|
||
|
||
def _execute_sweep_add(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None = None) -> FeatureResult:
|
||
profile = sketch or session.sketches.get(str(node.sketch_id))
|
||
if profile is None:
|
||
raise ValueError("sweep has no resolved profile sketch")
|
||
faces = session.adapter.faces_for_sketch(profile)
|
||
if len(faces) != 1:
|
||
raise ValueError("sweep requires exactly one closed profile region")
|
||
solid, topology_delta = session.adapter.sweep_with_topology_delta(
|
||
faces[0], _sweep_path(node, session),
|
||
is_frenet=bool(node.params.get("is_frenet", False)),
|
||
)
|
||
is_new_body = node.params.get("result_mode") == "new_body"
|
||
body = session.adapter.combine(session.body, solid) if is_new_body else session.adapter.fuse(session.body, solid)
|
||
# A union rebuilds topology, so the pipe-shell builder cannot prove the
|
||
# final aggregate's relations. The independent-body path retains its exact
|
||
# subshape identity and may expose evidence for the new member.
|
||
if session.body is not None and not is_new_body:
|
||
topology_delta = None
|
||
session.register_body(node.feature_id, body, replay_node=node, topology_delta=topology_delta)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_reference_plane(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 基准面特征(reference_plane)执行入口:从参数解析平面并登记为拓扑上下文。
|
||
|
||
# 1. 从特征参数 plane 中解析出平面定义 PlaneSpec(原点到法向)。
|
||
plane = PlaneSpec.from_mapping(node.params.get("plane") or {})
|
||
# 2. 将该平面注册到拓扑上下文,供后续特征(如草图基准、参考轴)引用。
|
||
session.topology.register_context(node.feature_id, plane)
|
||
# 3. 返回结果对象,并将该平面作为上下文一并携带。
|
||
return session.result(node, context=plane)
|
||
|
||
|
||
def _execute_reference_axis(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 基准轴特征(reference_axis)执行入口:由参数直接定义轴,或由两个基准平面求交线得到轴。
|
||
|
||
# 1. 尝试直接取参数:若同时给出原点 origin_mm 与方向 direction,则直接构造轴。
|
||
params = node.params.get("axis") or {}
|
||
if params.get("origin_mm") and params.get("direction"):
|
||
axis = AxisSpec.from_mapping(params)
|
||
else:
|
||
# 2. 否则从特征选择器中筛选出已解析的基准平面。
|
||
planes = [session.resolve(selector) for selector in node.selectors if selector.get("kind") == "plane"]
|
||
resolved = [item.record.value for item in planes if item.status == "resolved" and isinstance(item.record.value, PlaneSpec)]
|
||
# 3. 校验:轴需要两个非平行的平面,不足两个则报错。
|
||
if len(resolved) < 2:
|
||
raise ValueError("reference axis requires two uniquely resolved planes")
|
||
# 4. 用两平面法线叉积求交线方向;若方向长度接近 0 说明两平面平行,无法成轴。
|
||
first, second = resolved[0], resolved[1]
|
||
n1, n2 = first.normal, second.normal
|
||
direction = vector_cross(n1, n2)
|
||
squared_length = vector_dot(direction, direction)
|
||
if squared_length <= 1e-18:
|
||
raise ValueError("reference planes are parallel and cannot define an axis")
|
||
# 5. 求交线上的一点:两平面到各自原点的垂距参与线性组合,得到交线上的最近点。
|
||
d1 = vector_dot(n1, first.origin_mm)
|
||
d2 = vector_dot(n2, second.origin_mm)
|
||
point = vector_scale(vector_add(vector_scale(vector_cross(n2, direction), d1), vector_scale(vector_cross(direction, n1), d2)), 1 / squared_length)
|
||
# 6. 由该点与归一化的交线方向组合成基准轴 AxisSpec。
|
||
axis = AxisSpec(origin_mm=point, direction=vector_unit(direction, field_name="reference axis"))
|
||
# 7. 注册为拓扑上下文,并返回结果对象(携带该轴)。
|
||
session.topology.register_context(node.feature_id, axis)
|
||
return session.result(node, context=axis)
|
||
|
||
|
||
def _register_added_solid(
|
||
session: ExecutionSession,
|
||
node: FeaturePlanNode,
|
||
solid: Any,
|
||
) -> None:
|
||
"""Register an additive primitive solid (box/cyl/sphere/thread/gear/rack/bend).
|
||
|
||
When ``node.params['result_mode'] == "new_body"`` the primitive is kept as
|
||
an independent body member so that downstream ``boolean_bodies`` can
|
||
reference it without pulling in the accumulated fuse history. The current
|
||
body is replaced by a Compound that preserves both, matching the
|
||
``extrude_add_blind`` ``new_body`` semantics. Any other value (including
|
||
missing) falls back to the legacy fuse-into-body behavior.
|
||
"""
|
||
if node.params.get("result_mode") == "new_body":
|
||
combined = session.adapter.combine(session.body, solid)
|
||
members = {**session.body_members, node.feature_id: solid}
|
||
session.register_body(node.feature_id, combined, replay_node=node, body_members=members)
|
||
return
|
||
fused = session.adapter.fuse(session.body, solid)
|
||
session.register_body(node.feature_id, fused, replay_node=node)
|
||
|
||
|
||
def _execute_sphere(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 球体特征(sphere_add)执行入口:按球心与半径生成球体并并入当前主体。
|
||
|
||
# 1. 解析参数:半径 radius_mm 与球心 center_mm。
|
||
radius = float(node.params.get("radius_mm") or 0.0)
|
||
center = node.params.get("center_mm") or []
|
||
# 2. 校验:半径必须大于 0,球心必须是三维坐标。
|
||
if radius <= 0 or len(center) != 3:
|
||
raise ValueError("sphere_add requires radius_mm and a three-dimensional center_mm")
|
||
# 3. 由适配器创建球体实体。
|
||
solid = session.adapter.sphere(radius, (float(center[0]), float(center[1]), float(center[2])))
|
||
# 4. 球体与当前主体做布尔并(fuse)后登记为新主体,并返回该特征的结果对象。
|
||
_register_added_solid(session, node, solid)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_box(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 长方体特征(box_add)执行入口:以几何中心 center_mm 与三向尺寸生成原生长方体。
|
||
# 1. 解析并校验尺寸与中心,非法输入抛出带具体原因的 ValueError。
|
||
try:
|
||
length = float(node.params.get("length_mm") or 0.0)
|
||
width = float(node.params.get("width_mm") or 0.0)
|
||
height = float(node.params.get("height_mm") or 0.0)
|
||
center = node.params.get("center_mm") or []
|
||
except (TypeError, ValueError) as error:
|
||
raise ValueError("box dimensions must be numeric") from error
|
||
if length <= 0 or width <= 0 or height <= 0 or len(center) != 3:
|
||
raise ValueError("box_add requires positive length_mm/width_mm/height_mm and a three-dimensional center_mm")
|
||
# 2. 生成世界轴对齐的 plane frame:plane 原点是长方体的最小角点(中心减去半
|
||
# 尺寸),长/宽/高分别沿世界 x/y/z 生长(build123d Solid.make_box 语义)。
|
||
corner = (
|
||
float(center[0]) - length / 2,
|
||
float(center[1]) - width / 2,
|
||
float(center[2]) - height / 2,
|
||
)
|
||
plane = PlaneSpec.from_mapping({"origin_mm": corner, "x_dir": [1, 0, 0], "normal": [0, 0, 1]})
|
||
solid = session.adapter.box(length, width, height, plane)
|
||
# 3. 与当前主体做布尔并后登记为新主体,并返回该特征的结果对象。
|
||
_register_added_solid(session, node, solid)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_cylinder(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 圆柱特征(cylinder_add)执行入口:axis 的原点是底面圆心、方向为轴向;
|
||
# axis 缺省为世界 +Z 过原点(底面圆心落在 (0,0,0))。
|
||
# 1. 解析并校验半径与高度,非法输入抛出带具体原因的 ValueError。
|
||
try:
|
||
radius = float(node.params.get("radius_mm") or 0.0)
|
||
height = float(node.params.get("height_mm") or 0.0)
|
||
except (TypeError, ValueError) as error:
|
||
raise ValueError("cylinder dimensions must be numeric") from error
|
||
if radius <= 0 or height <= 0:
|
||
raise ValueError("cylinder_add requires positive radius_mm and height_mm")
|
||
raw_axis = node.params.get("axis")
|
||
if raw_axis is not None and not (
|
||
isinstance(raw_axis, dict) and raw_axis.get("origin_mm") is not None and raw_axis.get("direction") is not None
|
||
):
|
||
raise ValueError("cylinder_add axis must define origin_mm and direction")
|
||
axis = AxisSpec.from_mapping(raw_axis) if isinstance(raw_axis, dict) else None
|
||
# 2. 由适配器创建原生圆柱(axis=None 即世界 +Z 过原点)。
|
||
solid = session.adapter.cylinder(radius, height, axis)
|
||
# 3. 与当前主体做布尔并后登记为新主体,并返回该特征的结果对象。
|
||
_register_added_solid(session, node, solid)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_thread(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 螺纹特征(thread_add)执行入口:按规格生成参数化螺纹段并并入当前主体。
|
||
# 1. 解析并校验尺寸/牙距/轴,非法输入抛出带具体原因的 ValueError。
|
||
spec = ThreadSpec.from_feature(node.atomic_id, node.params)
|
||
# 2. 由适配器门面生成沿 spec.axis 放置的外螺纹实心段。
|
||
solid = session.adapter.thread_solid(spec)
|
||
# 3. 与当前主体做布尔并(fuse)后登记为新主体,并返回该特征的结果对象。
|
||
_register_added_solid(session, node, solid)
|
||
return session.result(node)
|
||
|
||
|
||
def _thread_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
# 螺纹特征(thread_add/thread_cut)不需要草图平面,丢弃该参数后执行。
|
||
# thread_cut 走布尔差分支:从已有主体切出内螺纹槽,而非并入外螺纹段。
|
||
del sketch
|
||
if node.atomic_id == "thread_cut":
|
||
return _execute_thread_cut(node, session)
|
||
return _execute_thread(node, session)
|
||
|
||
|
||
def _execute_thread_cut(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 内螺纹(thread_cut)执行入口:ThreadSpec.from_feature 对 thread_cut 恒置
|
||
# internal=True,生成牙顶外放 INTERNAL_CUT_OVERLAP_MM 的切削刀具,沿
|
||
# spec.axis 放置后从当前主体布尔差出全深螺旋牙槽(宿主通常已预打光孔,
|
||
# 刀具 core 落在孔腔中,仅外放的牙槽层切入孔壁)。
|
||
# 1. 解析并校验尺寸/牙距/轴,非法输入抛出带具体原因的 ValueError。
|
||
spec = ThreadSpec.from_feature(node.atomic_id, node.params)
|
||
# 2. 由适配器门面生成沿 spec.axis 放置的内螺纹切削刀具实心段。
|
||
tool = session.adapter.thread_solid(spec)
|
||
# 3. 从当前主体布尔差(cut)后登记为新主体,并返回该特征的结果对象。
|
||
session.register_body(node.feature_id, session.adapter.cut(session.body, tool), replay_node=node)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_bend(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 折弯特征(bend_add)执行入口:按规格生成等厚折弯板并并入当前主体。
|
||
# 1. 解析并校验板厚/宽度/折痕链与放置平面,非法输入抛出带具体原因的 ValueError。
|
||
spec = BendSpec.from_feature(node.params)
|
||
# 2. 由适配器门面生成沿 spec.frame 放置的折弯实心段。
|
||
solid = session.adapter.bend_solid(spec)
|
||
# 3. 与当前主体做布尔并(fuse)后登记为新主体,并返回该特征的结果对象。
|
||
_register_added_solid(session, node, solid)
|
||
return session.result(node)
|
||
|
||
|
||
def _bend_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
# 折弯特征(bend_add)不需要草图平面,丢弃该参数后执行。
|
||
del sketch
|
||
return _execute_bend(node, session)
|
||
|
||
|
||
def _execute_gear(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 齿轮特征(gear_add)执行入口:按规格生成渐开线齿轮并入当前主体。
|
||
# 1. 解析并校验模数/齿数/齿宽/螺旋角与放置轴,非法输入抛出带具体原因的 ValueError。
|
||
spec = GearSpec.from_feature(node.params)
|
||
# 2. 由适配器门面生成沿 spec.axis 放置的齿轮实体(直齿/斜齿/人字齿)。
|
||
solid = session.adapter.gear_solid(spec)
|
||
# 3. 与当前主体做布尔并(fuse)后登记为新主体,并返回该特征的结果对象。
|
||
_register_added_solid(session, node, solid)
|
||
# 4. 小齿数根切风险:不阻断执行,附加 info 级诊断供完成报告如实披露。
|
||
diagnostics: list[RuntimeDiagnostic] = []
|
||
if spec.teeth_count < 17:
|
||
diagnostics.append(RuntimeDiagnostic(
|
||
code="undercut_risk",
|
||
message=(
|
||
f"Gear with {spec.teeth_count} teeth and a 20 degree pressure angle is undercut-prone; "
|
||
"standard involute geometry is generated without profile shift"
|
||
),
|
||
feature_id=node.feature_id,
|
||
))
|
||
return session.result(node, diagnostics=diagnostics)
|
||
|
||
|
||
def _gear_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
# 齿轮特征(gear_add)不需要草图平面,丢弃该参数后执行。
|
||
del sketch
|
||
return _execute_gear(node, session)
|
||
|
||
|
||
def _execute_rack(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 齿条特征(rack_add)执行入口:按规格生成直线齿条并入当前主体。
|
||
# 1. 解析并校验模数/齿数/厚度/压力角与放置轴,非法输入抛出带具体原因的 ValueError。
|
||
spec = RackSpec.from_feature(node.params)
|
||
# 2. 由适配器门面生成沿 spec.axis 放置的齿条实体(齿沿轴方向伸出)。
|
||
solid = session.adapter.rack_solid(spec)
|
||
# 3. 与当前主体做布尔并(fuse)后登记为新主体,并返回该特征的结果对象。
|
||
_register_added_solid(session, node, solid)
|
||
return session.result(node)
|
||
|
||
|
||
def _rack_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
# 齿条特征(rack_add)不需要草图平面,丢弃该参数后执行。
|
||
del sketch
|
||
return _execute_rack(node, session)
|
||
|
||
|
||
def _host_plane(resolution: SelectorResolution) -> PlaneSpec:
|
||
if resolution.record is None:
|
||
raise ValueError(resolution.diagnostic.message if resolution.diagnostic else "host face was not resolved")
|
||
geometry = resolution.record.geometry
|
||
return PlaneSpec.from_mapping({
|
||
"origin_mm": geometry["center_mm"],
|
||
"x_dir": [1, 0, 0] if abs(float(geometry["normal"][0])) < 0.9 else [0, 1, 0],
|
||
"normal": geometry["normal"],
|
||
})
|
||
|
||
|
||
def _hole_starts(
|
||
spec: HoleSpec,
|
||
*,
|
||
host_plane: PlaneSpec,
|
||
positions_are_local: bool,
|
||
) -> list[Vector3]:
|
||
starts: list[Vector3] = []
|
||
for point in spec.positions_mm:
|
||
if positions_are_local:
|
||
start = vector_add(
|
||
vector_add(
|
||
vector_add(host_plane.origin_mm, vector_scale(host_plane.x_dir, point[0])),
|
||
vector_scale(host_plane.y_dir, point[1]),
|
||
),
|
||
vector_scale(host_plane.normal, point[2]),
|
||
)
|
||
else:
|
||
start = point
|
||
starts.append(start)
|
||
return starts
|
||
|
||
|
||
def _execute_hole(node: FeaturePlanNode, session: ExecutionSession, *, wizard: bool = False) -> FeatureResult:
|
||
# 孔特征(hole)执行入口:在指定宿主面上按孔规格生成切除工具,并从主体上减去。
|
||
|
||
# 1. 校验:孔是切除操作,必须先有主体。
|
||
if session.body is None:
|
||
raise ValueError("hole feature has no body")
|
||
# 2. 确定宿主面 host_face:
|
||
host_selector = node.params.get("host_face")
|
||
if isinstance(host_selector, dict) and isinstance(host_selector.get("frame"), dict):
|
||
# 若直接带 frame(平面定义),则以该平面为宿主,孔位按局部坐标解释。
|
||
host = PlaneSpec.from_mapping(host_selector["frame"])
|
||
positions_are_local = True
|
||
else:
|
||
# 否则从特征选择器中取 face,解析出宿主平面,孔位按世界坐标解释。
|
||
selectors = list(node.selectors)
|
||
if isinstance(host_selector, dict):
|
||
selectors.append(host_selector)
|
||
selector = next((item for item in selectors if item.get("kind") == "face"), None)
|
||
if selector is None:
|
||
raise ValueError("hole requires host_face selector or frame")
|
||
host = _host_plane(session.resolve(selector))
|
||
positions_are_local = False
|
||
# 3. 解析孔规格 HoleSpec(直径、深度、类型等,wizard 模式提供额外默认值)。
|
||
spec = HoleSpec.from_feature(node.atomic_id, node.params, wizard=wizard)
|
||
# 4. A host-face normal is an outward B-rep orientation, so its inverse
|
||
# always enters the material. Inferring direction from the global body
|
||
# centre fails for concave or multi-leg parts: for example, the top face
|
||
# of an L bracket can sit below the whole body's centre and the old rule
|
||
# drilled outward, producing a no-op feature reported as successful.
|
||
# The selected topology face is the local, authoritative orientation.
|
||
inward = vector_scale(host.normal, -1)
|
||
# 5. 生成孔切除工具:按孔规格、起始位置、内方向及“贯穿到主体底面”的深度构造工具实体。
|
||
tool = session.adapter.hole_tool(
|
||
spec,
|
||
_hole_starts(spec, host_plane=host, positions_are_local=positions_are_local),
|
||
inward,
|
||
session.adapter.body_span(session.body, inward) + 2.0,
|
||
)
|
||
# 6. 从主体上减去工具实体,登记新主体并返回结果。
|
||
# thread 是装饰螺纹(无螺距、不进实体几何,SolidWorks/STEP 的螺纹孔
|
||
# 即光滑孔):孔按光滑圆柱孔执行,同时记录 info 级诊断便于批量报告
|
||
# 追溯降级数量(issue #9,capabilities 已不再拒绝 thread)。
|
||
diagnostics: list[RuntimeDiagnostic] = []
|
||
if wizard and node.params.get("thread"):
|
||
diagnostics.append(RuntimeDiagnostic(
|
||
code="thread_decoration_ignored",
|
||
message="Thread decoration is not modeled; the hole falls back to a plain cylindrical bore",
|
||
feature_id=node.feature_id,
|
||
))
|
||
session.register_body(node.feature_id, session.adapter.cut(session.body, tool), replay_node=node)
|
||
return session.result(node, diagnostics=diagnostics)
|
||
|
||
|
||
def _selector_edges(node: FeaturePlanNode, session: ExecutionSession, *, tangent_propagation: bool = False) -> list[Any]:
|
||
resolved: list[SelectorResolution] = [session.resolve(selector) for selector in node.selectors]
|
||
failed = next((item for item in resolved if item.status != "resolved"), None)
|
||
if failed:
|
||
raise ValueError(failed.diagnostic.message if failed.diagnostic else "selector resolution failed")
|
||
|
||
def is_body_boundary(edge: Any) -> bool:
|
||
# 圆柱、圆锥等周期面会带一条仅属于自身的参数 seam。该线不是实体
|
||
# 边界;FeatureScript 以 FACE 选择倒角时不应将其当作额外的待倒角边,
|
||
# 否则连续的锥面会被错误切成两段。显式 EDGE selector 仍可表达真正的
|
||
# 单边选择,所以这里只约束由 FACE 展开的候选边。
|
||
face_count = sum(
|
||
1
|
||
for face in session.body.faces()
|
||
if any(candidate.is_same(edge) for candidate in face.edges())
|
||
)
|
||
return face_count >= 2
|
||
|
||
edges: list[Any] = []
|
||
for item in resolved:
|
||
if item.record.kind == "edge":
|
||
edges.append(item.record.value)
|
||
elif item.record.kind == "face":
|
||
edges.extend(edge for edge in item.record.value.edges() if is_body_boundary(edge))
|
||
if not edges:
|
||
raise ValueError("selectors did not resolve any edges")
|
||
return session.adapter.tangent_edges(session.body, edges) if tangent_propagation else edges
|
||
|
||
|
||
def _shell_target(node: FeaturePlanNode, session: ExecutionSession) -> tuple[Any, list[Any]]:
|
||
# shell 的 remove-face selector 必须全部属于同一实体。CADFS 允许一个
|
||
# Compound 中保留多个独立 body,不能将整组 body 交给 OCC 后由内核猜测
|
||
# 应抽壳的成员。
|
||
resolved = [session.resolve(selector) for selector in node.selectors]
|
||
failed = next((item for item in resolved if item.status != "resolved"), None)
|
||
if failed:
|
||
raise ValueError(failed.diagnostic.message if failed.diagnostic else "selector resolution failed")
|
||
records = [item.record for item in resolved if item.record is not None]
|
||
if not records or any(record.kind != "face" for record in records):
|
||
raise ValueError("shell selectors must resolve to faces")
|
||
target_ids = {record.body_id for record in records}
|
||
if len(target_ids) != 1:
|
||
raise ValueError("shell faces must belong to one target body")
|
||
target_id = next(iter(target_ids))
|
||
members = session.adapter.body_solids(session.body)
|
||
if len(members) == 1:
|
||
target = members[0]
|
||
else:
|
||
if target_id is None or session.body_id is None:
|
||
raise ValueError("shell target body is unresolved")
|
||
prefix = f"{session.body_id}:"
|
||
if not target_id.startswith(prefix):
|
||
raise ValueError("shell target body is outside the active body set")
|
||
try:
|
||
member_index = int(target_id[len(prefix):])
|
||
except ValueError as error:
|
||
raise ValueError("shell target body has an invalid member id") from error
|
||
if member_index < 0 or member_index >= len(members):
|
||
raise ValueError("shell target body member is unavailable")
|
||
target = members[member_index]
|
||
target_feature_id = node.params.get("target_feature_id")
|
||
if target_feature_id is not None:
|
||
if not isinstance(target_feature_id, str) or not target_feature_id:
|
||
raise ValueError("shell target_feature_id is invalid")
|
||
declared = session.body_members.get(target_feature_id)
|
||
if declared is None:
|
||
raise ValueError("shell target body is no longer an independently selectable member")
|
||
declared_solids = session.adapter.body_solids(declared)
|
||
if len(declared_solids) != 1:
|
||
raise ValueError("shell target body must resolve to exactly one active solid")
|
||
if not declared_solids[0].is_same(target):
|
||
raise ValueError("shell target body does not match the resolved face member")
|
||
return target, [record.value for record in records]
|
||
|
||
|
||
def _replace_shell_target(session: ExecutionSession, target: Any, replacement: Any) -> Any:
|
||
# 仅替换抽壳目标实体;其他独立实体保持原样和原有相对顺序。
|
||
members = session.adapter.body_solids(session.body)
|
||
if len(members) == 1:
|
||
return replacement
|
||
replaced = False
|
||
result = None
|
||
for member in members:
|
||
if member.is_same(target):
|
||
result = session.adapter.combine(result, replacement)
|
||
replaced = True
|
||
else:
|
||
result = session.adapter.combine(result, member)
|
||
if not replaced or result is None:
|
||
raise ValueError("shell target solid is no longer part of the active body")
|
||
return result
|
||
|
||
|
||
def _execute_shell(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 抽壳特征:移除 selector 所指面,并按 CADFS thickness 向实体内部偏置。
|
||
if session.body is None:
|
||
raise ValueError("shell has no body")
|
||
thickness = float(node.params.get("thickness_mm") or 0)
|
||
if thickness <= 0:
|
||
raise ValueError("shell thickness_mm must be > 0")
|
||
target, faces = _shell_target(node, session)
|
||
result, topology_delta = session.adapter.shell_with_topology_delta(
|
||
target, faces, thickness, inward=bool(node.params.get("inward", True)),
|
||
)
|
||
session.register_body(
|
||
node.feature_id, _replace_shell_target(session, target, result), replay_node=node,
|
||
topology_delta=topology_delta,
|
||
)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_fillet(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 圆角特征(fillet)执行入口:对选中边按半径做圆角,平滑尖角与棱边。
|
||
|
||
# 1. 校验:圆角作用于已有主体,必须先有主体。
|
||
if session.body is None:
|
||
raise ValueError("fillet has no body")
|
||
# 2. 解析圆角半径并校验必须大于 0。
|
||
radius = float(node.params.get("radius_mm") or 0)
|
||
if radius <= 0:
|
||
raise ValueError("fillet radius_mm must be > 0")
|
||
# 3. 解析目标边(支持 tangent_propagation 相切传播),并执行圆角。
|
||
body, topology_delta = session.adapter.fillet_with_topology_delta(
|
||
session.body, radius, _selector_edges(node, session, tangent_propagation=bool(node.params.get("tangent_propagation"))),
|
||
)
|
||
# 4. 登记新主体并返回结果。
|
||
session.register_body(node.feature_id, body, replay_node=node, topology_delta=topology_delta)
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_chamfer(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
# 倒角特征(chamfer)执行入口:对选中边按距离做倒角(可带第二距离形成不对称倒角)。
|
||
|
||
# 1. 校验:倒角作用于已有主体,必须先有主体。
|
||
if session.body is None:
|
||
raise ValueError("chamfer has no body")
|
||
# 2. 解析主距离并校验必须大于 0。
|
||
distance = float(node.params.get("distance_mm") or 0)
|
||
if distance <= 0:
|
||
raise ValueError("chamfer distance_mm must be > 0")
|
||
# 3. 解析第二距离与角度(importer 对 SolidWorks Distance-Angle 倒角产出
|
||
# angle_rad,单位为弧度)。第二距离 = 主距离 * tan(angle);angle=45° 时
|
||
# tan=1,退化为等距倒角(与历史行为一致,零回归)。
|
||
# 注意:build123d 的 length/length2 侧向分配依赖面的枚举顺序,对非 45°
|
||
# 倒角仅保证量级正确,距离所在侧可能反转。
|
||
distance_2 = node.params.get("distance_2_mm")
|
||
angle_rad = node.params.get("angle_rad")
|
||
if distance_2 is None and angle_rad is not None:
|
||
distance_2 = distance * math.tan(float(angle_rad))
|
||
# 4. 解析目标边(支持相切传播),执行倒角。
|
||
edges = _selector_edges(node, session, tangent_propagation=bool(node.params.get("tangent_propagation")))
|
||
diagnostics: list[RuntimeDiagnostic] = []
|
||
topology_delta: TopologyDelta | None = None
|
||
try:
|
||
body, topology_delta = session.adapter.chamfer_with_topology_delta(session.body, distance, distance_2, edges)
|
||
except ValueError as error:
|
||
# 显式 surfaceEntities 可以在后续实体上留下曲面分区边界。若标准
|
||
# OCC 倒角因环域宽度不足而拒绝,只允许在该 shell 给出同轴边界证据
|
||
# 时按原始距离构造受限倒角;没有证明时仍保留原始内核失败。
|
||
if distance_2 is not None or not session.surface_members:
|
||
raise
|
||
try:
|
||
body = session.adapter.surface_limited_chamfer(
|
||
session.body, distance, edges, list(session.surface_members.values()),
|
||
)
|
||
except ValueError:
|
||
raise error
|
||
diagnostics.append(RuntimeDiagnostic(
|
||
"chamfer_surface_limited",
|
||
"Chamfer was limited by an explicit coaxial surface boundary",
|
||
feature_id=node.feature_id,
|
||
detail={"distance_mm": distance, "surface_count": len(session.surface_members)},
|
||
))
|
||
# 5. 登记新主体并返回结果。
|
||
session.register_body(node.feature_id, body, replay_node=node, topology_delta=topology_delta)
|
||
return session.result(node, diagnostics=diagnostics)
|
||
|
||
|
||
def _execute_linear_pattern(node: FeaturePlanNode, session: ExecutionSession, execute: Callable[[FeaturePlanNode, ExecutionSession, dict[str, Any] | None], FeatureResult]) -> FeatureResult:
|
||
# 线性阵列特征(pattern)执行入口:沿两个方向按数量与间距重放源特征形成阵列。
|
||
|
||
# 1. 取源特征的 replay 定义(源特征按 feature_id 在会话中登记,供本阵列重放)。
|
||
params = node.params
|
||
sources = session.replay_sources(params.get("source_feature_ids") or [])
|
||
if not sources:
|
||
raise ValueError("pattern source features have no replay definitions")
|
||
# 2. 解析两个方向的实例数量。
|
||
count_1 = int(params.get("pattern_count_1") or 1)
|
||
count_2 = int(params.get("pattern_count_2") or 1)
|
||
# 3. 解析两个方向的步长向量(方向单位向量 × 间距),作为阵列位移基准。
|
||
direction_1 = vector_scale(vector_unit(tuple(float(value) for value in (params.get("direction_1") or [1, 0, 0])), field_name="pattern direction_1"), float(params.get("spacing_1_mm") or 0))
|
||
direction_2 = vector_scale(vector_unit(tuple(float(value) for value in (params.get("direction_2") or [0, 1, 0])), field_name="pattern direction_2"), float(params.get("spacing_2_mm") or 0))
|
||
# 4. 双重循环生成每个阵列实例(跳过原点 0,0 处,那里是源特征本身)。
|
||
for first in range(count_1):
|
||
for second in range(count_2):
|
||
if first == 0 and second == 0:
|
||
continue
|
||
# 计算当前实例相对源特征的偏移向量。
|
||
offset = vector_add(vector_scale(direction_1, first), vector_scale(direction_2, second))
|
||
for source in sources:
|
||
# 逐个源特征克隆并按偏移平移后重放执行(草图也同步平移)。
|
||
dependency = pattern_transform_blocker(source)
|
||
if dependency:
|
||
raise ValueError(f"pattern source uses an unsupported {dependency}")
|
||
cloned = _translated_node(source, f"{node.feature_id}.p{first}_{second}.{source.feature_id}", offset, session)
|
||
sketch = session.sketches.get(str(source.sketch_id))
|
||
execute(cloned, session, _translated_sketch(sketch, offset) if sketch else None)
|
||
# 5. 记录本阵列的 replay 定义:后续阵列若选中本阵列,按定义递归重放,
|
||
# 而非复制当前主体做近似。
|
||
# A later pattern may select this pattern feature. The definition is
|
||
# replayed recursively, never approximated by copying the current body.
|
||
session.replay_definitions[node.feature_id] = node
|
||
return session.result(node)
|
||
|
||
|
||
def _execute_mirror_pattern(node: FeaturePlanNode, session: ExecutionSession) -> FeatureResult:
|
||
mirror = node.params.get("mirror_plane") or {}
|
||
resolution = session.resolve(mirror)
|
||
if resolution.status != "resolved" or not isinstance(resolution.record.value, PlaneSpec):
|
||
raise ValueError(resolution.diagnostic.message if resolution.diagnostic else "mirror plane was not resolved")
|
||
source_ids = [str(value) for value in node.params.get("source_feature_ids") or ()]
|
||
if (
|
||
source_ids
|
||
and all(source_id in session.body_members for source_id in source_ids)
|
||
and all(
|
||
(source := session.nodes.get(source_id)) is not None
|
||
and source.params.get("result_mode") == "new_body"
|
||
for source_id in source_ids
|
||
)
|
||
):
|
||
# Only a direct NEW body has a standalone source identity after a
|
||
# mirror. A hole, dress-up, or ordinary additive source is merely an
|
||
# aggregate successor and must use the feature-replay path below.
|
||
# Keeping this condition identical to capability preflight prevents a
|
||
# downstream COPY body query from selecting an arbitrary aggregate.
|
||
members = dict(session.body_members)
|
||
body = session.body
|
||
for source_id in source_ids:
|
||
mirrored = session.adapter.mirror(session.body_members[source_id], resolution.record.value)
|
||
members[pattern_instance_member_id(node.feature_id, source_id, 1)] = mirrored
|
||
body = session.adapter.fuse(body, mirrored)
|
||
if body is None:
|
||
raise ValueError("mirror pattern produced no body")
|
||
session.register_body(node.feature_id, body, replay_node=node, body_members=members)
|
||
return session.result(node)
|
||
if node.params.get("mirror_current_body"):
|
||
# CADFS SWEPT_BODY 表示被后续 feature 持续修改的同一实体。这里复制
|
||
# 当前 B-rep 再镜像并合并,不能重放其初始 additive feature,否则会
|
||
# 丢失后续 cut/fillet 并生成独立错误实体。
|
||
if session.body is None:
|
||
raise ValueError("mirror current body has no active body")
|
||
mirrored = session.adapter.mirror(session.body, resolution.record.value)
|
||
session.register_body(node.feature_id, session.adapter.fuse(session.body, mirrored), replay_node=node)
|
||
return session.result(node)
|
||
sources = session.replay_sources(node.params.get("source_feature_ids") or [])
|
||
if not sources:
|
||
raise ValueError("mirror pattern source features have no replay definitions")
|
||
for source in sources:
|
||
dependency = pattern_transform_blocker(source)
|
||
if dependency:
|
||
raise ValueError(f"mirror pattern source uses an unsupported {dependency}")
|
||
if source.atomic_id == "box_add" and not _normal_is_coordinate_axis(resolution.record.value.normal):
|
||
# box_add 是固定世界轴对齐的原生图元:跨非坐标平面镜像会产生倾斜朝向,
|
||
# 当前参数语义无法表达,静默重放会得到错误几何 → 明确拒绝。跨坐标平面
|
||
# (法向平行于任一坐标轴)的镜像仍然精确。
|
||
raise ValueError("box_add mirror is exact only across coordinate-aligned mirror planes")
|
||
cloned = _mirrored_node(source, f"{node.feature_id}.m.{source.feature_id}", resolution.record.value, session)
|
||
sketch = session.sketches.get(str(source.sketch_id))
|
||
_execute_node(cloned, session, _mirrored_sketch(sketch, resolution.record.value) if sketch else None)
|
||
session.replay_definitions[node.feature_id] = node
|
||
return session.result(node)
|
||
|
||
|
||
def _circular_source_is_axisymmetric(node: FeaturePlanNode, session: ExecutionSession, axis: AxisSpec) -> bool:
|
||
"""Whether rotating a direct circular extrusion creates no new geometry."""
|
||
if node.atomic_id not in {"extrude_add_blind", "extrude_add_two_sided"}:
|
||
return False
|
||
sketch = session.sketches.get(str(node.sketch_id))
|
||
if sketch is None:
|
||
return False
|
||
profile = sketch.get("profile") or {}
|
||
circle = profile if profile.get("type") == "circle" else None
|
||
if circle is None:
|
||
contours = profile.get("contours") or []
|
||
segments = (contours[0] or {}).get("segments") if len(contours) == 1 else []
|
||
circle = segments[0] if isinstance(segments, list) and len(segments) == 1 and segments[0].get("type") == "circle" else None
|
||
center = (circle or {}).get("center")
|
||
if not isinstance(center, list) or len(center) != 2:
|
||
return False
|
||
try:
|
||
plane = PlaneSpec.from_mapping(sketch.get("workplane") or {})
|
||
except (TypeError, ValueError):
|
||
return False
|
||
if abs(vector_dot(plane.normal, axis.direction)) < 1 - 1e-7:
|
||
return False
|
||
world_center = vector_add(
|
||
plane.origin_mm,
|
||
vector_add(vector_scale(plane.x_dir, float(center[0])), vector_scale(plane.y_dir, float(center[1]))),
|
||
)
|
||
offset = vector_subtract(world_center, axis.origin_mm)
|
||
radial = vector_subtract(offset, vector_scale(axis.direction, vector_dot(offset, axis.direction)))
|
||
return math.sqrt(vector_dot(radial, radial)) <= 1e-6
|
||
|
||
|
||
def _advance_copy_topology_records(
|
||
records: list[TopologyRecord], topology_delta: TopologyDelta | None,
|
||
) -> list[TopologyRecord]:
|
||
"""Carry COPY provenance through one exact adapter-history operation.
|
||
|
||
Pattern copies are separate CDSL results even when their solids fuse into
|
||
a single final body. The temporary records here are never selector
|
||
candidates themselves. They only retain instance ownership while opaque
|
||
OCC history proves a unique subshape continuation to the final snapshot.
|
||
"""
|
||
if topology_delta is None:
|
||
return []
|
||
advanced: list[TopologyRecord] = []
|
||
for record in records:
|
||
values: list[Any] = []
|
||
for relation in topology_delta.relations:
|
||
if (
|
||
relation.kind != record.kind
|
||
or relation.event not in {"preserved", "modified"}
|
||
or not TopologyRegistry._same_topology_value(record.value, relation.source_value)
|
||
):
|
||
continue
|
||
for value in relation.result_values:
|
||
if not any(TopologyRegistry._same_topology_value(value, known) for known in values):
|
||
values.append(value)
|
||
# A split/merge has no unique COPY owner in the present selector
|
||
# contract. Keep the executable model, but do not make a claim that a
|
||
# later COPY selector can bind one arbitrary descendant.
|
||
if len(values) != 1:
|
||
continue
|
||
advanced.append(TopologyRecord(
|
||
record_id=record.record_id,
|
||
kind=record.kind,
|
||
feature_id=record.feature_id,
|
||
body_id=record.body_id,
|
||
geometry=dict(record.geometry),
|
||
value=values[0],
|
||
owner_feature_ids=record.owners,
|
||
output_roles=record.output_roles,
|
||
output_role_sources=record.output_role_sources,
|
||
))
|
||
return advanced
|
||
|
||
|
||
def _copy_snapshot_topology_delta(records: list[TopologyRecord]) -> TopologyDelta | None:
|
||
"""Bridge traced final COPY handles into the one registered body snapshot."""
|
||
if not records:
|
||
return None
|
||
return TopologyDelta(
|
||
operation="pattern_circular_copy_snapshot",
|
||
relations=tuple(
|
||
# ``record.value`` has already passed through every transform/fuse
|
||
# builder in this pattern and is an actual final-B-rep handle. The
|
||
# identity relation merely connects that evidence to the fresh
|
||
# adapter snapshot; it is not a geometric rebinding shortcut.
|
||
TopologyDeltaRelation("preserved", record.kind, record.value, (record.value,))
|
||
for record in records
|
||
),
|
||
)
|
||
|
||
|
||
def _has_usable_pattern_body(session: ExecutionSession, body: Any | None) -> bool:
|
||
"""Reject a formally valid but empty OCC boolean result before publishing it."""
|
||
if body is None or not session.adapter.body_solids(body):
|
||
return False
|
||
try:
|
||
return abs(float(body.volume)) > 1e-12
|
||
except (AttributeError, TypeError, ValueError):
|
||
return False
|
||
|
||
|
||
def _execute_circular_pattern(node: FeaturePlanNode, session: ExecutionSession, execute: Callable[[FeaturePlanNode, ExecutionSession, dict[str, Any] | None], FeatureResult]) -> FeatureResult:
|
||
# 环形阵列特征(pattern_circular)执行入口:绕显式轴按数量与包角重放源特征
|
||
# 形成环形阵列。源特征整体绕轴旋转(绝对坐标变换),非复制当前主体的近似。
|
||
params = node.params
|
||
raw_axis = params.get("axis")
|
||
if not (isinstance(raw_axis, dict) and raw_axis.get("origin_mm") is not None and raw_axis.get("direction") is not None):
|
||
raise ValueError("circular pattern requires an explicit axis with origin_mm and direction")
|
||
axis = AxisSpec.from_mapping(raw_axis)
|
||
count = int(params.get("pattern_count") or 1)
|
||
if count < 1:
|
||
raise ValueError("circular pattern pattern_count must be >= 1")
|
||
sweep_angle_deg = float(params.get("sweep_angle_deg") or 360.0)
|
||
operation_mode = str(params.get("operation_mode") or "add")
|
||
if operation_mode not in {"add", "remove"}:
|
||
raise ValueError("circular pattern operation_mode must be add or remove")
|
||
excluded = {int(value) for value in params.get("excluded_instance_indices") or []}
|
||
if any(instance < 1 or instance >= count for instance in excluded):
|
||
raise ValueError("circular pattern excluded instance is outside the generated range")
|
||
sources = session.replay_sources(params.get("source_feature_ids") or [])
|
||
if not sources:
|
||
raise ValueError("circular pattern source features have no replay definitions")
|
||
source_ids = [source.feature_id for source in sources]
|
||
pre_pattern_members = dict(session.body_members)
|
||
if operation_mode == "add" and all(source_id in session.body_members for source_id in source_ids):
|
||
# A pattern over explicit NEW/kept body members has a stronger contract
|
||
# than replay: each copy is an independently addressable rigid image of
|
||
# the named source member. Keep the instance keys in the body graph so
|
||
# a later CADFS COPY(BODY) transform/delete can name exactly one copy.
|
||
members = dict(session.body_members)
|
||
body = session.body
|
||
traced_copy_records: list[TopologyRecord] = []
|
||
for instance in range(1, count):
|
||
if instance in excluded:
|
||
continue
|
||
angle_deg = sweep_angle_deg * instance / count
|
||
transform = {
|
||
"type": "rotation",
|
||
"axis": {"origin_mm": list(axis.origin_mm), "direction": list(axis.direction)},
|
||
"angle_deg": angle_deg,
|
||
}
|
||
for source_id in source_ids:
|
||
member_id = pattern_instance_member_id(node.feature_id, source_id, instance)
|
||
owner_id = f"{node.feature_id}.c{instance}.{source_id}"
|
||
source_body = session.body_members[source_id]
|
||
copy, transform_delta = session.adapter.transform_with_topology_delta(source_body, transform)
|
||
source_records = session.adapter.topology_records(
|
||
source_body, owner_id, f"body:{node.feature_id}:copy:{instance}:{source_id}:source",
|
||
)
|
||
copy_records = _advance_copy_topology_records(source_records, transform_delta)
|
||
members[member_id] = copy
|
||
body, fuse_delta = session.adapter.fuse_with_topology_delta(body, copy)
|
||
traced_copy_records = _advance_copy_topology_records(
|
||
[*traced_copy_records, *copy_records], fuse_delta,
|
||
)
|
||
if _has_usable_pattern_body(session, body):
|
||
session.register_body(
|
||
node.feature_id, body, replay_node=node, body_members=members,
|
||
topology_delta=_copy_snapshot_topology_delta(traced_copy_records),
|
||
topology_predecessors=traced_copy_records,
|
||
)
|
||
return session.result(node)
|
||
# An OCC boolean may report IsDone/valid for an empty result when a
|
||
# copied fused body contains coincident internal topology. The normal
|
||
# pattern contract can replay the source feature contribution instead;
|
||
# it is the only sound fallback because it keeps source operation,
|
||
# sketch frame, and body lifecycle semantics intact.
|
||
for instance in range(1, count):
|
||
if instance in excluded:
|
||
continue
|
||
# 实例 i 位于包角 sweep_angle_deg 的 i/count 处(i=0 即源特征本身)。
|
||
angle_deg = sweep_angle_deg * instance / count
|
||
angle_rad = math.radians(angle_deg)
|
||
for source in sources:
|
||
# 与阵列轴同心、法向平行的圆形实体拉伸在任意环形实例中均与
|
||
# 原实体完全重合。重复执行它会把同一 B-rep 再次交给 OCC fuse,
|
||
# 后续非轴对称 source 可能因此丢失已生成的实体分支。
|
||
if _circular_source_is_axisymmetric(source, session, axis):
|
||
continue
|
||
dependency = pattern_transform_blocker(source)
|
||
if dependency:
|
||
raise ValueError(f"circular pattern source uses an unsupported {dependency}")
|
||
if source.atomic_id == "box_add" and not _box_circular_is_exact(axis, angle_rad):
|
||
raise ValueError(
|
||
"box_add circular pattern is exact only for coordinate-axis rotation "
|
||
"by multiples of 180 degrees"
|
||
)
|
||
cloned = _rotated_node(source, f"{node.feature_id}.c{instance}.{source.feature_id}", axis, angle_rad, session)
|
||
cloned = _pattern_operation_node(cloned, operation_mode)
|
||
# CADFS pattern instances are copies of the source result, not
|
||
# independent `NEW` operations. Replay them through normal add
|
||
# semantics: intersecting or face-sharing instances fuse, while
|
||
# spatially separate copies remain separate solids in the result.
|
||
if cloned.params.get("result_mode") == "new_body":
|
||
cloned = FeaturePlanNode(
|
||
cloned.feature_id, cloned.atomic_id, cloned.name, cloned.depends_on,
|
||
{key: value for key, value in cloned.params.items() if key != "result_mode"},
|
||
cloned.selectors, cloned.sketch_id, cloned.declared_status, cloned.source_feature,
|
||
)
|
||
sketch = session.sketches.get(str(source.sketch_id))
|
||
execute(cloned, session, _rotated_sketch(sketch, axis, angle_rad) if sketch else None)
|
||
# 环形阵列本身是完整 B-rep 结果的 producer。每个 replay 子特征都会更新
|
||
# active body;循环结束后必须用 pattern feature 重新登记最终快照,否则后续
|
||
# selector binding 会只保留最后一个实例的 body id,漏掉其它 COPY 实例。
|
||
if session.body is None:
|
||
raise ValueError("circular pattern produced no body")
|
||
# Replaying a fused sole-body source may be more robust than copying its
|
||
# full aggregate B-rep (for example, when a rotationally invariant base
|
||
# would otherwise be unioned with itself). If that replay still has one
|
||
# physical body, the direct source remains a proven alias of the current
|
||
# member. Preserve it for a following parts-scoped operation such as
|
||
# shell; do not extend this alias across multi-body patterns or multiple
|
||
# source members.
|
||
members = {node.feature_id: session.body}
|
||
if (
|
||
len(source_ids) == 1
|
||
and len(pre_pattern_members) == 1
|
||
and source_ids[0] in pre_pattern_members
|
||
and _has_usable_pattern_body(session, session.body)
|
||
and len(session.adapter.body_solids(session.body)) == 1
|
||
):
|
||
members[source_ids[0]] = session.body
|
||
session.register_body(node.feature_id, session.body, replay_node=node, body_members=members)
|
||
return session.result(node)
|
||
|
||
|
||
def _circular_pattern_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_circular_pattern(node, session, _execute_node)
|
||
|
||
|
||
def _execute_node(node: FeaturePlanNode, session: ExecutionSession, sketch_override: dict[str, Any] | None = None) -> FeatureResult:
|
||
executor = EXECUTORS.get(node.atomic_id)
|
||
if executor is None:
|
||
raise ValueError(f"No executor registered for {node.atomic_id!r}")
|
||
previous_feature_id = session.active_feature_id
|
||
session.active_feature_id = node.feature_id
|
||
try:
|
||
return executor(node, session, sketch_override)
|
||
finally:
|
||
session.active_feature_id = previous_feature_id
|
||
|
||
|
||
ExecutorFunction = Callable[[FeaturePlanNode, ExecutionSession, dict[str, Any] | None], FeatureResult]
|
||
|
||
|
||
def _primary_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
return _shape_from_primary(node, session, sketch=sketch)
|
||
|
||
|
||
def _revolve_surface_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_revolve_surface(node, session)
|
||
|
||
|
||
def _extrude_surface_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_extrude_surface(node, session)
|
||
|
||
|
||
def _loft_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_loft_add(node, session)
|
||
|
||
|
||
def _loft_cap_face_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_loft_add_with_cap_face(node, session)
|
||
|
||
|
||
def _sweep_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
return _execute_sweep_add(node, session, sketch)
|
||
|
||
|
||
def _reference_plane_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_reference_plane(node, session)
|
||
|
||
|
||
def _reference_axis_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_reference_axis(node, session)
|
||
|
||
|
||
def _sphere_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_sphere(node, session)
|
||
|
||
|
||
def _box_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_box(node, session)
|
||
|
||
|
||
def _cylinder_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_cylinder(node, session)
|
||
|
||
|
||
def _hole_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_hole(node, session)
|
||
|
||
|
||
def _hole_wizard_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_hole(node, session, wizard=True)
|
||
|
||
|
||
def _fillet_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_fillet(node, session)
|
||
|
||
|
||
def _chamfer_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_chamfer(node, session)
|
||
|
||
|
||
def _shell_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_shell(node, session)
|
||
|
||
|
||
def _linear_pattern_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_linear_pattern(node, session, _execute_node)
|
||
|
||
|
||
def _mirror_pattern_executor(node: FeaturePlanNode, session: ExecutionSession, sketch: dict[str, Any] | None) -> FeatureResult:
|
||
del sketch
|
||
return _execute_mirror_pattern(node, session)
|
||
|
||
|
||
EXECUTORS: dict[str, ExecutorFunction] = {
|
||
"reference_plane": _reference_plane_executor,
|
||
"reference_axis": _reference_axis_executor,
|
||
"sphere_add": _sphere_executor,
|
||
"box_add": _box_executor,
|
||
"cylinder_add": _cylinder_executor,
|
||
"thread_add": _thread_executor,
|
||
"thread_cut": _thread_executor,
|
||
"bend_add": _bend_executor,
|
||
"gear_add": _gear_executor,
|
||
"rack_add": _rack_executor,
|
||
"extrude_add_blind": _primary_executor,
|
||
"extrude_add_blind_with_hole": _primary_executor,
|
||
"extrude_add_two_sided": _primary_executor,
|
||
"extrude_cut_blind": _primary_executor,
|
||
"extrude_cut_two_sided": _primary_executor,
|
||
"extrude_cut_through": _primary_executor,
|
||
"extrude_from_face": lambda node, session, _sketch: _execute_extrude_from_face(node, session),
|
||
"loft_add": _loft_executor,
|
||
"loft_add_with_cap_face": _loft_cap_face_executor,
|
||
"sweep_add": _sweep_executor,
|
||
"revolve_add": _primary_executor,
|
||
"revolve_cut": _primary_executor,
|
||
"revolve_surface": _revolve_surface_executor,
|
||
"extrude_surface": _extrude_surface_executor,
|
||
"hole_blind": _hole_executor,
|
||
"hole_countersink": _hole_executor,
|
||
"hole_counterbore": _hole_executor,
|
||
"hole_wizard": _hole_wizard_executor,
|
||
"fillet": _fillet_executor,
|
||
"chamfer": _chamfer_executor,
|
||
"shell": _shell_executor,
|
||
"boolean_bodies": lambda node, session, sketch: _execute_boolean_bodies(node, session),
|
||
"transform_bodies": lambda node, session, sketch: _execute_transform_bodies(node, session),
|
||
"delete_bodies": lambda node, session, sketch: _execute_delete_bodies(node, session),
|
||
"pattern_linear": _linear_pattern_executor,
|
||
"pattern_mirror": _mirror_pattern_executor,
|
||
"pattern_circular": _circular_pattern_executor,
|
||
}
|
||
|
||
|
||
def analyze_cdsl(cdsl: dict[str, Any]):
|
||
"""Resolve profiles and return the current runtime capability analysis."""
|
||
sketch_errors: dict[str, str] = {}
|
||
resolved = resolve_required_sketches(
|
||
deepcopy(cdsl), sketch_ids_required_by_contract(cdsl), errors=sketch_errors,
|
||
)
|
||
analyzer = CapabilityAnalyzer(atomic_ids=EXECUTORS, profile_types=CORE_SHAPE_GENERATORS)
|
||
return analyzer.analyze(resolved, sketch_errors=sketch_errors)
|
||
|
||
|
||
def rebuild_cdsl(cdsl: dict[str, Any], out_step: Path, *, strict: bool = True) -> dict[str, Any]:
|
||
"""Rebuild CDSL through session-scoped atomic executors only."""
|
||
sketch_errors: dict[str, str] = {}
|
||
resolved = resolve_required_sketches(
|
||
deepcopy(cdsl), sketch_ids_required_by_contract(cdsl), errors=sketch_errors,
|
||
)
|
||
analysis = CapabilityAnalyzer(atomic_ids=EXECUTORS, profile_types=CORE_SHAPE_GENERATORS).analyze(
|
||
resolved, sketch_errors=sketch_errors,
|
||
)
|
||
if strict and not analysis.runtime_eligible:
|
||
first = next((result for result in analysis.feature_results if not result.executable), None)
|
||
if first is None:
|
||
raise ValueError(analysis.document_blockers[0].code)
|
||
if any(blocker.code == "unknown_atomic" for blocker in first.blockers):
|
||
raise ValueError(f"unsupported atomic_id: {first.atomic_id}")
|
||
detail = "; ".join(blocker.code for blocker in first.blockers)
|
||
raise ValueError(f"Feature {first.feature_id} is not runtime eligible: {detail}")
|
||
session = ExecutionSession(
|
||
sketches={str(sketch.get("id")): sketch for sketch in (resolved.get("geometry") or {}).get("sketches") or []},
|
||
nodes={node.feature_id: node for node in analysis.plan},
|
||
)
|
||
diagnostics: list[RuntimeDiagnostic] = []
|
||
for node, preflight in zip(analysis.plan, analysis.feature_results):
|
||
if not preflight.executable:
|
||
diagnostics.extend(preflight.blockers)
|
||
if strict:
|
||
break
|
||
continue
|
||
try:
|
||
_execute_node(node, session)
|
||
except Exception as error:
|
||
failed_resolution = next(
|
||
(item for item in reversed(session.selector_resolutions) if item["status"] != "resolved"), None,
|
||
)
|
||
diagnostic = (
|
||
RuntimeDiagnostic(error.code, str(error), feature_id=node.feature_id, detail=error.detail)
|
||
if isinstance(error, FeatureExecutionError)
|
||
else
|
||
RuntimeDiagnostic(
|
||
failed_resolution["diagnostic"]["code"], failed_resolution["diagnostic"]["message"],
|
||
feature_id=node.feature_id, detail=failed_resolution["diagnostic"].get("detail") or {},
|
||
)
|
||
if failed_resolution and failed_resolution.get("diagnostic")
|
||
else RuntimeDiagnostic("execution_failed", str(error), feature_id=node.feature_id)
|
||
)
|
||
diagnostics.append(diagnostic)
|
||
if strict:
|
||
raise RuntimeExecutionError(diagnostic, list(session.selector_resolutions)) from error
|
||
output = session.body
|
||
surface_geometry: dict[str, Any] | None = None
|
||
if output is None:
|
||
if not session.surface_members:
|
||
raise ValueError("CDSL execution produced no body")
|
||
# 纯曲面文档没有 active solid,但依然是可执行的 CAD 结果。只有在
|
||
# 没有实体时才将 surface members 作为 STEP 输出,混合模型继续只导出
|
||
# 实体,避免曲面意外改变既有实体比较和下游消费语义。
|
||
output = session.adapter.combine_surfaces(*session.surface_members.values())
|
||
surface_geometry = session.adapter.surface_geometry(output)
|
||
out_step.parent.mkdir(parents=True, exist_ok=True)
|
||
session.adapter.export(output, str(out_step))
|
||
geometry = session.adapter.body_geometry(session.body) if session.body is not None else surface_geometry
|
||
if geometry is None:
|
||
raise ValueError("CDSL execution produced no exportable geometry")
|
||
bbox = geometry["bbox_mm"]
|
||
return {
|
||
"engine": "cdsl_session_runtime",
|
||
"out_step": str(out_step),
|
||
"volume_mm3": float(geometry.get("volume_mm3") or 0.0),
|
||
"bbox_mm": {"min": bbox[:3], "max": bbox[3:]},
|
||
# #7 multi-body:重建结果里的独立实体数(Compound 成员数),
|
||
# 与 batch 验证的 document_truth.geometry.solid_body_count 对齐。
|
||
"solid_count": len(session.adapter.body_solids(session.body)) if session.body is not None else 0,
|
||
"surface_count": len(session.surface_members),
|
||
"surface_face_count": int(surface_geometry["face_count"]) if surface_geometry is not None else 0,
|
||
"surface_area_mm2": float(surface_geometry["area_mm2"]) if surface_geometry is not None else 0.0,
|
||
"feature_results": [result.as_dict() for result in session.results.values()],
|
||
"runtime_diagnostics": [diagnostic.as_dict() for diagnostic in diagnostics],
|
||
"topology_records": [record.public_dict() for record in session.topology.records()],
|
||
"topology_deltas": list(session.topology.topology_deltas()),
|
||
"selector_resolution": session.selector_resolutions,
|
||
}
|