mstk.trajectory.CombinedTrj

class mstk.trajectory.CombinedTrj(files, mode='r')

Read several trajectory files at the same time.

It is useful for processing truncated trajectories after restarting simulation. All trajectory files should contain the same number of atoms.

Methods

__init__(files[, mode])

close()

Close the handler.

get_handler_for_file(file)

Get the appropriate handler class for a trajectory file.

get_info()

Get the number of atoms and frames in the trajectory.

read_frame(i_frame, frame)

Read a single frame.

register_format(extension, Handler)

Register a handler class for a trajectory format based on the extension name

write_frame(frame, **kwargs)

Write a frame into the trajectory file opened by the handler.

get_info()

Get the number of atoms and frames in the trajectory.

Also record the offset of lines and frames, so that we can read arbitrary frame later. It assumes all frames have the same number of atoms.

Returns:

  • n_atom (int)

  • n_frame (int)

close()

Close the handler.

read_frame(i_frame, frame)

Read a single frame.

Parameters:
  • i_frame (int) – The index of the frame in the trajectory

  • frame (Frame) – The information read from the trajectory will be written into this Frame