Get rid of the TrainAI tagging system

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

diff --git a/source/engineer/trainpanel.cpp b/source/engineer/trainpanel.cpp
index 97592be1bed92dc2547c5fb79ca3eb2399ca38f2..d6066cb1c5a962e489032b4d9d8eda0ef6bbc7d5 100644 (file)
--- a/source/engineer/trainpanel.cpp
+++ b/source/engineer/trainpanel.cpp
@@ -66,12 +66,9 @@ TrainPanel::TrainPanel(Engineer &e, Train &t):
        pnl_extra->set_geometry(GLtk::Geometry(0, 10, geom.w, 135));
        pnl_extra->set_visible(false);
 
-       TrainRouter *router = dynamic_cast<TrainRouter *>(train.get_tagged_ai("router"));
+       TrainRouter *router = train.get_ai_of_type<TrainRouter>();
        if(!router)
-       {
                router = new TrainRouter(train);
-               router->set_tag("router");
-       }
 
        const Route *route = router->get_route();
        pnl_extra->add(*(lbl_route = new GLtk::Label((route ? route->get_name() : "Free run"))));
@@ -215,12 +212,9 @@ void TrainPanel::goto_clicked()
 
 void TrainPanel::timetable_clicked()
 {
-       Timetable *timetable = dynamic_cast<Timetable *>(train.get_tagged_ai("timetable"));
+       Timetable *timetable = train.get_ai_of_type<Timetable>();
        if(!timetable)
-       {
                timetable = new Timetable(train);
-               timetable->set_tag("timetable");
-       }
 
        TimetableDialog *dialog = new TimetableDialog(*timetable);
        engineer.get_root().add(*dialog);
@@ -284,7 +278,7 @@ void TrainPanel::go_to(Track *track, unsigned)
 {
        pick_conn.disconnect();
 
-       TrainRouter *router = dynamic_cast<TrainRouter *>(train.get_tagged_ai("router"));
+       TrainRouter *router = train.get_ai_of_type<TrainRouter>();
        if(!router || !router->go_to(*track))
                engineer.set_status("Could not set route");
 }