From d7b3b746532f03a3925c1b2eb5179f0263deeb9b Mon Sep 17 00:00:00 2001
From: Harry Braviner <harry.braviner@gmail.com>
Date: Thu, 7 Dec 2023 21:53:39 -0500
Subject: [PATCH] pyvista UniformGrid -> ImageData

pyvists 0.43.0 renames UniformGrid to ImageData. See https://github.com/pyvista/pyvista/pull/4518 for details.
Renamed all instances of this for compatability with pyvista 0.43.
Works with pyvista back to and including 0.40.
---
 src/utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/utils.py b/src/utils.py
index 61c11fb..1720ca6 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -123,7 +123,7 @@ def save_fields_vtk(timestep, fields, output_dir='.', prefix='fields'):
     if value.ndim == 2:
         dimensions = dimensions + (1,)
 
-    grid = pv.UniformGrid(dimensions=dimensions)
+    grid = pv.ImageData(dimensions=dimensions)
 
     # Add the fields to the grid
     for key, value in fields.items():
@@ -157,7 +157,7 @@ def live_volume_randering(timestep, field):
     if field.ndim != 3:
         raise ValueError("The input field must be 3D!")
     dimensions = field.shape
-    grid = pv.UniformGrid(dimensions=dimensions)
+    grid = pv.ImageData(dimensions=dimensions)
 
     # Add the field to the grid
     grid['field'] = field.flatten(order='F')
@@ -207,7 +207,7 @@ def save_BCs_vtk(timestep, BCs, gridInfo,  output_dir='.'):
         gridDimensions = (gridInfo['nx'] + 1, gridInfo['ny'] + 1, gridInfo['nz'] + 1)
         fieldDimensions = (gridInfo['nx'], gridInfo['ny'], gridInfo['nz'])
 
-    grid = pv.UniformGrid(dimensions=gridDimensions)
+    grid = pv.ImageData(dimensions=gridDimensions)
 
     # Dictionary to keep track of encountered BC names
     bcNamesCount = {}