Further adjustments to changes in GLtk

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

diff --git a/source/engineer/routeselect.cpp b/source/engineer/routeselect.cpp
index e5c090c9bcb26fbe7a9535cf80697264f0a8e667..ae3dd8340949304b22c4a08bf3ba313307a153b9 100644 (file)
--- a/source/engineer/routeselect.cpp
+++ b/source/engineer/routeselect.cpp
@@ -24,10 +24,10 @@ RouteSelect::RouteSelect(Engineer &e, Train &t):
        drp_route->set_selected_index(0);
 
        const Route *current_route = 0;
-       if(TrainRouter *router = dynamic_cast<TrainRouter *>(train.get_tagged_ai("router")))
+       if(TrainRouter *router = train.get_ai_of_type<TrainRouter>())
                current_route = router->get_route();
 
-       const set<Route *> &routes = engineer.get_layout().get_routes();
+       const set<Route *> &routes = engineer.get_layout().get_all<Route>();
        unsigned n = 1;
        for(set<Route *>::const_iterator i=routes.begin(); i!=routes.end(); ++i)
                if(!(*i)->is_temporary())
@@ -43,7 +43,7 @@ void RouteSelect::on_ok_clicked()
 {
        if(drp_route->get_selected_index()>0)
        {
-               const set<Route *> &routes = engineer.get_layout().get_routes();
+               const set<Route *> &routes = engineer.get_layout().get_all<Route>();
                set<Route *>::const_iterator i = routes.begin();
                unsigned n = drp_route->get_selected_index()-1;
                while(i!=routes.end())