Split a Brep With a Plane

../../_images/example_brep_split.png
from math import radians

from compas.colors import Color
from compas.geometry import Box
from compas.geometry import Brep
from compas.geometry import Plane
from compas.geometry import Rotation
from compas.geometry import is_point_infrontof_plane
from compas_viewer import Viewer

box = Box(1).to_brep()

R = Rotation.from_axis_and_angle([0, 1, 0], radians(30))
plane = Plane.worldXY()
plane.transform(R)
splitter = Brep.from_plane(plane, domain_u=(-2, +2), domain_v=(-2, +2))

result = box.split(splitter)

# =============================================================================
# Visualization
# =============================================================================

viewer = Viewer()

# viewer.view.camera.position = [2, -4, 1]
# viewer.view.camera.look_at([0, 0, 0])

viewer.scene.add(splitter, linewidth=2, opacity=0.3)

for brep in result:  # type: ignore
    if is_point_infrontof_plane(brep.centroid, plane):
        viewer.scene.add(
            brep,
            surfacecolor=Color.red().lightened(50),
            linecolor=Color.red(),
            linewidth=2,
            show_points=False,
        )
    else:
        viewer.scene.add(
            brep,
            surfacecolor=Color.blue().lightened(50),
            linecolor=Color.blue(),
            linewidth=2,
            show_points=False,
        )

viewer.show()