Trajectory

This module defines a class for handling multiple trajectories.

class Trajectory(name, **kwargs)[source]

A class for handling trajectories in multiple files.

Trajectory can be instantiated with a name or a filename. When name is a valid path to a trajectory file it will be opened for reading.

addFile(filename, **kwargs)[source]

Add a file to the trajectory instance. Currently only DCD files are supported.

close()[source]

Close trajectory file.

getAtoms()

Returns associated/selected atoms.

getCoords()

Returns a copy of reference coordinates for (selected) atoms.

getCoordsets(indices=None)[source]

Returns coordinate sets at given indices. indices may be an integer, a list of ordered integers or None. None returns all coordinate sets. If a list of indices is given, unique numbers will be selected and sorted. That is, this method will always return unique coordinate sets in the order they appear in the trajectory file. Shape of the coordinate set array is (n_sets, n_atoms, 3).

getFilenames(absolute=False)[source]

Returns list of filenames opened for reading.

getFirstTimestep()[source]

Returns list of first timestep values, one number from each file.

getFrameFreq()[source]

Returns list of timesteps between frames, one number from each file.

getLinked()

Returns linked AtomGroup instance, or None if a link is not established.

getTimestep()[source]

Returns list of timestep sizes, one number from each file.

getTitle()

Returns title of the ensemble.

getWeights()

Returns a copy of weights of (selected) atoms.

goto(n)[source]

Go to the frame at index n. n=0 will rewind the trajectory to the beginning, same as calling reset() method. n=-1 will go to the last frame. Frame n will not be parsed until one of next() or nextCoordset() methods is called.

hasUnitcell()[source]

Returns True if trajectory has unitcell data.

isLinked()

Returns True if trajectory is linked to an AtomGroup instance.

iterCoordsets()

Yield coordinate sets for (selected) atoms. Reference coordinates are not included. Iteration starts from the next frame in line.

Link, return, or unlink an AtomGroup instance. When a link to ag is established, coordinates of new frames parsed from the trajectory file will be set as the coordinates of ag and this will update coordinates of all selections and atom subsets pointing to it. At link time, if ag does not have any coordinate sets and reference coordinates of the trajectory is set, reference coordinates of the trajectory will be passed to ag. To break an established link, pass None argument, or to return the linked atom group instance, call with no arguments.

Warning

Every time a frame is parsed from the trajectory, all coordinate sets present in the linked AtomGroup will be overwritten.

next()

Returns next coordinate set in a Frame instance. Note that when atoms are set for the trajectory, this method will return the same frame instance after updating its coordinates.

nextCoordset()[source]

Returns next coordinate set.

nextIndex()

Returns the index of the next frame.

numAtoms()

Returns number of atoms.

numCoordsets()

Returns number of frames.

numFiles()[source]

Returns number of open trajectory files.

numFixed()[source]

Returns a list of fixed atom numbers, one from each file.

numFrames()

Returns number of frames.

numSelected()

Returns number of selected atoms. A subset of atoms can be selected by passing a selection to setAtoms().

reset()[source]

Go to first frame at index 0. First frame will not be parsed until one of next() or nextCoordset() methods is called.

setAtoms(atoms)[source]

Set atoms or specify a selection of atoms to be considered in calculations and coordinate requests. When a selection is set, corresponding subset of coordinates will be considered in, for example, alignments and RMSD calculations. Setting atoms also allows some functions to access atomic data when needed. For example, Trajectory and Frame instances become suitable arguments for writePDB(). Passing None as atoms argument will deselect atoms. Note that setting atoms does not change the reference coordinates of the trajectory. To change the reference, use setCoords() method.

setCoords(coords)

Set coords as the trajectory reference coordinate set. coords must be an object with getCoords() method, or a Numpy array with suitable data type, shape, and dimensionality.

setTitle(title)

Set title of the ensemble.

setWeights(weights)

Set atomic weights.

skip(n)[source]

Skip n frames. n must be a positive integer. Skipping some frames will only change the next frame index (nextIndex()) Next frame will not be parsed until one of next() or nextCoordset() methods is called.