def test_viz_to_binary_json(test_dir):
components = [
Beam("bm1", n1=[0, 0, 0], n2=[2, 0, 0], sec="IPE220", colour="red"),
Beam("bm2", n1=[0, 0, 1], n2=[2, 0, 1], sec="HP220x10", colour="blue"),
Beam("bm3",
n1=[0, 0, 2],
n2=[2, 0, 2],
sec="BG800x400x20x40",
colour="green"),
Beam("bm4", n1=[0, 0, 3], n2=[2, 0, 3], sec="CIRC200", colour="green"),
Beam("bm5",
n1=[0, 0, 4],
n2=[2, 0, 4],
sec="TUB200x10",
colour="green"),
Plate(
"pl1",
[(0, 0, 0), (0, 0, 1), (0, 1, 1), (0, 1, 0)],
0.01,
use3dnodes=True,
),
]
a = Assembly("my_test_assembly") / components
res = a.to_vis_mesh()
res.to_binary_and_json(test_dir / "viz/binjson/beams")
res.to_custom_json(test_dir / "viz/binjson/beams.json")
def test_viz_structural():
components = [
Beam("bm1", n1=[0, 0, 0], n2=[2, 0, 0], sec="IPE220", colour="red"),
Beam("bm2", n1=[0, 0, 1], n2=[2, 0, 1], sec="HP220x10", colour="blue"),
Beam("bm3",
n1=[0, 0, 2],
n2=[2, 0, 2],
sec="BG800x400x20x40",
colour="green"),
Beam("bm4", n1=[0, 0, 3], n2=[2, 0, 3], sec="CIRC200", colour="green"),
Beam("bm5",
n1=[0, 0, 4],
n2=[2, 0, 4],
sec="TUB200x10",
colour="green"),
Plate(
"pl1",
[(0, 0, 0), (0, 0, 1), (0, 1, 1), (0, 1, 0)],
0.01,
use3dnodes=True,
),
]
a = Assembly("my_test_assembly") / components
res = a.to_vis_mesh(auto_merge_by_color=False)
merged = res.merge_objects_in_parts_by_color()
assert res.num_polygons == 416
assert len(res.world[0].id_map.values()) == 6
assert merged.num_polygons == 416
assert len(merged.world[0].id_map.values()) == 4
