X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fengineer%2Fnewvehicledialog.cpp;h=e9c537ee0c21f8b0a73a8b665ea542752a5a250c;hb=81ee4fbd16b472892b67b35ea85956423a2247ed;hp=de4d14a7fda8786679bdfaff10bab774dea8402d;hpb=93c148ff88ba2c8266b44c336ba7ddd948010de7;p=r2c2.git

diff --git a/source/engineer/newvehicledialog.cpp b/source/engineer/newvehicledialog.cpp
index de4d14a..e9c537e 100644
--- a/source/engineer/newvehicledialog.cpp
+++ b/source/engineer/newvehicledialog.cpp
@@ -37,15 +37,14 @@ NewVehicleDialog::NewVehicleDialog(Train &t):
 
 	dynamic_cast<GLtk::Button *>(get_item(widgets, "btn_add"))->signal_clicked.connect(sigc::mem_fun(this, &NewVehicleDialog::add_clicked));
 
-	const Catalogue::ObjectMap &obj_types = train.get_layout().get_catalogue().get_all();
-	for(Catalogue::ObjectMap::const_iterator i=obj_types.begin(); i!=obj_types.end(); ++i)
-		if(const VehicleType *vt = dynamic_cast<const VehicleType *>(i->second))
-		{
-			if(vt->is_locomotive())
-				loco_types.append(vt);
-			else
-				wagon_types.append(vt);
-		}
+	list<VehicleType *> veh_types = train.get_layout().get_catalogue().get_list<VehicleType>();
+	for(list<VehicleType *>::iterator i=veh_types.begin(); i!=veh_types.end(); ++i)
+	{
+		if((*i)->is_locomotive())
+			loco_types.append(*i);
+		else
+			wagon_types.append(*i);
+	}
 }
 
 void NewVehicleDialog::toggle_list(bool show, GLtk::List *lst)