i-< d Z ddlmZmZ ddlZddlmZ ddlm Z ddl
m
Z
ddl m Z ddl
Z
ddlZdd lmZ d
d lmZ d d
lmZ d dlmZmZ ddZ ddZ ddZdS )a
Distance Matrix calculation --- :mod:`MDAnalysis.analysis.ensemble.confdistmatrix`
==================================================================================
The module contains a base class to easily compute, using
parallelization and shared memory, matrices of conformational
distance between the structures stored as frames in a Universe. A
class to compute an RMSD matrix in such a way is also available.
:Author: Matteo Tiberti, Wouter Boomsma, Tone Bengtsen
.. versionadded:: 0.16.0
.. deprecated:: 2.8.0
This module is deprecated in favour of the
MDAKit `mdaencore `_ and will be removed
in MDAnalysis 3.0.0.
)ParalleldelayedN)getuser) gethostname)datetime)sleep )Universe )rotation_matrix )PureRMSD)TriangularMatrix trm_indices TmassFc
t | j | | d }
|rt j t
t t t j
| j |
||dk fgdt fdt fdt fdt fdt fd t fd
t fd t fg }| j | | d |r7|r|} n|} | j | | d nd
t t t" t j f szdk rt| | j t j |r8| | j t j nd
nt j | j | | d j d t j |rHt j t0 d j d t j n:d
n7|r3t dk rt3 d d d nd
|
|
dz z dz } t j | t j t7 d|
dz
|
dz
f }
t9 ||d| fd|
D t; | S )a
Run the conformational distance matrix calculation.
args and kwargs are passed to conf_dist_function.
Parameters
----------
ensemble : Universe object
Universe object for which the conformational distance matrix will
be computed.
conf_dist_function : function object
Function that fills the matrix with conformational distance
values. See set_rmsd_matrix_elements for an example.
select : str, optional
use this selection for the calculation of conformational distance
superimposition_select : str, optional
use atoms from this selection for fitting instead of those of
`select`
pairwise_align : bool, optional
Whether to perform pairwise alignment between conformations.
Default is True (do the superimposition)
weights : str/array_like, optional
weights to be used for fit. Can be either 'mass' or an array_like
metadata : bool, optional
Whether to build a metadata dataset for the calculated matrix.
Default is True.
n_jobs : int, optional
Number of cores to be used for parallel calculation
Default is 1. -1 uses all available cores
max_nbytes : str, optional
Threshold on the size of arrays passed to the workers that triggers automated memory mapping in temp_folder (default is None).
See https://joblib.readthedocs.io/en/latest/generated/joblib.Parallel.html for detailed documentation.
verbose : bool, optional
enable verbose output
Returns
-------
conf_dist_matrix : encore.utils.TriangularMatrix object
Conformational distance matrix in triangular representation.
facorderr hostuserdatez
topology fileznumber of frameszpairwise superimpositionzsuperimposition subsetz
mass-weighted)dtypeNr r zzused pairwise alignment with custom weights. Please provide 2 tuple with weights for 'select' and 'superimposition_select'r
)r r sharedmem)n_jobsverboserequire
max_nbytesc 3 v K | ]3} t t j | V 4d S )N)r npint64).0elementconf_dist_functiondistmatfitting_coordinatesrmsd_coordinatessubset_weightsweightss s/srv/www/vhosts/g4struct/public_html/venv/lib/python3.11/site-packages/MDAnalysis/analysis/encore/confdistmatrix.py z1conformational_distance_matrix.. sj
$"## HW
)metadata) len
trajectory
timeseries select_atomsr! arrayr r strr nowfilenameobjectintbool
isinstancelisttuplendarraymassesastypefloat64onesshape
fit_coords RuntimeErroremptyr r r )ensembler% selectsuperimposition_selectr pairwise_alignr* r. r r framesn
subset_selectmatsizeindicesr& r' r( r) s ` ` @@@@r+ conformational_distance_matrixrN < s l && ! !& ) ) '
G
8 MMII '' %"v%
&)#S)+T2)62 $'
6 *55 f%%U 6
# ! # 2MM "M & 1 < < ! !- 0 0 =
#
wubj 9 : :%" v ''//6==bjII " % 2 2 VVBJ''
N "NN '#..x/D/DV/L/LMM
&
" Wjm&9!&< > > E E
NN "NN "7||q "/ %QZN ajGG !N 1%*Ghw
++G &7Q;! "<==GH
$ Gh 7 7 77r- c | \ }} ||t j | }
t || t j || t j || j d ||
||dz |z dz | z < dS |e|bt j | }
t j | t j | z } t j || d| } || | z
}
|| | z
}t j || d| }|| |z
}|| |z
}t ||| d }t j
t j |t j
|
}t | t j | t j || j d ||
||dz |z dz | z < dS t d )a
RMSD Matrix calculator
Parameters
----------
tasks : iterator of int of length 2
Given a triangular matrix, this function will calculate RMSD
values from element tasks[0] to tasks[1]. Since the matrix
is triangular, the trm_indices matrix automatically
calculates the corrisponding i,j matrix indices.
The matrix is written as an array in a row-major
order (see the TriangularMatrix class for details).
If fit_coords and fit_weights are specified, the structures
will be superimposed before calculating RMSD, and fit_coords and fit_weights
will be used to place both structures at their center of mass and
compute the rotation matrix. In this case, both fit_coords and fit_weights
must be specified.
coords : numpy.array
Array of the ensemble coordinates
weights : numpy.array
Array of atomic weights, having the same order as the
coordinates array
rmsdmat : encore.utils.TriangularMatrix
Memory-shared triangular matrix object
fit_coords : numpy.array or None, optional
Array of the coordinates used for fitting
fit_weights : numpy.array. optional
Array of atomic weights, having the same order as the
fit_coords array
Nr r
r )axisr* zIBoth fit_coords and fit_weights must be specified if one of them is given)
r! sumr r? r@ rB asarraymeanaverager transposedot TypeError)taskscoordsrmsdmatr* rC fit_weightsargskwargsij
sumweightsr) com_i translated_isubset1_coordscom_j translated_jsubset2_coordsrotamat rotated_is r+ set_rmsd_matrix_elementsri s R DAqk 1VG__
(0 1I RZ ( ( 1I RZ ( ( 1IOA )
)
Q! q 1$%%%
K$;VG__
K0027;3G3GG
:a=q+FFF ay5( #A .
:a=q+FFF ay5( #A .! NN
L l1K1K!L!LMM (0 RZ ( (
+ + 1IOA )
)
Q! q 1$%%%
&
r- name CAc
|r9t j d | t | j | | d j d | } t j d | j j j
D ]H}
t j d |
t | j |
d I| j | j | | d j d k st j
d d S n*| t |t t" t$ j f s |d
k rd }n|d }nd
}t j d t | t j d | |r't j d | t j d t) | t* |||||| } t j d |r| | | S )aq
Retrieves or calculates the conformational distance (RMSD)
matrix. The distance matrix is calculated between all the frames of all
the :class:`~MDAnalysis.core.universe.Universe` objects given as input.
The order of the matrix elements depends on the order of the coordinates
of the ensembles and on the order of the input ensembles themselves,
therefore the order of the input list is significant.
The distance matrix can either be calculated from input ensembles or
loaded from an input numpy binary file.
Please notice that the .npz file does not contain a bi-dimensional array,
but a flattened representation that is meant to represent the elements of
an encore.utils.TriangularMatrix object.
Parameters
----------
ensemble : Universe
select : str
Atom selection string in the MDAnalysis format. Default is "name CA"
load_matrix : str, optional
Load similarity/dissimilarity matrix from numpy binary file instead
of calculating it (default is None). A filename is required.
save_matrix : bool, optional
Save calculated matrix as numpy binary file (default is None). A
filename is required.
superimpose : bool, optional
Whether to superimpose structures before calculating distance
(default is True).
superimposition_subset : str, optional
Group for superimposition using MDAnalysis selection syntax
(default is CA atoms: "name CA")
weights : str/array_like, optional
weights to be used for fit. Can be either 'mass' or an array_like
n_jobs : int, optional
Maximum number of cores to be used (default is 1). If -1 use all cores.
max_nbytes : str, optional
Threshold on the size of arrays passed to the workers that triggers automated memory mapping in temp_folder (default is None).
See https://joblib.readthedocs.io/en/latest/generated/joblib.Parallel.html for detailed documentation.
verbose : bool, optional
print progress
Returns
-------
confdistmatrix : encore.utils.TriangularMatrix
Conformational distance matrix. .
z+ Loading similarity matrix from: {0}r r r )sizeloadfilez
Done!z {0} : {1}zEERROR: The size of the loaded matrix and of the ensemble do not matchNr MassNoneCustomz' Perform pairwise alignment: {0}z( weighted alignment and RMSD: {0}z' Atoms subset for alignment: {0}z' Calculating similarity matrix . . .)r% rG rI r* r r r z Done!)logginginfoformatr r0 r1 r2 rB r. r namesr4 rl errortransfer_to_memoryr: r; r< r! r= rN ri savez)rF rG load_matrix save_matrix superimposesuperimposition_subsetr* r r r conf_dist_argsconf_dist_kwargsconfdistmatrixkey weight_types r+ get_distance_matrixr 8 s B L. 9 @ @ M M
*$//%%f--U 0 A !
_%%%!*06 C L#**^ 4S 9! <==
#"--%%f--U . A
M
4 ##%%% 7T5"*$=>> #6!! KK
_ KK"K 5 < ??? 8 7&!
[!!! . - - - r- )r r r
Tr TFN)NN) rj NNTrj r r
NF)__doc__joblibr r numpyr! getpassr socketr r timer rq warnings
core.universer
alignr cutilsr utilsr r rN ri r r- r+ r sr . ( % $ $ $ $ $ $ $ % % % % % % # # # # # # 0 0 0 0 0 0 0 0
i8 i8 i8 i8b
M
M
M
M
d $ O O O O O Or-