| 1 | /*
 | 
|---|
| 2 |  * WorldIterators.cpp
 | 
|---|
| 3 |  *
 | 
|---|
| 4 |  *  Created on: Feb 25, 2010
 | 
|---|
| 5 |  *      Author: crueger
 | 
|---|
| 6 |  */
 | 
|---|
| 7 | 
 | 
|---|
| 8 | #include "Descriptors/AtomDescriptor.hpp"
 | 
|---|
| 9 | #include "Descriptors/AtomDescriptor_impl.hpp"
 | 
|---|
| 10 | #include "Descriptors/MoleculeDescriptor.hpp"
 | 
|---|
| 11 | #include "Descriptors/MoleculeDescriptor_impl.hpp"
 | 
|---|
| 12 | #include "atom.hpp"
 | 
|---|
| 13 | #include "molecule.hpp"
 | 
|---|
| 14 | #include "World.hpp"
 | 
|---|
| 15 | 
 | 
|---|
| 16 | /********************** Atoms *************************/
 | 
|---|
| 17 | 
 | 
|---|
| 18 | World::AtomIterator::AtomIterator(){
 | 
|---|
| 19 |   state = World::get()->atomEnd();
 | 
|---|
| 20 | }
 | 
|---|
| 21 | 
 | 
|---|
| 22 | World::AtomIterator::AtomIterator(AtomDescriptor _descr, World* _world) :
 | 
|---|
| 23 |     descr(_descr.get_impl()),
 | 
|---|
| 24 |     index(0),
 | 
|---|
| 25 |     world(_world)
 | 
|---|
| 26 | {
 | 
|---|
| 27 |   state = world->atoms.begin();
 | 
|---|
| 28 |   advanceState();
 | 
|---|
| 29 | }
 | 
|---|
| 30 | 
 | 
|---|
| 31 | World::AtomIterator::AtomIterator(const AtomIterator& rhs) :
 | 
|---|
| 32 |     state(rhs.state),
 | 
|---|
| 33 |     descr(rhs.descr),
 | 
|---|
| 34 |     index(rhs.index),
 | 
|---|
| 35 |     world(rhs.world)
 | 
|---|
| 36 |   {}
 | 
|---|
| 37 | 
 | 
|---|
| 38 | World::AtomIterator& World::AtomIterator::operator=(const AtomIterator& rhs)
 | 
|---|
| 39 | {
 | 
|---|
| 40 |   if(&rhs!=this){
 | 
|---|
| 41 |     state=rhs.state;
 | 
|---|
| 42 |     descr=rhs.descr;
 | 
|---|
| 43 |     index=rhs.index;
 | 
|---|
| 44 |     world=rhs.world;
 | 
|---|
| 45 |   }
 | 
|---|
| 46 |   return *this;
 | 
|---|
| 47 | }
 | 
|---|
| 48 | 
 | 
|---|
| 49 | World::AtomIterator& World::AtomIterator::operator++(){
 | 
|---|
| 50 |   ++state;
 | 
|---|
| 51 |   ++index;
 | 
|---|
| 52 |   advanceState();
 | 
|---|
| 53 |   return *this;
 | 
|---|
| 54 | }
 | 
|---|
| 55 | 
 | 
|---|
| 56 | World::AtomIterator World::AtomIterator::operator++(int){
 | 
|---|
| 57 |   AtomIterator res(*this);
 | 
|---|
| 58 |   ++(*this);
 | 
|---|
| 59 |   return res;
 | 
|---|
| 60 | }
 | 
|---|
| 61 | 
 | 
|---|
| 62 | bool World::AtomIterator::operator==(const AtomIterator& rhs){
 | 
|---|
| 63 |   return state==rhs.state;
 | 
|---|
| 64 | }
 | 
|---|
| 65 | 
 | 
|---|
| 66 | bool World::AtomIterator::operator==(const World::AtomSet::iterator& rhs){
 | 
|---|
| 67 |   return state==rhs;
 | 
|---|
| 68 | }
 | 
|---|
| 69 | 
 | 
|---|
| 70 | bool World::AtomIterator::operator!=(const AtomIterator& rhs){
 | 
|---|
| 71 |   return state!=rhs.state;
 | 
|---|
| 72 | }
 | 
|---|
| 73 | 
 | 
|---|
| 74 | bool World::AtomIterator::operator!=(const World::AtomSet::iterator& rhs){
 | 
|---|
| 75 |   return state!=rhs;
 | 
|---|
| 76 | }
 | 
|---|
| 77 | 
 | 
|---|
| 78 | atom* World::AtomIterator::operator*(){
 | 
|---|
| 79 |   return (*state).second;
 | 
|---|
| 80 | }
 | 
|---|
| 81 | 
 | 
|---|
| 82 | void World::AtomIterator::advanceState(){
 | 
|---|
| 83 |   // go forward until we have a matching atom or the end is reached
 | 
|---|
| 84 |   while((state!=world->atoms.end()) && (!descr->predicate(*state))){
 | 
|---|
| 85 |     ++state;
 | 
|---|
| 86 |     ++index;
 | 
|---|
| 87 |   }
 | 
|---|
| 88 | }
 | 
|---|
| 89 | 
 | 
|---|
| 90 | int World::AtomIterator::getCount(){
 | 
|---|
| 91 |   return index;
 | 
|---|
| 92 | }
 | 
|---|
| 93 | 
 | 
|---|
| 94 | /*********************************** Molecules ************************/
 | 
|---|
| 95 | 
 | 
|---|
| 96 | World::MoleculeIterator::MoleculeIterator(){
 | 
|---|
| 97 |   state = World::get()->moleculeEnd();
 | 
|---|
| 98 | }
 | 
|---|
| 99 | 
 | 
|---|
| 100 | World::MoleculeIterator::MoleculeIterator(MoleculeDescriptor _descr, World* _world) :
 | 
|---|
| 101 |     descr(_descr.get_impl()),
 | 
|---|
| 102 |     index(0),
 | 
|---|
| 103 |     world(_world)
 | 
|---|
| 104 | {
 | 
|---|
| 105 |   state = world->molecules.begin();
 | 
|---|
| 106 |   advanceState();
 | 
|---|
| 107 | }
 | 
|---|
| 108 | 
 | 
|---|
| 109 | World::MoleculeIterator::MoleculeIterator(const MoleculeIterator& rhs) :
 | 
|---|
| 110 |     state(rhs.state),
 | 
|---|
| 111 |     descr(rhs.descr),
 | 
|---|
| 112 |     index(rhs.index),
 | 
|---|
| 113 |     world(rhs.world)
 | 
|---|
| 114 |   {}
 | 
|---|
| 115 | 
 | 
|---|
| 116 | World::MoleculeIterator& World::MoleculeIterator::operator=(const MoleculeIterator& rhs)
 | 
|---|
| 117 | {
 | 
|---|
| 118 |   if(&rhs!=this){
 | 
|---|
| 119 |     state=rhs.state;
 | 
|---|
| 120 |     descr=rhs.descr;
 | 
|---|
| 121 |     index=rhs.index;
 | 
|---|
| 122 |     world=rhs.world;
 | 
|---|
| 123 |   }
 | 
|---|
| 124 |   return *this;
 | 
|---|
| 125 | }
 | 
|---|
| 126 | 
 | 
|---|
| 127 | World::MoleculeIterator& World::MoleculeIterator::operator++(){
 | 
|---|
| 128 |   ++state;
 | 
|---|
| 129 |   ++index;
 | 
|---|
| 130 |   advanceState();
 | 
|---|
| 131 |   return *this;
 | 
|---|
| 132 | }
 | 
|---|
| 133 | 
 | 
|---|
| 134 | World::MoleculeIterator World::MoleculeIterator::operator++(int){
 | 
|---|
| 135 |   MoleculeIterator res(*this);
 | 
|---|
| 136 |   ++(*this);
 | 
|---|
| 137 |   return res;
 | 
|---|
| 138 | }
 | 
|---|
| 139 | 
 | 
|---|
| 140 | bool World::MoleculeIterator::operator==(const MoleculeIterator& rhs){
 | 
|---|
| 141 |   return state==rhs.state;
 | 
|---|
| 142 | }
 | 
|---|
| 143 | 
 | 
|---|
| 144 | bool World::MoleculeIterator::operator==(const World::MoleculeSet::iterator& rhs){
 | 
|---|
| 145 |   return state==rhs;
 | 
|---|
| 146 | }
 | 
|---|
| 147 | 
 | 
|---|
| 148 | bool World::MoleculeIterator::operator!=(const MoleculeIterator& rhs){
 | 
|---|
| 149 |   return state!=rhs.state;
 | 
|---|
| 150 | }
 | 
|---|
| 151 | 
 | 
|---|
| 152 | bool World::MoleculeIterator::operator!=(const World::MoleculeSet::iterator& rhs){
 | 
|---|
| 153 |   return state!=rhs;
 | 
|---|
| 154 | }
 | 
|---|
| 155 | 
 | 
|---|
| 156 | molecule* World::MoleculeIterator::operator*(){
 | 
|---|
| 157 |   return (*state).second;
 | 
|---|
| 158 | }
 | 
|---|
| 159 | 
 | 
|---|
| 160 | void World::MoleculeIterator::advanceState(){
 | 
|---|
| 161 |   while((state!=world->molecules.end()) && (!descr->predicate(*state))){
 | 
|---|
| 162 |     ++state;
 | 
|---|
| 163 |     ++index;
 | 
|---|
| 164 |   }
 | 
|---|
| 165 | }
 | 
|---|
| 166 | 
 | 
|---|
| 167 | int World::MoleculeIterator::getCount(){
 | 
|---|
| 168 |   return index;
 | 
|---|
| 169 | }
 | 
|---|
| 170 | 
 | 
|---|