1 #include <msp/strings/format.h>
2 #include <msp/time/units.h>
15 Timetable::Timetable(Train &t):
24 train.signal_advanced.connect(sigc::mem_fun(this, &Timetable::train_advanced));
25 train.signal_ai_event.connect(sigc::mem_fun(this, &Timetable::event));
26 Layout &layout = train.get_layout();
27 layout.signal_block_state_changed.connect(sigc::mem_fun(this, &Timetable::block_state_changed));
28 layout.signal_block_reserved.connect(sigc::mem_fun(this, &Timetable::block_reserved));
31 void Timetable::set_enabled(bool e)
36 void Timetable::reset()
39 wait_timeout = Time::TimeStamp();
44 void Timetable::clear()
50 void Timetable::append(const Row &row)
55 void Timetable::insert(unsigned i, const Row &row)
58 throw out_of_range("Timetable::insert");
60 rows.insert(rows.begin()+i, row);
65 const Timetable::Row &Timetable::get_row(unsigned i) const
68 throw out_of_range("Timetable::get_row");
72 void Timetable::tick(const Time::TimeStamp &t, const Time::TimeDelta &)
74 if(rows.empty() || !enabled)
77 if(wait_timeout && t>=wait_timeout)
79 wait_timeout = Time::TimeStamp();
80 current_row = (current_row+1)%rows.size();
86 Row &row = rows[current_row];
91 train.ai_message(Message("go-to-track", &get_sensor(row.get_param<unsigned>(0))));
95 train.ai_message(Message("go-to-zone", &get_zone(row.get_param<string>(0))));
99 Block *block = &get_sensor(row.get_param<unsigned>(0)).get_block();
100 if(block->get_train()!=&train || block->get_state()<Block::MAYBE_ACTIVE)
102 pending_block = block;
103 pending_train = &train;
109 pending_block = &get_turnout(row.get_param<unsigned>(0)).get_block();
110 pending_train = (pending_block->get_train()==&train ? 0 : &train);
114 wait_timeout = t+row.get_param<unsigned>(0)*Time::sec;
119 unsigned unixtime = t.to_unixtime();
120 unsigned mod = row.get_param<unsigned>(1);
121 unsigned secs = ((mod+row.get_param<unsigned>(0))-(unixtime%mod))%mod;
122 wait_timeout = t+secs*Time::sec;
128 Train *other_train = &train.get_layout().get_train(row.get_param<unsigned>(0));
129 Block *block = &get_sensor(row.get_param<unsigned>(1)).get_block();
130 if(block->get_train()!=other_train || block->get_state()<Block::MAYBE_ACTIVE)
132 pending_train = other_train;
133 pending_block = block;
146 float speed = row.get_param<unsigned>(0)/3.6*train.get_layout().get_catalogue().get_scale();
147 train.ai_message(Message("set-target-speed", speed));
151 train.ai_message(Message("toggle-reverse"));
154 train.ai_message(Message("set-route", &train.get_layout().get_route(row.get_param<string>(0))));
159 current_row = (current_row+1)%rows.size();
163 void Timetable::save(list<DataFile::Statement> &st) const
165 for(vector<Row>::const_iterator i=rows.begin(); i!=rows.end(); ++i)
166 st.push_back(i->save());
169 Track &Timetable::get_sensor(unsigned id)
171 Block &block = train.get_layout().get_block(id|0x1000);
172 return **block.get_tracks().begin();
175 Track &Timetable::get_turnout(unsigned id)
177 Block &block = train.get_layout().get_block(id|0x2000);
178 return **block.get_tracks().begin();
181 Zone &Timetable::get_zone(const string &name)
183 string::size_type space = name.rfind(' ');
184 if(space==string::npos || space==0)
185 throw invalid_argument("Timetable::get_zone");
186 unsigned number = lexical_cast<unsigned>(name.substr(space+1));
187 return train.get_layout().get_zone(name.substr(0, space), number);
190 void Timetable::block_state_changed(Block &block, Block::State state)
192 if(rows.empty() || !enabled)
195 if(&block==pending_block && block.get_train()==pending_train && state>=Block::MAYBE_ACTIVE)
198 current_row = (current_row+1)%rows.size();
203 void Timetable::block_reserved(Block &block, Train *trn)
205 if(rows.empty() || !enabled)
208 if(&block==pending_block && trn==pending_train)
210 Row &row = rows[current_row];
211 if(row.type==TRAVEL_PAST && !pending_train)
214 current_row = (current_row+1)%rows.size();
220 void Timetable::train_advanced(Block &block)
222 if(rows.empty() || !enabled)
225 Row &row = rows[current_row];
226 if(row.type==TRAVEL_PAST && &block==pending_block && pending_train)
230 void Timetable::event(TrainAI &, const Message &ev)
232 if(ev.type=="arrived")
234 if(rows.empty() || !enabled)
237 Row &row = rows[current_row];
240 current_row = (current_row+1)%rows.size();
249 Timetable::Row::Row(RowType t):
254 Timetable::Row::Row(RowType t, const T &p):
261 const T &Timetable::Row::get_param(unsigned i) const
264 throw out_of_range("Timetable::Row::get_param");
265 return params[i].value<T>();
268 string Timetable::Row::str() const
273 return format("set route to sensor %d", get_param<unsigned>(0));
275 return "set route to "+get_param<string>(0);
277 return format("travel to sensor %d", get_param<unsigned>(0));
279 return format("travel past turnout %d", get_param<unsigned>(0));
281 return format("wait for %d seconds", get_param<unsigned>(0));
283 return format("wait until %d mod %d seconds", get_param<unsigned>(0), get_param<unsigned>(1));
285 return format("wait for train %d at sensor %d", get_param<unsigned>(0), get_param<unsigned>(1));
287 return "travel until arrival";
289 return format("set speed %d km/h", get_param<unsigned>(0));
293 return "set route "+get_param<string>(0);
295 return "invalid row";
299 DataFile::Statement Timetable::Row::save() const
304 return DataFile::Statement("goto_sensor"), get_param<unsigned>(0);
306 return DataFile::Statement("goto_zone"), get_param<string>(0);
308 return DataFile::Statement("travel_to"), get_param<unsigned>(0);
310 return DataFile::Statement("travel_past"), get_param<unsigned>(0);
312 return DataFile::Statement("wait"), get_param<unsigned>(0);
314 return DataFile::Statement("wait_until"), get_param<unsigned>(0), get_param<unsigned>(1);
316 return DataFile::Statement("wait_train"), get_param<unsigned>(0), get_param<unsigned>(1);
318 return DataFile::Statement("arrive");
320 return DataFile::Statement("speed"), get_param<unsigned>(0);
322 return DataFile::Statement("reverse");
324 return DataFile::Statement("route"), get_param<string>(0);
326 return DataFile::Statement();
330 Timetable::Row Timetable::Row::parse(const string &s)
332 if(!s.compare(0, 7, "travel "))
334 if(!s.compare(7, 10, "to sensor "))
335 return Row(TRAVEL_TO, lexical_cast<unsigned>(s.substr(17)));
336 else if(!s.compare(7, 13, "past turnout "))
337 return Row(TRAVEL_PAST, lexical_cast<unsigned>(s.substr(20)));
338 else if(!s.compare(7, string::npos, "until arrival"))
341 else if(!s.compare(0, 9, "wait for "))
345 unsigned nondigit = 10;
346 while(nondigit<s.size() && isdigit(s[nondigit]))
348 return Row(WAIT_TIME, lexical_cast<unsigned>(s.substr(9, nondigit-9)));
350 else if(!s.compare(9, 6, "train "))
352 string::size_type at = s.find(" at sensor ", 15);
355 Row row(WAIT_TRAIN, lexical_cast<unsigned>(s.substr(15, at-15)));
356 row.params.push_back(lexical_cast<unsigned>(s.substr(at+11)));
361 else if(!s.compare(0, 11, "wait until "))
363 string::size_type mod = s.find(" mod ", 11);
364 unsigned nondigit = (mod!=string::npos ? mod+5 : 11);
365 while(nondigit<s.size() && isdigit(s[nondigit]))
367 if(mod!=string::npos)
369 unsigned time = lexical_cast<unsigned>(s.substr(11, mod-11));
370 Row row(WAIT_UNTIL, time);
371 row.params.push_back(lexical_cast<unsigned>(s.substr(mod+5, nondigit-mod-5)));
376 unsigned time = lexical_cast<unsigned>(s.substr(11, nondigit-11));
377 Row row(WAIT_UNTIL, time);
378 row.params.push_back(3600);
382 else if(!s.compare(0, 10, "set speed "))
384 unsigned nondigit = 11;
385 while(nondigit<s.size() && (isdigit(s[nondigit]) || s[nondigit]=='-'))
387 return Row(SPEED, lexical_cast<unsigned>(s.substr(10, nondigit-10)));
389 else if(s=="reverse")
391 else if(!s.compare(0, 10, "set route "))
393 if(!s.compare(10, 3, "to "))
395 if(!s.compare(13, 7, "sensor "))
396 return Row(GOTO_SENSOR, lexical_cast<unsigned>(s.substr(20)));
398 return Row(GOTO_ZONE, s.substr(13));
400 return Row(ROUTE, s.substr(10));
403 throw invalid_argument("Timetable::Row::parse");
407 Timetable::Loader::Loader(Timetable &tt):
408 DataFile::ObjectLoader<Timetable>(tt)
410 add("arrive", &Loader::arrive);
411 add("goto_sensor", &Loader::goto_sensor);
412 add("goto_zone", &Loader::goto_zone);
413 add("route", &Loader::route);
414 add("speed", &Loader::speed);
415 add("reverse", &Loader::reverse);
416 add("travel_to", &Loader::travel_to);
417 add("travel_past", &Loader::travel_past);
418 add("wait", &Loader::wait);
419 add("wait_train", &Loader::wait_train);
420 add("wait_until", &Loader::wait_until);
423 add("goto", &Loader::goto_sensor_str);
424 add("travel", &Loader::travel_to);
427 void Timetable::Loader::arrive()
429 obj.rows.push_back(Row(ARRIVE));
432 void Timetable::Loader::goto_sensor(unsigned s)
434 obj.rows.push_back(Row(GOTO_SENSOR, s));
437 void Timetable::Loader::goto_sensor_str(const string &s)
439 if(!s.compare(0, 7, "sensor "))
440 obj.rows.push_back(Row(GOTO_SENSOR, lexical_cast<unsigned>(s.substr(7))));
443 void Timetable::Loader::goto_zone(const string &z)
445 obj.rows.push_back(Row(GOTO_ZONE, z));
448 void Timetable::Loader::route(const string &r)
450 obj.rows.push_back(Row(ROUTE, r));
453 void Timetable::Loader::reverse()
455 obj.rows.push_back(Row(REVERSE));
458 void Timetable::Loader::speed(unsigned s)
460 obj.rows.push_back(Row(SPEED, s));
463 void Timetable::Loader::travel_to(unsigned s)
465 obj.rows.push_back(Row(TRAVEL_TO, s));
468 void Timetable::Loader::travel_past(unsigned s)
470 obj.rows.push_back(Row(TRAVEL_PAST, s));
473 void Timetable::Loader::wait(unsigned t)
475 obj.rows.push_back(Row(WAIT_TIME, t));
478 void Timetable::Loader::wait_train(unsigned t, unsigned s)
480 Row row(WAIT_TRAIN, t);
481 row.params.push_back(s);
482 obj.rows.push_back(row);
485 void Timetable::Loader::wait_until(unsigned t, unsigned m)
487 Row row(WAIT_UNTIL, t);
488 row.params.push_back(m);
489 obj.rows.push_back(row);