B ž3Rc-ã@sfddlZddlZddlmZdd„Zddd„Zedkrbeejƒd krXe d ƒe  d ¡e  ejd ¡Z iZ xDe D] list Returns a single, flat list which contains all elements retrieved from the sequence and all nested sub-sequences (iterables). Examples: >>> [1, 2, [3,4], (5,6)] [1, 2, [3, 4], (5, 6)] >>> flatten([[[1,2,3], (42,None)], [4,5], [6], 7, MyVector(8,9,10)]) [1, 2, 3, 42, None, 4, 5, 6, 7, 8, 9, 10]Ú__iter__)ÚhasattrÚ isinstanceÚstrÚextendÚflattenÚappend)ÚxÚresultZel©r ú'share/RDKit/Contrib/M_Kossner/Frames.pyrs  rÚScaffcsÖ| ¡}t| ¡ƒ‰‡fdd„| ¡Dƒ}g‰g}xN|D]F}x@| |¡ ¡D].}| ¡ˆkrLˆ |¡| | ¡|g¡PqLWq8W‡fdd„|Dƒ}dd„ˆDƒ}g} xæt|ƒdkrŒg} xÂ|D]º} | dkrÖt dƒqÀx¢| | d¡ ¡D]Œ}| ¡| krê| ¡|kr6| dd…} |  | ¡¡|  | dd…¡qê| ¡ˆkrê| dd…} |  | ¡¡|  | ¡|| dd…}qêWqÀW| dd…}q¨W|d krNt t |ƒƒ}g} |j d d t  |¡}d d„| Dƒ}g}x<|D]4}||krÖ| |d|d t jj¡| |¡qÖWx|D]}|  |¡qWx| D]}| |¡q.W| ¡}|S|dkrÒt t |ƒƒ}g} |j d d x,|D]$}|dk rz||krz|  |¡qzWt  |¡}x| D]}| |¡q²W| ¡}|SdS)a$return a ganeric molecule defining the reduced scaffold of the input mol. mode can be 'Scaff' or 'RedScaff': Scaff -> chop off the side chains and return the scaffold RedScaff -> remove all linking chains and connect the rings directly at the atoms where the linker was cs g|]}| ¡ˆkr| ¡‘qSr )ÚGetIdx)Ú.0Zatom)Ú RingAtomsr r ú 8szGetFrame..csg|]}|ˆkr|‘qSr r )rr )Ú RingNeighborsr r rBscSsg|]}|‘qSr r )rr r r r rFsrNZoohéÿÿÿÿZRedScaffT)ÚreversecSs g|]}t|d|dgƒ‘qS)rr)Úsorted)rÚir r r rhsér)Z GetRingInforZ AtomRingsZGetAtomsZGetAtomWithIdxZ GetNeighborsrr ÚlenÚprintÚlistÚsetÚsortÚChemZ EditableMolZAddBondZBondTypeZSINGLEZ RemoveAtomZGetMol)ÚmolÚmodeZringZ NonRingAtomsZPathsZ NonRingAtomZneighborZ PosConnectorsZ FrameworkZLinkersZNewPathsÚPÚnZtodelZemZ BondsToAddZmemrÚmr )rrr ÚGetFrame,sz                               r$Ú__main__éz4No input file provided: Frames.py filetosprocess.extrT)ZisomericSmilesz frames.smiÚwz.sdfZScaffoldZClusterú Ú znumber of Clusters: %d)r) ÚosÚsysrrrr$Ú__name__rÚargvrÚexitZ SDMolSupplierZsupplZ FrameDictrr#Z MolToSmilesZ cansmilesr ZcounterÚopenr'ÚitemsÚkeyÚitemZSDWriterrÚdrZSetPropÚwriteÚcloser r r r Ús8  n      "