Skip to content

Commit ba84542

Browse files
Use add_composite plotting API (#470)
Co-authored-by: Roberto Pastor Muela <37798125+RobPasMue@users.noreply.github.com>
1 parent 840ae1d commit ba84542

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

src/ansys/geometry/core/plotting/plotter.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,11 @@ def add_body(
223223
"""
224224
# Use the default PyGeometry add_mesh arguments
225225
self.__set_add_mesh_defaults(plotting_options)
226-
self.scene.add_mesh(body.tessellate(merge=merge), **plotting_options)
226+
dataset = body.tessellate(merge=merge)
227+
if isinstance(dataset, pv.MultiBlock):
228+
self.scene.add_composite(dataset, **plotting_options)
229+
else:
230+
self.scene.add_mesh(dataset, **plotting_options)
227231

228232
def add_component(
229233
self,
@@ -253,7 +257,10 @@ def add_component(
253257
# Use the default PyGeometry add_mesh arguments
254258
self.__set_add_mesh_defaults(plotting_options)
255259
dataset = component.tessellate(merge_component=merge_component, merge_bodies=merge_bodies)
256-
self.scene.add_mesh(dataset, **plotting_options)
260+
if isinstance(dataset, pv.MultiBlock):
261+
self.scene.add_composite(dataset, **plotting_options)
262+
else:
263+
self.scene.add_mesh(dataset, **plotting_options)
257264

258265
def add_sketch_polydata(self, polydata_entries: List[pv.PolyData], **plotting_options) -> None:
259266
"""Add sketches to the scene from PyVista polydata.

0 commit comments

Comments
 (0)