fix(plotting): Improve interface of the plotting class.

doc/changelog.d/3702.fixed.md

@@ -0,0 +1 @@
+fix(plotting): Improve interface of the plotting class.

src/ansys/mapdl/core/plotting/visualizer.py

@@ -22,13 +22,14 @@
 
 """Module for the MapdlPlotter class."""
 from collections import OrderedDict
-from typing import Any, Iterable, Optional, Union
+from typing import Any, Dict, Iterable, Optional, Union
 
 from ansys.tools.visualization_interface import Plotter
 from ansys.tools.visualization_interface.backends.pyvista import PyVistaBackendInterface
 import numpy as np
 from numpy.typing import NDArray
 
+from ansys.mapdl.core import LOG as logger
 from ansys.mapdl.core import _HAS_VISUALIZER
 from ansys.mapdl.core.misc import get_bounding_box
 from ansys.mapdl.core.plotting.consts import (
@@ -280,9 +281,9 @@
 
     def add_mesh(
         self,
-        meshes,
-        points,
-        labels,
+        meshes: Union[pv.PolyData, Dict[str, Any]] = [],
+        points=[],
+        labels=[],
         *,
         cpos=None,
         show_bounds=False,
@@ -331,6 +332,10 @@
         plotter_kwargs : dict, optional
             Extra kwargs, by default {}
         """
+        if not meshes and not points and not labels:
+            logger.warning("No meshes, points or labels to plot.")
+            return
+
         if theme is None:
             theme = MapdlTheme()
 
@@ -378,31 +383,43 @@
                 **(add_points_kwargs or {}),
             )
 
+        if isinstance(meshes, pv.PolyData):
+            meshes = [meshes]
         for mesh in meshes:
-            scalars: Optional[NDArray[Any]] = mesh.get("scalars")
-
-            if (
-                "scalars" in mesh
-                and scalars.ndim == 2
-                and (scalars.shape[1] == 3 or scalars.shape[1] == 4)
-            ):
-                # for the case we are using scalars for plotting
-                rgb = True
+            rgb = False
+            if isinstance(mesh, Dict):
+                scalars: Optional[NDArray[Any]] = mesh.get("scalars")
+
+                if (
+                    "scalars" in mesh
+                    and scalars.ndim == 2
+                    and (scalars.shape[1] == 3 or scalars.shape[1] == 4)
+                ):
+                    # for the case we are using scalars for plotting
+                    rgb = True
+
+                # To avoid index error.
+                mesh_ = mesh["mesh"]
+                if not isinstance(mesh_, list):
+                    mesh_ = [mesh_]
             else:
-                rgb = False
-
-            # To avoid index error.
-            mesh_ = mesh["mesh"]
-            if not isinstance(mesh_, list):
-                mesh_ = [mesh_]
-
+                scalars = None
+                mesh_ = meshes
             for each_mesh in mesh_:
                 self.scene.add_mesh(
                     each_mesh,
                     scalars=scalars,
                     scalar_bar_args=scalar_bar_args,
-                    color=mesh.get("color", color),
-                    style=mesh.get("style", style),
+                    color=(
+                        mesh.get("color", color)
+                        if isinstance(mesh, Dict) and "color" in mesh
+                        else color
+                    ),
+                    style=(
+                        mesh.get("style", style)
+                        if isinstance(mesh, Dict) and "style" in mesh
+                        else style
+                    ),
                     show_edges=show_edges,
                     edge_color=edge_color,
                     smooth_shading=smooth_shading,
@@ -651,9 +668,9 @@
 
     def plot(
         self,
-        meshes,
-        points,
-        labels,
+        meshes: Union[pv.PolyData, Dict[str, Any]] = [],
+        points=[],
+        labels=[],
         *,
         title="",
         cpos=None,