mwtkrayer
/
suspendtools
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

simplification for unraveled trajectories

This commit is contained in:
Michael Krayer 2022-02-22 13:55:13 +01:00
parent 649f8b4161
commit aec0cdead6
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
particle.py
View File

@ -536,12 +536,17 @@ class Trajectories:
self.unraveled = False
return
def to_vtk(self):
def to_vtk(self,slice_part=slice(None),slice_time=slice(None),force_ravel=False):
import pyvista as pv
mesh = pv.PolyData()
for part in self.get_trajectories_segmented():
for seg in part:
mesh += pv.helpers.lines_from_points(seg.transpose())
if force_ravel or not self.unraveled:
for part in self.get_trajectories_segmented(slice_part=slice_part,slice_time=slice_time):
for seg in part:
mesh += pv.helpers.lines_from_points(seg.transpose())
else:
tmp = self.get_trajectories(slice_part=slice_part,slice_time=slice_time)
for ipart in range(self.numpart):
mesh += pv.helpers.lines_from_points(np.array([tmp[0][ipart],tmp[1][ipart],tmp[2][ipart]]).transpose())
return mesh
def _make_data_array(self):