From 8cc815d4f2ebf56460f0f0f6f9195bf46afc968a Mon Sep 17 00:00:00 2001 From: Patrick Shriwise Date: Thu, 21 Jul 2022 22:24:17 -0500 Subject: [PATCH] Add numpy arrays to a list so they persist to the time of the VTK write --- openmc/mesh.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/openmc/mesh.py b/openmc/mesh.py index 1fe5429d0..d08213f77 100644 --- a/openmc/mesh.py +++ b/openmc/mesh.py @@ -248,8 +248,10 @@ class StructuredMesh(MeshBase): # create VTK arrays for each of # the data sets + datasets_out = [] for label, dataset in datasets.items(): dataset = np.asarray(dataset).flatten() + datasets_out.append(dataset) if volume_normalization: dataset /= self.volumes.flatten() @@ -692,7 +694,7 @@ class RegularMesh(StructuredMesh): root_cell.fill = lattice return root_cell, cells - + def write_data_to_vtk(self, filename, datasets, volume_normalization=True): """Creates a VTK object of the mesh @@ -1605,7 +1607,7 @@ class UnstructuredMesh(MeshBase): Raises ------ RuntimeError - when the size of a dataset doesn't match the number of cells + when the size of a dataset doesn't match the number of cells """ import vtk