<commit_before>##############################################################################
# MDTraj: A Python Library for Loading, Saving, and Manipulating
#         Molecular Dynamics Trajectories.
# Copyright 2012-2013 Stanford University and the Authors
#
# Authors: <NAME>
# Contributors: <NAME>
#
# MDTraj is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 2.1
# of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with MDTraj. If not, see <http://www.gnu.org/licenses/>.
##############################################################################

from __future__ import print_function, division

__all__ = ['baker_hubbard', 'shrake_rupley', 'kabsch_sander', 'compute_distances',
           'compute_displacements', 'compute_angles', 'compute_dihedrals',
           'compute_phi', 'compute_psi', 'compute_chi1', 'compute_chi2',
           'compute_chi3', 'compute_chi4', 'compute_omega', 'compute_rg',
           'compute_contacts', 'compute_drid']

from mdtraj.geometry.rg import *
from mdtraj.geometry.angle import *
from mdtraj.geometry.distance import *
from mdtraj.geometry.dihedral import *
from mdtraj.geometry.hbond import *
from mdtraj.geometry.sasa import *
from mdtraj.geometry.contact import *
try:
    from mdtraj.geometry.drid import compute_drid
except ImportError:
    def compute_drid(traj, atom_indices=None):
        raise NotImplementedError<commit_msg>check cpu support<commit_after>##############################################################################
# MDTraj: A Python Library for Loading, Saving, and Manipulating
#         Molecular Dynamics Trajectories.
# Copyright 2012-2013 Stanford University and the Authors
#
# Authors: <NAME>
# Contributors: <NAME>
#
# MDTraj is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 2.1
# of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with MDTraj. If not, see <http://www.gnu.org/licenses/>.
##############################################################################

from __future__ import print_function, division

__all__ = ['baker_hubbard', 'shrake_rupley', 'kabsch_sander', 'compute_distances',
           'compute_displacements', 'compute_angles', 'compute_dihedrals',
           'compute_phi', 'compute_psi', 'compute_chi1', 'compute_chi2',
           'compute_chi3', 'compute_chi4', 'compute_omega', 'compute_rg',
           'compute_contacts', 'compute_drid']

from mdtraj.geometry.rg import *
from mdtraj.geometry.angle import *
from mdtraj.geometry.distance import *
from mdtraj.geometry.dihedral import *
from mdtraj.geometry.hbond import *
from mdtraj.geometry.sasa import *
from mdtraj.geometry.contact import *
try:
    from mdtraj.geometry.drid import compute_drid
    if not _geometry._processor_supports_sse41():
        raise ImportError()
except ImportError:
    def compute_drid(traj, atom_indices=None):
        """compute_drid(traj, atom_indices=None)

        Distribution of reciprocal interatomic distances (DRID) representation
        of an MD trajectory

        Parameters
        ----------
        traj : md.Trajectory
            A trajectory
        atom_indices : np.ndarray, dtype=int, shape=(n_atom_indices,) or None
            The indices (zero-based) of the atoms to use in the DRID representation.
            If None, all atoms will be used.

        Returns
        -------
        X : np.ndarray, shape=(n_frames, n_atom_indices*3)

        Notes
        -----
        This function is stub for the actual function, and indicates a failure
        to compile or import the DRID function, due to the fact that the CPU
        does not support the required instruction set (SSE4.1).

        References
        ----------
        ..[1] <NAME>; Distribution of Reciprocal of Interatomic Distances:
        A Fast Structural Metric. JCTC 2012 doi:10.1021/ct3003145
        """
        raise NotImplementedError
{'fraction_non_alphanumeric': 0.143693787910479, 'fraction_numerical': 0.010667224430035557, 'mean_word_length': 3.810865191146881, 'pattern_counts': {'":': 0, '<': 10, '<?xml version=': 0, '>': 10, 'https://': 0, 'lorem ipsum': 0, 'www.': 2, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '6058834', 'n_tokens_mistral': 1321, 'n_tokens_neox': 1248, 'n_words': 465}