B 3RcI @sddlZddlZddlmZddlmZddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZedkrdZdZdZiZiZiZiZeddZejdddddd d!ejd"d#d$d%d&d'd(ejd)d*d$d+d,d-d(e\ZZejdk rejZn2ejdk r4ejZed.kr0ed/ed.d0ZxTejD]HZ e !Z e "d1\Z#Z$Z%Z&e$ee#<e#ee$<dZ'ere$ekree#ee$<ee$Z'e(e%dkre(e&dkrq>e(e%dkrPe&"d2Z)ee)d.d.e're)dZ&e)d.Z%d3e$e%fZ*e+e&g,e*ee)dd.e're)d.Z&e)dZ%d3e$e%fZ*e+e&g,e*n6e%-d4Z.ee%e.e'r>d3e$e%fZ*e+e&g,e*q>WexeD]Z/e(ee/Z0e0d.krqxe1e0D]Z2xe1e2e0D]Z3e2e3kree/e2"d5\Z4Z5ee/e3"d5\Z6Z7e4e6kre5e7kre e5e7e/\Z8Z&ed6ee4ee6e4e6e8e&fej9re e7e5e/\Z8Z&ed6ee6ee4e6e4e8e&fqWqWqWdS)7N)Chem) OptionParsercCst|}|S)N)r MolFromSmilesZ GetNumAtoms)smimr$share/RDKit/Contrib/mmpa/indexing.pyheavy_atom_count's r cCsBd}t||}tr2t|t|}|tkr>d}n |tkr>d}|S)NFT)r use_ratiofloatratiomax_size)r attachments cmpd_heavyresultZ core_sizeZ core_ratiorrr add_to_index-s rcCsRg}t|}tj|dd}x0t||D]\}}|dkr,||q,W|S)NF)Z breakTiesr)rrZCanonicalRankAtomszipZGetAtomsZGetMassappend)rZsymmetryrZsymmetry_classesZatomsymmetry_classrrrget_symmetry_class=s  rcCsZi}|d}|dkr&t|}t|}|dkrD|d|dkr||d|dkr|t||}t|}t|||}t|||}n|d|dkr|d|dkrt|}t|}n|d|dkr|d|dkrt|}t||}t|}t|||}nJ|d|dkrF|d|dkrFt||}t|}t|||}t|}n|dkrF|d|dkr|d|dkr|d|dkr|d|dkr|d|dkr|d|dkrt|}t|}nx|d|dkrd|d|dkrd|d|dkrd|d|dkrd|d|dkrd|d|dkrdt|}t||}t|}t|||}n|d|dkr|d|dkr|d|dkr|d|dkr|d|dkr|d|dkrt||}t|}t|||}t|||}nH|d|dkr|d|dkr|d|dkr|d|dkr|d|dkr|d|dkrt||}t|}t|||}t|}n|d|dkrZ|d|dkrZ|d|dkrZt||}t|}t|||}t|||}|d|dkrt||dd}nB|d|dkr8t||dd}n |d|dkrFt||dd}n|d|dkr|d|dkr|d|dkrt|}t||}t|}t|||}nt||}t|}t|||}t|||}|d|dkrt||dd}nB|d|dkr&t||dd}n |d|dkrFt||dd}d||f}||fS)N*rz%s>>%s)countrbuild_track_dictionaryswitch_labels_on_position switch_labels"switch_specific_labels_on_symmetry)ZlhsZrhscontext isotope_trackstarsZlhs_symZrhs_symZsmirkrrrcansmirkMs          $    $6 66 66  66   6   6     r"cCs||d||dkrtd|}|rt||t||krtd||dd||d|}td||dd||d|}tdd|}|S) Nrz1\[\*\:([123])\].*\[\*\:([123])\].*\[\*\:([123])\]z\[\*\:z\]z[*:XXzXX]XX)researchintgroupsub)rrabmatchObjrrrrs **rcCsBtdd|d}tdd|d}tdd|d}tdd|}|S)Nz \[\*\:[123]\]z [*:XX1XX]rz [*:XX2XX]z [*:XX3XX]r#r$)r%r))rrrrr#s rcCs|dkr|ddkr.tdd|dd|}|ddkrTtdd|dd|}|dkr|d d krtd d|d d|}td d |}|S) Nr1z \[\*\:1\]z[*:XXzXX]2z \[\*\:2\]r3z \[\*\:3\]r#r$)r%r))Ztrackr!rrrrr.s   rcCsi}|dkr:td|}|r|d||d<d||d<nB|dkr|td|}|r|d||d<d||d<d||d<|S) Nrz \[\*\:([123])\].*\[\*\:([123])\]r-rr.rz1\[\*\:([123])\].*\[\*\:([123])\].*\[\*\:([123])\]r/)r%r&r()rr!r r,rrrrEs  rcCsxtD]}|d}|dkr|}tdd|}t|}|dkrVtjd||fqtj |dd}|t krd}t |}d ||f}t| |qWdS) Nrrz \[\*\:1\]z[H]z Error with key: %s, Added H: %s T)ZisomericSmilesz[*:1][H]z%s;t%s) indexrr%r)rrsysstderrwriteZ MolToSmiles smi_to_idr)keyrrZtempZc_smicoreidvaluerrrindex_hydrogen_change[s    r9__main__ g333333?FzProgram to generate MMPs)Z descriptionz-sz --symmetric store_truesymzkOutput symmetrically equivalent MMPs, i.e output both cmpd1,cmpd2, SMIRKS:A>>B and cmpd2,cmpd1, SMIRKS:B>>A)defaultactiondesthelpz-mz --maxsizeZstoremaxsizer'zMaximum size of change (in heavy atoms) allowed in matched molecular pairs identified. DEFAULT=10. Note: This option overrides the ratio option if both are specified.)r?r@typerAz-rz--ratior r aMaximum ratio of change allowed in matched molecular pairs identified. The ratio is: size of change / size of cmpd (in terms of heavy atoms). DEFAULT=0.3. Note: If this option is used with the maxsize option, the maxsize option will be used.rz3Ratio specified: %s. Ratio needs to be less than 1.T,.z%s;t%srz;tz%s,%s,%s,%s,%s,%s):r1r%ZrdkitrZoptparserr rrr"rrrrr9__name__r r r r0r4Z id_to_smiZ id_to_heavyparserZ add_option parse_argsZoptionsargsrBprintexitstdinlinerstripsplitrr7r6rrlenZ side_chainsr8 setdefaultrrrr5ZtotalrangeZxaZxbZid_aZcore_aZid_bZcore_bZsmirksr=rrrr!s  B *