def _handle_slice(self, idx):
idx = np.index_exp[idx]
frames, dims = None, None
if isinstance(idx, (tuple, list)):
if len(idx) == 1:
frames, dims = idx[0], slice(None, None, None)
if len(idx) == 2:
frames, dims = idx[0], idx[1]
if len(idx) > 2:
raise IndexError(
"Slice was more than two-dimensional, not supported.")
cumsum = np.cumsum(self._source.trajectory_lengths())
if not isinstance(frames, (list, np.ndarray)):
frames = self._get_indices(frames, cumsum[-1])
dims = self._get_indices(dims, self._source.ndim)
nframes = len(frames)
ndims = len(dims)
data = np.empty((nframes, ndims), dtype=self._source.output_type())
from pyemma.coordinates.clustering import UniformTimeClustering
for i, x in enumerate(frames):
traj, idx = UniformTimeClustering._idx_to_traj_idx(x, cumsum)
data[i, :] = self._source.data[traj][idx, dims]
return data
def _get_itraj_random_accessible(self, itrajs, frames, dims):
itrajs = self._get_indices(itrajs, self._source.ntraj)
frames = self._get_indices(frames, sum(self._source.trajectory_lengths()[itrajs]))
dims = self._get_indices(dims, self._source.ndim)
nframes = len(frames)
ndims = len(dims)
if max(dims) > self._source.ndim:
raise IndexError("Data only has %s dimensions, wanted to slice by dimension %s."
% (self._source.ndim, max(dims)))
cumsum = np.cumsum(self._source.trajectory_lengths()[itrajs])
from pyemma.coordinates.clustering import UniformTimeClustering
ra = np.array([self._map_to_absolute_traj_idx(UniformTimeClustering._idx_to_traj_idx(x, cumsum), itrajs)
for x in frames])
indices = np.lexsort((ra[:, 1], ra[:, 0]))
ra = ra[indices]
data = np.empty((nframes, ndims), dtype=self._source.output_type())
curr = 0
for X in self._source.iterator(stride=ra, lag=0, chunk=0, return_trajindex=False):
L = len(X)
data[indices[curr:curr + L]] = X
curr += L
return data
def _handle_slice(self, idx):
idx = np.index_exp[idx]
frames, dims = None, None
if isinstance(idx, (tuple, list)):
if len(idx) == 1:
frames, dims = idx[0], slice(None, None, None)
if len(idx) == 2:
frames, dims = idx[0], idx[1]
if len(idx) > 2:
raise IndexError("Slice was more than two-dimensional, not supported.")
cumsum = np.cumsum(self._source.trajectory_lengths())
frames = self._get_indices(frames, cumsum[-1])
dims = self._get_indices(dims, self._source.ndim)
nframes = len(frames)
ndims = len(dims)
frames_order = frames.argsort().argsort()
frames_sorted = np.sort(frames)
from pyemma.coordinates.clustering import UniformTimeClustering
ra_stride = np.array([UniformTimeClustering._idx_to_traj_idx(x, cumsum) for x in frames_sorted])
data = np.empty((nframes, ndims), dtype=self._source.output_type())
offset = 0
for X in self._source.iterator(stride=ra_stride, lag=0, chunk=0, return_trajindex=False):
L = len(X)
data[offset:offset + L, :] = X[:, dims]
offset += L
return data[frames_order]
def _get_itraj_random_accessible(self, itrajs, frames, dims):
itrajs = self._get_indices(itrajs, self._source.ntraj)
frames = self._get_indices(frames, sum(self._source.trajectory_lengths()[itrajs]))
dims = self._get_indices(dims, self._source.ndim)
nframes = len(frames)
ndims = len(dims)
if max(dims) > self._source.ndim:
raise IndexError("Data only has %s dimensions, wanted to slice by dimension %s."
% (self._source.ndim, max(dims)))
cumsum = np.cumsum(self._source.trajectory_lengths()[itrajs])
data = np.empty((nframes, ndims), dtype=self._source.output_type())
from pyemma.coordinates.clustering import UniformTimeClustering
for i, x in enumerate(frames):
traj, idx = self._map_to_absolute_traj_idx(UniformTimeClustering._idx_to_traj_idx(x, cumsum), itrajs)
data[i, :] = self._source.data[traj][idx, dims]
return data
