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 self.unraveled = False
return return
def to_vtk(self): def to_vtk(self,slice_part=slice(None),slice_time=slice(None),force_ravel=False):
import pyvista as pv import pyvista as pv
mesh = pv.PolyData() mesh = pv.PolyData()
for part in self.get_trajectories_segmented(): if force_ravel or not self.unraveled:
for seg in part: for part in self.get_trajectories_segmented(slice_part=slice_part,slice_time=slice_time):
mesh += pv.helpers.lines_from_points(seg.transpose()) 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 return mesh
def _make_data_array(self): def _make_data_array(self):