Source code for molsysmt.structure.principal_component_analysis

from molsysmt._private.exceptions import NotImplementedMethodError
from molsysmt._private.digestion import digest
from molsysmt import lib as msmlib
from molsysmt._private.variables import is_all, is_iterable_of_iterables
from molsysmt import pyunitwizard as puw
import numpy as np
import gc


[docs]
@digest()
def principal_component_analysis(molecular_system, selection='all', structure_indices='all',
        weights=None, syntax='MolSysMT', engine='MolSysMT', skip_digestion=False):

    from molsysmt.basic import select, get

    if engine=='MolSysMT':

        atom_indices = select(molecular_system, selection=selection, syntax=syntax)

        coordinates = get(molecular_system, element='atom', selection=atom_indices,
                structure_indices=structure_indices, coordinates=True)
        coordinates, length_unit = puw.get_value_and_unit(coordinates)

        if weights is None:
            weights = np.ones((coordinates.shape[1]), dtype=np.float64)

        variances, principal_components = msmlib.structure.principal_component_analysis(coordinates, weights)

        del(coordinates, atom_indices, weights)

        gc.collect()

        return principal_components, variances

    else:

        raise NotImplementedMethodError()


# https://manual.gromacs.org/documentation/2019-rc1/reference-manual/analysis/covariance-analysis.html