B
ž3RcE ã
@ sü d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
dd„ Z e d ƒZ e
d
¡ e d ¡ e d ¡ ed
dƒZg Zx€eD ]xZe ¡ d¡Zeed ƒ ¡ Ze ej¡ ee ƒZe e¡ e de e e e ¡ ƒƒ¡Z!e "e!ed ed ed f¡ q–W xäe j#D ]ÚZe de¡Z$e$d Z%e$d Z&ee%ƒ ¡ Ze eƒZ'e' ej¡ ee ƒZ(e( e¡ e de e e e( ¡ ƒƒ¡Z)xjeD ]bZ*e*d Z+e*d Z,e*d Z-e
.d¡Z/e/ 0e*d ¡ e/ 1e)¡Z2e2dkrŒe3de-e+e,e%e&e2f ƒ qŒW qW dS )é N)Ú
MolHandler)ÚMolecule)ÚChemicalFingerprint)Ú CFParameters)ÚSimilarityCalculatorFactoryc C sR | }| d¡}d}| d¡}x0|D ](}t|ƒ ¡ } | ¡ }||kr"|}| }q"W |S )NZsmilesr Ú.)ZtoFormatÚsplitr Ú getMoleculeZ getAtomCount)ÚmolZparmolZsmiZparcountZmsmiÚcount© r ú*share/RDKit/Contrib/fraggle/cxn_tversky.pyÚdesalt/ s
r a1 i é é Zfrag_q_split_outÚrú,é Úié z\s|,é zTversky;0.95;0.05gÍÌÌÌÌÌì?z%s,%s,%s,%s,%s,%s)4ÚsysÚreZarrayZ
chemaxon.utilr Zchemaxon.strucr Zchemaxon.descriptorsr r r r ZcfpZ setLengthZ setBondCountZ setBitCountÚopenZ
q_split_inputZqueriesÚlineÚrstripr Úinfor r
Z aromatizeZ AROM_GENERALZqfpZgenerateÚlistÚmapÚintZ toFloatArrayZqintfpÚappendÚstdinZ line_fieldsZdbsmiZdbidZ mol_desaltedÚfpZintfpÚqZqsmiZqidZqsubZcreateZscZsetQueryFingerprintZ
getSimilarityZtverskyÚprintr r r r
Ú$ sT
"