Jerry
94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
"""OCP-native boolean helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import List, Optional, Sequence
|
|
|
|
from OCP.BOPAlgo import BOPAlgo_GlueOff, BOPAlgo_GlueShift
|
|
from OCP.BRepAlgoAPI import BRepAlgoAPI_Common, BRepAlgoAPI_Cut, BRepAlgoAPI_Fuse
|
|
from OCP.ShapeUpgrade import ShapeUpgrade_UnifySameDomain
|
|
from OCP.TopAbs import TopAbs_SOLID
|
|
from OCP.TopExp import TopExp_Explorer
|
|
from OCP.TopoDS import TopoDS, TopoDS_Shape, TopoDS_Solid
|
|
from OCP.TopTools import TopTools_ListOfShape
|
|
|
|
|
|
def _list_of(shapes: Sequence[TopoDS_Shape]) -> TopTools_ListOfShape:
|
|
out = TopTools_ListOfShape()
|
|
for shape in shapes:
|
|
out.Append(shape)
|
|
return out
|
|
|
|
|
|
def solids_of(shape: TopoDS_Shape) -> List[TopoDS_Solid]:
|
|
out: List[TopoDS_Solid] = []
|
|
explorer = TopExp_Explorer(shape, TopAbs_SOLID)
|
|
while explorer.More():
|
|
out.append(TopoDS.Solid_s(explorer.Current()))
|
|
explorer.Next()
|
|
if not out and shape.ShapeType() == TopAbs_SOLID:
|
|
out.append(TopoDS.Solid_s(shape))
|
|
return out
|
|
|
|
|
|
def clean_shape(shape: TopoDS_Shape) -> TopoDS_Shape:
|
|
unifier = ShapeUpgrade_UnifySameDomain(shape, True, True, True)
|
|
unifier.Build()
|
|
return unifier.Shape()
|
|
|
|
|
|
def fuse_shapes(shapes: Sequence[TopoDS_Shape], *, glue: bool = True, tol: Optional[float] = None, clean: bool = True) -> TopoDS_Shape:
|
|
if not shapes:
|
|
raise ValueError("fuse_shapes requires at least one shape")
|
|
if len(shapes) == 1:
|
|
return shapes[0]
|
|
builder = BRepAlgoAPI_Fuse()
|
|
builder.SetRunParallel(True)
|
|
builder.SetUseOBB(True)
|
|
builder.SetArguments(_list_of([shapes[0]]))
|
|
builder.SetTools(_list_of(list(shapes[1:])))
|
|
if tol is not None:
|
|
builder.SetFuzzyValue(float(tol))
|
|
# Match CadQuery's Shape.fuse(glue=True) behavior: CadQuery maps glue=True
|
|
# to OCC's GlueShift, not GlueFull. GlueFull can leave overlapping solids
|
|
# separate in cases where CQ would return one fused solid.
|
|
builder.SetGlue(BOPAlgo_GlueShift if glue else BOPAlgo_GlueOff)
|
|
builder.Build()
|
|
if not builder.IsDone():
|
|
raise ValueError("OCP fuse failed")
|
|
result = builder.Shape()
|
|
return clean_shape(result) if clean else result
|
|
|
|
|
|
def cut_shapes(body: TopoDS_Shape, tools: Sequence[TopoDS_Shape]) -> TopoDS_Shape:
|
|
if not tools:
|
|
return body
|
|
builder = BRepAlgoAPI_Cut()
|
|
builder.SetRunParallel(True)
|
|
builder.SetUseOBB(True)
|
|
builder.SetArguments(_list_of([body]))
|
|
builder.SetTools(_list_of(tools))
|
|
builder.Build()
|
|
if not builder.IsDone():
|
|
raise ValueError("OCP cut failed")
|
|
return builder.Shape()
|
|
|
|
|
|
def common_shapes(shapes: Sequence[TopoDS_Shape]) -> TopoDS_Shape:
|
|
if not shapes:
|
|
raise ValueError("common_shapes requires at least one shape")
|
|
if len(shapes) == 1:
|
|
return shapes[0]
|
|
result = shapes[0]
|
|
for tool in shapes[1:]:
|
|
builder = BRepAlgoAPI_Common()
|
|
builder.SetRunParallel(True)
|
|
builder.SetUseOBB(True)
|
|
builder.SetArguments(_list_of([result]))
|
|
builder.SetTools(_list_of([tool]))
|
|
builder.Build()
|
|
if not builder.IsDone():
|
|
raise ValueError("OCP common failed")
|
|
result = builder.Shape()
|
|
return result
|