crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 13:45:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add numpy arrays to a list so they persist to the time of the VTK write

Browse source
This commit is contained in:
Patrick Shriwise 2022-07-21 22:24:17 -05:00
parent 925a5f6bb6
commit 8cc815d4f2
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
openmc/mesh.py
View file

@ -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