From aec0cdead6e55f6eb4cecf71244e1d247bb41c00 Mon Sep 17 00:00:00 2001 From: Michael Krayer Date: Tue, 22 Feb 2022 13:55:13 +0100 Subject: [PATCH] simplification for unraveled trajectories --- particle.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/particle.py b/particle.py index c214e69..a462aab 100644 --- a/particle.py +++ b/particle.py @@ -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):