Fix for PVPythonViewConfigurator to handle new MultiblockDataSet input. Can still crash in some cases and will need separate PR to deal with it.

Author: alexbenedicto

geos-pv/src/geos/pv/plugins/PVPythonViewConfigurator.py

@@ -16,6 +16,7 @@
 
 update_paths()
 
+from geos.mesh.utils.multiblockModifiers import mergeBlocks
 import geos.pv.utils.paraviewTreatments as pvt
 from geos.pv.utils.checkboxFunction import (  # type: ignore[attr-defined]
     createModifiedCallback, )
@@ -44,7 +45,10 @@
 from paraview.util.vtkAlgorithm import (  # type: ignore[import-not-found]
     VTKPythonAlgorithmBase )
 
-from vtkmodules.vtkCommonDataModel import vtkDataObject
+from vtkmodules.vtkCommonDataModel import (
+    vtkDataObject,
+    vtkMultiBlockDataSet,
+)
 from geos.pv.utils.details import SISOFilter, FilterCategory
 
 __doc__ = """
@@ -84,6 +88,9 @@ def __init__( self: Self ) -> None:
         # Input source and curve names.
         inputSource = GetActiveSource()
         dataset = servermanager.Fetch( inputSource )
+        # Handle vtkMultiBlockDataSet by merging blocks first
+        if isinstance( dataset, vtkMultiBlockDataSet ):
+            dataset = mergeBlocks( dataset, keepPartialAttributes=True )
         dataframe: pd.DataFrame = pvt.vtkToDataframe( dataset )
         self.m_pathPythonViewScript: Path = geos_pv_path / "src/geos/pv/pythonViewUtils/mainPythonView.py"
 

geos-pv/src/geos/pv/utils/paraviewTreatments.py

@@ -37,6 +37,7 @@
     vtkTable,
     vtkUnstructuredGrid,
 )
+from vtkmodules.vtkFiltersParallelDIY2 import vtkGenerateGlobalIds
 
 from geos.utils.GeosOutputsConstants import (
     ComponentNameEnum,
@@ -122,8 +123,18 @@ def vtkUnstructuredGridCellsToDataframe( grid: vtkUnstructuredGrid ) -> pd.DataF
     Returns:
         pd.DataFrame: Pandas dataframe.
     """
-    cellIdAttributeName = GeosMeshOutputsEnum.VTK_ORIGINAL_CELL_ID.attributeName
+    cellIdAttributeName: str = GeosMeshOutputsEnum.VTK_ORIGINAL_CELL_ID.attributeName
     cellData = grid.GetCellData()
+    if not cellData.HasArray( GeosMeshOutputsEnum.VTK_ORIGINAL_CELL_ID.attributeName ):
+        print( "We have to create global ids." )
+        idFilter = vtkGenerateGlobalIds()
+        idFilter.SetInputData( grid )
+        idFilter.Update()
+        grid = idFilter.GetOutput()
+        cellData = grid.GetCellData()  # Update cellData to point to the new grid's cell data
+        cellIdAttributeName = "GlobalCellIds"
+        assert cellData.HasArray(cellIdAttributeName), "Invalid global ids array name selected."
+
     numberCells: int = grid.GetNumberOfCells()
     data: dict[ str, Any ] = {}
     for i in range( cellData.GetNumberOfArrays() ):