crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 14:15:42 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

producing vtp vtk files

Browse source
This commit is contained in:
Jonathan Shimwell 2022-08-15 13:22:44 +01:00
parent 239fc8d9c5
commit 2263549b9b
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
openmc/tracks.py
View file

@ -266,7 +266,7 @@ class Tracks(list):
track.plot(ax)
return ax
def write_tracks_to_vtk(self, filename):
def write_tracks_to_vtk(self, filename='tracks.vtk'):
"""Creates a VTK file of the tracks
Parameters
@ -286,12 +286,14 @@ class Tracks(list):
points = vtk.vtkPoints()
cells = vtk.vtkCellArray()
point_offset = 0
for particle in self:
for state in particle.states:
points.InsertNextPoint(state['r'])
for pt in particle.particle_tracks:
for state in pt.states:
points.InsertNextPoint(state['r'])
# Create VTK line and assign points to line.
n = particle.states.size
n = pt.states.size
line = vtk.vtkPolyLine()
line.GetPointIds().SetNumberOfIds(n)
for i in range(n):