o Rŀg @srdZddlZddlmZddlmZddlmZgdZddZ d d Z d d Z e d kr7ddl mZedSdS)z"Selection of atoms, residues, etc.N)Atom)Entity) PDBException)ARCMScCs tt|S)zZReturn a list of the unique items in the given iterable. Order is NOT preserved. )listset)itemsr E/var/www/html/myenv/lib/python3.10/site-packages/Bio/PDB/Selection.py uniqueifys rcCsdd|D}t|S)zATranslate a list of entities to a list of their (unique) parents.cSsh|]}|qSr  get_parent.0entityr r r sz%get_unique_parents..)r ) entity_listunique_parentsr r rget_unique_parentssrcs|tvr t|d|gkrgSt|ttfr|g}|dtfdd|Ds0tdt|}t}||kr@|S|}||krXt||D]}t j |}qKt |St||D] }dd|D}q]t |S)aUnfold 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: Not an entity level.rc3s|] }|kVqdSN) get_levelrlevelr r <sz"unfold_entities..zEntity list is not homogeneous.cSsi|]}|dqSrrrr r r Msz#unfold_entities..) entity_levelsr isinstancerrrallindexrange itertoolschain from_iterabler )r target_level target_index level_indexentitiesir rrunfold_entities!s*   r,__main__) run_doctest)__doc__r$ Bio.PDB.AtomrBio.PDB.EntityrBio.PDB.PDBExceptionsrrrrr,__name__ Bio._utilsr.r r r rs   1