2 files changed, 25 insertions, 3 deletions
diff --git a/libdimensionxx/dimensionxx/scene.hpp b/libdimensionxx/dimensionxx/scene.hpp
index 73cbc2e..acb2cb1 100644
--- a/libdimensionxx/dimensionxx/scene.hpp
+++ b/libdimensionxx/dimensionxx/scene.hpp
@@ -25,7 +25,7 @@
 
 namespace Dimension
 {
-  // Iterator class for scene objects
+  // Iterator class for scene objects - never invalidated unless removed
   class Scene_Iterator;
 
   // Base scene class.  Wraps a dmnsn_scene*.
@@ -55,6 +55,7 @@ namespace Dimension
     Iterator end();
 
     void push_object(Object& object);
+    void remove_object(Iterator i);
 
     // Access the wrapped C object.
     dmnsn_scene*       dmnsn();
@@ -72,8 +73,10 @@ namespace Dimension
 
   class Scene_Iterator
   {
+    typedef std::list<std::tr1::shared_ptr<Object> >::iterator Iterator;
+
   public:
-    Scene_Iterator(std::list<std::tr1::shared_ptr<Object> >::iterator i)
+    Scene_Iterator(Iterator i)
       : m_i(i) { }
     // Scene_Iterator(const Scene_Iterator& i);
     // ~Scene_Iterator();
@@ -91,8 +94,10 @@ namespace Dimension
     Scene_Iterator& operator--()    { --m_i; return *this; }
     Scene_Iterator  operator--(int) { return Scene_Iterator(m_i--); }
 
+    Iterator iterator() const { return m_i; }
+
   private:
-    std::list<std::tr1::shared_ptr<Object> >::iterator m_i;
+    Iterator m_i;
   };
 }
 
diff --git a/libdimensionxx/scene.cpp b/libdimensionxx/scene.cpp
index 4ff19e1..1bff351 100644
--- a/libdimensionxx/scene.cpp
+++ b/libdimensionxx/scene.cpp
@@ -99,6 +99,23 @@ namespace Dimension
     dmnsn_array_push(dmnsn()->objects, &cobject);
   }
 
+  // Remove an object
+  void
+  Scene::remove_object(Iterator i)
+  {
+    // Find it in the dmnsn_array* of objects and remove it
+    for (unsigned int j = 0; j < dmnsn_array_size(dmnsn()->objects); ++j) {
+      dmnsn_object* cobject;
+      dmnsn_array_get(dmnsn()->objects, j, &cobject);
+      if (cobject == i->dmnsn()) {
+        dmnsn_array_remove(dmnsn()->objects, j);
+        break;
+      }
+    }
+    // Remove it from the std::list
+    m_objects.erase(i.iterator());
+  }
+
   // Access the wrapped C object.
 
   dmnsn_scene*