Source code for MDAnalysis.transformations.nojump

# -*- Mode: python; tab-width: 4; indent-tabs-mode:nil; coding:utf-8 -*-
# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4
#
# MDAnalysis --- https://www.mdanalysis.org
# Copyright (c) 2006-2017 The MDAnalysis Development Team and contributors
# (see the file AUTHORS for the full list of names)
#
# Released under the Lesser GNU Public Licence, v2.1 or any higher version
#
# Please cite your use of MDAnalysis in published work:
#
# R. J. Gowers, M. Linke, J. Barnoud, T. J. E. Reddy, M. N. Melo, S. L. Seyler,
# D. L. Dotson, J. Domanski, S. Buchoux, I. M. Kenney, and O. Beckstein.
# MDAnalysis: A Python package for the rapid analysis of molecular dynamics
# simulations. In S. Benthall and S. Rostrup editors, Proceedings of the 15th
# Python in Science Conference, pages 102-109, Austin, TX, 2016. SciPy.
# doi: 10.25080/majora-629e541a-00e
#
# N. Michaud-Agrawal, E. J. Denning, T. B. Woolf, and O. Beckstein.
# MDAnalysis: A Toolkit for the Analysis of Molecular Dynamics Simulations.
# J. Comput. Chem. 32 (2011), 2319--2327, doi:10.1002/jcc.21787
#

"""\
No Jump Trajectory Unwrapping --- :mod:`MDAnalysis.transformations.nojump`
=======================================================================================

Unwraps the trajectory such that atoms never move more than half a periodic box length.
The consequence of this is that particles will diffuse across periodic boundaries when
needed. This unwrapping method is suitable as a preprocessing step to calculate
molecular diffusion, or more commonly to keep multi-domain proteins whole during trajectory
analysis.
The algorithm used is based on :footcite:p:`Kulke2022`.

.. autoclass:: NoJump

"""
import numpy as np
import warnings

from .base import TransformationBase
from ..due import due, Doi
from ..exceptions import NoDataError


[docs] class NoJump(TransformationBase): """ Returns transformed coordinates for the given timestep so that an atom does not move more than half the periodic box size between two timesteps, and will move across periodic boundary edges. The algorithm used is based on :footcite:p:`Kulke2022`, equation B6 for non-orthogonal systems, so it is general to most applications where molecule trajectories should not "jump" from one side of a periodic box to another. Note that this transformation depends on a periodic box dimension being set for every frame in the trajectory, and that this box dimension can be transformed to an orthonormal unit cell. If not, an error is emitted. Since it is typical to transform all frames sequentially when unwrapping trajectories, warnings are emitted if non-sequential timesteps are transformed using NoJump. Example ------- Suppose you had a molecular trajectory with a protein dimer that moved across a periodic boundary. This will normally appear to make the dimer split apart. This transformation uses the molecular motions between frames in a wrapped trajectory to create an unwrapped trajectory where molecules never move more than half a periodic box dimension between subsequent frames. Thus, the normal use case is to apply the transformation to every frame of a trajectory, and to do so sequentially. .. code-block:: python transformation = NoJump() u.trajectory.add_transformations(transformation) for ts in u.trajectory: print(ts.positions) In this case, ``ts.positions`` will return the NoJump unwrapped trajectory, which would keep protein dimers together if they are together in the inital timestep. The unwrapped trajectory may also be useful to compute diffusion coefficients via the Einstein relation. To reverse the process, wrap the coordinates again. Returns ------- MDAnalysis.coordinates.timestep.Timestep References ---------- .. footbibliography:: """ @due.dcite( Doi("10.1021/acs.jctc.2c00327"), description="Works through the orthogonal case for unwrapping, " "and proposes the non-orthogonal approach.", path=__name__, ) def __init__( self, check_continuity=True, max_threads=None, ): # NoJump transforms are inherently unparallelizable, since # it depends on the previous frame's unwrapped coordinates super().__init__(max_threads=max_threads, parallelizable=False) self.prev = None self.old_frame = 0 self.older_frame = "A" self.check_c = check_continuity def _transform(self, ts): L = ts.triclinic_dimensions if L is None: msg = f"Periodic box dimensions not provided at step {ts.frame}" raise NoDataError(msg) try: Linverse = np.linalg.inv(L) except np.linalg.LinAlgError: msg = f"Periodic box dimensions are not invertible at step {ts.frame}" raise NoDataError(msg) if ts.frame == 0: # We don't need to apply the transformation here. However, we need to # ensure we have the 0th frame coordinates in reduced form. We also need to # set an an appropriate value for self.older frame. This is so that on the # following frame we don't return early when we check # `self.older_frame != "A"`. If we return early, then the transformation is # not applied, and any jumps across boundaries that occur at that frame will # not be accounted for. self.prev = ts.positions @ Linverse self.old_frame = 0 self.older_frame = -1 return ts if ( self.check_c and self.older_frame != "A" and (self.old_frame - self.older_frame) != (ts.frame - self.old_frame) ): warnings.warn( "NoJump detected that the interval between frames is unequal." "We are resetting the reference frame to the current timestep.", UserWarning, ) self.prev = ts.positions @ Linverse self.old_frame = ts.frame self.older_frame = "A" return ts if self.check_c and np.abs(self.old_frame - ts.frame) != 1: warnings.warn( "NoJump transform is only accurate when positions" "do not move by more than half a box length." "Currently jumping between frames with a step of more than 1." "This might be fine, but depending on the trajectory stride," "this might be inaccurate.", UserWarning, ) # Convert into reduced coordinate space fcurrent = ts.positions @ Linverse fprev = ( self.prev ) # Previous unwrapped coordinates in reduced box coordinates. # Calculate the new positions in reduced coordinate space (Equation B6 from # 10.1021/acs.jctc.2c00327). As it turns out, the displacement term can # be moved inside the round function in this coordinate space, as the # difference between wrapped and unwrapped coordinates is an integer. newpositions = fcurrent - np.round(fcurrent - fprev) # Convert back into real space ts.positions = newpositions @ L # Set things we need to save for the next frame. self.prev = ( newpositions # Note that this is in reduced coordinate space. ) self.older_frame = self.old_frame self.old_frame = ts.frame return ts