Rework how widget ownership works in Container

[libs/gltk.git] / source / container.cpp

diff --git a/source/container.cpp b/source/container.cpp
index 88e59524e8ddac754a69b7506d230d6d019957be..f1f1934c071d724caaf17086076ccde92c111369 100644 (file)
--- a/source/container.cpp
+++ b/source/container.cpp
@@ -14,19 +14,21 @@ hierarchy_error::hierarchy_error(const string &w):
 
 Container::~Container()
 {
+       // Clear children here while members are still valid
        while(!children.empty())
-               delete children.front()->widget;
+       {
+               if(children.back()->own_widget)
+                       /* Avoid destroying the unique_ptr for the widget from within its own
+                       reset() function */
+                       unique_ptr<Widget> w = move(children.back()->own_widget);
+               else
+                       children.pop_back();
+       }
 }
 
-void Container::add(Widget &wdg)
+void Container::add(unique_ptr<Widget> wdg)
 {
-       wdg.set_parent(this);
-       children.push_back(make_unique<Child>(*this, &wdg));
-       if(wdg.get_animation_interval())
-               check_animation_interval();
-       children_rebuild_needed = true;
-       signal_rebuild_needed.emit();
-       on_child_added(wdg);
+       add_child(*wdg).own_widget = move(wdg);
 }
 
 void Container::remove(Widget &wdg)
@@ -35,6 +37,7 @@ void Container::remove(Widget &wdg)
        if(i==children.end())
                throw hierarchy_error("widget not in container");
 
+       unique_ptr<Widget> owned = move((*i)->own_widget);
        if(&wdg==saved_input_focus)
                saved_input_focus = nullptr;
        wdg.set_parent(nullptr);
@@ -44,6 +47,18 @@ void Container::remove(Widget &wdg)
        on_child_removed(wdg);
 }
 
+Container::Child &Container::add_child(Widget &wdg)
+{
+       wdg.set_parent(this);
+       children.push_back(make_unique<Child>(*this, &wdg));
+       if(wdg.get_animation_interval())
+               check_animation_interval();
+       children_rebuild_needed = true;
+       signal_rebuild_needed.emit();
+       on_child_added(wdg);
+       return *children.back();
+}
+
 Geometry Container::determine_child_geometry(const Widget &child, const Part &part) const
 {
        Geometry pgeom = part.get_geometry();
@@ -381,6 +396,12 @@ Container::Child::Child(Container &c, Widget *w):
        widget->signal_rebuild_needed.connect(sigc::mem_fun(this, &Child::rebuild_needed));
 }
 
+Container::Child::Child(Container &c, unique_ptr<Widget> w):
+       Child(c, w.get())
+{
+       own_widget = move(w);
+}
+
 Container::Child::~Child()
 {
        visibility_changed(false);