jmwright / cq-annotate Goto Github PK

import cadquery as cq
from cq_annotate.views import explode_assembly

# Create the first assembly component for first assembly
box1 = cq.Workplane().workplane(offset=0.0).box(10, 10, 10)

# Create a second assembly component to be assembled with the first
box2 = cq.Workplane().workplane(offset=0.0).box(10, 10, 10)

# The first assembly
assy1 = cq.Assembly()


assy1.add(
    box1,
    loc=cq.Location((0, 0, 5)),
    metadata={"explode_loc": cq.Location((0, 0, 10))},
    color=cq.Color(1, 0, 0, 1),
)
assy1.add(
    box2,
    loc=cq.Location((0, 0, -5)),
    metadata={"explode_loc": cq.Location((0, 0, -10))},
    color=cq.Color(0, 1, 0, 1),
)


# Create the first assembly component for second assembly
box3 = cq.Workplane().workplane(offset=0.0).box(10, 10, 10)

# Create a second assembly component to be assembled with the first
box4 = cq.Workplane().workplane(offset=0.0).box(10, 10, 10)

# The second assembly
assy2 = cq.Assembly()

assy2.add(
    box3,
    loc=cq.Location((0, 0, 5)),
    metadata={"explode_loc": cq.Location((0, 0, 10))},
    color=cq.Color(1, 0, 0, 1),
)
assy2.add(
    box4,
    loc=cq.Location((0, 0, -5)),
    metadata={"explode_loc": cq.Location((0, 0, -10))},
    color=cq.Color(0, 1, 0, 1),
)

# The main assembly
assy3 = cq.Assembly()

assy3.add(
    assy1,
    loc=cq.Location((0, 5, 0)),
    metadata={"explode_loc": cq.Location((0, 10, 0))},
    color=cq.Color(1, 0, 0, 1),
)
assy3.add(
    assy2,
    loc=cq.Location((0, -5, 0)),
    metadata={"explode_loc": cq.Location((0, -10, 0))},
    color=cq.Color(0, 1, 0, 1),
)

explode_assembly(assy3)

# This only exists so that users running the example in CQ-editor
# can see the result
if "show_object" in globals():
    show_object(assy3)