Persist most dialogs across runs

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

diff --git a/source/engineer/traindialog.cpp b/source/engineer/traindialog.cpp
index 432f3dd22c99b98f7ea8d212b2dfeb73db861b9c..985d4c94ea09eca8bb358e516c19b5641b58619e 100644 (file)
--- a/source/engineer/traindialog.cpp
+++ b/source/engineer/traindialog.cpp
@@ -9,6 +9,7 @@
 #include "libr2c2/layout.h"
 #include "libr2c2/trainstatus.h"
 #include "controlpanel.h"
+#include "engineer.h"
 #include "routerpanel.h"
 #include "timetablepanel.h"
 #include "traindialog.h"
@@ -19,6 +20,7 @@ using namespace Msp;
 using namespace R2C2;
 
 TrainDialog::TrainDialog(Engineer &e, R2C2::Train &t):
+       DynamicDialog(e.get_user_interface()),
        engineer(e),
        train(t),
        updating(false)
@@ -134,8 +136,13 @@ void TrainDialog::ui_forward_toggled(bool value)
 
 void TrainDialog::expand_clicked()
 {
-       pnl_expander->set_visible(!pnl_expander->is_visible());
-       btn_expand->set_style(pnl_expander->is_visible() ? "arrow_up" : "arrow_down");
+       set_expanded(!pnl_expander->is_visible());
+}
+
+void TrainDialog::set_expanded(bool e)
+{
+       pnl_expander->set_visible(e);
+       btn_expand->set_style(e ? "arrow_up" : "arrow_down");
        signal_autosize_changed.emit();
        find_ancestor<GLtk::Root>()->get_layout()->update();
 }
@@ -144,3 +151,25 @@ void TrainDialog::toggle_panel(bool show, GLtk::Panel *panel)
 {
        panel->set_visible(show);
 }
+
+bool TrainDialog::save_state(DataFile::Statement &st) const
+{
+       st.keyword = "traindialog";
+       st.append(train.get_address());
+       st.sub.push_back((DataFile::Statement("expanded"), pnl_expander->is_visible()));
+       save_position(st.sub);
+
+       return true;
+}
+
+
+TrainDialog::StateLoader::StateLoader(TrainDialog &td):
+       DataFile::DerivedObjectLoader<TrainDialog, DynamicDialog::StateLoader>(td)
+{
+       add("expanded", &StateLoader::expanded);
+}
+
+void TrainDialog::StateLoader::expanded(bool e)
+{
+       obj.set_expanded(e);
+}