:>"^ã@s¦dZddlZddlmZddlmZddlmZmZiZ de d)Ú get_resnameÚget_id)rrÚhetflagÚresseqÚicodeÚfull_idrrrÚ__repr__#s zResidue.__repr__cCsE|jƒ}|j|ƒr1td||fƒ‚tj||ƒdS)z}Add an Atom object. Checks for adding duplicate atoms, and raises a PDBConstructionException if so. z#Atom %s defined twice in residue %sN)rZhas_idrrÚadd)rÚatomZatom_idrrrr *s  z Residue.addcCs!tjdtƒ|jjƒdS)zÓSort child atoms. Atoms N, CA, C, O always come first, thereafter alphabetically by name, with any alternative location specifier for disordered atoms (altloc) as a tie-breaker. zˆThe custom sort() method will be removed in the future in favour of rich comparison methods. Use the built-in sorted() function instead.N)ÚwarningsÚwarnrZ child_listÚsort)rrrrr$7sz Residue.sortcCs d|_dS)zSet the disordered flag.rN)r)rrrrÚflag_disorderedEszResidue.flag_disorderedcCs|jS)z2Return 1 if the residue contains disordered atoms.)r)rrrrÚ is_disorderedIszResidue.is_disorderedcCs|jS)zReturn the residue name.)r)rrrrrMszResidue.get_resnamecCsS|jƒ}g}x:|D]2}|jƒr>||jƒ7}q|j|ƒqW|S)z5Return the list of all atoms, unpack DisorderedAtoms.)Zget_listr&Údisordered_get_listÚappend)rZ atom_listZundisordered_atom_listr!rrrÚget_unpacked_listQs   zResidue.get_unpacked_listcCs|jS)zReturn the segment identifier.)r)rrrrÚ get_segid\szResidue.get_segidccsx|D] }|VqWdS)z Return atoms.Nr)rÚarrrÚ get_atoms`s zResidue.get_atomsccs*tjdtƒx|D] }|VqWdS)z Return atom.z{`get_atom` has been deprecated and we intend to remove it in a future release of Biopython. Please use `get_atoms` instead.N)r"r#r)rr+rrrÚget_atomes  zResidue.get_atomN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr r$r%r&rr)r*r,r-rrrrrs         rc@sReZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚDisorderedResiduezÁDisorderedResidue is a wrapper around two or more Residue objects. It is used to represent point mutations (e.g. there is a Ser 60 and a Cys 60 residue, each with 50 % occupancy). cCstj||ƒdS)zInitialize the class.N)rr)rrrrrrwszDisorderedResidue.__init__cCs;|jƒ}|jƒ\}}}||||f}d|S)z*Return disordered residue full identifier.z0)rr)rrrrrrrrrr{s zDisorderedResidue.__repr__cCsy|jƒ}|jƒdksh|jƒ}|jƒ\}}}|j|ƒtd||||fƒ‚|j|ƒdS)zAdd atom to residue.rz2Blank altlocs in duplicate residue %s (%s, %i, %s)N)Zdisordered_getr&rrr r)rr!ÚresiduerZhetrrrrrr ‚s   zDisorderedResidue.addcCs%x|jƒD]}|jƒq WdS)z,Sort the atoms in the child Residue objects.N)r'r$)rr3rrrr$’szDisorderedResidue.sortcCsV|jƒ}|jƒ}|j|ƒ|j|ƒ s;t‚|||<|j|ƒdS)zrAdd a residue object and use its resname as key. Arguments: - residue - Residue object N)rZ get_parentZ set_parentZdisordered_has_idÚAssertionErrorZdisordered_select)rr3rÚchainrrrÚdisordered_add—s     z DisorderedResidue.disordered_addN) r.r/r0r1rrr r$r6rrrrr2ps     r2) r1r"ZBiorZBio.PDB.PDBExceptionsrZBio.PDB.EntityrrZ_atom_name_dictrr2rrrrÚs     X