Added BindingManager::visitAllPyObjects function, used to visit all PyObjects with a function.

Reviewer: Marcelo Lira <marcelo.lira@openbossa.org>
          Luciano Wolf <luciano.wolf@openbossa.org>

2 files changed, 21 insertions, 0 deletions

diff --git a/libshiboken/bindingmanager.cpp b/libshiboken/bindingmanager.cpp
index a43852614..2bcf2debe 100644
--- a/libshiboken/bindingmanager.cpp
+++ b/libshiboken/bindingmanager.cpp
@@ -277,5 +277,14 @@ std::set<SbkObject*> BindingManager::getAllPyObjects()
     return pyObjects;
 }
 
+void BindingManager::visitAllPyObjects(ObjectVisitor visitor, void* data)
+{
+    WrapperMap copy = m_d->wrapperMapper;
+    for (WrapperMap::iterator it = copy.begin(); it != copy.end(); ++it) {
+        if (hasWrapper(it->first))
+            visitor(it->second, data);
+    }
+}
+
 } // namespace Shiboken
 
diff --git a/libshiboken/bindingmanager.h b/libshiboken/bindingmanager.h
index b6d4ac42b..e12558b4f 100644
--- a/libshiboken/bindingmanager.h
+++ b/libshiboken/bindingmanager.h
@@ -33,6 +33,8 @@ struct SbkObjectType;
 namespace Shiboken
 {
 
+typedef void (*ObjectVisitor)(SbkObject*, void*);
+
 class LIBSHIBOKEN_API BindingManager
 {
 public:
@@ -50,6 +52,16 @@ public:
     SbkObjectType* resolveType(void* cptr, SbkObjectType* type);
 
     std::set<SbkObject*> getAllPyObjects();
+
+    /**
+     * Calls the function \p visitor for each object registered on binding manager.
+     * \note As various C++ pointers can point to the same PyObject due to multiple inheritance
+     *       a PyObject can be called more than one time for each PyObject.
+     * \param visitor function called for each object.
+     * \param data user data passed as second argument to the visitor function.
+     */
+    void visitAllPyObjects(ObjectVisitor visitor, void* data);
+
 private:
     ~BindingManager();
     // disable copy