# $Id$
#
# Copyright (C) 2004-2006 Rational Discovery LLC
#
#   @@ All Rights Reserved @@
#  This file is part of the RDKit.
#  The contents are covered by the terms of the BSD license
#  which is included in the file license.txt, found at the root
#  of the RDKit source tree.
#
from rdkit import RDConfig

# change this to use another viewer:
if RDConfig.molViewer in ('WEBLAB','DSVIEWER'):
  from rdkit.Chem.DSViewer import *
elif RDConfig.molViewer=='PYMOL':
  from rdkit.Chem.PyMol import *
else:
  raise ValueError('invalid RD_MOLVIEWER specified')

  
  
if __name__=='__main__':
  import AllChem
  import sys
  if len(sys.argv)<2:
    smi ='c1cccc2c1cccc2CC(=O)N'
  else:
    smi = sys.argv[1]

  m = Chem.MolFromSmiles(smi)
  m = Chem.AddHs(m)
  AllChem.EmbedMolecule(m)
  v = MolViewer()
  v.ShowMol(m,'raw')
  AllChem.UFFOptimizeMolecule(m)
  v.ShowMol(m,'opt',showOnly=0,highlightFeatures=[(0,),(2,),(3,4)])