Ensure that Timetable has necessary helper AIs

[r2c2.git] / source / libr2c2 / timetable.cpp

diff --git a/source/libr2c2/timetable.cpp b/source/libr2c2/timetable.cpp
index 96081820a56a54f236ee88d40032354cded852b0..42414f63d0baf3ecfe14fcbda9afb4de8c3669e3 100644 (file)
--- a/source/libr2c2/timetable.cpp
+++ b/source/libr2c2/timetable.cpp
@@ -1,8 +1,10 @@
 #include <msp/strings/format.h>
+#include "aicontrol.h"
 #include "clock.h"
 #include "layout.h"
 #include "timetable.h"
 #include "train.h"
+#include "trainrouter.h"
 #include "zone.h"
 
 using namespace std;
@@ -15,6 +17,11 @@ Timetable::Timetable(Train &t):
        current_row(rows.end()),
        update_pending(false)
 {
+       if(!train.get_ai_of_type<AIControl>())
+               new AIControl(train);
+       if(!train.get_ai_of_type<TrainRouter>())
+               new TrainRouter(train);
+
        train.signal_ai_event.connect(sigc::mem_fun(this, &Timetable::event));
 }