ReactionSubstructureSearch.hpp

Go to the documentation of this file.

/* 
 * ReactionSubstructureSearch.hpp 
 *
 * This file is part of the Chemical Data Processing Toolkit
 *
 * Copyright (C) 2003 Thomas Seidel <thomas.seidel@univie.ac.at>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; see the file COPYING. If not, write to
 * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */
 
#ifndef CDPL_CHEM_REACTIONSUBSTRUCTURESEARCH_HPP
#define CDPL_CHEM_REACTIONSUBSTRUCTURESEARCH_HPP
 
#include <vector>
#include <deque>
#include <set>
#include <utility>
#include <cstddef>
 
#include <boost/iterator/indirect_iterator.hpp>
 
#include "CDPL/Chem/APIPrefix.hpp"
#include "CDPL/Chem/AtomBondMapping.hpp"
#include "CDPL/Chem/MatchExpression.hpp"
#include "CDPL/Util/BitSet.hpp"
#include "CDPL/Util/ObjectStack.hpp"
 
 
namespace CDPL
{
 
    namespace Chem
    {
 
        class Reaction;
        class MolecularGraph;
        class Atom;
        class Bond;
 
        class CDPL_CHEM_API ReactionSubstructureSearch
        {
 
            typedef std::vector<AtomBondMapping*> ABMappingList;
 
          public:
            typedef boost::indirect_iterator<ABMappingList::iterator, AtomBondMapping> MappingIterator;
 
            typedef boost::indirect_iterator<ABMappingList::const_iterator, const AtomBondMapping> ConstMappingIterator;
 
            ReactionSubstructureSearch();
 
            ReactionSubstructureSearch(const Reaction& query);
 
            ~ReactionSubstructureSearch();
 
            void setQuery(const Reaction& query);
 
            bool mappingExists(const Reaction& target);
 
            bool findMappings(const Reaction& target);
 
            std::size_t getNumMappings() const;
 
            AtomBondMapping& getMapping(std::size_t idx);
 
            const AtomBondMapping& getMapping(std::size_t idx) const;
 
            MappingIterator getMappingsBegin();
 
            ConstMappingIterator getMappingsBegin() const;
 
            MappingIterator getMappingsEnd();
 
            ConstMappingIterator getMappingsEnd() const;
 
            MappingIterator begin();
 
            ConstMappingIterator begin() const;
 
            MappingIterator end();
 
            ConstMappingIterator end() const;
 
            void uniqueMappingsOnly(bool unique);
 
            bool uniqueMappingsOnly() const;
 
            void setMaxNumMappings(std::size_t max_num_mappings);
 
            std::size_t getMaxNumMappings() const;
 
            void setEnabledReactionRoles(unsigned int roles);
 
            unsigned int getEnabledReactionRoles() const;
 
          private:
            typedef std::pair<std::size_t, std::size_t> IndexOffsetPair;
 
            ReactionSubstructureSearch(const ReactionSubstructureSearch&);
 
            ReactionSubstructureSearch& operator=(const ReactionSubstructureSearch&);
 
            bool init(const Reaction&);
 
            void initQueryData();
            void initTargetData();
 
            bool findEquivAtoms();
            bool findEquivBonds();
 
            bool mapAtoms();
 
            std::size_t nextQueryAtom() const;
            bool        nextTargetAtom(std::size_t, unsigned int, std::size_t&, std::size_t&) const;
 
            bool atomMappingAllowed(std::size_t, std::size_t) const;
 
            bool mapBonds(const Atom*, const Atom*, const IndexOffsetPair&);
            bool mapAtoms(std::size_t);
            bool mapAtoms(std::size_t, std::size_t);
 
            bool mappingFound();
 
            bool hasPostMappingMatchExprs() const;
            bool foundMappingMatches(const AtomBondMapping*) const;
 
            bool foundMappingUnique();
 
            void freeAtomBondMappings();
            void freeAtomBondMapping();
 
            AtomBondMapping* createAtomBondMapping();
 
            class ABMappingMask
            {
 
              public:
                void initAtomMask(std::size_t);
                void initBondMask(std::size_t);
 
                void setAtomBit(std::size_t);
                void resetAtomBit(std::size_t);
 
                bool testAtomBit(std::size_t) const;
 
                void setBondBit(std::size_t);
                void resetBondMask();
 
                bool operator<(const ABMappingMask&) const;
                bool operator>(const ABMappingMask&) const;
 
              private:
                Util::BitSet atomMask;
                Util::BitSet bondMask;
            };
 
            typedef MatchExpression<Reaction>::SharedPointer ReactionMatchExprPtr;
 
            typedef std::vector<Util::BitSet>                                         BitMatrix;
            typedef std::vector<const Atom*>                                          AtomMappingTable;
            typedef std::vector<std::size_t>                                          BondMappingTable;
            typedef std::deque<std::size_t>                                           AtomQueue;
            typedef std::set<ABMappingMask>                                           UniqueMappingList;
            typedef std::vector<const Atom*>                                          AtomList;
            typedef std::vector<const Bond*>                                          BondList;
            typedef std::vector<std::size_t>                                          IndexList;
            typedef std::vector<MatchExpression<Atom, MolecularGraph>::SharedPointer> AtomMatchExprTable;
            typedef std::vector<MatchExpression<Bond, MolecularGraph>::SharedPointer> BondMatchExprTable;
            typedef std::vector<IndexOffsetPair>                                      IndexOffsetTable;
            typedef Util::ObjectStack<AtomBondMapping>                                MappingCache;
 
            const Reaction*      query;
            const Reaction*      target;
            AtomList             queryAtoms;
            BondList             queryBonds;
            AtomList             targetAtoms;
            BondList             targetBonds;
            IndexOffsetTable     queryABIndexOffsets;
            IndexOffsetTable     targetABIndexOffsets;
            BitMatrix            atomEquivMatrix;
            BitMatrix            bondEquivMatrix;
            AtomQueue            termQueryAtoms;
            AtomMappingTable     queryAtomMapping;
            BondMappingTable     queryBondMapping;
            Util::BitSet         queryMappingMask;
            ABMappingMask        targetMappingMask;
            ABMappingList        foundMappings;
            UniqueMappingList    uniqueMappings;
            AtomMatchExprTable   atomMatchExprTable;
            BondMatchExprTable   bondMatchExprTable;
            ReactionMatchExprPtr rxnMatchExpression;
            IndexList            postMappingMatchAtoms;
            IndexList            postMappingMatchBonds;
            MappingCache         mappingCache;
            bool                 queryChanged;
            bool                 initQueryMappingData;
            bool                 uniqueMatches;
            bool                 saveMappings;
            unsigned int         enabledRxnRoles;
            std::size_t          numQueryAtoms;
            std::size_t          numQueryBonds;
            std::size_t          numTargetAtoms;
            std::size_t          numTargetBonds;
            std::size_t          maxNumMappings;
        };
    } // namespace Chem
} // namespace CDPL
 
#endif // CDPL_CHEM_REACTIONSUBSTRUCTURESEARCH_HPP