Changes in src/Descriptors/AtomDescriptor.cpp [6d574a:8cce2b]

File:

• src/Descriptors/AtomDescriptor.cpp (modified) (4 diffs)

src/Descriptors/AtomDescriptor.cpp

@@ -13 +13 @@
 #include "World.hpp"
 #include "atom.hpp"
+#include "Patterns/ObservedContainer_impl.hpp"
 
 #include <boost/bind.hpp>
@@ -20 +21 @@
 using namespace std;
 
-typedef World::AtomSet::iterator atoms_iter_t;
+typedef World::AtomSet::internal_iterator atoms_iter_t;
 
 /************************ Forwarding object **************************************/
@@ -73 +74 @@
 
 atom* AtomDescriptor_impl::find() {
-  World::AtomSet atoms = getAtoms();
-  atoms_iter_t res = find_if(atoms.begin(),atoms.end(),boost::bind(&AtomDescriptor_impl::predicate,this,_1));
-  return (res!=atoms.end())?((*res).second):0;
+  World::AtomSet &atoms = getAtoms();
+  atoms_iter_t res = find_if(atoms.begin_internal(),atoms.end_internal(),boost::bind(&AtomDescriptor_impl::predicate,this,_1));
+  return (res!=atoms.end_internal())?((*res).second):0;
 }
 
@@ -81 +82 @@
   vector<atom*> res;
   World::AtomSet atoms = getAtoms();
-  atoms_iter_t iter;
-  for(iter=atoms.begin();iter!=atoms.end();++iter) {
-    if(predicate(*iter)){
-      res.push_back((*iter).second);
-    }
+  for_each(atoms.begin_internal(),
+           atoms.end_internal(),
+           boost::bind(&AtomDescriptor_impl::checkAndAdd,
+                       this,&res,_1));
+  return res;
+}
+
+void AtomDescriptor_impl::checkAndAdd(std::vector<atom*> *v,std::pair<atomId_t,atom*> p){
+  if(predicate(p)){
+    v->push_back(p.second);
   }
-  return res;
 }