Add a UI for editing train timetables

[r2c2.git] / source / libmarklin / timetable.cpp

diff --git a/source/libmarklin/timetable.cpp b/source/libmarklin/timetable.cpp
index bd9d4b37313123c3b87f323f56e95e76a6f749f7..cfe2ef6c0d6226eaec15ee2896c096f4372557a2 100644 (file)
--- a/source/libmarklin/timetable.cpp
+++ b/source/libmarklin/timetable.cpp
@@ -5,6 +5,7 @@ Copyright © 2010  Mikkosoft Productions, Mikko Rasa
 Distributed under the GPL
 */
 
+#include <msp/strings/formatter.h>
 #include <msp/time/units.h>
 #include "block.h"
 #include "catalogue.h"
@@ -20,6 +21,7 @@ namespace Marklin {
 
 Timetable::Timetable(Train &t):
        train(t),
+       enabled(false),
        current_row(0),
        executing(true),
        pending_block(0)
@@ -28,8 +30,44 @@ Timetable::Timetable(Train &t):
        train.get_layout().get_driver().signal_sensor.connect(sigc::mem_fun(this, &Timetable::sensor_event));
 }
 
+void Timetable::set_enabled(bool e)
+{
+       enabled = e;
+}
+
+void Timetable::clear()
+{
+       rows.clear();
+       current_row = 0;
+}
+
+void Timetable::append(const Row &row)
+{
+       rows.push_back(row);
+}
+
+void Timetable::insert(unsigned i, const Row &row)
+{
+       if(i>rows.size())
+               throw InvalidParameterValue("Insert position out of range");
+
+       rows.insert(rows.begin()+i, row);
+       if(i<=current_row)
+               ++current_row;
+}
+
+const Timetable::Row &Timetable::get_row(unsigned i) const
+{
+       if(i>=rows.size())
+               throw InvalidParameterValue("Row index out of range");
+       return rows[i];
+}
+
 void Timetable::tick(const Time::TimeStamp &t)
 {
+       if(rows.empty() || !enabled)
+               return;
+
        if(wait_timeout && t>=wait_timeout)
        {
                wait_timeout = Time::TimeStamp();
@@ -131,6 +169,51 @@ Timetable::Row::Row(RowType t, const string &p):
        strparam(p)
 { }
 
+string Timetable::Row::str() const
+{
+       switch(type)
+       {
+       case GOTO:
+               return "go to "+strparam;
+       case TRAVEL:
+               return "travel to "+strparam;
+       case WAIT:
+               return format("wait for %d seconds", intparam);
+       case SPEED:
+               return format("set speed %d km/h", intparam);
+       case ROUTE:
+               return "set route "+strparam;
+       default:
+               return "invalid row";
+       }
+}
+
+Timetable::Row Timetable::Row::parse(const string &s)
+{
+       if(!s.compare(0, 6, "go to "))
+               return Row(GOTO, s.substr(6));
+       else if(!s.compare(0, 10, "travel to "))
+               return Row(TRAVEL, s.substr(10));
+       else if(!s.compare(0, 9, "wait for ") && isdigit(s[9]))
+       {
+               unsigned nondigit = 10;
+               while(nondigit<s.size() && isdigit(s[nondigit]))
+                       ++nondigit;
+               return Row(WAIT, lexical_cast<unsigned>(s.substr(9, nondigit-9)));
+       }
+       else if(!s.compare(0, 10, "set speed "))
+       {
+               unsigned nondigit = 11;
+               while(nondigit<s.size() && isdigit(s[nondigit]))
+                       ++nondigit;
+               return Row(SPEED, lexical_cast<unsigned>(s.substr(10, nondigit-10)));
+       }
+       else if(!s.compare(0, 10, "set route "))
+               return Row(ROUTE, s.substr(10));
+
+       throw InvalidParameterValue("Invalid row");
+}
+
 
 Timetable::Loader::Loader(Timetable &tt):
        DataFile::ObjectLoader<Timetable>(tt)