Add a utility function to get path length from Track

[r2c2.git] / source / engineer / newtraindialog.cpp

diff --git a/source/engineer/newtraindialog.cpp b/source/engineer/newtraindialog.cpp
index 11fd5c04d6f1331c607926c8d84256216d9e44df..78a3f113b6b0a161ed1ea96e2d6ddddb83a28f60 100644 (file)
--- a/source/engineer/newtraindialog.cpp
+++ b/source/engineer/newtraindialog.cpp
@@ -67,12 +67,8 @@ void NewTrainDialog::on_response(int code)
                Train *train = new Train(engineer.get_layout(), type, address, protocol);
                train->set_name(ent_name->get_text());
 
-               GLtk::Container *root = parent;
-               while(root->get_parent())
-                       root = root->get_parent();
-
                TrainDialog *dlg = new TrainDialog(engineer, *train);
-               root->add(*dlg);
+               find_ancestor<GLtk::Root>()->add(*dlg);
                dlg->autosize();
        }
 }