
     i%                         d Z ddlZddlZddlmZ ddlmZ ddlm	Z	 ddl
mZmZmZmZmZmZmZmZmZmZmZ  G d d	e          Z G d
 de          ZdS )a  OpenMM topology parser :mod:`MDAnalysis.converters.OpenMMParser`
===================================================================

.. versionadded:: 2.0.0
.. versionchanged:: 2.8.0
   Removed type and mass guessing (attributes guessing takes place
   now through universe.guess_TopologyAttrs() API)


Converts an
`OpenMM topology <http://docs.openmm.org/latest/api-python/generated/openmm.app.topology.Topology.html#openmm.app.topology.Topology>`_
:class:`openmm.app.topology.Topology` into a :class:`MDAnalysis.core.Topology`.

Also converts some objects within the
`OpenMM Application layer <http://docs.openmm.org/latest/api-python/app.html>`_

- `openmm.app.pdbfile.PDBFile <http://docs.openmm.org/latest/api-python/generated/openmm.app.pdbfile.PDBFile.html#openmm.app.pdbfile.PDBFile>`_
- `openmm.app.simulation.Simulation <http://docs.openmm.org/latest/api-python/generated/openmm.app.simulation.Simulation.html#openmm.app.simulation.Simulation>`_
- `openmm.app.modeller.Modeller <http://docs.openmm.org/latest/api-python/generated/openmm.app.modeller.Modeller.html#openmm.app.modeller.Modeller>`_
- `openmm.app.pdbxfile.PDBxFile <http://docs.openmm.org/latest/api-python/generated/openmm.app.pdbxfile.PDBxFile.html#openmm.app.pdbxfile.PDBxFile>`_

The :class:`OpenMMTopologyParser` generates a topology from an OpenMM Topology object.


Classes
-------

.. autoclass:: OpenMMTopologyParser
   :members:
   :inherited-members:

.. autoclass:: OpenMMAppTopologyParser
   :members:
   :inherited-members:

    N   )TopologyReaderBase)SYMB2Z)Topology)Atomids	Atomnames	AtomtypesBondsChainIDsElementsMassesResidsResnumsResnamesSegidsc                   4    e Zd ZdZed             Zd Zd ZdS )OpenMMTopologyParserOPENMMTOPOLOGYc                     	 ddl m} t          | |j                  S # t          $ r 	 ddlm} Y dS # t          $ r Y Y dS w xY ww xY wz$Can this Parser read object *thing*?r   )appFN)openmmr   
isinstancer   ImportErrorsimtk.openmmthingr   s     l/srv/www/vhosts/g4struct/public_html/venv/lib/python3.11/site-packages/MDAnalysis/converters/OpenMMParser.py_format_hintz!OpenMMTopologyParser._format_hintU   s    	3"""""" eS\222  	 	 	,,,,,,,,,   uuu	s"    
A1
A AA  Ac           
      
   	 ddl m} n7# t          $ r* 	 ddlm} n# t          $ r d}t          |          w xY wY nw xY wd |                                D             }d |                                D             }d |                                D             }d |                                D             }d |                                D             }d	 |                                D             }	d
 |                                D             }
|
                                }d |                                D             }d |                                D             }d |                                D             }d |                                D             }t          t          j        |t          j                            t          t          j        |t                              t          |||d          t!          t          j        |t                              t#          |
          t%          |          t'          |	          t)          |          g}g }g }g }|                                D ]}|j        }||j                                        t0          v r|                    |j                   n|                    d           |                    |j                   |                    |j                            |                     |                    d           |                    d           |                    d           t9          |          st;          |          r}t=          j        d           t;          d |D                       rt=          j        d           |                    tA          t          j        |t                                         nRd}t=          j        |           n;|                    tA          t          j        |t                                         |                    tC          t          j        |t                                         |                    tE          t          j        |                               tG          |          }tG          |
          }tG          |          }tI          ||||||          }|S )a
  Construct mda topology from omm topology

        Can be used for any openmm object that contains a topology object

        Parameters
        ----------
        omm_topology: openmm.Topology

        Returns
        -------
        top : MDAnalysis.core.topology.Topology


        When partial elements are present, values from available elements
        are used whereas the absent elements are assigned an empty string
        with their atomtype set to 'X' and mass set to 0.0.

        For elements with invalid and unreal symbols, the symbol is used
        as it is for atomtypes but an empty string is used for elements.

        .. versionchanged:: 2.2.0
           The parser now works when element attribute is missing in some or
           all the atoms.

        r   )daltonsz{OpenMM is required for the OpenMMParser but it's not installed. Try installing it with 
conda install -c conda-forge openmmc                 &    g | ]}|j         j        S  )residueindex.0as     r   
<listcomp>zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    GGGQGGG    c                 &    g | ]}|j         j        S r#   )chainr%   r'   rs     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    KKKaAGMKKKr*   c                     g | ]	}|j         
S r#   )idr&   s     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    666A14666r*   c                     g | ]	}|j         
S r#   namer&   s     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    :::QV:::r*   c                 0    g | ]}|j         j        j        S r#   )r$   r,   r0   r&   s     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    EEE1AIO&EEEr*   c                     g | ]	}|j         
S r#   r2   r-   s     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    <<<qAF<<<r*   c                 "    g | ]}|j         d z   S )   )r%   r-   s     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    ???!!'A+???r*   c                 6    g | ]}t          |j                  S r#   )strr%   )r'   cs     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s     >>>1#ag,,>>>r*   c                 >    g | ]}|j         j        |j        j        fS r#   )atom1r%   atom2r'   bs     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s%    NNNA!'-/NNNr*   c                     g | ]	}|j         
S r#   )orderr>   s     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    ===1qw===r*   c                     g | ]	}|j         
S r#   )typer>   s     r   r)   zHOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<listcomp>   s    ;;;af;;;r*   )dtypeF)typesrA   guessedN g        XzZElement information missing for some atoms. These have been given an empty element record c              3   "   K   | ]
}|d k    V  dS )rH   Nr#   )r'   is     r   	<genexpr>zGOpenMMTopologyParser._mda_topology_from_omm_topology.<locals>.<genexpr>   s&      33AqCx333333r*   a	  For absent elements, atomtype has been  set to 'X' and mass has been set to 0.0. If needed these can be guessed using universe.guess_TopologyAttrs(to_guess=['masses', 'types']). (for MDAnalysis version 2.x this is done automatically, but it will be removed in 3.0).a  Element information is missing for all the atoms. Elements attribute will not be populated. Atomtype attribute will be guessed using atom name and mass will be guessed using atomtype.For MDAnalysis version 2.x this is done automatically, but it will be removed in MDAnalysis v3.0. These can be guessed using universe.guess_TopologyAttrs(to_guess=['masses', 'types']) See MDAnalysis.guessers.)attrsatom_resindexresidue_segindex)%openmm.unitr!   r   
simtk.unitatomsresiduescopychainsbondsr   nparrayint32r   objectr
   r   r   r   r   r   elementsymbol
capitalizer   appendmassvalue_in_unitallanywarningswarnr   r	   r   lenr   )selfomm_topologyr!   msgrM   rN   atomids	atomnameschainidsresnamesresidsresnumssegidsrU   bond_orders
bond_typesrL   validated_elementsmasses	atomtypesr(   elemwmsgn_atoms
n_residues
n_segmentstops                              r   _mda_topology_from_omm_topologyz4OpenMMTopologyParser._mda_topology_from_omm_topologyb   s   6	'+++++++ 		' 		' 		''....... ' ' ': 
 "#&&&' /.		' HG,2D2D2F2FGGGKK<3H3H3J3JKKK66!3!3!5!5666::\%7%7%9%9:::	EE0B0B0D0DEEE<<L$9$9$;$;<<<??|'<'<'>'>???++-->>(;(;(=(=>>>NN9K9K9M9MNNN==(:(:(<(<===;;l&8&8&:&:;;;
 BHWBH55566bhy77788%zeLLLRXhf555666NNGX6NN	
  	##%% 	& 	&A9D;))++v55&--dk::::&--b111  ---di55g>>????"))"---c"""  %%%%%&& &	O%&& #$E   3333333 
M;	 	 	 RX&8GGGHH   	/  d####LL"(+=V"L"L"LMMNNNYrx	@@@AABBBVBHV,,--...g,,[[
[[
'-
 
 
 
s   	 
==7==c                 >    | j         }|                     |          }|S N)filenamerz   re   kwargsrf   ry   s       r   parsezOpenMMTopologyParser.parse   s!    }22<@@
r*   N)__name__
__module____qualname__formatstaticmethodr   rz   r   r#   r*   r   r   r   R   sT        F
3 
3 \
3G G GR    r*   r   c                   .    e Zd ZdZed             Zd ZdS )OpenMMAppTopologyParser	OPENMMAPPc                     	 ddl m} t          | |j        |j        |j        |j        f          S # t          $ r 	 ddlm} Y dS # t          $ r Y Y dS w xY ww xY wr   )	r   r   r   PDBFileModeller
SimulationPDBxFiler   r   r   s     r   r   z$OpenMMAppTopologyParser._format_hint   s    	"""""" clCNCLI    	 	 	,,,,,,,,,   uuu	s&   0 
AA
AAAAc                     	 | j                                         }n# t          $ r | j         j        }Y nw xY w|                     |          }|S r|   )r}   getTopologyAttributeErrortopologyrz   r~   s       r   r   zOpenMMAppTopologyParser.parse  s]    	2=4466LL 	2 	2 	2=1LLL	222<@@
s    55N)r   r   r   r   r   r   r   r#   r*   r   r   r      sA        F  \    r*   r   )__doc__numpyrV   rb   topology.baser   guesser.tablesr   core.topologyr   core.topologyattrsr   r   r	   r
   r   r   r   r   r   r   r   r   r   r#   r*   r   <module>r      sH  0# #J      . . . . . . # # # # # # $ $ $ $ $ $                         ] ] ] ] ]- ] ] ]@    2     r*   