Get rid of the TrainAI tagging system

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

diff --git a/source/engineer/trainproperties.cpp b/source/engineer/trainproperties.cpp
index ecc9b2b73e555c718927ea626d540f00744ba6af..d92750e41725fd7d65e7071fbcfefc7e785c5cb1 100644 (file)
--- a/source/engineer/trainproperties.cpp
+++ b/source/engineer/trainproperties.cpp
@@ -90,7 +90,7 @@ TrainProperties::TrainProperties(Engineer &e, Train *t):
        {
                ent_addr->set_text(lexical_cast<string>(train->get_address()));
                ent_name->set_text(train->get_name());
-               if(TrainRouter *router = dynamic_cast<TrainRouter *>(train->get_tagged_ai("router")))
+               if(TrainRouter *router = train->get_ai_of_type<TrainRouter>())
                        drp_priority->set_selected_index(router->get_priority()+2);
 
                unsigned n_vehicles = train->get_n_vehicles();
@@ -117,7 +117,7 @@ void TrainProperties::on_ok_clicked()
        }
 
        train->set_name(ent_name->get_text());
-       if(TrainRouter *router = dynamic_cast<TrainRouter *>(train->get_tagged_ai("router")))
+       if(TrainRouter *router = train->get_ai_of_type<TrainRouter>())
                router->set_priority(drp_priority->get_selected_index()-2);
 
        // The locomotive is vehicle 0 so we need to add 1