Skip to content

Commit 5c2bfe7

Browse files
authored
Write particle states as separate lines in track VTK files. (#3628)
1 parent 8cd3911 commit 5c2bfe7

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

openmc/tracks.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -296,16 +296,16 @@ def write_to_vtk(self, filename=Path('tracks.vtp')):
296296
for state in pt.states:
297297
points.InsertNextPoint(state['r'])
298298

299-
# Create VTK line and assign points to line.
300-
n = pt.states.size
301-
line = vtk.vtkPolyLine()
302-
line.GetPointIds().SetNumberOfIds(n)
303-
for i in range(n):
304-
line.GetPointIds().SetId(i, point_offset + i)
305-
point_offset += n
306-
307-
# Add line to cell array
308-
cells.InsertNextCell(line)
299+
# Create VTK line and assign points to line.
300+
n = pt.states.size
301+
line = vtk.vtkPolyLine()
302+
line.GetPointIds().SetNumberOfIds(n)
303+
for i in range(n):
304+
line.GetPointIds().SetId(i, point_offset + i)
305+
point_offset += n
306+
307+
# Add line to cell array
308+
cells.InsertNextCell(line)
309309

310310
data = vtk.vtkPolyData()
311311
data.SetPoints(points)

0 commit comments

Comments
 (0)