mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-04 21:54:27 +08:00
282 lines
9.1 KiB
C++
282 lines
9.1 KiB
C++
// $Id$
|
|
//
|
|
// Copyright (C) 2003-2010 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 <boost/foreach.hpp>
|
|
|
|
// our stuff
|
|
#include <RDGeneral/Invariant.h>
|
|
#include <RDGeneral/RDLog.h>
|
|
#include "RWMol.h"
|
|
#include "Atom.h"
|
|
#include "Bond.h"
|
|
#include "BondIterators.h"
|
|
#include "RingInfo.h"
|
|
|
|
namespace RDKit{
|
|
void RWMol::destroy() {
|
|
ROMol::destroy();
|
|
d_partialBonds.clear();
|
|
d_partialBonds.resize(0);
|
|
};
|
|
|
|
void RWMol::insertMol(const ROMol &other)
|
|
{
|
|
VERTEX_ITER firstA,lastA;
|
|
boost::tie(firstA,lastA) = boost::vertices(other.d_graph);
|
|
while(firstA!=lastA){
|
|
Atom *newAt = other.d_graph[*firstA]->copy();
|
|
addAtom(newAt,false,true);
|
|
// set a bookmark for this atom's original index so we can update bonds
|
|
setAtomBookmark(newAt,ci_ATOM_HOLDER + other.d_graph[*firstA]->getIdx());
|
|
++firstA;
|
|
}
|
|
|
|
EDGE_ITER firstB,lastB;
|
|
boost::tie(firstB,lastB) = boost::edges(other.d_graph);
|
|
while(firstB != lastB){
|
|
Bond *bond_p = other.d_graph[*firstB]->copy();
|
|
unsigned int idx1,idx2;
|
|
idx1 = getAtomWithBookmark(ci_ATOM_HOLDER + bond_p->getBeginAtomIdx())->getIdx();
|
|
idx2 = getAtomWithBookmark(ci_ATOM_HOLDER + bond_p->getEndAtomIdx())->getIdx();
|
|
bond_p->setOwningMol(this);
|
|
bond_p->setBeginAtomIdx(idx1);
|
|
bond_p->setEndAtomIdx(idx2);
|
|
addBond(bond_p,true);
|
|
++firstB;
|
|
}
|
|
// blow out those bookmarks we set
|
|
boost::tie(firstA,lastA) = boost::vertices(other.d_graph);
|
|
while(firstA!=lastA){
|
|
clearAtomBookmark(ci_ATOM_HOLDER + *firstA);
|
|
++firstA;
|
|
}
|
|
|
|
// and clear our computed properties since they are no longer accurate:
|
|
clearComputedProps(true);
|
|
}
|
|
|
|
unsigned int RWMol::addAtom(bool updateLabel){
|
|
Atom *atom_p = new Atom();
|
|
atom_p->setOwningMol(this);
|
|
MolGraph::vertex_descriptor which = boost::add_vertex(d_graph);
|
|
d_graph[which].reset(atom_p);
|
|
atom_p->setIdx(which);
|
|
if(updateLabel){
|
|
clearAtomBookmark(ci_RIGHTMOST_ATOM);
|
|
setAtomBookmark(atom_p,ci_RIGHTMOST_ATOM);
|
|
}
|
|
return which;
|
|
}
|
|
|
|
void RWMol::replaceAtom(unsigned int idx,Atom *atom_pin,bool updateLabel){
|
|
PRECONDITION(atom_pin,"bad atom passed to replaceAtom");
|
|
RANGE_CHECK(0,idx,getNumAtoms()-1);
|
|
Atom *atom_p = atom_pin->copy();
|
|
atom_p->setOwningMol(this);
|
|
atom_p->setIdx(idx);
|
|
MolGraph::vertex_descriptor vd = boost::vertex(idx,d_graph);
|
|
d_graph[vd].reset(atom_p);
|
|
// FIX: do something about bookmarks
|
|
};
|
|
|
|
Atom *RWMol::getActiveAtom() {
|
|
if(hasAtomBookmark(ci_RIGHTMOST_ATOM)) return getAtomWithBookmark(ci_RIGHTMOST_ATOM);
|
|
else return getLastAtom();
|
|
};
|
|
|
|
void RWMol::setActiveAtom(Atom *at) {
|
|
clearAtomBookmark(ci_RIGHTMOST_ATOM);
|
|
setAtomBookmark(at,ci_RIGHTMOST_ATOM);
|
|
};
|
|
void RWMol::setActiveAtom(unsigned int idx) {
|
|
setActiveAtom(getAtomWithIdx(idx));
|
|
};
|
|
|
|
|
|
void RWMol::removeAtom(unsigned int idx) {
|
|
Atom *oatom = getAtomWithIdx(idx);
|
|
// remove any bookmarks which point to this atom:
|
|
ATOM_BOOKMARK_MAP *marks = getAtomBookmarks();
|
|
ATOM_BOOKMARK_MAP::iterator markI=marks->begin();
|
|
while(markI != marks->end()){
|
|
const ATOM_PTR_LIST &atoms=markI->second;
|
|
// we need to copy the iterator then increment it, because the
|
|
// deletion we're going to do in clearAtomBookmark will invalidate
|
|
// it.
|
|
ATOM_BOOKMARK_MAP::iterator tmpI=markI;
|
|
++markI;
|
|
if(std::find(atoms.begin(),atoms.end(),oatom)!=atoms.end()){
|
|
clearAtomBookmark(tmpI->first,oatom);
|
|
}
|
|
}
|
|
|
|
// remove bonds attached to the atom
|
|
std::vector<std::pair<unsigned int,unsigned int> > nbrs;
|
|
ADJ_ITER b1,b2;
|
|
boost::tie(b1,b2)=getAtomNeighbors(oatom);
|
|
while(b1!=b2){
|
|
nbrs.push_back(std::make_pair(oatom->getIdx(),*b1));
|
|
++b1;
|
|
}
|
|
for(unsigned int i=0;i<nbrs.size();++i){
|
|
removeBond(nbrs[i].first,nbrs[i].second);
|
|
}
|
|
|
|
// loop over all atoms with higher indices and update their indices
|
|
for(unsigned int i=idx+1;i<getNumAtoms();i++){
|
|
Atom *atom = getAtomWithIdx(i);
|
|
atom->setIdx(i-1);
|
|
}
|
|
|
|
// do the same with the coordinates in the conformations
|
|
BOOST_FOREACH(CONFORMER_SPTR conf,d_confs){
|
|
RDGeom::POINT3D_VECT &positions = conf->getPositions();
|
|
RDGeom::POINT3D_VECT_I pi = positions.begin();
|
|
for (unsigned int i = 0; i < getNumAtoms()-1;i++) {
|
|
++pi;
|
|
if (i >= idx) {
|
|
positions[i] = positions[i+1];
|
|
}
|
|
}
|
|
positions.erase(pi);
|
|
}
|
|
// now deal with bonds:
|
|
// their end indices may need to be decremented and their
|
|
// indices will need to be handled
|
|
unsigned int nBonds=0;
|
|
EDGE_ITER beg,end;
|
|
boost::tie(beg,end)=getEdges();
|
|
while(beg!=end){
|
|
BOND_SPTR bond = d_graph[*beg++];
|
|
unsigned int tmpIdx = bond->getBeginAtomIdx();
|
|
if( tmpIdx > idx) bond->setBeginAtomIdx(tmpIdx-1);
|
|
tmpIdx = bond->getEndAtomIdx();
|
|
if( tmpIdx > idx) bond->setEndAtomIdx(tmpIdx-1);
|
|
bond->setIdx(nBonds++);
|
|
}
|
|
|
|
// reset our ring info structure, because it is pretty likely
|
|
// to be wrong now:
|
|
dp_ringInfo->reset();
|
|
|
|
|
|
// remove all connections to the atom:
|
|
MolGraph::vertex_descriptor vd = boost::vertex(idx,d_graph);
|
|
boost::clear_vertex(vd,d_graph);
|
|
// finally remove the vertex itself
|
|
boost::remove_vertex(vd,d_graph);
|
|
}
|
|
|
|
void RWMol::removeAtom(Atom *atom) {
|
|
removeAtom(atom->getIdx());
|
|
}
|
|
|
|
unsigned int RWMol::addBond(unsigned int atomIdx1,unsigned int atomIdx2,
|
|
Bond::BondType bondType){
|
|
RANGE_CHECK(0,atomIdx1,getNumAtoms()-1);
|
|
RANGE_CHECK(0,atomIdx2,getNumAtoms()-1);
|
|
PRECONDITION(atomIdx1!=atomIdx2,"attempt to add self-bond");
|
|
PRECONDITION(!getBondBetweenAtoms(atomIdx1,atomIdx2),"bond already exists");
|
|
Bond *b = new Bond(bondType);
|
|
b->setOwningMol(this);
|
|
if(bondType==Bond::AROMATIC){
|
|
b->setIsAromatic(1);
|
|
//
|
|
// assume that aromatic bonds connect aromatic atoms
|
|
// This is relevant for file formats like MOL, where there
|
|
// is no such thing as an aromatic atom, but bonds can be
|
|
// marked aromatic.
|
|
//
|
|
getAtomWithIdx(atomIdx1)->setIsAromatic(1);
|
|
getAtomWithIdx(atomIdx2)->setIsAromatic(1);
|
|
}
|
|
bool ok;
|
|
MolGraph::edge_descriptor which;
|
|
boost::tie(which,ok) = boost::add_edge(atomIdx1,atomIdx2,d_graph);
|
|
d_graph[which].reset(b);
|
|
unsigned int res = boost::num_edges(d_graph);
|
|
b->setIdx(res-1);
|
|
b->setBeginAtomIdx(atomIdx1);
|
|
b->setEndAtomIdx(atomIdx2);
|
|
|
|
// if both atoms have a degree>1, reset our ring info structure,
|
|
// because there's a non-trivial chance that it's now wrong.
|
|
if(boost::out_degree(atomIdx1,d_graph)>1 && boost::out_degree(atomIdx2,d_graph)>1){
|
|
dp_ringInfo->reset();
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
unsigned int RWMol::addBond(Atom *atom1,Atom *atom2,
|
|
Bond::BondType bondType){
|
|
PRECONDITION(atom1&&atom2,"NULL atom passed in");
|
|
return addBond(atom1->getIdx(),atom2->getIdx(),bondType);
|
|
}
|
|
|
|
unsigned int RWMol::addBond(Atom::ATOM_SPTR atom1,Atom::ATOM_SPTR atom2,
|
|
Bond::BondType bondType){
|
|
return addBond(atom1->getIdx(),atom2->getIdx(),bondType);
|
|
}
|
|
|
|
void RWMol::removeBond(unsigned int aid1, unsigned int aid2) {
|
|
RANGE_CHECK(0,aid1,getNumAtoms()-1);
|
|
RANGE_CHECK(0,aid2,getNumAtoms()-1);
|
|
Bond *bnd = getBondBetweenAtoms(aid1, aid2);
|
|
if(!bnd) return;
|
|
|
|
// remove any bookmarks which point to this bond:
|
|
BOND_BOOKMARK_MAP *marks = getBondBookmarks();
|
|
BOND_BOOKMARK_MAP::iterator markI=marks->begin();
|
|
while(markI != marks->end()){
|
|
BOND_PTR_LIST &bonds=markI->second;
|
|
// we need to copy the iterator then increment it, because the
|
|
// deletion we're going to do in clearBondBookmark will invalidate
|
|
// it.
|
|
BOND_BOOKMARK_MAP::iterator tmpI=markI;
|
|
markI++;
|
|
if(std::find(bonds.begin(),bonds.end(),bnd)!=bonds.end()){
|
|
clearBondBookmark(tmpI->first,bnd);
|
|
}
|
|
}
|
|
|
|
// reset our ring info structure, because it is pretty likely
|
|
// to be wrong now:
|
|
dp_ringInfo->reset();
|
|
|
|
MolGraph::vertex_descriptor vd1 = boost::vertex(aid1,d_graph);
|
|
MolGraph::vertex_descriptor vd2 = boost::vertex(aid2,d_graph);
|
|
boost::remove_edge(vd1, vd2, d_graph);
|
|
}
|
|
|
|
Bond *RWMol::createPartialBond(unsigned int atomIdx1,Bond::BondType bondType){
|
|
RANGE_CHECK(0,atomIdx1,getNumAtoms()-1);
|
|
|
|
Bond *b = new Bond(bondType);
|
|
b->setOwningMol(this);
|
|
b->setBeginAtomIdx(atomIdx1);
|
|
return b;
|
|
}
|
|
|
|
unsigned int RWMol::finishPartialBond(unsigned int atomIdx2,int bondBookmark,
|
|
Bond::BondType bondType){
|
|
PRECONDITION(hasBondBookmark(bondBookmark),"no such partial bond");
|
|
RANGE_CHECK(0,atomIdx2,getNumAtoms()-1);
|
|
|
|
Bond *bsp = getBondWithBookmark(bondBookmark);
|
|
if(bondType==Bond::UNSPECIFIED){
|
|
bondType = bsp->getBondType();
|
|
}
|
|
return addBond(bsp->getBeginAtomIdx(),atomIdx2,bondType);
|
|
}
|
|
|
|
} // end o' namespace
|