Refactor 3D plotting utilities and improve color mapping.

Reordered actor assignment for scalar bars, corrected colormap flipping logic, and updated scalar range handling. Enhanced surface plotting with custom colormap support and fixed lithology scalar data calculation. These changes improve clarity, functionality, and visual consistency in 3D plots.

Author: Leguark

gempy_viewer/modules/plot_3d/drawer_input_3d.py

@@ -3,6 +3,8 @@
 from gempy.core.data import GeoModel
 from gempy.core.data.orientations import OrientationsTable
 from gempy.core.data.surface_points import SurfacePointsTable
+from matplotlib.colors import ListedColormap
+
 from gempy_viewer.modules.plot_2d.plot_2d_utils import get_geo_model_cmap
 from gempy_viewer.modules.plot_3d.vista import GemPyToVista
 from gempy_viewer.optional_dependencies import require_pyvista
@@ -22,7 +24,8 @@ def plot_data(gempy_vista: GemPyToVista,
 
     plot_surface_points(
         gempy_vista=gempy_vista,
-        surface_points=surface_points_copy
+        surface_points=surface_points_copy,
+        element_colors=model.structural_frame.elements_colors
     )
 
     plot_orientations(
@@ -37,6 +40,7 @@ def plot_surface_points(
         gempy_vista: GemPyToVista,
         surface_points: SurfacePointsTable,
         render_points_as_spheres=True,
+        element_colors=None,
         point_size=10
 ):
     # Selecting the surfaces to plot
@@ -53,14 +57,17 @@ def plot_surface_points(
     vectorize_ids = _vectorize_ids(ids, ids)
     poly['id'] = vectorize_ids
 
+    custom_cmap = ListedColormap(element_colors)
+    
     gempy_vista.surface_points_mesh = poly
     gempy_vista.surface_points_actor = gempy_vista.p.add_mesh(
         mesh=poly,
         scalars='id',
         render_points_as_spheres=render_points_as_spheres,
         point_size=point_size,
         show_scalar_bar=False,
-        # clim=(0, ids.max())
+        cmap=custom_cmap,
+        clim=(0, np.unique(vectorize_ids).shape[0])
     )
 
 

gempy_viewer/modules/plot_3d/drawer_structured_grid_3d.py

@@ -100,7 +100,7 @@ def set_scalar_data(
     # Substitute the madness of the previous if with match
     match scalar_data_type:
         case ScalarDataType.LITHOLOGY | ScalarDataType.ALL:
-            max_lith = data.n_surfaces + 1 # (for basement)
+            max_lith = data.n_surfaces # (for basement)
             block_ = max_lith - (data.lith_block - 1)
             structured_grid.cell_data['id'] = block_
         case ScalarDataType.SCALAR_FIELD | ScalarDataType.ALL:

gempy_viewer/modules/plot_3d/plot_3d_utils.py

@@ -40,10 +40,10 @@ def set_scalar_bar(gempy_vista: GemPyToVista, elements_names: list[str],
     import pyvista as pv
 
     # Get mapper actor 
-    if gempy_vista.surface_points_actor is not None:
-        mapper_actor: pv.Actor = gempy_vista.surface_points_actor
-    elif gempy_vista.regular_grid_actor is not None:
+    if gempy_vista.regular_grid_actor is not None:
         mapper_actor = gempy_vista.regular_grid_actor
+    elif gempy_vista.surface_points_actor is not None:
+        mapper_actor: pv.Actor = gempy_vista.surface_points_actor
     else:
         return None  # * Not a good mapper for the scalar bar
 
@@ -71,7 +71,7 @@ def set_scalar_bar(gempy_vista: GemPyToVista, elements_names: list[str],
 
         custom_cmap = ListedColormap(custom_colors)
         # Apply the custom colormap to the lookup table
-        lut.apply_cmap(cmap=custom_cmap, n_values=n_colors, flip=True)
+        lut.apply_cmap(cmap=custom_cmap, n_values=n_colors, flip=False)
 
     else:
         # Apply a default colormap if no custom colors are provided
@@ -81,6 +81,7 @@ def set_scalar_bar(gempy_vista: GemPyToVista, elements_names: list[str],
     sargs = gempy_vista.scalar_bar_arguments
     min_id, max_id = surfaces_ids.min(), surfaces_ids.max()
     mapper_actor.mapper.scalar_range = (min_id - .4, max_id + .5)
+
     sargs["mapper"] = mapper_actor.mapper
     sargs["n_labels"] = 0