Source code for mordred.AdjacencyMatrix

from ._base import Descriptor
from ._graph_matrix import AdjacencyMatrix as A
from ._matrix_attributes import methods, get_method

__all__ = ('AdjacencyMatrix',)


class AdjacencyMatrix(Descriptor):
    r"""adjacency matrix descriptor.

    :type type: :py:class:`str`
    :param type: :ref:`matrix_aggregating_methods`
    """
    __slots__ = ('_type',)
    explicit_hydrogens = False

    @classmethod
    def preset(cls):
        return map(cls, methods)

    def __str__(self):
        return '{}_A'.format(self._type.__name__)

    def parameters(self):
        return self._type,

    def __init__(self, type='SpMax'):
        self._type = get_method(type)

    def dependencies(self):
        return {
            'result': self._type(
                A(self.explicit_hydrogens),
                self.explicit_hydrogens,
                self.kekulize,
            )
        }

    def calculate(self, result):
        return result

    rtype = float