Fix volume pipeline range check to work with cell_data

topikuu · topikuu · commit 1b4a11af3ff2 · 2021-03-24T13:37:07.000+02:00
Previously trying to render volume views of UnstructuredGrids with
cell_data as the rendered scalar field caused constant error
messages although there are no real issues with them.

Signed-off-by: Topi Kuutela &lt;topi.kuutela@aalto.fi&gt;
diff --git a/mayavi/modules/volume.py b/mayavi/modules/volume.py
@@ -455,10 +455,15 @@ def _setup_current_range(self):
         # Set the current range.
         dataset = mm.source.get_output_dataset()
         dsh = DataSetHelper(dataset)
-        name, rng = dsh.get_range('scalars', 'point')
-        if name is None:
+        point_name, point_rng = dsh.get_range('scalars', 'point')
+        cell_name, cell_rng = dsh.get_range('scalars', 'cell')
+        if point_name is None and cell_name is None:
             error('No scalars in input data!')
             rng = (0, 255)
+        elif not point_name is None:
+            rng = point_rng
+        else:
+            rng = cell_rng
 
         if self.current_range != rng:
             self.current_range = rng
