:>"^ ã@s²dZddlmZddlZddlmZddlmZddlm Z ddd d d gZ d d „Z dd„Z dd„Z edkr®ddlmZeƒdS)z"Selection of atoms, residues, etc.é)Úprint_functionN)ÚAtom)ÚEntity)Ú PDBExceptionÚAÚRÚCÚMÚScCstt|ƒƒS)zZReturn a list of the unique items in the given iterable. Order is NOT preserved. )ÚlistÚset)Úitems©rú6/tmp/pip-build-ww9dw3qa/biopython/Bio/PDB/Selection.pyÚ uniqueifysrcCsdd„|Dƒ}t|ƒS)zATranslate a list of entities to a list of their (unique) parents.cSsh|]}|jƒ’qSr)Ú get_parent)Ú.0Úentityrrrú s z%get_unique_parents..)r )Ú entity_listZunique_parentsrrrÚget_unique_parentssrcs%|tkrtd|ƒ‚|gkr,gSt|ttfƒrJ|g}|djƒ‰t‡fdd†|Dƒƒs…tdƒ‚tj|ƒ}tjˆƒ}||kr³|S||krîxYt||ƒD]}t j 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!|]}|jƒˆkVqdS)N)Ú get_level)rr)Úlevelrrú ?sz"unfold_entities..zEntity list is not homogeneous.cSsh|]}|jƒ’qSr)r)rrrrrrNs z"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__Ú __future__rrZ Bio.PDB.AtomrZBio.PDB.EntityrZBio.PDB.PDBExceptionsrrrrr#Ú__name__Z Bio._utilsr%rrrrÚs    .