handle shell_layers in DpfPlotter

Author: PProfizi

src/ansys/dpf/core/plotter.py

@@ -233,6 +233,7 @@ def add_field(
         scale_factor=1.0,
         scale_factor_legend=None,
         as_linear=True,
+        shell_layer=eshell_layers.top,
         **kwargs,
     ):
         # Get the field name
@@ -275,6 +276,21 @@ def add_field(
             mesh_location = meshed_region.elements
         else:
             raise ValueError("Only elemental, nodal or faces location are supported for plotting.")
+
+        # Treat multilayered shells
+        if not isinstance(shell_layer, eshell_layers):
+            raise TypeError("shell_layer attribute must be a core.shell_layers instance.")
+        if field.shell_layers in [
+            eshell_layers.topbottom,
+            eshell_layers.topbottommid,
+        ]:
+            change_shell_layer_op = core.operators.utility.change_shell_layers()
+            change_shell_layer_op.inputs.fields_container.connect(field)
+            change_shell_layer_op.inputs.e_shell_layer.connect(
+                shell_layer.value
+            )  # top layers taken
+            field = change_shell_layer_op.get_output(0, core.types.field)
+
         component_count = field.component_count
         if component_count > 1:
             overall_data = np.full((len(mesh_location), component_count), np.nan)
@@ -603,6 +619,7 @@ def add_field(
         label_point_size=20,
         deform_by=None,
         scale_factor=1.0,
+        shell_layer=eshell_layers.top,
         **kwargs,
     ):
         """Add a field containing data to the plotter.
@@ -627,6 +644,9 @@ def add_field(
             Defaults to None.
         scale_factor : float, optional
             Scaling factor to apply when warping the mesh. Defaults to 1.0.
+        shell_layer: core.shell_layers, optional
+            Enum used to set the shell layer if the field to plot
+            contains shell elements. Defaults to top layer.
         **kwargs : optional
             Additional keyword arguments for the plotter. More information
             are available at :func:`pyvista.plot`.
@@ -653,6 +673,7 @@ def add_field(
             deform_by=deform_by,
             scale_factor=scale_factor,
             as_linear=True,
+            shell_layer=eshell_layers.top,
             **kwargs,
         )