B ‰°bà ã@stdZddlZddlmZddlmZddlmZdddd d gZd d „Z d d„Z dd„Z e dkrpddl mZeƒdS)z"Selection of atoms, residues, etc.éN)ÚAtom)ÚEntity)Ú PDBExceptionÚAÚRÚCÚMÚScCs tt|ƒƒS)zZReturn a list of the unique items in the given iterable. Order is NOT preserved. )ÚlistÚset)Úitems©r ú0lib/python3.7/site-packages/Bio/PDB/Selection.pyÚ uniqueifysrcCsdd„|Dƒ}t|ƒS)zATranslate a list of entities to a list of their (unique) parents.cSsh|] }| ¡’qSr )Ú get_parent)Ú.0Úentityr r rú sz%get_unique_parents..)r )Ú entity_listZunique_parentsr r rÚget_unique_parentssrcsÒ|tkrtd|ƒ‚|gkr gSt|ttfƒr4|g}|d ¡‰t‡fdd„|Dƒƒs^tdƒ‚t |¡}t ˆ¡}||kr~|S||kr¨xBt||ƒD]}t j   |¡}q’Wn"x t||ƒD]}dd„|Dƒ}q´Wt |ƒS)aâUnfold entities list to a child level (e.g. residues in chain). Unfold a list of entities to a list of entities of another level. E.g.: list of atoms -> list of residues list of modules -> list of atoms list of residues -> list of chains - entity_list - list of entities or a single entity - target_level - char (A, R, C, M, S) Note that if entity_list is an empty list, you get an empty list back: >>> unfold_entities([], "A") [] z%s: Not an entity level.rc3s|]}| ¡ˆkVqdS)N)Ú get_level)rr)Úlevelr rú >sz"unfold_entities..zEntity list is not homogeneous.cSsh|] }| ¡’qSr )r)rrr r rrMsz"unfold_entities..) Ú entity_levelsrÚ isinstancerrrÚallÚindexÚrangeÚ itertoolsÚchainÚ from_iterabler )rZ target_levelZ target_indexZ level_indexÚir )rrÚunfold_entities#s&    r"Ú__main__)Ú run_doctest)Ú__doc__rZ Bio.PDB.AtomrZBio.PDB.EntityrZBio.PDB.PDBExceptionsrrrrr"Ú__name__Z Bio._utilsr$r r r rÚs   .