gras: figured out how to do the python reference

1 files changed, 15 insertions, 1 deletions

diff --git a/include/gras/detail/element.hpp b/include/gras/detail/element.hpp
index 35033e1..22a63a0 100644
--- a/include/gras/detail/element.hpp
+++ b/include/gras/detail/element.hpp
@@ -5,6 +5,19 @@
 
 namespace gras
 {
+    struct WeakElementSharedPtr : WeakElement
+    {
+        WeakElementSharedPtr(boost::weak_ptr<Element> weak_self)
+        {
+            _weak_self = weak_self;
+        }
+        boost::shared_ptr<void> lock(void)
+        {
+            return _weak_self.lock();
+        }
+        boost::weak_ptr<Element> _weak_self;
+    };
+
     template <typename T>
     inline Element::Element(const boost::shared_ptr<T> &elem)
     {
@@ -16,7 +29,8 @@ namespace gras
     {
         try
         {
-            this->weak_self = this->shared_from_this();
+            if (not this->weak_self)
+                this->weak_self.reset(new WeakElementSharedPtr(this->shared_from_this()));
         }
         catch(...){}
         return *this;