//
//  Copyright (C) 2002-2018 Greg Landrum and 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.
//
#include "SmartsWrite.h"
#include <sstream>
#include <cstdint>
#include <boost/algorithm/string.hpp>
#include "SmilesWrite.h"
#include <GraphMol/RDKitBase.h>
#include <GraphMol/RDKitQueries.h>
#include <GraphMol/Canon.h>
#include <GraphMol/new_canon.h>
#include <RDGeneral/RDLog.h>

namespace RDKit {
using namespace Canon;

// local utility namespace
namespace {

std::string _recurseGetSmarts(const QueryAtom *qatom,
                              const QueryAtom::QUERYATOM_QUERY *node,
                              bool negate);
std::string _recurseBondSmarts(const Bond *bond,
                               const QueryBond::QUERYBOND_QUERY *node,
                               bool negate, int atomToLeftIdx);

bool _checkForOrAndLowAnd(std::string smarts) {
  size_t orLoc, andLoc;
  // if we're a pure recursive smarts, we don't need to worry about this
  if (smarts[0] == '$' && smarts[smarts.size() - 1] == ')') return false;
  orLoc = smarts.find(",");
  andLoc = smarts.find(";");
  if ((orLoc != std::string::npos) && (andLoc != std::string::npos)) {
    return true;
  } else {
    return false;
  }
}

std::string _combineChildSmarts(std::string cs1, std::string cs2,
                                std::string descrip) {
  std::string res = "";
  if ((descrip.find("Or") > 0) && (descrip.find("Or") < descrip.length())) {
    // if either of child smarts already have a "," and ";" we can't have one
    // more OR here
    if (_checkForOrAndLowAnd(cs1) || _checkForOrAndLowAnd(cs2)) {
      throw "This is a non-smartable query - OR above and below AND in the binary tree";
    }
    res += cs1;
    res += ",";
    res += cs2;
  } else if ((descrip.find("And") > 0) &&
             (descrip.find("And") < descrip.length())) {
    size_t orLoc1, orLoc2;
    std::string symb;
    orLoc1 = cs1.find(',');
    orLoc2 = cs2.find(',');
    if ((orLoc1 != std::string::npos) || (orLoc2 != std::string::npos)) {
      symb = ";";
    } else {
      symb = "&";
    }
    res += cs1;
    res += symb;
    res += cs2;
  } else {
    std::stringstream err;
    err << "Don't know how to combine using " << descrip;
    throw err.str();
  }
  return res;
}

template <typename T>
void describeQuery(const T *query, std::string leader = "\t") {
  // BOOST_LOG(rdInfoLog) << leader << query->getDescription() << std::endl;
  typename T::CHILD_VECT_CI iter;
  for (iter = query->beginChildren(); iter != query->endChildren(); ++iter) {
    describeQuery(iter->get(), leader + "\t");
  }
}

// called with the children of AND queries:
std::string smartsOrganicAtom(const QueryAtom::QUERYATOM_QUERY *child1,
                              const QueryAtom::QUERYATOM_QUERY *child2) {
  PRECONDITION(child1 && child2, "bad query");

  std::string desc1 = child1->getDescription();
  std::string desc2 = child2->getDescription();
  const QueryAtom::QUERYATOM_QUERY *origA, *otherA;
  if (desc1 == "AtomAtomicNum") {
    origA = child1;
    otherA = child2;
  } else {
    origA = child2;
    otherA = child1;
  }

  std::string odsc = otherA->getDescription();
  CHECK_INVARIANT(((odsc == "AtomIsAliphatic") || (odsc == "AtomIsAromatic")),
                  "Should be either an aromatic or Aliphatic atom");

  const ATOM_EQUALS_QUERY *torig =
      static_cast<const ATOM_EQUALS_QUERY *>(origA);
  int val = torig->getVal();
  std::string res = PeriodicTable::getTable()->getElementSymbol(val);
  if (odsc == "AtomIsAromatic") {
    // if aromatic convert the first leter to a small letter
    res[0] += ('a' - 'A');
    if (res.length() > 1) {
      res = "[" + res + "]";
    }
  }
  if (torig->getNegation()) {
    res = "!" + res;
  }
  return res;
}
const static std::string _qatomHasStereoSet = "_qatomHasStereoSet";
std::string getAtomSmartsSimple(const QueryAtom *qatom,
                                const ATOM_EQUALS_QUERY *query,
                                bool &needParen) {
  PRECONDITION(query, "bad query");

  std::string descrip = query->getDescription();
  bool hasVal = false;
  enum class Modifiers : std::uint8_t { NONE, RANGE, LESS, GREATER };
  Modifiers mods = Modifiers::NONE;
  if (boost::starts_with(descrip, "range_")) {
    mods = Modifiers::RANGE;
    descrip = descrip.substr(6);
  } else if (boost::starts_with(descrip, "less_")) {
    mods = Modifiers::LESS;
    descrip = descrip.substr(5);
  } else if (boost::starts_with(descrip, "greater_")) {
    mods = Modifiers::GREATER;
    descrip = descrip.substr(8);
  }
  std::stringstream res;
  if (descrip == "AtomImplicitHCount") {
    res << "h";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomHasImplicitH") {
    res << "h";
    needParen = true;
  } else if (descrip == "AtomTotalValence") {
    res << "v";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomAtomicNum") {
    res << "#";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomExplicitDegree") {
    res << "D";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomTotalDegree") {
    res << "X";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomHasRingBond") {
    res << "x";
    needParen = true;
  } else if (descrip == "AtomHCount") {
    res << "H";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomIsAliphatic") {
    res << "A";
    needParen = false;
  } else if (descrip == "AtomIsAromatic") {
    res << "a";
    needParen = false;
  } else if (descrip == "AtomNull") {
    res << "*";
    needParen = false;
  } else if (descrip == "AtomInRing") {
    res << "R";
    needParen = true;
  } else if (descrip == "AtomMinRingSize") {
    res << "r";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomInNRings") {
    res << "R";
    if (mods == Modifiers::NONE && query->getVal() >= 0) {
      hasVal = true;
    }
    needParen = true;
  } else if (descrip == "AtomHasHeteroatomNeighbors") {
    res << "z";
    needParen = true;
  } else if (descrip == "AtomNumHeteroatomNeighbors") {
    res << "z";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomHasAliphaticHeteroatomNeighbors") {
    res << "Z";
    needParen = true;
  } else if (descrip == "AtomNumAliphaticHeteroatomNeighbors") {
    res << "Z";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomFormalCharge") {
    int val = query->getVal();
    if (val < 0) {
      res << "-";
    } else {
      res << "+";
    }
    if (abs(val) != 1) {
      res << abs(val);
    }
    needParen = true;
  } else if (descrip == "AtomHybridization") {
    res << "^";
    switch (query->getVal()) {
      case Atom::S:
        res << "0";
        break;
      case Atom::SP:
        res << "1";
        break;
      case Atom::SP2:
        res << "2";
        break;
      case Atom::SP3:
        res << "3";
        break;
      case Atom::SP3D:
        res << "4";
        break;
      case Atom::SP3D2:
        res << "5";
        break;
    }
  } else if (descrip == "AtomMass") {
    res << query->getVal() / massIntegerConversionFactor << "*";
    needParen = true;
  } else if (descrip == "AtomIsotope") {
    res << query->getVal() << "*";
    needParen = true;
  } else if (descrip == "AtomHasRingBond") {
    res << "x";
    needParen = true;
  } else if (descrip == "AtomRingBondCount") {
    res << "x";
    hasVal = true;
    needParen = true;
  } else if (descrip == "AtomUnsaturated") {
    res << "$(*=,:,#*)";
    needParen = true;
  } else if (descrip == "AtomType") {
    int atNum;
    bool isAromatic;
    parseAtomType(query->getVal(), atNum, isAromatic);
    std::string symbol = PeriodicTable::getTable()->getElementSymbol(atNum);
    if (isAromatic) symbol[0] += ('a' - 'A');
    res << symbol;
  } else {
    BOOST_LOG(rdWarningLog)
        << "Cannot write SMARTS for query type : " << descrip
        << ". Ignoring it." << std::endl;
    res << "*";
  }

  if (mods != Modifiers::NONE) {
    res << "{";
    switch (mods) {
      case Modifiers::LESS:
        res << ((const ATOM_LESSEQUAL_QUERY *)query)->getVal() << "-";
        break;
      case Modifiers::RANGE:
        res << ((const ATOM_RANGE_QUERY *)query)->getLower() << "-"
            << ((const ATOM_RANGE_QUERY *)query)->getUpper();
        break;
      case Modifiers::GREATER:
        res << "-" << ((const ATOM_GREATEREQUAL_QUERY *)query)->getVal();
        break;
      default:
        break;
    }
    res << "}";
  } else if (hasVal) {
    res << query->getVal();
  }

  // handle atomic stereochemistry
  if (qatom->getOwningMol().hasProp(common_properties::_doIsoSmiles)) {
    if (qatom->getChiralTag() != Atom::CHI_UNSPECIFIED &&
        !qatom->hasProp(_qatomHasStereoSet) &&
        !qatom->hasProp(common_properties::_brokenChirality)) {
      qatom->setProp(_qatomHasStereoSet, 1);
      switch (qatom->getChiralTag()) {
        case Atom::CHI_TETRAHEDRAL_CW:
          res << "@@";
          needParen = true;
          break;
        case Atom::CHI_TETRAHEDRAL_CCW:
          res << "@";
          needParen = true;
          break;
        default:
          break;
      }
    }
  }

  return res.str();
}

std::string getRecursiveStructureQuerySmarts(
    const QueryAtom::QUERYATOM_QUERY *query) {
  PRECONDITION(query, "bad query");
  PRECONDITION(query->getDescription() == "RecursiveStructure", "bad query");
  const RecursiveStructureQuery *rquery =
      static_cast<const RecursiveStructureQuery *>(query);
  ROMol *qmol = const_cast<ROMol *>(rquery->getQueryMol());
  std::string res = MolToSmarts(*qmol);
  res = "$(" + res + ")";
  if (rquery->getNegation()) {
    res = "!" + res;
  }
  return res;
}

std::string getBasicBondRepr(Bond::BondType typ, Bond::BondDir dir,
                             bool doIsomericSmiles, bool reverseDative) {
  std::string res;
  switch (typ) {
    case Bond::SINGLE:
      res = "-";
      if (doIsomericSmiles) {
        if (dir == Bond::ENDDOWNRIGHT) {
          res = "\\";
        } else if (dir == Bond::ENDUPRIGHT) {
          res = "/";
        }
      }
      break;
    case Bond::DOUBLE:
      res = "=";
      break;
    case Bond::TRIPLE:
      res = "#";
      break;
    case Bond::AROMATIC:
      res = ":";
      break;
    case Bond::DATIVE:
      if (reverseDative)
        res = "<-";
      else
        res = "->";
      break;
    default:
      res = "";
  }
  return res;
}  // namespace

std::string getBondSmartsSimple(const Bond *bond,
                                const BOND_EQUALS_QUERY *bquery,
                                int atomToLeftIdx) {
  PRECONDITION(bond, "bad bond");

  PRECONDITION(bquery, "bad query");
  std::string descrip = bquery->getDescription();
  std::string res = "";
  if (descrip == "BondNull") {
    res += "~";
  } else if (descrip == "BondInRing") {
    res += "@";
  } else if (descrip == "SingleOrAromaticBond") {
    // don't need to do anything here... :-)
  } else if (descrip == "BondDir") {
    int val = bquery->getVal();
    if (val == static_cast<int>(Bond::ENDDOWNRIGHT)) {
      res += "\\";
    } else if (val == static_cast<int>(Bond::ENDUPRIGHT)) {
      res += "/";
    } else {
      throw "Can't write smarts for this bond dir type";
    }
  } else if (descrip == "BondOrder") {
    bool reverseDative =
        (atomToLeftIdx >= 0 &&
         bond->getBeginAtomIdx() == static_cast<unsigned int>(atomToLeftIdx));
    res += getBasicBondRepr(
        static_cast<Bond::BondType>(bquery->getVal()), bond->getBondDir(),
        bond->getOwningMol().hasProp(common_properties::_doIsoSmiles),
        reverseDative);
  } else {
    std::stringstream msg;
    msg << "Canot write smarts for this query bond type : " << descrip;
    throw msg.str().c_str();
  }
  return res;
}

std::string _recurseGetSmarts(const QueryAtom *qatom,
                              const QueryAtom::QUERYATOM_QUERY *node,
                              bool negate) {
  PRECONDITION(node, "bad node");
  // the algorithm goes something like this
  // - recursively get the smarts for the child queries
  // - combine the child smarts using the following rules:
  //      - if we are currently at an OR query, combine the subqueries with a
  //      ",",
  //        but only if neither of child smarts do not contain "," and ";"
  //        This situation leads to a no smartable situation and throw an
  //        error
  //      - if we are currently at an and query, combine the child smarts with
  //      "&"
  //        if neither of the child smarts contain a "," - otherwise combine
  //        them
  //        the child smarts with a ";"
  //
  // There is an additional complication with composite nodes that carry a
  // negation - in this
  // case we will propogate the neagtion to the child nodes using the
  // following rules
  //   NOT (a AND b) = ( NOT (a)) AND ( NOT (b))
  //   NOT (a OR b) = ( NOT (a)) OR ( NOT (b))

  std::string descrip = node->getDescription();
  std::string res = "";

  const QueryAtom::QUERYATOM_QUERY *child1;
  const QueryAtom::QUERYATOM_QUERY *child2;
  QueryAtom::QUERYATOM_QUERY::CHILD_VECT_CI chi;
  chi = node->beginChildren();
  child1 = chi->get();
  chi++;
  child2 = chi->get();
  chi++;
  // OK we should be at the end of vector by now - since we can have only two
  // children,
  // well - at least in this case
  CHECK_INVARIANT(chi == node->endChildren(), "Too many children on the query");

  std::string dsc1, dsc2;
  dsc1 = child1->getDescription();
  dsc2 = child2->getDescription();
  std::string csmarts1, csmarts2;

  bool needParen;

  // deal with any special AND cases
  //  1. This "node" is an AtomAnd between a AliphaticAtom (or AromaticAtom)
  //  and
  //      an organic atom e.g. "C"
  if (descrip == "AtomAnd") {
    bool specialCase = false;
    // case 1
    if ((!child1->getNegation() && !child2->getNegation()) &&
        (((dsc1 == "AtomAtomicNum") &&
          ((dsc2 == "AtomIsAliphatic") || (dsc2 == "AtomIsAromatic"))) ||
         ((dsc2 == "AtomAtomicNum") &&
          ((dsc1 == "AtomIsAliphatic") || (dsc1 == "AtomIsAromatic"))))) {
      // we trap this one because it's nicer to see
      //   "CC" in the output than "[#6&A][#6&A]"
      res = smartsOrganicAtom(child1, child2);
      specialCase = true;
    }
    if (specialCase) {
      if (negate) {
        res = "!" + res;
      }
      return res;
    }
  }

  // deal with the first child
  if (dsc1 == "RecursiveStructure") {
    csmarts1 = getRecursiveStructureQuerySmarts(child1);
  } else if ((dsc1 != "AtomOr") && (dsc1 != "AtomAnd")) {
    // child 1 is a simple node
    const ATOM_EQUALS_QUERY *tchild =
        static_cast<const ATOM_EQUALS_QUERY *>(child1);
    csmarts1 = getAtomSmartsSimple(qatom, tchild, needParen);
    bool nneg = (negate) ^ (tchild->getNegation());
    if (nneg) {
      csmarts1 = "!" + csmarts1;
    }
  } else {
    // child 1 is composite node - recurse
    bool nneg = (negate) ^ (child1->getNegation());
    csmarts1 = _recurseGetSmarts(qatom, child1, nneg);
  }

  // deal with the second child
  if (dsc2 == "RecursiveStructure") {
    csmarts2 = getRecursiveStructureQuerySmarts(child2);
  } else if ((dsc2 != "AtomOr") && (dsc2 != "AtomAnd")) {
    // child 2 is a simple node
    const ATOM_EQUALS_QUERY *tchild =
        static_cast<const ATOM_EQUALS_QUERY *>(child2);
    csmarts2 = getAtomSmartsSimple(qatom, tchild, needParen);
    bool nneg = (negate) ^ (tchild->getNegation());
    if (nneg) {
      csmarts2 = "!" + csmarts2;
    }
  } else {
    bool nneg = (negate) ^ (child2->getNegation());
    csmarts2 = _recurseGetSmarts(qatom, child2, nneg);
  }

  // ok if we have a negation and we have an OR , we have to change to
  // an AND since we propogated the negation
  // i.e NOT (A OR B) = (NOT (A)) AND (NOT(B))
  if (negate) {
    if (descrip == "AtomOr") {
      descrip = "AtomAnd";
    } else if (descrip == "AtomAnd") {
      descrip = "AtomOr";
    }
  }

  res += _combineChildSmarts(csmarts1, csmarts2, descrip);
  return res;
}

std::string _recurseBondSmarts(const Bond *bond,
                               const QueryBond::QUERYBOND_QUERY *node,
                               bool negate, int atomToLeftIdx) {
  // the algorithm goes something like this
  // - recursively get the smarts for the child query bonds
  // - combine the child smarts using the following rules:
  //      - if we are currently at an OR query, combine the subqueries with a
  //      ",",
  //        but only if neither of child smarts do not contain "," and ";"
  //        This situation leads to a no smartable situation and throw an
  //        error
  //      - if we are currently at an and query, combine the child smarts with
  //      "&"
  //        if neither of the child smarts contain a "," - otherwise combine
  //        them
  //        the child smarts with a ";"
  //
  // There is an additional complication with composite nodes that carry a
  // negation - in this
  // case we will propogate the neagtion to the child nodes using the
  // following rules
  //   NOT (a AND b) = ( NOT (a)) AND ( NOT (b))
  //   NOT (a OR b) = ( NOT (a)) OR ( NOT (b))
  PRECONDITION(bond, "bad bond");
  PRECONDITION(node, "bad node");
  std::string descrip = node->getDescription();
  std::string res = "";

  const QueryBond::QUERYBOND_QUERY *child1;
  const QueryBond::QUERYBOND_QUERY *child2;
  QueryBond::QUERYBOND_QUERY::CHILD_VECT_CI chi;

  chi = node->beginChildren();
  child1 = chi->get();
  chi++;
  child2 = chi->get();
  chi++;
  // OK we should be at the end of vector by now - since we can have only two
  // children,
  // well - atleat in this case
  CHECK_INVARIANT(chi == node->endChildren(), "Too many children on the query");

  std::string dsc1, dsc2;
  dsc1 = child1->getDescription();
  dsc2 = child2->getDescription();
  std::string csmarts1, csmarts2;

  if ((dsc1 != "BondOr") && (dsc1 != "BondAnd")) {
    // child1 is  simple node get the smarts directly
    const BOND_EQUALS_QUERY *tchild =
        static_cast<const BOND_EQUALS_QUERY *>(child1);
    csmarts1 = getBondSmartsSimple(bond, tchild, atomToLeftIdx);
    bool nneg = (negate) ^ (tchild->getNegation());
    if (nneg) {
      csmarts1 = "!" + csmarts1;
    }
  } else {
    // child1 is a composite node recurse further
    bool nneg = (negate) ^ (child1->getNegation());
    csmarts1 = _recurseBondSmarts(bond, child1, nneg, atomToLeftIdx);
  }

  // now deal with the second child
  if ((dsc2 != "BondOr") && (dsc2 != "BondAnd")) {
    // child 2 is a simple node
    const BOND_EQUALS_QUERY *tchild =
        static_cast<const BOND_EQUALS_QUERY *>(child2);
    csmarts2 = getBondSmartsSimple(bond, tchild, atomToLeftIdx);
    bool nneg = (negate) ^ (tchild->getNegation());
    if (nneg) {
      csmarts2 = "!" + csmarts2;
    }
  } else {
    // child two is a composite node - recurse
    bool nneg = (negate) ^ (child2->getNegation());
    csmarts1 = _recurseBondSmarts(bond, child2, nneg, atomToLeftIdx);
  }

  // ok if we have a negation and we have to change the underlying logic,
  // since we propogated the negation i.e NOT (A OR B) = (NOT (A)) AND
  // (NOT(B))
  if (negate) {
    if (descrip == "BondOr") {
      descrip = "BondAnd";
    } else if (descrip == "BondAnd") {
      descrip = "BondOr";
    }
  }
  res += _combineChildSmarts(csmarts1, csmarts2, descrip);
  return res;
}

std::string FragmentSmartsConstruct(ROMol &mol, unsigned int atomIdx,
                                    std::vector<Canon::AtomColors> &colors,
                                    UINT_VECT &ranks) {
  Canon::MolStack molStack;
  molStack.reserve(mol.getNumAtoms() + mol.getNumBonds());
  std::stringstream res;

  // this is dirty trick get around the fact that canonicalizeFragment
  // thinks we already called findSSSR - to do some atom ranking
  // but for smarts we are going to ignore that part. We will artificially
  // set the "SSSR" property to an empty property
  VECT_INT_VECT rings;
  mol.getRingInfo()->reset();
  mol.getRingInfo()->initialize();
  for (auto &atom : mol.atoms()) {
    atom->updatePropertyCache(false);
  }
  Canon::canonicalizeFragment(mol, atomIdx, colors, ranks, molStack);

  // now clear the "SSSR" property
  mol.getRingInfo()->reset();
  Canon::MolStack::const_iterator msCI;
  for (msCI = molStack.begin(); msCI != molStack.end(); msCI++) {
    switch (msCI->type) {
      case Canon::MOL_STACK_ATOM: {
        QueryAtom *qatm = static_cast<QueryAtom *>(msCI->obj.atom);
        res << SmartsWrite::GetAtomSmarts(qatm);
        break;
      }
      case Canon::MOL_STACK_BOND: {
        QueryBond *qbnd = static_cast<QueryBond *>(msCI->obj.bond);
        res << SmartsWrite::GetBondSmarts(qbnd, msCI->number);
        break;
      }
      case Canon::MOL_STACK_RING: {
        if (msCI->number < 10)
          res << msCI->number;
        else
          res << "%" << msCI->number;
        break;
      }
      case Canon::MOL_STACK_BRANCH_OPEN: {
        res << "(";
        break;
      }
      case Canon::MOL_STACK_BRANCH_CLOSE: {
        res << ")";
        break;
      }
      default:
        break;
    }
  }
  return res.str();
}

// this is the used when converting a SMILES or
// non-query atom from a mol file into SMARTS.
std::string getNonQueryAtomSmarts(const QueryAtom *qatom) {
  PRECONDITION(qatom, "bad atom");
  PRECONDITION(!qatom->hasQuery(), "atom should not have query");
  std::stringstream res;
  res << "[";

  int isotope = qatom->getIsotope();
  if (isotope) {
    res << isotope;
  }

  if (SmilesWrite::inOrganicSubset(qatom->getAtomicNum())) {
    res << "#" << qatom->getAtomicNum();
  } else {
    res << qatom->getSymbol();
  }

  if (qatom->getOwningMol().hasProp(common_properties::_doIsoSmiles)) {
    if (qatom->getChiralTag() != Atom::CHI_UNSPECIFIED &&
        !qatom->hasProp(_qatomHasStereoSet) &&
        !qatom->hasProp(common_properties::_brokenChirality)) {
      qatom->setProp(_qatomHasStereoSet, 1);
      switch (qatom->getChiralTag()) {
        case Atom::CHI_TETRAHEDRAL_CW:
          res << "@@";
          break;
        case Atom::CHI_TETRAHEDRAL_CCW:
          res << "@";
          break;
        default:
          break;
      }
    }
  }

  int hs = qatom->getNumExplicitHs();
  // FIX: probably should be smarter about Hs:
  if (hs) {
    res << "H";
    if (hs > 1) res << hs;
  }
  int chg = qatom->getFormalCharge();
  if (chg) {
    if (chg == -1) {
      res << "-";
    } else if (chg == 1) {
      res << "+";
    } else if (chg < 0) {
      res << qatom->getFormalCharge();
    } else {
      res << "+" << qatom->getFormalCharge();
    }
  }
  int mapNum;
  if (qatom->getPropIfPresent(common_properties::molAtomMapNumber, mapNum)) {
    res << ":";
    res << mapNum;
  }
  res << "]";
  return res.str();
}

// this is the used when converting a SMILES or
// non-query bond from a mol file into SMARTS.
std::string getNonQueryBondSmarts(const QueryBond *qbond, int atomToLeftIdx) {
  PRECONDITION(qbond, "bad bond");
  PRECONDITION(!qbond->hasQuery(), "bond should not have query");
  RDUNUSED_PARAM(atomToLeftIdx);
  std::string res;

  if (qbond->getIsAromatic()) {
    res = ":";
  } else {
    bool reverseDative =
        (atomToLeftIdx >= 0 &&
         qbond->getBeginAtomIdx() == static_cast<unsigned int>(atomToLeftIdx));
    res = getBasicBondRepr(
        qbond->getBondType(), qbond->getBondDir(),
        qbond->getOwningMol().hasProp(common_properties::_doIsoSmiles),
        reverseDative);
  }

  return res;
}

}  // namespace

namespace SmartsWrite {
std::string GetAtomSmarts(const QueryAtom *qatom) {
  PRECONDITION(qatom, "bad atom");
  std::string res;
  bool needParen = false;

  // BOOST_LOG(rdInfoLog)<<"Atom: " <<qatom->getIdx()<<std::endl;
  if (!qatom->hasQuery()) {
    res = getNonQueryAtomSmarts(qatom);
    // BOOST_LOG(rdInfoLog)<<"\tno query:" <<res;
    return res;
  }
  QueryAtom::QUERYATOM_QUERY *query = qatom->getQuery();
  // describeQuery(query);

  PRECONDITION(query, "atom has no query");
  std::string descrip = qatom->getQuery()->getDescription();
  if (descrip == "") {
    // we have simple atom - just generate the smiles and return
    res = SmilesWrite::GetAtomSmiles(qatom);
    if (res[0] == '[') {
      // chop the brackets off, we'll put them back on later:
      needParen = true;
      res = res.substr(1, res.size() - 2);
    }
  } else if ((descrip == "AtomOr") || (descrip == "AtomAnd")) {
    // we have a composite query
    needParen = true;
    res = _recurseGetSmarts(qatom, query, query->getNegation());
    if (res.length() == 1) {  // single atom symbol we don't need parens
      needParen = false;
    }
  } else if (descrip == "RecursiveStructure") {
    // it's a bare recursive structure query:
    res = getRecursiveStructureQuerySmarts(query);
    needParen = true;
  } else {  // we have a simple smarts
    ATOM_EQUALS_QUERY *tquery =
        static_cast<ATOM_EQUALS_QUERY *>(qatom->getQuery());
    res = getAtomSmartsSimple(qatom, tquery, needParen);
    if (tquery->getNegation()) {
      res = "!" + res;
    }
  }
  std::string mapNum;
  if (qatom->getPropIfPresent(common_properties::molAtomMapNumber, mapNum)) {
    needParen = true;
    res += ":" + mapNum;
  }
  if (needParen) {
    res = "[" + res + "]";
  }
  return res;
}

std::string GetBondSmarts(const QueryBond *bond, int atomToLeftIdx) {
  PRECONDITION(bond, "bad bond");
  std::string res = "";

  // BOOST_LOG(rdInfoLog) << "bond: " << bond->getIdx() << std::endl;
  ;
  // it is possible that we are regular single bond and we don't need to write
  // anything
  if (!bond->hasQuery()) {
    res = getNonQueryBondSmarts(bond, atomToLeftIdx);
    // BOOST_LOG(rdInfoLog) << "\tno query:" << res << std::endl;
    return res;
  }
  // describeQuery(bond->getQuery());
  if ((typeid(*bond) == typeid(Bond)) &&
      ((bond->getBondType() == Bond::SINGLE) ||
       (bond->getBondType() == Bond::AROMATIC))) {
    BOOST_LOG(rdInfoLog) << "\tbasic:" << res << std::endl;
    return res;
  }
  const QueryBond::QUERYBOND_QUERY *query = bond->getQuery();
  PRECONDITION(query, "bond has no query");
  std::string descrip = query->getDescription();

  if ((descrip == "BondAnd") || (descrip == "BondOr")) {
    // composite query
    res = _recurseBondSmarts(bond, query, query->getNegation(), atomToLeftIdx);
  } else {
    // simple query
    if (query->getNegation()) {
      res = "!";
    }
    const BOND_EQUALS_QUERY *tquery =
        static_cast<const BOND_EQUALS_QUERY *>(query);
    res += getBondSmartsSimple(bond, tquery, atomToLeftIdx);
  }
  // BOOST_LOG(rdInfoLog) << "\t  query:" << descrip << " " << res << std::endl;
  return res;
}
}  // end of namespace SmartsWrite

std::string MolToSmarts(ROMol &inmol, bool doIsomericSmiles) {
  std::string res;
  unsigned int nAtoms = inmol.getNumAtoms();
  if (!nAtoms) return "";

  ROMol mol(inmol);
  UINT_VECT ranks;
  ranks.resize(nAtoms);
  // For smiles writing we would be canonicalizing but we will not do that here.
  // We will simple use the atom indices as the rank
  for (ROMol::AtomIterator atIt = mol.beginAtoms(); atIt != mol.endAtoms();
       atIt++) {
    ranks.push_back((*atIt)->getIdx());
  }

  // clean up the chirality on any atom that is marked as chiral,
  // but that should not be:
  if (doIsomericSmiles) {
    mol.setProp(common_properties::_doIsoSmiles, 1);
  }

  std::vector<AtomColors> colors;
  colors.resize(nAtoms);
  std::vector<AtomColors>::iterator colorIt;
  for (colorIt = colors.begin(); colorIt != colors.end(); colorIt++)
    *colorIt = Canon::WHITE_NODE;

  colorIt = std::find(colors.begin(), colors.end(), Canon::WHITE_NODE);
  while (colorIt != colors.end()) {
    unsigned int nextAtomIdx = 0;
    unsigned int nextRank;
    std::string subSmi;
    nextRank = nAtoms + 1;
    for (unsigned int i = 0; i < nAtoms; i++) {
      if (colors[i] == Canon::WHITE_NODE && ranks[i] < nextRank) {
        nextRank = ranks[i];
        nextAtomIdx = i;
      }
    }

    subSmi = FragmentSmartsConstruct(mol, nextAtomIdx, colors, ranks);
    res += subSmi;

    colorIt = std::find(colors.begin(), colors.end(), Canon::WHITE_NODE);
    if (colorIt != colors.end()) {
      res += ".";
    }
  }
  return res;
}
}  // namespace RDKit
