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

some bugfixes

This commit is contained in:
Michael Krayer 2021-05-26 09:53:55 +02:00
parent 7bcd6b0394
commit 77f56261be
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
particle.py
View File

@ -207,6 +207,7 @@ class Trajectories:
traj = Trajectories(part[slice_part],unraveled=self.unraveled) traj = Trajectories(part[slice_part],unraveled=self.unraveled)
else: else:
traj += Trajectories(part[slice_part]) traj += Trajectories(part[slice_part])
traj.frame_velocity = self.frame_velocity
return traj return traj
def add_particles(self,part): def add_particles(self,part):
import numpy import numpy
@ -306,7 +307,7 @@ class Trajectories:
lastPeriod = thisPeriod lastPeriod = thisPeriod
# Restore previous ravel state # Restore previous ravel state
if restore_ravel_state and not was_unraveled: if restore_ravel_state and not was_unraveled:
self.ravel() self.enforce_periodicity()
return out return out
def slice_time(self,time_beg=None,time_end=None): def slice_time(self,time_beg=None,time_end=None):
return return
@ -321,7 +322,7 @@ class Trajectories:
def unravel(self): def unravel(self):
import numpy import numpy
if self.unraveled: return if self.unraveled: return
raise NotImplementedError('Implemented but untested!') #raise NotImplementedError('Implemented but untested!')
self._make_data_array() self._make_data_array()
for axis in range(0,3): for axis in range(0,3):
if self.period[axis] is not None: if self.period[axis] is not None: