"""
Complete morphing example create from the tutorial code.
"""
from molpher.core import ExplorationTree as ETree
from molpher.core.operations import *
cocaine = 'CN1[C@H]2CC[C@@H]1[C@@H](C(=O)OC)[C@@H](OC(=O)c1ccccc1)C2'
procaine = 'O=C(OCCN(CC)CC)c1ccc(N)cc1'
[docs]class MyFilterMorphs(TreeOperation):
def __init__(self):
super(MyFilterMorphs, self).__init__()
def __call__(self):
mask = [False for x in self.tree.candidates_mask]
mask[0] = True
mask[1] = True
mask[2] = True
self.tree.candidates_mask = mask
[docs]def main():
iteration = [
GenerateMorphsOper()
, SortMorphsOper()
, MyFilterMorphs()
, ExtendTreeOper()
, PruneTreeOper()
]
tree = ETree.create(source=cocaine, target=procaine)
counter = 0
while not tree.path_found:
for oper in iteration:
tree.runOperation(oper)
counter+=1
print("Iteration", counter)
print(
sorted(
[
(x.getSMILES(), x.getDistToTarget())
for x in tree.leaves
], key=lambda x : x[1]
)
)
path = []
current = tree.fetchMol(tree.params['target'])
path.append(current)
while current != '':
current = current.getParentSMILES()
if current:
current = tree.fetchMol(current)
path.append(current)
path.reverse()
print("Path found: ")
for mol in path:
print(mol.getSMILES(), mol.getDistToTarget())
if __name__ == "__main__":
exit(main())