X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fengineer%2Ftrainproperties.cpp;h=d92750e41725fd7d65e7071fbcfefc7e785c5cb1;hb=ca9d278f9472206ad9a01190dcef9f0eb1bcc274;hp=360923514b40297dd46440d1ea0864cccd064bf1;hpb=c06f3dde10411cd34aae026d8bf22b7771bdc3e0;p=r2c2.git

diff --git a/source/engineer/trainproperties.cpp b/source/engineer/trainproperties.cpp
index 3609235..d92750e 100644
--- a/source/engineer/trainproperties.cpp
+++ b/source/engineer/trainproperties.cpp
@@ -2,6 +2,7 @@
 #include <msp/strings/format.h>
 #include <msp/strings/lexicalcast.h>
 #include "libr2c2/driver.h"
+#include "libr2c2/trainrouter.h"
 #include "libr2c2/vehicle.h"
 #include "libr2c2/vehicletype.h"
 #include "engineer.h"
@@ -89,7 +90,8 @@ TrainProperties::TrainProperties(Engineer &e, Train *t):
 	{
 		ent_addr->set_text(lexical_cast<string>(train->get_address()));
 		ent_name->set_text(train->get_name());
-		drp_priority->set_selected_index(train->get_priority()+2);
+		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();
 		for(unsigned i=1; i<n_vehicles; ++i)
@@ -115,7 +117,8 @@ void TrainProperties::on_ok_clicked()
 	}
 
 	train->set_name(ent_name->get_text());
-	train->set_priority(drp_priority->get_selected_index()-2);
+	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
 	for(set<unsigned>::const_iterator i=rem_vehicles.end(); i!=rem_vehicles.begin();)