Separate train routing logic to a class derived from TrainAI

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

diff --git a/source/engineer/trainproperties.cpp b/source/engineer/trainproperties.cpp
index 360923514b40297dd46440d1ea0864cccd064bf1..ecc9b2b73e555c718927ea626d540f00744ba6af 100644 (file)
--- 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 = dynamic_cast<TrainRouter *>(train->get_tagged_ai("router")))
+                       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 = dynamic_cast<TrainRouter *>(train->get_tagged_ai("router")))
+               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();)