Make use of the unified storage class for other parts of Layout

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

diff --git a/source/engineer/routeselect.cpp b/source/engineer/routeselect.cpp
index 770a455a4130b7406e22a6f348c38b680b757626..ae3dd8340949304b22c4a08bf3ba313307a153b9 100644 (file)
--- a/source/engineer/routeselect.cpp
+++ b/source/engineer/routeselect.cpp
@@ -27,7 +27,7 @@ RouteSelect::RouteSelect(Engineer &e, Train &t):
        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())