#!/usr/local2/bin/python2.3
import os, string
from openeye.oechem import *

rxn = 7
ifs = oemolistream()
ifs.open('bb'+str(rxn)+'_ln.ism')
ofs_1 = oemolostream()
ofs_1.open('bb'+str(rxn)+'_lp.ism')

# could produce some doubles
# Leaving group X protonated
libgen2 = OELibraryGen("[R:4][#6R:1]([N:2])([R&!#6:3])[O-:100].[H:98]>>[*:4][*:1]([N:2])([O-:100])[AR+:3][H:98]")
# Leaving  amine N protonated
libgen3 = OELibraryGen("[R:4][#6R:1]([N:2])([R&!#6:3])[O-:100].[H:98]>>[*:4][C:1]([N+:2][H:98])([O-:100])[AR:3]")
# Leaving group A:4 protonated (if 3 heteroatoms at the attacked center!)
libgen4 = OELibraryGen("[#7,#8,#15,#16:4][#6R:1]([N:2])([R&!#6:3])[O-:100].[H:98]>>[A+:4]([H:98])[C:1]([N:2])([O-:100])[AR:3]")

libgen2.SetExplicitHydrogens(True)
libgen2.SetValenceCorrection(False)
libgen3.SetExplicitHydrogens(True)
libgen3.SetValenceCorrection(False)
libgen4.SetExplicitHydrogens(True)
libgen4.SetValenceCorrection(False)

mol_prot = OEGraphMol()
OEParseSmiles(mol_prot, "[H]")
for mol in ifs.GetOEMols():
    libgen2.AddStartingMaterial(mol, 0)
    libgen2.AddStartingMaterial(mol_prot, 1)
    libgen3.AddStartingMaterial(mol, 0)
    libgen3.AddStartingMaterial(mol_prot, 1)
    libgen4.AddStartingMaterial(mol, 0)
    libgen4.AddStartingMaterial(mol_prot, 1)
    mol.Clear()
    mol_prot.Clear()

title_list = []
rxn_n = 1
while rxn_n < rxn:
    try:
        old_title_file = open('bb'+str(rxn_n)+'_ln.ism','r')
        for title in old_title_file.readlines():
            title_list.append(string.split(title)[1])
        old_title_file.close()
    except IOError:
        print 'bb'+str(rxn_n)+'_ln.ism not existent\n'
    try:
        old_title_file = open('bb'+str(rxn_n)+'_lp.ism','r')
        for title in old_title_file.readlines():
            title_list.append(string.split(title)[1])
        old_title_file.close()
    except IOError:
        print 'bb'+str(rxn_n)+'_lp.ism not existent\n'
    rxn_n+=1


product_smiles = []

for products in libgen2.GetProducts():
    if OECreateIsoSmiString(products) in product_smiles:
        continue
    else:
        product_smiles.append(OECreateIsoSmiString(products))
    prot = 0
    title = products.GetTitle()[:string.find(products.GetTitle(), 'p0')]+'p'+str(prot)
    while title in title_list:
        prot+=1
        title = products.GetTitle()[:string.find(products.GetTitle(), 'p')]+'p'+str(prot)
    title_list.append(title)
    products.SetTitle(title)
    OEWriteMolecule(ofs_1, products)

for products in libgen3.GetProducts():
    if OECreateIsoSmiString(products) in product_smiles:
        continue
    else:
        product_smiles.append(OECreateIsoSmiString(products))
    prot = 0
    title = products.GetTitle()[:string.find(products.GetTitle(), 'p0')]+'p'+str(prot)
    while title in title_list:
        prot+=1
        title = products.GetTitle()[:string.find(products.GetTitle(), 'p')]+'p'+str(prot)
    title_list.append(title)
    products.SetTitle(title)
    OEWriteMolecule(ofs_1, products)

for products in libgen4.GetProducts():
    if OECreateIsoSmiString(products) in product_smiles:
        continue
    else:
        product_smiles.append(OECreateIsoSmiString(products))
    prot = 0
    title = products.GetTitle()[:string.find(products.GetTitle(), 'p0')]+'p'+str(prot)
    while title in title_list:
        prot+=1
        title = products.GetTitle()[:string.find(products.GetTitle(), 'p')]+'p'+str(prot)
    title_list.append(title)
    products.SetTitle(title)
    OEWriteMolecule(ofs_1, products)

ofs_1.close()
ifs.close()