Source code for MDAnalysis.analysis.hydrogenbonds.wbridge_analysis

# -*- 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 GNU Public Licence, v2 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
#

# Water Bridge Analysis
r"""Water Bridge analysis ---
mod:`MDAnalysis.analysis.hydrogenbonds.WaterBridgeAnalysis`
===============================================================================

:Author: Zhiyi Wu
:Year: 2017-2018
:Copyright: GNU Public License v3
:Maintainer: Zhiyi Wu <[email protected]>,  `@xiki-tempula`_ on GitHub


.. _`@xiki-tempula`: https://github.com/xiki-tempula


Given a :class:`~MDAnalysis.core.universe.Universe` (simulation
trajectory with 1 or more frames) measure all water bridges for each
frame between selections 1 and 2.
Water bridge is defined as a bridging water which simultaneously forms
two hydrogen bonds with atoms from both selection 1 and selection 2.

A water bridge can form between two hydrogen bond acceptors.

e.g. -CO\ :sub:`2`\ :sup:`-`:···H−O−H···:\ :sup:`-`\ O\ :sub:`2`\ C-

A water bridge can also form between two hydrogen bond donors.

e.g. -NH···:O:···HN- (where O is the oxygen of a bridging water)

A hydrogen bond acceptor and another hydrogen bond donor can be bridged by a
water.

e.g. -CO\ :sub:`2`\ :sup:`-`:···H−O:···HN- (where H−O is part of **H−O**\ −H)

A higher order water bridge is defined as more than one water bridging
hydrogen bond acceptor and donor. An example of a second order water bridge:

e.g. -CO\ :sub:`2`\ :sup:`-`:···H−O:···H−O:···HN-
(where H−O is part of **H−O**\ −H)

The following keyword arguments are important to control the behaviour of the
water bridge analysis:

 - *water_selection* (``resname SOL``): the selection string for the bridging
   water
 - *order* the maximum number of water bridging both ends
 - donor-acceptor *distance* (Å): 3.0
 - Angle *cutoff* (degrees): 120.0
 - *forcefield* to switch between default values for different force fields
 - *donors* and *acceptors* atom types (to add additional atom names)

Theory
------

This module attempts to find multi-order water bridge by an approach similar
to breadth-first search, where the first solvation shell of selection 1 is
selected, followed by the selection of the second solvation shell as well as
any hydrogen bonding partner from selection 1. After that, the third solvation
shell, as well as any binding partners from selection 2, are detected. This
process is repeated until the maximum order of water bridges is reached.

.. _wb_Analysis_Network:

Output as Network
-----------------

Since the waters connecting the two ends of the selections are by nature a
network. We provide a network representation of the water network. Water bridge
data are returned per frame, which is stored in
:attr:`WaterBridgeAnalysis.network`. Each frame is represented as a dictionary,
where the keys are the hydrogen bonds originating from selection 1 and the
values are new dictionaries representing the hydrogen bonds coming out of the
corresponding molecules making hydrogen bonds with selection 1.

As for the hydrogen bonds which reach the selection 2, the values of the
corresponding keys are None. One example where selection 1 and selection 2 are
joined by one water molecule (A) which also hydrogen bond to another water (B)
which also hydrogen bond to selection 2 would be represented as ::

    # (selection 1)-O:···H-O(water 1):···H-(selection 2)
    #                      |             :
    #                      H·············O-H(water2)
    #                                    H
    {(sele1_acceptor, None, water1_donor, water1_donor_heavy, distance, angle):
         {(water1_acceptor, None, sele2_donor, sele2_donor_heavy,
         distance, angle): None},
         {(water1_donor, water1_donor_heavy, water2_acceptor, None,
         distance, angle):
              {(water2_acceptor, None, sele2_donor, sele2_donor_heavy,
              distance, angle): None}
          },
    }

The atoms are represented by atom index and if the atom is hydrogen bond donor,
it is followed by the index of the corresponding heavy atom
``(donor_proton, donor_heavy_atom)``.
If the atom is a hydrogen bond acceptor, it is followed by none.

.. _wb_Analysis_Timeseries:

Output as Timeseries
--------------------

For lower order water bridges, it might be desirable to represent the
connections as :attr:`WaterBridgeAnalysis.timeseries`. The results are returned
per frame and are a list of hydrogen bonds between the selection 1 or selection
2 and the bridging waters. Due to the complexity of the higher order water
bridge and the fact that one hydrogen bond between two waters can appear in
both third and fourth order water bridge, the hydrogen bonds in the
:attr:`WaterBridgeAnalysis.timeseries` attribute are generated in a depth-first
search manner to avoid duplication. Example code of how
:attr:`WaterBridgeAnalysis.timeseries` is generated::

    def network2timeseries(network, timeseries):
        '''Traverse the network in a depth-first fashion.
        expand_timeseries will expand the compact representation to the
        familiar timeseries representation.'''

        if network is None:
            return
        else:
            for node in network:
                timeseries.append(expand_timeseries(node))
                network2timeseries(network[node], timeseries)

    timeseries = []
    network2timeseries(network, timeseries)

An example would be. ::

    results = [
        [ # frame 1
           [ <donor index>, <acceptor index>,
            (<donor residue name>, <donor residue number>, <donor atom name>),
            (<acceptor residue name>, <acceptor residue number>,
            <acceptor atom name>),
             <distance>, <angle>],
           ....
        ],
        [ # frame 2
          [ ... ], [ ... ], ...
        ],
        ...
    ]

Using the :meth:`WaterBridgeAnalysis.generate_table` method one can reformat
the results as a flat "normalised" table that is easier to import into a
database or dataframe for further processing.

Detection of water bridges
--------------------------
Water bridges are recorded if a bridging water simultaneously forms
hydrogen bonds with selection 1 and selection 2.

Hydrogen bonds are detected based on a geometric criterion:

1. The distance between acceptor and hydrogen is less than or equal to
   `distance` (default is 3 Å).

2. The angle between donor-hydrogen-acceptor is greater than or equal to
   `angle` (default is 120º).

The cut-off values `angle` and `distance` can be set as keywords to
:class:`WaterBridgeAnalysis`.

Donor and acceptor heavy atoms are detected from atom names. The current
defaults are appropriate for the CHARMM27 and GLYCAM06 force fields as defined
in Table `Default atom names for water bridge analysis`_.

Hydrogen atoms bonded to a donor are searched based on its distance to the
donor. The algorithm searches for all hydrogens
(name "H*" or name "[123]H" or type "H") in the same residue as the donor atom
within a cut-off distance of 1.2 Å.

.. _Default atom names for water bridge analysis:

.. table:: Default heavy atom names for CHARMM27 force field.

   =========== ==============  =========== ====================================
   group       donor           acceptor    comments
   =========== ==============  =========== ====================================
   main chain  N               O, OC1, OC2 OC1, OC2 from amber99sb-ildn
                                           (Gromacs)
   water       OH2, OW         OH2, OW     SPC, TIP3P, TIP4P (CHARMM27,Gromacs)

   ARG         NE, NH1, NH2
   ASN         ND2             OD1
   ASP                         OD1, OD2
   CYS         SG
   CYH                         SG          possible false positives for CYS
   GLN         NE2             OE1
   GLU                         OE1, OE2
   HIS         ND1, NE2        ND1, NE2    presence of H determines if donor
   HSD         ND1             NE2
   HSE         NE2             ND1
   HSP         ND1, NE2
   LYS         NZ
   MET                         SD          see e.g. [Gregoret1991]_
   SER         OG              OG
   THR         OG1             OG1
   TRP         NE1
   TYR         OH              OH
   =========== ==============  =========== ====================================

.. table:: Heavy atom types for GLYCAM06 force field.

   =========== =========== ==================
   element     donor       acceptor
   =========== =========== ==================
   N           N,NT,N3     N,NT
   O           OH,OW       O,O2,OH,OS,OW,OY
   S                       SM
   =========== =========== ==================

Donor and acceptor names for the CHARMM27 force field will also work for e.g.
OPLS/AA or amber (tested in Gromacs). Residue names in the table are for
information only and are not taken into account when determining acceptors and
donors. This can potentially lead to some ambiguity in the assignment of
donors/acceptors for residues such as histidine or cytosine.

For more information about the naming convention in GLYCAM06 have a look at the
`Carbohydrate Naming Convention in Glycam`_.

.. _`Carbohydrate Naming Convention in Glycam`:
   http://glycam.ccrc.uga.edu/documents/FutureNomenclature.htm

The lists of donor and acceptor names can be extended by providing lists of
atom names in the `donors` and `acceptors` keywords to
:class:`WaterBridgeAnalysis`. If the lists are entirely inappropriate
(e.g. when analysing simulations done with a force field that uses very
different atom names) then one should either use the value "other" for
`forcefield` to set no default values or derive a new class and set the
default list oneself::

 class WaterBridgeAnalysis_OtherFF(WaterBridgeAnalysis):
       DEFAULT_DONORS = {"OtherFF": tuple(set([...]))}
       DEFAULT_ACCEPTORS = {"OtherFF": tuple(set([...]))}

Then simply use the new class instead of the parent class and call it with
```forcefield` = "OtherFF"``. Please also consider contributing the list of
heavy atom names to MDAnalysis.

How to perform WaterBridgeAnalysis
----------------------------------

All water bridges between arginine and aspartic acid can be analysed with ::

  import MDAnalysis
  import MDAnalysis.analysis.hbonds

  u = MDAnalysis.Universe('topology', 'trajectory')
  w = MDAnalysis.analysis.hbonds.WaterBridgeAnalysis(u, 'resname ARG',
  'resname ASP')
  w.run()

The maximum number of bridging waters detected can be changed using the order
keyword. ::

  w = MDAnalysis.analysis.hbonds.WaterBridgeAnalysis(u, 'resname ARG',
  'resname ASP', order=3)

Thus, a maximum of three bridging waters will be detected.

An example of using the :attr:`~WaterBridgeAnalysis` would be
detecting the percentage of time a certain water bridge exits.

Trajectory :code:`u` has two frames, where the first frame contains a water
bridge from the oxygen of the first arginine to one of the oxygens in the
carboxylic group of aspartate (ASP3:OD1). In the second frame, the same water
bridge forms but is between the oxygen of the arginine and the other oxygen in
the carboxylic group (ASP3:OD2). ::

  # index residue id residue name atom name
  #     0          1          ARG         O
  #     1          2          SOL        OW
  #     2          2          SOL       HW1
  #     3          2          SOL       HW2
  #     4          3          ASP       OD1
  #     5          3          ASP       OD2
  print(w.timeseries)

prints out. ::

  [ # frame 1
    # A water bridge SOL2 links O from ARG1 to the carboxylic group OD1 of ASP3
   [[0,2,('ARG',1,  'O'),('SOL',2,'HW1'),  3.0,180],
    [3,4,('SOL',2,'HW2'),('ASP',3,'OD1'),  3.0,180],
   ],
    # frame 2
    # Another water bridge SOL2 links O from ARG1 to the other oxygen of the
    # carboxylic group OD2 of ASP3
   [[0,2,('ARG',1,  'O'),('SOL',2,'HW1'),  3.0,180],
    [3,5,('SOL',2,'HW2'),('ASP',3,'OD2'),  3.0,180],
   ],
  ]


.. _wb_count_by_type:

Use count_by_type
-----------------

We can use the :meth:`~WaterBridgeAnalysis.count_by_type` to
generate the frequence of all water bridges in the simulation. ::

  w.count_by_type()

Returns ::

  [(0, 3, 'ARG', 1, 'O', 'ASP', 3, 'OD1', 0.5),
   (0, 4, 'ARG', 1, 'O', 'ASP', 3, 'OD2', 0.5),]

You might think that the OD1 and OD2 are the same oxygen and the aspartate has
just flipped and thus, they should be counted as the same type of water bridge.
The type of the water bridge can be customised by supplying an analysis
function to :meth:`~WaterBridgeAnalysis.count_by_type`.

The analysis function has two parameters. The current and the output. The
current is a list of hydrogen bonds from selection 1 to selection 2, formatted
in the same fashion as :attr:`WaterBridgeAnalysis.network`, and an example will
be ::

  [
  # sele1 acceptor idx,   , water donor index, donor heavy atom idx, dist, ang.
   [                 0, None,                 2,                   1, 3.0,180],
  # water donor idx, donor heavy atom idx, sele2 acceptor idx, distance, angle.
   [              3,                    1,                  4, None, 3.0,180],]

Where ``current[0]`` is the first hydrogen bond originating from selection 1
and ``current[-1]`` is the final hydrogen bond ending in selection 2. The
output sums up all the information in the current frame and is a dictionary
with a user-defined key and the value is the weight of the corresponding key.
During the analysis phase, the function analysis iterates through all the water
bridges and modify the output in-place. At the end of the analysis, the keys
from all the frames are collected and the corresponding values will be summed
up and returned. ::

  def analysis(current, output, u):
      r'''This function defines how the type of water bridge should be
      specified.

        Parameters
        ----------
        current : list
            The current water bridge being analysed is a list of hydrogen bonds
            from selection 1 to selection 2.
        output : dict
            A dictionary which is modified in-place where the key is the type
            of the water bridge and the value is the weight of this type of
            water bridge.
        u : MDAnalysis.universe
            The current Universe for looking up atoms.'''

      # decompose the first hydrogen bond.
      sele1_index, sele1_heavy_index, atom2, heavy_atom2, dist, angle =
      current[0]
      # decompose the last hydrogen bond.
      atom1, heavy_atom1, sele2_index, sele2_heavy_index, dist, angle =
      current[-1]
      # expand the atom index to the resname, resid, atom names
      sele1 = u.atoms[sele1_index]
      sele2 = u.atoms[sele2_index]
      (s1_resname, s1_resid, s1_name) = (sele1.resname, sele1.resid,
      sele1.name)
      (s2_resname, s2_resid, s2_name) = (sele2.resname, sele2.resid,
      sele2.name)
      # if the residue name is ASP and the atom name is OD2 or OD1,
      # the atom name is changed to OD
      if s2_resname == 'ASP' and (s2_name == 'OD1' or s2_name == 'OD2'):
          s2_name = 'OD'
      # setting up the key which defines this type of water bridge.
      key = (s1_resname, s1_resid, s1_name, s2_resname, s2_resid, s2_name)
      # The number of this type of water bridge is incremented by 1.
      output[key] += 1

  w.count_by_type(analysis_func=analysis)

Returns ::

  [(('ARG', 1, 'O', 'ASP', 3, 'OD'), 1.0),]

Note that the result is arranged in the format of
``(key, the proportion of time)``. When no custom analysis function is supplied
, the key is expanded and is formatted as ::

  [('ARG', 1, 'O', 'ASP', 3, 'OD', 1.0),]

Some people might only interested in contacts between residues and pay no
attention to the details regarding the atom name. However, since multiple water
bridges can exist between two residues, which sometimes can give a result such
that the water bridge between two residues exists 300% of the time. Though this
might be a desirable result for some people, others might want the water bridge
between two residues to be only counted once per frame. This can also be
achieved by supplying an analysis function to
:meth:`~WaterBridgeAnalysis.count_by_type`. ::

  def analysis(current, output, u):
      '''This function defines how the type of water bridge should be specified
      .

        Parameters
        ----------
        current : list
            The current water bridge being analysed is a list of hydrogen bonds
            from selection 1 to selection 2.
        output : dict
            A dictionary which is modified in-place where the key is the type
            of the water bridge and the value is the weight of this type of
            water bridge.
        u : MDAnalysis.universe
            The current Universe for looking up atoms.
      '''

      # decompose the first hydrogen bond.
      sele1_index, sele1_heavy_index, atom2, heavy_atom2, dist, angle =
      current[0]
      # decompose the last hydrogen bond.
      atom1, heavy_atom1, sele2_index, sele2_heavy_index, dist, angle =
      current[-1]
      # expand the atom index to the resname, resid, atom names
      sele1 = u.atoms[sele1_index]
      sele2 = u.atoms[sele2_index]
      (s1_resname, s1_resid, s1_name) = (sele1.resname, sele1.resid,
      sele1.name)
      (s2_resname, s2_resid, s2_name) = (sele2.resname, sele2.resid,
      sele2.name)
      # s1_name and s2_name are not included in the key
      key = (s1_resname, s1_resid, s2_resname, s2_resid)

      # Each residue is only counted once per frame
      output[key] = 1

  w.count_by_type(analysis_func=analysis)

Returns ::

  [(('ARG', 1, 'ASP', 3), 1.0),]

On the other hand, other people may insist that the first order and
second-order water bridges shouldn't be mixed together, which can also be
achieved by supplying an analysis function to
:meth:`~WaterBridgeAnalysis.count_by_type`.  ::

  def analysis(current, output, u):
      '''This function defines how the type of water bridge should be specified
      .

        Parameters
        ----------
        current : list
            The current water bridge being analysed is a list of hydrogen bonds
            from selection 1 to selection 2.
        output : dict
            A dictionary which is modified in-place where the key is the type
            of the water bridge and the value is the weight of this type of
            water bridge.
        u : MDAnalysis.universe
            The current Universe for looking up atoms.
      '''

      # decompose the first hydrogen bond.
      sele1_index, sele1_heavy_index, atom2, heavy_atom2, dist, angle =
      current[0]
      # decompose the last hydrogen bond.
      atom1, heavy_atom1, sele2_index, sele2_heavy_index, dist, angle =
      current[-1]
      # expand the atom index to the resname, resid, atom names
      sele1 = u.atoms[sele1_index]
      sele2 = u.atoms[sele2_index]
      (s1_resname, s1_resid, s1_name) = (sele1.resname, sele1.resid,
      sele1.name)
      (s2_resname, s2_resid, s2_name) = (sele2.resname, sele2.resid,
      sele2.name)
      # order of the current water bridge is computed
      order_of_water_bridge = len(current) - 1
      # and is included in the key
      key = (s1_resname, s1_resid, s2_resname, s2_resid, order_of_water_bridge)
      # The number of this type of water bridge is incremented by 1.
      output[key] += 1

  w.count_by_type(analysis_func=analysis)

The extra number 1 precede the 1.0 indicate that this is a first order water
bridge ::

  [(('ARG', 1, 'ASP', 3, 1), 1.0),]

Some people might not be interested in the interactions related to arginine.
The undesirable interactions can be discarded by supplying an analysis function
to :meth:`~WaterBridgeAnalysis.count_by_type`.  ::

  def analysis(current, output, u):
      '''This function defines how the type of water bridge should be
      specified.

        Parameters
        ----------
        current : list
            The current water bridge being analysed is a list of hydrogen bonds
            from selection 1 to selection 2.
        output : dict
            A dictionary which is modified in-place where the key is the type
            of the water bridge and the value is the number of this type of
            water bridge.
        u : MDAnalysis.universe
            The current Universe for looking up atoms.
      '''

      # decompose the first hydrogen bond.
      sele1_index, sele1_heavy_index, atom2, heavy_atom2, dist, angle =
      current[0]
      # decompose the last hydrogen bond.
      atom1, heavy_atom1, sele2_index, sele2_heavy_index, dist, angle =
      current[-1]
      # expand the atom index to the resname, resid, atom names
      sele1 = u.atoms[sele1_index]
      sele2 = u.atoms[sele2_index]
      (s1_resname, s1_resid, s1_name) = (sele1.resname, sele1.resid,
      sele1.name)
      (s2_resname, s2_resid, s2_name) = (sele2.resname, sele2.resid,
      sele2.name)
      if not s1_resname == 'ARG':
          key = (s1_resname, s1_resid, s2_resname, s2_resid)
          output[key] += 1

  w.count_by_type(analysis_func=analysis)

Returns nothing in this case ::

  [,]

Additional keywords can be supplied to the analysis function by passing through
:meth:`~WaterBridgeAnalysis.count_by_type`.  ::

  def analysis(current, output, **kwargs):
      ...
  w.count_by_type(analysis_func=analysis, **kwargs)


.. _wb_count_by_time:

Use count_by_time
-----------------

:meth:`~WaterBridgeAnalysis.count_by_type` aggregates data across frames, which
might be desirable in some cases but not the others.
:meth:`~WaterBridgeAnalysis.count_by_time` provides additional functionality
for aggregating results for each frame.

The default behaviour of :meth:`~WaterBridgeAnalysis.count_by_time` is counting
the number of water bridges from selection 1 to selection 2 for each frame.
Take the previous ASP, ARG salt bridge for example:  ::

  w.count_by_time()

As one water bridge is found in both frames, the method returns ::

  [(1.0, 1), (2.0, 1), ]

Similar to :meth:`~WaterBridgeAnalysis.count_by_type`
The behaviour of :meth:`~WaterBridgeAnalysis.count_by_time` can also be
modified by supplying an analysis function.

Suppose we want to count

  - the **number** of water molecules involved in bridging selection 1 to
    selection 2.
  - only if water bridge terminates in atom name **OD1 of ASP**.
  - only when water bridge is joined by less than **two** water.

The analysis function can be written as::

  def analysis(current, output, u, **kwargs):
      '''This function defines how the counting of water bridge should be
      specified.

        Parameters
        ----------
        current : list
            The current water bridge being analysed is a list of hydrogen bonds
            from selection 1 to selection 2.
        output : dict
            A dictionary which is modified in-place where the key is the type
            of the water bridge and the value is the number of this type of
            water  bridge.
        u : MDAnalysis.universe
            The current Universe for looking up atoms.
      '''

      # decompose the first hydrogen bond.
      sele1_index, sele1_heavy_index, atom2, heavy_atom2, dist, angle =
      current[0]
      # decompose the last hydrogen bond.
      atom1, heavy_atom1, sele2_index, sele2_heavy_index, dist, angle =
      current[-1]
      # expand the atom index to the resname, resid, atom names
      sele1 = u.atoms[sele1_index]
      sele2 = u.atoms[sele2_index]
      (s1_resname, s1_resid, s1_name) =
      (sele1.resname, sele1.resid, sele1.name)
      (s2_resname, s2_resid, s2_name) =
      (sele2.resname, sele2.resid, sele2.name)

      # only the residue name is ASP and the atom name is OD1,
      if s2_resname == 'ASP' and s2_name == 'OD1':
          # only if the order of water bridge is less than 2
          if len(current) -1 < 2:
              # extract all water molecules involved in the water bridge
              # extract the first water from selection 1
              s1_index, to_index, (s1_resname, s1_resid, s1_name),
              (to_resname, to_resid, to_name), dist, angle = current[0]
              key = (to_resname, to_resid)
              output[key] = 1

              # extract all the waters between selection 1 and selection 2
              for hbond in current[1:-1]:
                  # decompose the hydrogen bond.
                  from_index, to_index, (from_resname, from_resid, from_name),
                  (to_resname, to_resid, to_name), dist, angle = hbond
                  # add first water
                  key1 = (from_resname, from_resid)
                  output[key1] = 1
                  # add second water
                  key2 = (to_resname, to_resid)
                  output[key2] = 1

              # extract the last water to selection 2
              from_index, s2_index, (from_resname, from_resid, from_name),
              (s2_resname, s2_resid, s2_name), dist, angle = current[-1]
              key = (from_resname, from_resid)
              output[key] = 1

  w.count_by_time(analysis_func=analysis)

Returns ::

  [(1.0, 1), (2.0, 0),]

Classes
-------

.. autoclass:: WaterBridgeAnalysis
   :members:

   .. attribute:: timesteps

      List of the times of each timestep. This can be used together with
      :attr:`~WaterBridgeAnalysis.timeseries` to find the specific time point
      of a water bridge existence.

"""
from __future__ import print_function, absolute_import, division

from collections import defaultdict
import logging
import warnings
import six
from six.moves import range, zip
import numpy as np

from ..base import AnalysisBase
from MDAnalysis.lib.NeighborSearch import AtomNeighborSearch
from MDAnalysis.lib.distances import capped_distance, calc_angles
from MDAnalysis import NoDataError, MissingDataWarning, SelectionError
from MDAnalysis.lib import distances

logger = logging.getLogger('MDAnalysis.analysis.WaterBridgeAnalysis')


[docs]class WaterBridgeAnalysis(AnalysisBase): """Perform a water bridge analysis The analysis of the trajectory is performed with the :meth:`WaterBridgeAnalysis.run` method. The result is stored in :attr:`WaterBridgeAnalysis.timeseries`. See :meth:`~WaterBridgeAnalysis.run` for the format. .. versionadded:: 0.17.0 """ # use tuple(set()) here so that one can just copy&paste names from the # table; set() takes care for removing duplicates. At the end the # DEFAULT_DONORS and DEFAULT_ACCEPTORS should simply be tuples. #: default heavy atom names whose hydrogens are treated as *donors* #: (see :ref:`Default atom names for water bridge analysis`); #: use the keyword `donors` to add a list of additional donor names. DEFAULT_DONORS = { 'CHARMM27': tuple( {'N', 'OH2', 'OW', 'NE', 'NH1', 'NH2', 'ND2', 'SG', 'NE2', 'ND1', 'NZ', 'OG', 'OG1', 'NE1', 'OH'}), 'GLYCAM06': tuple({'N', 'NT', 'N3', 'OH', 'OW'}), 'other': tuple(set([]))} #: default atom names that are treated as hydrogen *acceptors* #: (see :ref:`Default atom names for water bridge analysis`); #: use the keyword `acceptors` to add a list of additional acceptor names. DEFAULT_ACCEPTORS = { 'CHARMM27': tuple( {'O', 'OC1', 'OC2', 'OH2', 'OW', 'OD1', 'OD2', 'SG', 'OE1', 'OE1', 'OE2', 'ND1', 'NE2', 'SD', 'OG', 'OG1', 'OH'}), 'GLYCAM06': tuple({'N', 'NT', 'O', 'O2', 'OH', 'OS', 'OW', 'OY', 'SM'}), 'other': tuple(set([]))} #: A :class:`collections.defaultdict` of covalent radii of common donors #: (used in :meth`_get_bonded_hydrogens_list` to check if a hydrogen is #: sufficiently close to its donor heavy atom). Values are stored for #: N, O, P, and S. Any other heavy atoms are assumed to have hydrogens #: covalently bound at a maximum distance of 1.5 Å. r_cov = defaultdict(lambda: 1.5, # default value N=1.31, O=1.31, P=1.58, S=1.55) def __init__(self, universe, selection1='protein', selection2='not resname SOL', water_selection='resname SOL', order=1, selection1_type='both', update_selection=False, update_water_selection=True, filter_first=True, distance_type='hydrogen', distance=3.0, angle=120.0, forcefield='CHARMM27', donors=None, acceptors=None, output_format="sele1_sele2", debug=None, pbc=False, **kwargs): """Set up the calculation of water bridges between two selections in a universe. The timeseries is accessible as the attribute :attr:`WaterBridgeAnalysis.timeseries`. If no hydrogen bonds are detected or if the initial check fails, look at the log output (enable with :func:`MDAnalysis.start_logging` and set `verbose` ``=True``). It is likely that the default names for donors and acceptors are not suitable (especially for non-standard ligands). In this case, either change the `forcefield` or use customized `donors` and/or `acceptors`. Parameters ---------- universe : Universe Universe object selection1 : str (optional) Selection string for first selection ['protein'] selection2 : str (optional) Selection string for second selection ['not resname SOL'] This string selects everything except water where water is assumed to have a residue name as SOL. water_selection : str (optional) Selection string for bridging water selection ['resname SOL'] The default selection assumes that the water molecules have residue name "SOL". Change it to the appropriate selection for your specific force field. However, in theory, this selection can be anything which forms a hydrogen bond with selection 1 and selection 2. order : int (optional) The maximum number of water bridges linking both selections. if the order is set to 3, then all the residues linked with less than three water molecules will be detected. [1] Computation of high order water bridges can be very time-consuming. Think carefully before running the calculation, do you really want to compute the 20th order water bridge between domain A and domain B or you just want to know the third order water bridge between two residues. selection1_type : {"donor", "acceptor", "both"} (optional) Selection 1 can be 'donor', 'acceptor' or 'both'. Note that the value for `selection1_type` automatically determines how `selection2` handles donors and acceptors: If `selection1` contains 'both' then `selection2` will also contain 'both'. If `selection1` is set to 'donor' then `selection2` is 'acceptor' (and vice versa). ['both']. update_selection : bool (optional) Update selection 1 and 2 at each frame. Setting to ``True`` if the selection is not static. Selections are filtered first to speed up performance. Thus, setting to ``True`` is recommended if contact surface between selection 1 and selection 2 is constantly changing. [``False``] update_water_selection : bool (optional) Update selection of water at each frame. Setting to ``False`` is **only** recommended when the total amount of water molecules in the simulation are small and when water molecules remain static across the simulation. However, in normal simulations, only a tiny proportion of water is engaged in the formation of water bridge. It is recommended to update the water selection and set keyword `filter_first` to ``True`` so as to filter out water not residing between the two selections. [``True``] filter_first : bool (optional) Filter the water selection to only include water within 4 Å + `order` * (2 Å + `distance`) away from `both` selection 1 and selection 2. Selection 1 and selection 2 are both filtered to only include atoms with the same distance away from the other selection. [``True``] distance : float (optional) Distance cutoff for hydrogen bonds; only interactions with a H-A distance <= `distance` (and the appropriate D-H-A angle, see `angle`) are recorded. (Note: `distance_type` can change this to the D-A distance.) [3.0] angle : float (optional) Angle cutoff for hydrogen bonds; an ideal H-bond has an angle of 180º. A hydrogen bond is only recorded if the D-H-A angle is >= `angle`. The default of 120º also finds fairly non-specific hydrogen interactions and possibly better value is 150º. [120.0] forcefield : {"CHARMM27", "GLYCAM06", "other"} (optional) Name of the forcefield used. Switches between different :attr:`~DEFAULT_DONORS` and :attr:`~DEFAULT_ACCEPTORS` values. ["CHARMM27"] donors : sequence (optional) Extra H donor atom types (in addition to those in :attr:`~DEFAULT_DONORS`), must be a sequence. acceptors : sequence (optional) Extra H acceptor atom types (in addition to those in :attr:`~DEFAULT_ACCEPTORS`), must be a sequence. distance_type : {"hydrogen", "heavy"} (optional) Measure hydrogen bond lengths between donor and acceptor heavy atoms ("heavy") or between donor hydrogen and acceptor heavy atom ("hydrogen"). If using "heavy" then one should set the *distance* cutoff to a higher value such as 3.5 Å. ["hydrogen"] output_format: {"sele1_sele2", "donor_acceptor"} (optional) Setting the output format for timeseries and table. If set to "sele1_sele2", for each hydrogen bond, the one close to selection 1 will be placed before selection 2. If set to "donor_acceptor", the donor will be placed before acceptor. "sele1_sele2"] debug : bool (optional) If set to ``True`` enables per-frame debug logging. This is disabled by default because it generates a very large amount of output in the log file. (Note that a logger must have been started to see the output, e.g. using :func:`MDAnalysis.start_logging`.) verbose : bool (optional) Toggle progress output. (Can also be given as keyword argument to :meth:`run`.) Notes ----- In order to speed up processing, atoms are filtered by a coarse distance criterion before a detailed hydrogen bonding analysis is performed (`filter_first` = ``True``). If selection 1 and selection 2 are very mobile during the simulation and the contact surface is constantly changing (i.e. residues are moving farther than 4 Å + `order` * (2 Å + `distance`)), you might consider setting the `update_selection` keywords to ``True`` to ensure correctness. .. versionchanged 0.20.0 The :attr:`WaterBridgeAnalysis.timeseries` has been updated see :attr:`WaterBridgeAnalysis.timeseries` for detail. This class is now based on :class:`~MDAnalysis.analysis.base.AnalysisBase`. """ super(WaterBridgeAnalysis, self).__init__(universe.trajectory, **kwargs) self.water_selection = water_selection self.update_water_selection = update_water_selection # per-frame debugging output? self.debug = debug # set the output format self.output_format = output_format self.u = universe self.selection1 = selection1 self.selection2 = selection2 self.selection1_type = selection1_type # if the selection 1 and selection 2 are the same if selection1 == selection2: # eliminate the duplication self.selection1_type = "donor" self.update_selection = update_selection self.filter_first = filter_first self.distance = distance self.distance_type = distance_type # note: everything except 'heavy' # will give the default behavior self.angle = angle self.pbc = pbc and all(self.u.dimensions[:3]) self.order = order # set up the donors/acceptors lists if donors is None: donors = () if acceptors is None: acceptors = () self.forcefield = forcefield self.donors = tuple(set(self.DEFAULT_DONORS[forcefield]).union(donors)) self.acceptors = tuple(set(self.DEFAULT_ACCEPTORS[forcefield]).union( acceptors)) if self.selection1_type not in ('both', 'donor', 'acceptor'): raise ValueError('WaterBridgeAnalysis: ' 'Invalid selection type {0!s}'.format( self.selection1_type)) self._network = [] # final result accessed as self.network self.timesteps = None # time for each frame self._log_parameters() def _log_parameters(self): """Log important parameters to the logfile.""" logger.info("WaterBridgeAnalysis: selection = %r (update: %r)", self.selection2, self.update_selection) logger.info("WaterBridgeAnalysis: water selection = %r (update: %r)", self.water_selection, self.update_water_selection) logger.info("WaterBridgeAnalysis: criterion: donor %s atom and " "acceptor atom distance <= %.3f A", self.distance_type, self.distance) logger.info("WaterBridgeAnalysis: criterion: " "angle D-H-A >= %.3f degrees", self.angle) logger.info("WaterBridgeAnalysis: force field %s to guess donor and \ acceptor names", self.forcefield) def _build_residue_dict(self, selection): # Build the residue_dict where the key is the residue name # The content is a dictionary where hydrogen bond donor heavy atom # names is the key # The content is the hydrogen bond donor hydrogen atom names atom_group = self.u.select_atoms(selection) for residue in atom_group.residues: if residue.resname not in self._residue_dict: self._residue_dict[residue.resname] = defaultdict(set) for atom in residue.atoms: if atom.name in self.donors: self._residue_dict[residue.resname][atom.name].update( self._get_bonded_hydrogens(atom).names) def _update_donor_h(self, atom_ix, h_donors, donors_h): atom = self.u.atoms[atom_ix] residue = atom.residue hydrogen_names = self._residue_dict[residue.resname][atom.name] if hydrogen_names: hydrogens = residue.atoms.select_atoms('name {0}'.format( ' '.join(hydrogen_names))).ix for atom in hydrogens: h_donors[atom] = atom_ix donors_h[atom_ix].append(atom) def _update_selection(self): self._s1_donors = [] self._s1_h_donors = {} self._s1_donors_h = defaultdict(list) self._s1_acceptors = [] self._s2_donors = [] self._s2_h_donors = {} self._s2_donors_h = defaultdict(list) self._s2_acceptors = [] self._s1 = self.u.select_atoms(self.selection1).ix self._s2 = self.u.select_atoms(self.selection2).ix if self.filter_first and len(self._s1): self.logger_debug('Size of selection 1 before filtering:' ' {} atoms'.format(len(self._s1))) ns_selection_1 = AtomNeighborSearch(self.u.atoms[self._s1], box=self.box) self._s1 = ns_selection_1.search(self.u.atoms[self._s2], self.selection_distance).ix self.logger_debug("Size of selection 1: {0} atoms".format( len(self._s1))) if len(self._s1) == 0: logger.warning('Selection 1 "{0}" did not select any atoms.' .format(str(self.selection1)[:80])) return if self.filter_first and len(self._s2): self.logger_debug('Size of selection 2 before filtering:' ' {} atoms'.format(len(self._s2))) ns_selection_2 = AtomNeighborSearch(self.u.atoms[self._s2], box=self.box) self._s2 = ns_selection_2.search(self.u.atoms[self._s1], self.selection_distance).ix self.logger_debug('Size of selection 2: {0} atoms'.format( len(self._s2))) if len(self._s2) == 0: logger.warning('Selection 2 "{0}" did not select any atoms.' .format(str(self.selection2)[:80])) return if self.selection1_type in ('donor', 'both'): self._s1_donors = self.u.atoms[self._s1].select_atoms( 'name {0}'.format(' '.join(self.donors))).ix for atom_ix in self._s1_donors: self._update_donor_h(atom_ix, self._s1_h_donors, self._s1_donors_h) self.logger_debug("Selection 1 donors: {0}".format( len(self._s1_donors))) self.logger_debug("Selection 1 donor hydrogens: {0}".format( len(self._s1_h_donors))) if self.selection1_type in ('acceptor', 'both'): self._s1_acceptors = self.u.atoms[self._s1].select_atoms( 'name {0}'.format(' '.join(self.acceptors))).ix self.logger_debug("Selection 1 acceptors: {0}".format( len(self._s1_acceptors))) if len(self._s2) == 0: return None if self.selection1_type in ('donor', 'both'): self._s2_acceptors = self.u.atoms[self._s2].select_atoms( 'name {0}'.format(' '.join(self.acceptors))).ix self.logger_debug("Selection 2 acceptors: {0:d}".format( len(self._s2_acceptors))) if self.selection1_type in ('acceptor', 'both'): self._s2_donors = self.u.atoms[self._s2].select_atoms( 'name {0}'.format(' '.join(self.donors))).ix for atom_ix in self._s2_donors: self._update_donor_h(atom_ix, self._s2_h_donors, self._s2_donors_h) self.logger_debug("Selection 2 donors: {0:d}".format( len(self._s2_donors))) self.logger_debug("Selection 2 donor hydrogens: {0:d}".format( len(self._s2_h_donors))) def _update_water_selection(self): self._water_donors = [] self._water_h_donors = {} self._water_donors_h = defaultdict(list) self._water_acceptors = [] self._water = self.u.select_atoms(self.water_selection).ix self.logger_debug('Size of water selection before filtering:' ' {} atoms'.format(len(self._water))) if len(self._water) and self.filter_first: filtered_s1 = AtomNeighborSearch(self.u.atoms[self._water], box=self.box).search( self.u.atoms[self._s1], self.selection_distance) if filtered_s1: self._water = AtomNeighborSearch(filtered_s1, box=self.box).search( self.u.atoms[self._s2], self.selection_distance).ix self.logger_debug("Size of water selection: {0} atoms".format( len(self._water))) if len(self._water) == 0: logger.warning("Water selection '{0}' did not select any atoms." .format(str(self.water_selection)[:80])) else: self._water_donors = self.u.atoms[self._water].select_atoms( 'name {0}'.format(' '.join(self.donors))).ix for atom_ix in self._water_donors: self._update_donor_h(atom_ix, self._water_h_donors, self._water_donors_h) self.logger_debug("Water donors: {0}".format( len(self._water_donors))) self.logger_debug("Water donor hydrogens: {0}".format( len(self._water_h_donors))) self._water_acceptors = self.u.atoms[self._water].select_atoms( 'name {0}'.format(' '.join(self.acceptors))).ix self.logger_debug("Water acceptors: {0}".format( len(self._water_acceptors))) def _get_bonded_hydrogens(self, atom): """Find hydrogens bonded within cutoff to `atom`. Hydrogens are detected by either name ("H*", "[123]H*") or type ("H"); this is not fool-proof as the atom type is not always a character but the name pattern should catch most typical occurrences. The distance from `atom` is calculated for all hydrogens in the residue and only those within a cutoff are kept. The cutoff depends on the heavy atom (more precisely, on its element, which is taken as the first letter of its name ``atom.name[0]``) and is parameterized in :attr:`WaterBridgeAnalysis.r_cov`. If no match is found then the default of 1.5 Å is used. Parameters ---------- atom : groups.Atom heavy atom Returns ------- hydrogen_atoms : AtomGroup or [] list of hydrogens (can be a :class:`~MDAnalysis.core.groups.AtomGroup`) or empty list ``[]`` if none were found. """ try: return atom.residue.atoms.select_atoms( "(name H* 1H* 2H* 3H* or type H) and around {0:f} name {1!s}" "".format(self.r_cov[atom.name[0]], atom.name)) except NoDataError: return [] def logger_debug(self, *args): if self.debug: logger.debug(*args) def _prepare(self): # The distance for selection is defined as twice the maximum bond # length of an O-H bond (2A) plus order of water bridge times the # length of OH bond plus hydrogne bond distance self.selection_distance = (2 * 2 + self.order * (2 + self.distance)) self.box = self.u.dimensions if self.pbc else None self._residue_dict = {} self._build_residue_dict(self.selection1) self._build_residue_dict(self.selection2) self._build_residue_dict(self.water_selection) self._update_selection() self.timesteps = [] if len(self._s1) and len(self._s2): self._update_water_selection() else: logger.info("WaterBridgeAnalysis: " "no atoms found in the selection.") logger.info("WaterBridgeAnalysis: initial checks passed.") logger.info("WaterBridgeAnalysis: starting") logger.debug("WaterBridgeAnalysis: donors %r", self.donors) logger.debug("WaterBridgeAnalysis: acceptors %r", self.acceptors) logger.debug("WaterBridgeAnalysis: water bridge %r", self.water_selection) if self.debug: logger.debug("Toggling debug to %r", self.debug) else: logger.debug("WaterBridgeAnalysis: For full step-by-step " "debugging output use debug=True") logger.info("Starting analysis " "(frame index start=%d stop=%d, step=%d)", self.start, self.stop, self.step) def _donor2acceptor(self, donors, h_donors, acceptor): if len(donors) == 0 or len(acceptor) == 0: return [] if self.distance_type != 'heavy': donors_idx = list(h_donors.keys()) else: donors_idx = list(donors.keys()) result = [] # Code modified from p-j-smith pairs, distances = capped_distance(self.u.atoms[donors_idx].positions, self.u.atoms[acceptor].positions, max_cutoff=self.distance, box=self.box, return_distances=True) if self.distance_type != 'heavy': tmp_donors = [h_donors[donors_idx[idx]] for idx in pairs[:, 0]] tmp_hydrogens = [donors_idx[idx] for idx in pairs[:, 0]] tmp_acceptors = [acceptor[idx] for idx in pairs[:, 1]] else: tmp_donors = [] tmp_hydrogens = [] tmp_acceptors = [] for idx in range(len(pairs[:, 0])): for h in donors[donors_idx[pairs[idx, 0]]]: tmp_donors.append(donors_idx[pairs[idx, 0]]) tmp_hydrogens.append(h) tmp_acceptors.append(acceptor[pairs[idx, 1]]) angles = np.rad2deg( calc_angles( self.u.atoms[tmp_donors].positions, self.u.atoms[tmp_hydrogens].positions, self.u.atoms[tmp_acceptors].positions, box=self.box ) ) hbond_indices = np.where(angles > self.angle)[0] for index in hbond_indices: h = tmp_hydrogens[index] d = tmp_donors[index] a = tmp_acceptors[index] result.append((h, d, a, self._expand_index(h), self._expand_index(a), distances[index], angles[index])) return result def _single_frame(self): self.timesteps.append(self._ts.time) self.box = self.u.dimensions if self.pbc else None if self.update_selection: self._update_selection() if len(self._s1) and len(self._s2): if self.update_water_selection: self._update_water_selection() else: self._network.append(defaultdict(dict)) return selection_1 = [] water_pool = defaultdict(list) next_round_water = set([]) selection_2 = [] if self.selection1_type in ('donor', 'both'): # check for direct hbond from s1 to s2 self.logger_debug("Selection 1 Donors <-> Selection 2 Acceptors") results = self._donor2acceptor( self._s1_donors_h, self._s1_h_donors, self._s2_acceptors) for line in results: h_index, d_index, a_index, (h_resname, h_resid, h_name), \ (a_resname, a_resid, a_name), dist, angle = line water_pool[(a_resname, a_resid)] = None selection_1.append( (h_index, d_index, a_index, None, dist, angle)) selection_2.append((a_resname, a_resid)) if self.order > 0: self.logger_debug("Selection 1 Donors <-> Water Acceptors") results = self._donor2acceptor( self._s1_donors_h, self._s1_h_donors, self._water_acceptors) for line in results: h_index, d_index, a_index, (h_resname, h_resid, h_name), ( a_resname, a_resid, a_name), dist, angle = line selection_1.append( (h_index, d_index, a_index, None, dist, angle)) self.logger_debug("Water Donors <-> Selection 2 Acceptors") results = self._donor2acceptor( self._water_donors_h, self._water_h_donors, self._s2_acceptors) for line in results: h_index, d_index, a_index, (h_resname, h_resid, h_name), ( a_resname, a_resid, a_name), dist, angle = line water_pool[(h_resname, h_resid)].append( (h_index, d_index, a_index, None, dist, angle)) selection_2.append((a_resname, a_resid)) if self.selection1_type in ('acceptor', 'both'): self.logger_debug("Selection 2 Donors <-> Selection 1 Acceptors") results = self._donor2acceptor(self._s2_donors_h, self._s2_h_donors, self._s1_acceptors) for line in results: h_index, d_index, a_index, (h_resname, h_resid, h_name), \ (a_resname, a_resid, a_name), dist, angle = line water_pool[(h_resname, h_resid)] = None selection_1.append( (a_index, None, h_index, d_index, dist, angle)) selection_2.append((h_resname, h_resid)) if self.order > 0: self.logger_debug("Selection 2 Donors <-> Water Acceptors") results = self._donor2acceptor( self._s2_donors_h, self._s2_h_donors, self._water_acceptors) for line in results: h_index, d_index, a_index, (h_resname, h_resid, h_name), ( a_resname, a_resid, a_name), dist, angle = line water_pool[(a_resname, a_resid)].append( (a_index, None, h_index, d_index, dist, angle)) selection_2.append((h_resname, h_resid)) self.logger_debug("Selection 1 Acceptors <-> Water Donors") results = self._donor2acceptor( self._water_donors_h, self._water_h_donors, self._s1_acceptors) for line in results: h_index, d_index, a_index, (h_resname, h_resid, h_name), ( a_resname, a_resid, a_name), dist, angle = line selection_1.append( (a_index, None, h_index, d_index, dist, angle)) if self.order > 1: self.logger_debug("Water donor <-> Water Acceptors") results = self._donor2acceptor(self._water_donors_h, self._water_h_donors, self._water_acceptors) for line in results: h_index, d_index, a_index, (h_resname, h_resid, h_name), ( a_resname, a_resid, a_name), dist, angle = line water_pool[(a_resname, a_resid)].append( (a_index, None, h_index, d_index, dist, angle)) water_pool[(h_resname, h_resid)].append( (h_index, d_index, a_index, None, dist, angle)) # solve the connectivity network # The following code attempt to generate a water network which is # formed by the class dict. # Suppose we have a water bridge connection ARG1 to ASP3 via the two # hydrogen bonds. # [0,1,('ARG',1,'O'), ('SOL',2,'HW1'), 3.0,180], # [2,3,('SOL',2,'HW2'),('ASP',3,'OD1'), 3.0,180], # The resulting network will be # {(0,1,('ARG',1,'O'), ('SOL',2,'HW1'), 3.0,180): # {(2,3,('SOL',2,'HW2'),('ASP',3,'OD1'), 3.0,180): None}} # Where the key of the a dict will be all the hydrogen bonds starting # from this nodes. # The corresponding value of a certain key will be a dictionary whose # key will be all the hydrogen bonds from # the destination of in the key. # If the value of a certain key is None, which means it is reaching # selection 2. result = {'start': defaultdict(dict), 'water': defaultdict(dict)} def add_route(result, route): if len(route) == 1: result['start'][route[0]] = None else: # exclude the the selection which goes back to itself if (sorted(route[0][0:3:2]) == sorted(route[-1][0:3:2])): return # selection 2 to water result['water'][route[-1]] = None # water to water for i in range(1, len(route) - 1): result['water'][route[i]][route[i+1]] = \ result['water'][route[i+1]] # selection 1 to water result['start'][route[0]][route[1]] = result['water'][route[1]] def traverse_water_network(graph, node, end, route, maxdepth, result): if len(route) > self.order + 1: return else: if node in end: # check if any duplication happens if len(route) == len(set(route)): add_route(result, route) else: for new_node in graph[node]: new_route = route[:] new_route.append(new_node) new_node = self._expand_timeseries( new_node, 'sele1_sele2')[3][:2] traverse_water_network(graph, new_node, end, new_route, maxdepth, result) for s1 in selection_1: route = [s1, ] next_mol = self._expand_timeseries(s1, 'sele1_sele2')[3][:2] traverse_water_network(water_pool, next_mol, selection_2, route[:], self.order, result) self._network.append(result['start']) def _traverse_water_network(self, graph, current, analysis_func=None, output=None, link_func=None, **kwargs): ''' This function recursively traverses the water network self._network and finds the hydrogen bonds which connect the current atom to the next atom. The newly found hydrogen bond will be appended to the hydrogen bonds connecting the selection 1 to the current atom via link_func. When selection 2 is reached, the full list of hydrogen bonds connecting the selection 1 to selection 2 will be fed into analysis_func, which will then modify the output in place. :param graph: The connection network describes the connection between the atoms in the water network. :param current: The hydrogen bonds from selection 1 until now. :param analysis_func: The analysis function which is called to analysis the hydrogen bonds. :param output: where the result is stored. :param link_func: The new hydrogen bonds will be appended to current. :param kwargs: the keywords which are passed into the analysis_func. :return: ''' if link_func is None: # If no link_func is provided, the default link_func will be used link_func = self._full_link if graph is None: # if selection 2 is reached if analysis_func is not None: # the result is analysed by analysis_func which will change the # output analysis_func(current, output, self.u, **kwargs) else: # make sure no loop can occur if len(current) <= self.order: for node in graph: # the new hydrogen bond will be added to the existing bonds new = link_func(current, node) self._traverse_water_network(graph[node], new, analysis_func, output, link_func, **kwargs) def _expand_index(self, index): ''' Expand the index into (resname, resid, name). ''' atom = self.u.atoms[index] return (atom.resname, atom.resid, atom.name) def _expand_timeseries(self, entry, output_format=None): ''' Expand the compact data format into the old timeseries form. The old is defined as the format for release up to 0.19.2. As is discussed in Issue #2177, the internal storage of the hydrogen bond information has been changed to the compact format. The function takes in the argument `output_format` to see which output format will be chosen. if `output_format` is not specified, the value will be taken from :attr:`output_format`. If `output_format` is 'sele1_sele2', the output will be the old water bridge analysis format:: # donor from selection 1 to acceptor in selection 2 [sele1_index, sele2_index, (sele1_resname, sele1_resid, sele1_name), (sele2_resname, sele2_resid, sele2_name), dist, angle] If `output_format` is 'donor_acceptor', the output will be the old hydrogen bond analysis format:: # From donor to acceptor [donor_index, acceptor_index, (donor_resname, donor_resid, donor_name), (acceptor_resname, acceptor_resid, acceptor_name), dist, angle] ''' output_format = output_format or self.output_format # Expand the compact entry into atom1, which is the first index in the # output and atom2, which is the second # entry. atom1, heavy_atom1, atom2, heavy_atom2, dist, angle = entry if output_format == 'sele1_sele2': # If the output format is the sele1_sele2, no change will be # executed atom1, atom2 = atom1, atom2 elif output_format == 'donor_acceptor': # If the output format is donor_acceptor, use heavy atom position # to check which is donor and which is # acceptor if heavy_atom1 is None: # atom1 is hydrogen bond acceptor and thus, the position of # atom1 and atom2 are swapped. atom1, atom2 = atom2, atom1 else: # atom1 is hydrogen bond donor, position not swapped. atom1, atom2 = atom1, atom2 else: raise KeyError( "Only 'sele1_sele2' or 'donor_acceptor' are allowed as output " "format") return (atom1, atom2, self._expand_index(atom1), self._expand_index(atom2), dist, angle) def _generate_timeseries(self, output_format=None): r'''Time series of water bridges. The output is generated per frame as is explained in :ref:`wb_Analysis_Timeseries`. The format of output can be changed via the output_format selection. If ``output_format="sele1_sele2"``, the hydrogen bond forms a directional link from selection 1 to selection 2. If ``output_format="donor_acceptor"``, for each hydrogen bond, the donor is always written before the acceptor. Note ---- To find an acceptor atom in :attr:`Universe.atoms` by *index* one would use ``u.atoms[acceptor_index]``. The :attr:`timeseries` is a managed attribute and it is generated from the underlying data in :attr:`_network` every time the attribute is accessed. It is therefore costly to call and if :attr:`timeseries` is needed repeatedly it is recommended that you assign to a variable:: w = WaterBridgeAnalysis(u) w.run() timeseries = w.timeseries .. versionchanged 0.20.0 The :attr:`WaterBridgeAnalysis.timeseries` has been updated where the donor and acceptor string has been changed to tuple (resname string, resid, name_string). ''' output_format = output_format or self.output_format def analysis(current, output, *args, **kwargs): output = current timeseries = [] for frame in self._network: new_frame = [] self._traverse_water_network(frame, new_frame, analysis_func=analysis, output=new_frame, link_func=self._compact_link) timeseries.append([ self._expand_timeseries(entry, output_format) for entry in new_frame]) return timeseries timeseries = property(_generate_timeseries) def _get_network(self): r'''Network representation of the water network. The output is generated per frame as is explained in :ref:`wb_Analysis_Network`. Each hydrogen bond has a compact representation of :: [sele1_acceptor_idx, None, sele2_donor_idx, donor_heavy_idx, distance, angle] or :: [sele1_donor_idx, donor_heavy_idx, sele1_acceptor_idx, None, distance, angle] The donor_heavy_idx is the heavy atom bonding to the proton and atoms can be retrived from the universe:: atom = u.atoms[idx] .. versionadded:: 0.20.0 ''' return self._network def set_network(self, network): self._network = network network = property(_get_network, set_network) @classmethod def _full_link(self, output, node): ''' A function used in _traverse_water_network to add the new hydrogen bond to the existing bonds. :param output: The existing hydrogen bonds from selection 1 :param node: The new hydrogen bond :return: The hydrogen bonds from selection 1 with the new hydrogen bond added ''' result = output[:] result.append(node) return result @classmethod def _compact_link(self, output, node): ''' A function used in _traverse_water_network to add the new hydrogen bond to the existing bonds. In this form no new list is created and thus, one bridge will only appear once. :param output: The existing hydrogen bonds from selection 1 :param node: The new hydrogen bond :return: The hydrogen bonds from selection 1 with the new hydrogen bond added ''' output.append(node) return output def _count_by_type_analysis(self, current, output, *args, **kwargs): ''' Generates the key for count_by_type analysis. :return: ''' s1_index, to_index, s1, to_residue, dist, angle = \ self._expand_timeseries(current[0]) s1_resname, s1_resid, s1_name = s1 from_index, s2_index, from_residue, s2, dist, angle = \ self._expand_timeseries(current[-1]) s2_resname, s2_resid, s2_name = s2 key = (s1_index, s2_index, s1_resname, s1_resid, s1_name, s2_resname, s2_resid, s2_name) output[key] += 1
[docs] def count_by_type(self, analysis_func=None, **kwargs): """Counts the frequency of water bridge of a specific type. If one atom *A* from *selection 1* is linked to atom *B* from *selection 2* through one or more bridging waters, an entity will be created and the proportion of time that this linkage exists in the whole simulation will be calculated. The identification of a specific type of water bridge can be modified by supplying the analysis_func function. See :ref:`wb_count_by_type` for detail. Returns ------- counts : list Returns a :class:`list` containing atom indices for *A* and *B*, residue names, residue numbers, atom names (for both A and B) and the fraction of the total time during which the water bridge was detected. This method returns None if method :meth:`WaterBridgeAnalysis.run` was not executed first. """ output = None if analysis_func is None: analysis_func = self._count_by_type_analysis output = 'combined' if self._network: length = len(self._network) result_dict = defaultdict(int) for frame in self._network: frame_dict = defaultdict(int) self._traverse_water_network(frame, [], analysis_func=analysis_func, output=frame_dict, link_func=self._full_link, **kwargs) for key, value in frame_dict.items(): result_dict[key] += frame_dict[key] if output == 'combined': result = [[i for i in key] for key in result_dict] [result[i].append(result_dict[key]/length) for i, key in enumerate(result_dict)] else: result = [(key, result_dict[key]/length) for key in result_dict] return result else: return None
def _count_by_time_analysis(self, current, output, *args, **kwargs): s1_index, to_index, s1, to_residue, dist, angle = \ self._expand_timeseries(current[0]) s1_resname, s1_resid, s1_name = s1 from_index, s2_index, from_residue, s2, dist, angle = \ self._expand_timeseries(current[-1]) s2_resname, s2_resid, s2_name = s2 key = (s1_index, s2_index, s1_resname, s1_resid, s1_name, s2_resname, s2_resid, s2_name) output[key] += 1
[docs] def count_by_time(self, analysis_func=None, **kwargs): """Counts the number of water bridges per timestep. The counting behaviour can be adjusted by supplying analysis_func. See :ref:`wb_count_by_time` for details. Returns ------- counts : list Returns a time series ``N(t)`` where ``N`` is the total number of observed water bridges at time ``t``. """ if analysis_func is None: analysis_func = self._count_by_time_analysis if self._network: result = [] for time, frame in zip(self.timesteps, self._network): result_dict = defaultdict(int) self._traverse_water_network(frame, [], analysis_func=analysis_func, output=result_dict, link_func=self._full_link, **kwargs) result.append((time, sum([result_dict[key] for key in result_dict]))) return result else: return None
def _timesteps_by_type_analysis(self, current, output, *args, **kwargs): s1_index, to_index, s1, to_residue, dist, angle = \ self._expand_timeseries(current[0]) s1_resname, s1_resid, s1_name = s1 from_index, s2_index, from_residue, s2, dist, angle = \ self._expand_timeseries(current[-1]) s2_resname, s2_resid, s2_name = s2 key = (s1_index, s2_index, s1_resname, s1_resid, s1_name, s2_resname, s2_resid, s2_name) output[key].append(kwargs.pop('time'))
[docs] def timesteps_by_type(self, analysis_func=None, **kwargs): """Frames during which each water bridges existed, sorted by each water bridges. Processes :attr:`WaterBridgeAnalysis._network` and returns a :class:`list` containing atom indices, residue names, residue numbers (from selection 1 and selection 2) and each timestep at which the water bridge was detected. Similar to :meth:`~WaterBridgeAnalysis.count_by_type` and :meth:`~WaterBridgeAnalysis.count_by_time`, the behavior can be adjusted by supplying an analysis_func. Returns ------- data : list """ output = None if analysis_func is None: analysis_func = self._timesteps_by_type_analysis output = 'combined' if self._network: result = defaultdict(list) if self.timesteps is None: timesteps = range(len(self._network)) else: timesteps = self.timesteps for time, frame in zip(timesteps, self._network): self._traverse_water_network(frame, [], analysis_func=analysis_func, output=result, link_func=self._full_link, time=time, **kwargs) result_list = [] for key, time_list in six.iteritems(result): for time in time_list: if output == 'combined': key = list(key) key.append(time) result_list.append(key) else: result_list.append((key, time)) return result_list else: return None
[docs] def generate_table(self, output_format=None): """Generate a normalised table of the results. The table is stored as a :class:`numpy.recarray` in the attribute :attr:`~WaterBridgeAnalysis.table`. The output format of :attr:`~WaterBridgeAnalysis.table` can also be changed using output_format in a fashion similar to :attr:`WaterBridgeAnalysis.timeseries` """ output_format = output_format or self.output_format if self._network == []: msg = "No data computed, do run() first." warnings.warn(msg, category=MissingDataWarning) logger.warning(msg) return None timeseries = self._generate_timeseries(output_format) num_records = np.sum([len(hframe) for hframe in timeseries]) # build empty output table if output_format == 'sele1_sele2': dtype = [ ("time", float), ("sele1_index", int), ("sele2_index", int), ("sele1_resnm", "|U4"), ("sele1_resid", int), ("sele1_atom", "|U4"), ("sele2_resnm", "|U4"), ("sele2_resid", int), ("sele2_atom", "|U4"), ("distance", float), ("angle", float)] elif output_format == 'donor_acceptor': dtype = [ ("time", float), ("donor_index", int), ("acceptor_index", int), ("donor_resnm", "|U4"), ("donor_resid", int), ("donor_atom", "|U4"), ("acceptor_resnm", "|U4"), ("acceptor_resid", int), ("acceptor_atom", "|U4"), ("distance", float), ("angle", float)] # according to Lukas' notes below, using a recarray at this stage is # ineffective and speedups of ~x10 can be achieved by filling a # standard array, like this: out = np.empty((num_records,), dtype=dtype) cursor = 0 # current row for t, hframe in zip(self.timesteps, timeseries): for (donor_index, acceptor_index, donor, acceptor, distance, angle) in hframe: # donor|acceptor = (resname, resid, atomid) out[cursor] = (t, donor_index, acceptor_index) + \ donor + acceptor + (distance, angle) cursor += 1 assert cursor == num_records, \ "Internal Error: Not all wb records stored" table = out.view(np.recarray) logger.debug( "WBridge: Stored results as table with %(num_records)d entries.", vars()) self.table = table