X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Ftrain.cpp;h=0084f38fab8d4764c79c15873134b9f4a31d3049;hb=975ea87cc7be179618b06291cb2506a2523cad1f;hp=ccb7cc25bbf8714706e549c3bb4cee6ff07e2b32;hpb=4c787ea0227ee891b3f7dac3d071e607fbf8d109;p=r2c2.git diff --git a/source/libmarklin/train.cpp b/source/libmarklin/train.cpp index ccb7cc2..0084f38 100644 --- a/source/libmarklin/train.cpp +++ b/source/libmarklin/train.cpp @@ -16,6 +16,7 @@ Distributed under the GPL #include "route.h" #include "simplecontroller.h" #include "timetable.h" +#include "trackiter.h" #include "tracktype.h" #include "train.h" #include "vehicle.h" @@ -39,19 +40,22 @@ struct SetFlag namespace Marklin { -Train::Train(Layout &l, const VehicleType &t, unsigned a): +Train::Train(Layout &l, const VehicleType &t, unsigned a, const string &p): layout(l), loco_type(t), address(a), + protocol(p), priority(0), yielding_to(0), + cur_blocks_end(blocks.end()), + clear_blocks_end(blocks.end()), pending_block(0), reserving(false), advancing(false), controller(new AIControl(*this, new SimpleController)), timetable(0), active(false), - current_speed(0), + current_speed_step(0), speed_changing(false), reverse(false), functions(0), @@ -59,7 +63,7 @@ Train::Train(Layout &l, const VehicleType &t, unsigned a): status("Unplaced"), travel_dist(0), pure_speed(false), - real_speed(15), + real_speed(layout.get_driver().get_protocol_speed_steps(protocol)+1), accurate_position(false), overshoot_dist(false) { @@ -70,16 +74,20 @@ Train::Train(Layout &l, const VehicleType &t, unsigned a): layout.add_train(*this); - layout.get_driver().add_loco(address); + layout.get_driver().add_loco(address, protocol); layout.get_driver().signal_loco_speed.connect(sigc::mem_fun(this, &Train::loco_speed_event)); layout.get_driver().signal_loco_function.connect(sigc::mem_fun(this, &Train::loco_func_event)); layout.signal_block_reserved.connect(sigc::mem_fun(this, &Train::block_reserved)); layout.get_driver().signal_sensor.connect(sigc::mem_fun(this, &Train::sensor_event)); - layout.get_driver().signal_turnout.connect(sigc::mem_fun(this, &Train::turnout_event)); layout.get_driver().signal_halt.connect(sigc::mem_fun(this, &Train::halt_event)); + const set &tracks = layout.get_tracks(); + for(set::const_iterator i=tracks.begin(); i!=tracks.end(); ++i) + if((*i)->get_turnout_id()) + (*i)->signal_path_changed.connect(sigc::hide(sigc::bind(sigc::mem_fun(this, &Train::turnout_path_changed), sigc::ref(**i)))); + controller->signal_control_changed.connect(sigc::mem_fun(this, &Train::control_changed)); } @@ -203,56 +211,59 @@ void Train::set_timetable(Timetable *tt) timetable = tt; } -void Train::set_route(const Route *r) +bool Train::set_route(const Route *r) { free_noncritical_blocks(); - routes.clear(); - if(r) - routes.push_back(r); - end_of_route = false; - - if(r && !cur_blocks.empty()) + Route *lead = 0; + if(r && !blocks.empty()) { - BlockRef &first = cur_blocks.front(); - BlockRef &last = (rsv_blocks.empty() ? cur_blocks.back() : rsv_blocks.back()); - BlockRef next = last.next(); - const Block::Endpoint &first_ep = first.block->get_endpoints()[first.entry]; - const Block::Endpoint &next_ep = next.block->get_endpoints()[next.entry]; - if(!r->get_tracks().count(next_ep.track)) + TrackIter first = blocks.front().track_iter(); + TrackIter next = blocks.back().next().track_iter(); + if(!r->has_track(*next)) { - Route *lead = Route::find(*next_ep.track, next_ep.track_ep, *r); + lead = Route::find(next, *r); + if(!lead) + return false; create_lead_route(lead, lead); routes.push_front(lead); } - else if(!r->get_tracks().count(first_ep.track)) - routes.push_front(create_lead_route(0, r)); + else if(!r->has_track(*first)) + lead = create_lead_route(0, r); } + routes.clear(); + if(lead) + routes.push_back(lead); + if(r) + routes.push_back(r); + end_of_route = false; + reserve_more(); signal_route_changed.emit(get_route()); + + return true; } -void Train::go_to(const Track &to) +bool Train::go_to(Track &to) { - for(list::const_iterator i=cur_blocks.begin(); i!=cur_blocks.end(); ++i) - if(i->block->get_tracks().count(const_cast(&to))) + for(BlockList::const_iterator i=blocks.begin(); i!=cur_blocks_end; ++i) + if((*i)->has_track(to)) { signal_arrived.emit(); - set_route(0); - return; + return set_route(0); } free_noncritical_blocks(); - BlockRef &last = (rsv_blocks.empty() ? cur_blocks.back() : rsv_blocks.back()); - BlockRef next = last.next(); - const Block::Endpoint &ep = next.block->get_endpoints()[next.entry]; + TrackIter next = blocks.back().next().track_iter(); - Route *route = Route::find(*ep.track, ep.track_ep, to); + Route *route = Route::find(next, to); + if(!route) + return false; create_lead_route(route, route); - set_route(route); + return set_route(route); } bool Train::divert(Track &from) @@ -262,102 +273,79 @@ bool Train::divert(Track &from) if(routes.empty()) return false; - int path = -1; - unsigned from_ep = 0; + unsigned path = 0; + unsigned entry = 0; list::iterator route = routes.begin(); - Block *block = cur_blocks.back().block; - unsigned entry = cur_blocks.back().entry; - set visited; // Follow our routes to find out where we're entering the turnout - while(1) + for(TrackLoopIter track = blocks.front().track_iter();;) { - Block *link = block->get_link(block->traverse(entry, route->route)); - entry = link->get_endpoint_by_link(*block); - block = link; - - const Block::Endpoint &entry_ep = block->get_endpoints()[entry]; - - if(visited.count(entry_ep.track)) - return false; - visited.insert(entry_ep.track); - - if(!advance_route(route, *entry_ep.track)) + if(!advance_route(route, *track)) return false; - if(entry_ep.track==&from) + if(&*track==&from) { - if(block->get_train()==this && !free_block(*block)) + Block &block = track->get_block(); + if(block.get_train()==this && !free_block(block)) return false; - from_ep = entry_ep.track_ep; - path = route->route->get_turnout(from.get_turnout_id()); - break; - } - } + int route_path = route->route->get_turnout(from.get_turnout_id()); - // Check that more than one path is available - unsigned ep_paths = from.get_type().get_endpoints()[from_ep].paths; - if(!(ep_paths&(ep_paths-1))) - return false; + // Check that more than one path is available + unsigned ep_paths = track.endpoint().paths; + if(!(ep_paths&(ep_paths-1))) + return false; - // Choose some other path - for(int i=0; ep_paths>>i; ++i) - if((ep_paths&(1<>i; ++i) + if((ep_paths&(1<route->get_path(*track)); + + if(!track || track.looped()) + return false; + } + + TrackIter track = TrackIter(&from, entry).next(path); if(!track) return false; - unsigned ep = track->get_endpoint_by_link(from); - - set tracks; + set tracks; for(list::iterator i=routes.begin(); i!=routes.end(); ++i) tracks.insert(i->route->get_tracks().begin(), i->route->get_tracks().end()); - Route *diversion = 0; - try - { - diversion = Route::find(*track, ep, tracks); - } - catch(const Msp::Exception &) - { + RefPtr diversion = Route::find(track, tracks); + if(!diversion) return false; - } diversion->set_name("Diversion"); diversion->add_track(from); diversion->set_turnout(from.get_turnout_id(), path); - if(!is_valid_diversion(*diversion, from, from_ep)) - { - delete diversion; + if(!is_valid_diversion(*diversion, TrackIter(&from, entry))) return false; - } // Follow the diversion route until we get back to the original route list::iterator end = routes.end(); while(1) { - path = 0; - if(track->get_turnout_id()) - path = diversion->get_turnout(track->get_turnout_id()); - Track *next = track->get_link(track->traverse(ep, path)); - for(list::iterator i=route; (end==routes.end() && i!=routes.end()); ++i) - if(i->route->get_tracks().count(next)) + if(i->route->has_track(*track)) end = i; if(end!=routes.end()) break; - else if(!diversion->get_tracks().count(next)) - throw Exception("Pathfinder returned a bad route"); + else if(!diversion->has_track(*track)) + throw LogicError("Pathfinder returned a bad route"); - ep = next->get_endpoint_by_link(*track); - track = next; + track = track.next(diversion->get_path(*track)); } if(route==end) @@ -368,7 +356,7 @@ bool Train::divert(Track &from) ++route; routes.erase(route, end); } - routes.insert(end, RouteRef(diversion, from.get_turnout_id())); + routes.insert(end, RouteRef(diversion.release(), from.get_turnout_id())); return true; } @@ -385,8 +373,7 @@ void Train::place(Block &block, unsigned entry) if(controller->get_speed()) throw InvalidState("Must be stopped before placing"); - release_blocks(rsv_blocks); - release_blocks(cur_blocks); + release_blocks(); set_active(false); accurate_position = false; @@ -397,18 +384,15 @@ void Train::place(Block &block, unsigned entry) return; } - cur_blocks.push_back(BlockRef(&block, entry)); + blocks.push_back(BlockIter(&block, entry)); if(reverse) { - unsigned exit = block.traverse(entry); - const Block::Endpoint &bep = block.get_endpoints()[exit]; - Track *track = bep.track->get_link(bep.track_ep); - unsigned ep = track->get_endpoint_by_link(*bep.track); - vehicles.front()->place(*track, ep, 0, Vehicle::FRONT_BUFFER); + TrackIter track = BlockIter(&block, entry).reverse().track_iter(); + vehicles.front()->place(*track, track.entry(), 0, Vehicle::FRONT_BUFFER); } else { - const Block::Endpoint &bep = block.get_endpoints()[entry]; + const Block::Endpoint &bep = block.get_endpoint(entry); vehicles.back()->place(*bep.track, bep.track_ep, 0, Vehicle::BACK_BUFFER); } } @@ -418,8 +402,7 @@ void Train::unplace() if(controller->get_speed()) throw InvalidState("Must be stopped before unplacing"); - release_blocks(rsv_blocks); - release_blocks(cur_blocks); + release_blocks(); set_active(false); accurate_position = false; @@ -437,16 +420,16 @@ bool Train::free_block(Block &block) return false; unsigned nsens = 0; - for(list::iterator i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i) + for(BlockList::iterator i=cur_blocks_end; i!=blocks.end(); ++i) { - if(i->block==&block) + if(i->block()==&block) { if(nsens<1) return false; - release_blocks(rsv_blocks, i, rsv_blocks.end()); + release_blocks(i, blocks.end()); return true; } - else if(i->block->get_sensor_id()) + else if((*i)->get_sensor_id()) ++nsens; } @@ -455,12 +438,12 @@ bool Train::free_block(Block &block) void Train::free_noncritical_blocks() { - if(cur_blocks.empty() || rsv_blocks.empty()) + if(blocks.empty()) return; if(controller->get_speed()==0) { - release_blocks(rsv_blocks); + release_blocks(cur_blocks_end, blocks.end()); return; } @@ -469,23 +452,19 @@ void Train::free_noncritical_blocks() Vehicle &veh = *(reverse ? vehicles.back() : vehicles.front()); - Track *track = veh.get_track(); - list::iterator block = cur_blocks.begin(); + TrackIter track(veh.get_track(), veh.get_entry()); + BlockList::iterator block = blocks.begin(); bool in_rsv = false; - while(block!=rsv_blocks.end() && !block->block->get_tracks().count(track)) + while(block!=blocks.end() && !(*block)->has_track(*track)) { ++block; - if(block==cur_blocks.end()) - { - block = rsv_blocks.begin(); + if(block==cur_blocks_end) in_rsv = true; - } } - unsigned entry = veh.get_entry(); float dist = veh.get_offset(); if(reverse) - entry = track->traverse(entry); + track.reverse(); else dist = track->get_type().get_path_length(track->get_active_path())-dist; dist -= veh.get_type().get_length()/2; @@ -493,28 +472,23 @@ void Train::free_noncritical_blocks() bool nsens = 0; while(1) { - Track *next = track->get_link(track->traverse(entry)); - entry = next->get_endpoint_by_link(*track); - track = next; + track = track.next(); - if(!block->block->get_tracks().count(track)) + if(!(*block)->has_track(*track)) { ++block; - if(block==cur_blocks.end()) - { - block = rsv_blocks.begin(); + if(block==cur_blocks_end) in_rsv = true; - } - if(block==rsv_blocks.end()) + if(block==blocks.end()) return; if(dist>min_dist && nsens>0) { - release_blocks(rsv_blocks, block, rsv_blocks.end()); + release_blocks(block, blocks.end()); return; } - if(in_rsv && block->block->get_sensor_id()) + if(in_rsv && (*block)->get_sensor_id()) ++nsens; } @@ -524,12 +498,9 @@ void Train::free_noncritical_blocks() int Train::get_entry_to_block(Block &block) const { - for(list::const_iterator i=cur_blocks.begin(); i!=cur_blocks.end(); ++i) - if(i->block==&block) - return i->entry; - for(list::const_iterator i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i) - if(i->block==&block) - return i->entry; + for(BlockList::const_iterator i=blocks.begin(); i!=blocks.end(); ++i) + if(i->block()==&block) + return i->entry(); return -1; } @@ -542,8 +513,7 @@ void Train::tick(const Time::TimeStamp &t, const Time::TimeDelta &dt) { if(!active && stop_timeout && t>=stop_timeout) { - release_blocks(rsv_blocks); - end_of_route = false; + release_blocks(cur_blocks_end, blocks.end()); stop_timeout = Time::TimeStamp(); } @@ -553,26 +523,26 @@ void Train::tick(const Time::TimeStamp &t, const Time::TimeDelta &dt) timetable->tick(t); controller->tick(dt); float speed = controller->get_speed(); - unsigned speed_notch = find_speed(speed); + unsigned speed_step = find_speed_step(speed); if(controller->get_reverse()!=reverse) { reverse = controller->get_reverse(); driver.set_loco_reverse(address, reverse); - release_blocks(rsv_blocks); - reverse_blocks(cur_blocks); + release_blocks(cur_blocks_end, blocks.end()); + reverse_blocks(blocks); reserve_more(); } - if(speed_notch!=current_speed && !speed_changing && !driver.is_halted() && driver.get_power()) + if(speed_step!=current_speed_step && !speed_changing && !driver.is_halted() && driver.get_power()) { speed_changing = true; - driver.set_loco_speed(address, speed_notch); + driver.set_loco_speed(address, speed_step); pure_speed = false; - if(speed_notch) + if(speed_step) set_status(format("Traveling %d kmh", get_travel_speed())); else set_status("Waiting"); @@ -587,10 +557,14 @@ void Train::tick(const Time::TimeStamp &t, const Time::TimeDelta &dt) Track *track = vehicle.get_track(); bool ok = false; - for(list::const_iterator i=cur_blocks.begin(); (!ok && i!=cur_blocks.end()); ++i) - ok = i->block->get_tracks().count(track); + for(BlockList::const_iterator i=blocks.begin(); (!ok && i!=cur_blocks_end); ++i) + ok = (*i)->has_track(*track); - float d = get_real_speed(current_speed)*(dt/Time::sec); + float d; + if(real_speed.size()>1) + d = get_real_speed(current_speed_step)*(dt/Time::sec); + else + d = speed*(dt/Time::sec); if(ok) { SetFlag setf(advancing); @@ -606,21 +580,21 @@ void Train::tick(const Time::TimeStamp &t, const Time::TimeDelta &dt) } } } - else if(end_of_route && rsv_blocks.empty()) + else if(end_of_route && cur_blocks_end==blocks.end()) { set_active(false); signal_arrived.emit(); set_route(0); } - if(!cur_blocks.empty() && !cur_blocks.front().block->get_sensor_id()) + if(!blocks.empty() && !blocks.front()->get_sensor_id()) { - float dist = get_reserved_distance_until(cur_blocks.front().block, true); + float dist = get_reserved_distance_until(&*blocks.front(), true); if(dist>10*layout.get_catalogue().get_scale()) { - cur_blocks.front().block->reserve(0); - cur_blocks.pop_front(); + blocks.front()->reserve(0); + blocks.pop_front(); } } } @@ -635,21 +609,21 @@ void Train::save(list &st) const if(i!=vehicles.begin()) st.push_back((DataFile::Statement("vehicle"), (*i)->get_type().get_article_number())); - for(unsigned i=0; i<=14; ++i) + for(unsigned i=0; i blocks = cur_blocks; + BlockList blks(blocks.begin(), BlockList::const_iterator(cur_blocks_end)); if(reverse) - reverse_blocks(blocks); + reverse_blocks(blks); - Block *prev = blocks.front().block->get_endpoints()[blocks.front().entry].link; + BlockIter prev = blks.front().flip(); st.push_back((DataFile::Statement("block_hint"), prev->get_id())); - for(list::const_iterator i=blocks.begin(); i!=blocks.end(); ++i) - st.push_back((DataFile::Statement("block"), i->block->get_id())); + for(BlockList::const_iterator i=blks.begin(); i!=blks.end(); ++i) + st.push_back((DataFile::Statement("block"), (*i)->get_id())); } if(!routes.empty()) @@ -677,7 +651,7 @@ void Train::loco_speed_event(unsigned addr, unsigned speed, bool) { if(addr==address) { - current_speed = speed; + current_speed_step = speed; speed_changing = false; pure_speed = false; } @@ -703,12 +677,12 @@ void Train::sensor_event(unsigned addr, bool state) if(state) { // Find the first sensor block from our reserved blocks that isn't this sensor - list::iterator i; + BlockList::iterator end; unsigned result = 0; - for(i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i) - if(i->block->get_sensor_id()) + for(end=cur_blocks_end; end!=blocks.end(); ++end) + if((*end)->get_sensor_id()) { - if(i->block->get_sensor_id()!=addr) + if((*end)->get_sensor_id()!=addr) { if(result==0) result = 2; @@ -721,39 +695,36 @@ void Train::sensor_event(unsigned addr, bool state) result = 3; } - if(result==1 && i!=rsv_blocks.begin()) + if(result==1) { // Compute speed and update related state float travel_time_secs = (Time::now()-last_entry_time)/Time::sec; if(pure_speed) { - if(current_speed) + if(current_speed_step>0) { - RealSpeed &rs = real_speed[current_speed]; + RealSpeed &rs = real_speed[current_speed_step]; rs.add(travel_dist/travel_time_secs, travel_time_secs); } set_status(format("Traveling %d kmh", get_travel_speed())); } travel_dist = 0; - float block_len; - for(list::iterator j=rsv_blocks.begin(); j!=i; ++j) + for(BlockList::iterator j=cur_blocks_end; j!=end; ++j) { - j->block->traverse(j->entry, &block_len); - travel_dist += block_len; + travel_dist += (*j)->get_path_length(j->entry()); - if(j->block->get_sensor_id()==addr && !advancing) + if((*j)->get_sensor_id()==addr && !advancing) { - const Block::Endpoint &bep = j->block->get_endpoints()[j->entry]; + TrackIter track = j->track_iter(); if(reverse) { - Track *track = bep.track->get_link(bep.track_ep); - unsigned ep = track->get_endpoint_by_link(*bep.track); - vehicles.back()->place(*track, ep, 0, Vehicle::BACK_AXLE); + track = track.flip(); + vehicles.back()->place(*track, track.entry(), 0, Vehicle::BACK_AXLE); } else - vehicles.front()->place(*bep.track, bep.track_ep, 0, Vehicle::FRONT_AXLE); + vehicles.front()->place(*track, track.entry(), 0, Vehicle::FRONT_AXLE); } } last_entry_time = Time::now(); @@ -764,9 +735,9 @@ void Train::sensor_event(unsigned addr, bool state) // Check if we've reached the next route if(routes.size()>1) { - const set &rtracks = (++routes.begin())->route->get_tracks(); - for(list::iterator j=rsv_blocks.begin(); j!=i; ++j) - if(rtracks.count(j->block->get_endpoints()[j->entry].track)) + const Route &route = *(++routes.begin())->route; + for(BlockList::iterator j=cur_blocks_end; j!=end; ++j) + if(route.has_track(*j->track_iter())) { routes.pop_front(); // XXX Exceptions? @@ -776,7 +747,7 @@ void Train::sensor_event(unsigned addr, bool state) } // Move blocks up to the next sensor to our current blocks - cur_blocks.splice(cur_blocks.end(), rsv_blocks, rsv_blocks.begin(), i); + cur_blocks_end = end; // Try to get more blocks if we're moving if(active) @@ -790,14 +761,14 @@ void Train::sensor_event(unsigned addr, bool state) const Vehicle &veh = *(reverse ? vehicles.front() : vehicles.back()); // Find the first sensor in our current blocks that's still active - list::iterator end = cur_blocks.begin(); - for(list::iterator i=cur_blocks.begin(); i!=cur_blocks.end(); ++i) + BlockList::iterator end = blocks.begin(); + for(BlockList::iterator i=blocks.begin(); i!=cur_blocks_end; ++i) { - if(i->block->get_tracks().count(veh.get_track())) + if((*i)->has_track(*veh.get_track())) break; - if(i->block->get_sensor_id()) + if((*i)->get_sensor_id()) { - if(layout.get_driver().get_sensor(i->block->get_sensor_id())) + if(layout.get_driver().get_sensor((*i)->get_sensor_id())) break; else { @@ -807,26 +778,17 @@ void Train::sensor_event(unsigned addr, bool state) } } - if(end!=cur_blocks.begin() && end!=cur_blocks.end()) + if(end!=blocks.begin() && end!=cur_blocks_end) // Free blocks up to the last inactive sensor - release_blocks(cur_blocks, cur_blocks.begin(), end); + release_blocks(blocks.begin(), end); } } -void Train::turnout_event(unsigned addr, bool) +void Train::turnout_path_changed(Track &track) { - if(pending_block && (!pending_block->get_train() || pending_block->get_train()==this)) - { - unsigned pending_addr = pending_block->get_turnout_id(); - bool double_addr = (*pending_block->get_tracks().begin())->get_type().is_double_address(); - if(addr==pending_addr || (double_addr && addr==pending_addr+1)) - { - if(reserving) - pending_block = 0; - else - reserve_more(); - } - } + for(list::iterator i=blocks.begin(); i!=blocks.end(); ++i) + if((*i)->get_turnout_id()==track.get_turnout_id() && !reserving) + check_turnout_paths(false); } void Train::halt_event(bool h) @@ -841,114 +803,88 @@ void Train::block_reserved(const Block &block, const Train *train) reserve_more(); } -unsigned Train::reserve_more() +void Train::reserve_more() { - if(!active) - return 0; + if(!active || blocks.empty() || end_of_route) + return; - BlockRef *start = 0; - if(!rsv_blocks.empty()) - start = &rsv_blocks.back(); - else if(!cur_blocks.empty()) - start = &cur_blocks.back(); - if(!start) - return 0; + BlockIter start = blocks.back(); pending_block = 0; // See how many sensor blocks and how much track we already have unsigned nsens = 0; float dist = 0; - for(list::const_iterator i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i) + for(BlockList::const_iterator i=cur_blocks_end; i!=blocks.end(); ++i) { - if(i->block->get_sensor_id()) + if((*i)->get_sensor_id()) ++nsens; if(nsens>0) - { - float length = 0; - i->block->traverse(i->entry, &length); - dist += length; - } + dist += (*i)->get_path_length(i->entry()); } - - if(end_of_route) - return nsens; list::iterator cur_route = routes.begin(); - advance_route(cur_route, *start->block->get_endpoints()[start->entry].track); + advance_route(cur_route, *start.track_iter()); float approach_margin = 50*layout.get_catalogue().get_scale(); float min_dist = controller->get_braking_distance()*1.3+approach_margin*2; - BlockRef *last = start; - BlockRef *good = start; + BlockIter block = start; + list::iterator good_end = blocks.end(); Track *divert_track = 0; bool try_divert = false; - unsigned good_sens = nsens; - float good_dist = dist; Train *blocking_train = 0; - std::list contested_blocks; + BlockList contested_blocks; SetFlag setf(reserving); - while(good_sens<3 || good_distroute : 0); + if(!block || block->get_endpoints().size()<2) { - const Route *route = (cur_route!=routes.end() ? cur_route->route : 0); - unsigned exit = last->block->traverse(last->entry, route, &length); - link = last->block->get_link(exit); - } - if(!link) + if(!blocking_train) + good_end = blocks.end(); break; + } - int entry = link->get_endpoint_by_link(*last->block); - if(entry<0) - throw LogicError("Block links are inconsistent!"); - - const Block::Endpoint &entry_ep = link->get_endpoints()[entry]; + TrackIter track = block.track_iter(); if(cur_route!=routes.end()) { - if(!advance_route(cur_route, *entry_ep.track)) + if(!advance_route(cur_route, *track)) { // Keep the blocks if we arrived at the end of the route if(!blocking_train) { - good = last; - good_sens = nsens; - good_dist = dist; + good_end = blocks.end(); end_of_route = true; } break; } } - else if(!routes.empty() && routes.front().route->get_tracks().count(entry_ep.track)) - cur_route = routes.begin(); - if(link->get_endpoints().size()<2) + if(block->get_turnout_id() && !last->get_turnout_id()) { - if(!blocking_train) - { - good = last; - good_sens = nsens; - good_dist = dist; - } - break; + /* We can keep the blocks if we arrive at a turnout from a non-turnout + block. Having a turnout block as our last reserved block is not good + as it would limit our diversion possibilities for little benefit. */ + good_end = blocks.end(); + if(nsens>=3 && dist>=min_dist) + break; } if(blocking_train) { - if(link->get_train()!=blocking_train) + if(block->get_train()!=blocking_train) { - if(blocking_train->free_block(*contested_blocks.back().block)) + if(blocking_train->free_block(*contested_blocks.back())) { // Roll back and start actually reserving the blocks - last = &rsv_blocks.back(); + block = blocks.back(); cur_route = routes.begin(); - advance_route(cur_route, *last->block->get_endpoints()[last->entry].track); + advance_route(cur_route, *block.track_iter().track()); if(blocking_train->get_priority()==priority) blocking_train->yield_to(*this); blocking_train = 0; @@ -957,41 +893,38 @@ unsigned Train::reserve_more() else { yield_to(*blocking_train); - pending_block = contested_blocks.front().block; + pending_block = contested_blocks.front().block(); try_divert = divert_track; break; } } else { - contested_blocks.push_back(BlockRef(link, entry)); - last = &contested_blocks.back(); + contested_blocks.push_back(block); continue; } } - bool reserved = link->reserve(this); + bool reserved = block->reserve(this); if(!reserved) { /* We've found another train. If it wants to exit the block from the same endpoint we're trying to enter from or the other way around, treat it as coming towards us. Otherwise treat it as going in the same direction. */ - Train *other_train = link->get_train(); - int other_entry = other_train->get_entry_to_block(*link); + Train *other_train = block->get_train(); + int other_entry = other_train->get_entry_to_block(*block); if(other_entry<0) throw LogicError("Block reservation inconsistency"); - bool entry_conflict = (static_cast(entry)==link->traverse(other_entry)); - bool exit_conflict = (link->traverse(entry)==static_cast(other_entry)); - if(!entry_conflict && !last->block->get_turnout_id()) - { + unsigned exit = block.reverse().entry(); + unsigned other_exit = BlockIter(block.block(), other_entry).reverse().entry(); + bool entry_conflict = (block.entry()==other_exit); + bool exit_conflict = (exit==static_cast(other_entry)); + if(!entry_conflict && !last->get_turnout_id()) /* The other train is not coming to the blocks we're holding, so we can keep them. */ - good = last; - good_sens = nsens; - good_dist = dist; - } + good_end = blocks.end(); int other_prio = other_train->get_priority(); @@ -999,8 +932,8 @@ unsigned Train::reserve_more() { /* Ask a lesser priority train going to the same direction to free the block for us */ - if(other_train->free_block(*link)) - reserved = link->reserve(this); + if(other_train->free_block(*block)) + reserved = block->reserve(this); } else if(other_train!=yielding_to && (other_prioget_turnout_id()) + if(block->get_turnout_id()) { - const Endpoint &track_ep = entry_ep.track->get_type().get_endpoints()[entry_ep.track_ep]; + const TrackType::Endpoint &track_ep = track.endpoint(); bool multiple_paths = (track_ep.paths&(track_ep.paths-1)); - if(multiple_paths || !last->block->get_turnout_id()) - { - /* We can keep the blocks reserved so far if we are facing the - points or if there was no turnout immediately before this one. - With multiple successive turnouts (as is common in crossovers) it's - best to hold at one we can divert from. */ - good = last; - good_sens = nsens; - good_dist = dist; - } - - // Figure out what path we'd like to take on the turnout - int path = -1; - for(list::iterator i=cur_route; (path<0 && i!=routes.end()); ++i) - path = i->route->get_turnout(link->get_turnout_id()); - if(path<0) - path = entry_ep.track->get_active_path(); - if(!(track_ep.paths&(1<>i; ++i) - if(track_ep.paths&(1<(entry_ep.track->get_active_path())) - { - // The turnout is set to wrong path - switch and wait for it - pending_block = link; - entry_ep.track->set_active_path(path); - if(pending_block) - { - link->reserve(0); - break; - } - } - - if(multiple_paths && cur_route!=routes.end() && cur_route->diversion!=link->get_turnout_id()) + if(multiple_paths && cur_route!=routes.end() && cur_route->diversion!=block->get_turnout_id()) /* There's multiple paths to be taken and we are on a route - take note of the diversion possibility */ - divert_track = entry_ep.track; + divert_track = &*track; } - if(!contested_blocks.empty() && contested_blocks.front().block==link) + if(!contested_blocks.empty() && contested_blocks.front()==block) contested_blocks.pop_front(); - rsv_blocks.push_back(BlockRef(link, entry)); - last = &rsv_blocks.back(); - if(last->block->get_sensor_id()) + blocks.push_back(block); + + if(cur_blocks_end==blocks.end()) + --cur_blocks_end; + if(clear_blocks_end==blocks.end()) + --clear_blocks_end; + if(good_end==blocks.end()) + --good_end; + + if(block->get_sensor_id()) ++nsens; if(nsens>0) - dist += length; + dist += block->get_path_length(block.entry()); } // Unreserve blocks that were not good - while(!rsv_blocks.empty() && &rsv_blocks.back()!=good) - { - rsv_blocks.back().block->reserve(0); - rsv_blocks.pop_back(); - } + release_blocks(good_end, blocks.end()); - if(!rsv_blocks.empty() && &rsv_blocks.back()!=start) + if(blocks.back()!=start) // We got some new blocks, so no longer need to yield yielding_to = 0; + check_turnout_paths(true); + // Make any sensorless blocks at the beginning immediately current - list::iterator i; - for(i=rsv_blocks.begin(); (i!=rsv_blocks.end() && !i->block->get_sensor_id()); ++i) ; - if(i!=rsv_blocks.begin()) - cur_blocks.splice(cur_blocks.end(), rsv_blocks, rsv_blocks.begin(), i); + while(cur_blocks_end!=clear_blocks_end && !(*cur_blocks_end)->get_sensor_id()) + ++cur_blocks_end; if(try_divert && divert(*divert_track)) - return reserve_more(); + reserve_more(); +} + +void Train::check_turnout_paths(bool set) +{ + if(clear_blocks_end==blocks.end()) + return; + + for(list::iterator i=clear_blocks_end; i!=blocks.end(); ++i) + { + if((*i)->get_turnout_id()) + { + TrackIter track = i->track_iter(); + const TrackType::Endpoint &track_ep = track.endpoint(); + + unsigned path = 0; + list::iterator j = i; + if(++j!=blocks.end()) + { + TrackIter rev = j->track_iter().flip(); + unsigned mask = rev.endpoint().paths&track_ep.paths; + for(path=0; mask>1; mask>>=1, ++path) ; + } + else + return; + + if(path!=track->get_active_path()) + { + if(set) + track->set_active_path(path); + + /* Check again, in case the driver was able to service the request + instantly */ + if(!set || path!=track->get_active_path()) + continue; + } + } - return good_sens; + if(i==clear_blocks_end) + ++clear_blocks_end; + } } float Train::get_reserved_distance_until(const Block *until_block, bool back) const { - if(cur_blocks.empty()) + if(blocks.empty()) return 0; Vehicle &veh = *(reverse!=back ? vehicles.back() : vehicles.front()); const VehicleType &vtype = veh.get_type(); - Track *track = veh.get_track(); - if(!track) + TrackIter track(veh.get_track(), veh.get_entry()); + if(!track) // XXX Probably unnecessary return 0; - list::const_iterator block = cur_blocks.begin(); - while(block!=rsv_blocks.end() && !block->block->get_tracks().count(track)) - { + BlockList::const_iterator block = blocks.begin(); + while(block!=clear_blocks_end && !(*block)->has_track(*track)) ++block; - if(block==cur_blocks.end()) - { - if(back) - return 0; - block = rsv_blocks.begin(); - } - } - if(block==rsv_blocks.end() || block->block==until_block) + if(block==clear_blocks_end || &**block==until_block) return 0; - unsigned entry = veh.get_entry(); - float result = veh.get_offset(); if(reverse!=back) - entry = track->traverse(entry); + track = track.reverse(); else result = track->get_type().get_path_length(track->get_active_path())-result; result -= vtype.get_length()/2; while(1) { - if(track->get_type().get_endpoints().size()<2) + track = track.next(); + if(!track) break; - Track *next = track->get_link(track->traverse(entry)); - - if(!block->block->get_tracks().count(next)) + if(!(*block)->has_track(*track)) { if(back) { - if(block==cur_blocks.begin()) + if(block==blocks.begin()) break; --block; } else { ++block; - if(block==cur_blocks.end()) - block = rsv_blocks.begin(); - if(block==rsv_blocks.end()) + if(block==clear_blocks_end) break; } - if(block->block==until_block) + if(&**block==until_block) break; } - entry = next->get_endpoint_by_link(*track); - track = next; - result += track->get_type().get_path_length(track->get_active_path()); } @@ -1178,6 +1099,8 @@ float Train::get_reserved_distance_until(const Block *until_block, bool back) co float Train::get_real_speed(unsigned i) const { + if(i==0) + return 0; if(real_speed[i].weight) return real_speed[i].speed; @@ -1186,7 +1109,7 @@ float Train::get_real_speed(unsigned i) const for(low=i; low>0; --low) if(real_speed[low].weight) break; - for(high=i; high<14; ++high) + for(high=i; high(low*real/real_speed[low].speed), 14U), last+3); + return min(min(static_cast(low*real/real_speed[low].speed), real_speed.size()-1), last+limit); } float f = (real-real_speed[low].speed)/(real_speed[high].speed-real_speed[low].speed); @@ -1241,7 +1167,7 @@ unsigned Train::find_speed(float real) const float Train::get_travel_speed() const { - float speed = get_real_speed(current_speed); + float speed = get_real_speed(current_speed_step); float scale = layout.get_catalogue().get_scale(); return static_cast(round(speed/scale*3.6/5))*5; } @@ -1252,38 +1178,46 @@ void Train::set_status(const string &s) signal_status_changed.emit(s); } -void Train::release_blocks(list &blocks) +void Train::release_blocks() { - release_blocks(blocks, blocks.begin(), blocks.end()); + release_blocks(blocks.begin(), blocks.end()); } -void Train::release_blocks(list &blocks, list::iterator begin, list::iterator end) +void Train::release_blocks(BlockList::iterator begin, BlockList::iterator end) { while(begin!=end) { - Block *block = begin->block; + if(begin==cur_blocks_end) + cur_blocks_end = end; + if(begin==clear_blocks_end) + clear_blocks_end = end; + + Block &block = **begin; blocks.erase(begin++); - block->reserve(0); + block.reserve(0); + + if(begin==blocks.end()) + end_of_route = false; } } -void Train::reverse_blocks(list &blocks) const +void Train::reverse_blocks(BlockList &blks) const { - blocks.reverse(); - for(list::iterator i=blocks.begin(); i!=blocks.end(); ++i) - i->entry = i->block->traverse(i->entry); + blks.reverse(); + for(BlockList::iterator i=blks.begin(); i!=blks.end(); ++i) + *i = i->reverse(); } -bool Train::advance_route(list::iterator &iter, const Track &track) +bool Train::advance_route(list::iterator &iter, Track &track) { - while(iter!=routes.end() && !iter->route->get_tracks().count(&track)) + while(iter!=routes.end() && !iter->route->has_track(track)) ++iter; if(iter==routes.end()) return false; list::iterator next = iter; ++next; - if(next!=routes.end() && next->diversion && next->route->get_tracks().count(&track)) + if(next!=routes.end() && next->diversion && next->route->has_track(track)) iter = next; return true; @@ -1298,16 +1232,13 @@ Route *Train::create_lead_route(Route *lead, const Route *target) lead->set_temporary(true); } - set tracks; - for(list::iterator i=cur_blocks.begin(); i!=rsv_blocks.end(); ) + set tracks; + for(BlockList::iterator i=blocks.begin(); i!=blocks.end(); ++i) { - const set &btracks = i->block->get_tracks(); + const set &btracks = (*i)->get_tracks(); for(set::const_iterator j=btracks.begin(); j!=btracks.end(); ++j) - if(!target || !target->get_tracks().count(*j)) + if(!target || !target->has_track(**j)) tracks.insert(*j); - - if(++i==cur_blocks.end()) - i = rsv_blocks.begin(); } lead->add_tracks(tracks); @@ -1315,76 +1246,51 @@ Route *Train::create_lead_route(Route *lead, const Route *target) return lead; } -bool Train::is_valid_diversion(const Route &diversion, const Track &from, unsigned from_ep) +bool Train::is_valid_diversion(const Route &diversion, const TrackIter &from) { float diversion_len = 0; - const Track *track = &from; - unsigned ep = from_ep; - while(diversion.get_tracks().count(track)) + TrackLoopIter track1 = from; + while(diversion.has_track(*track1)) { - unsigned path = 0; - if(track->get_turnout_id()) - path = diversion.get_turnout(track->get_turnout_id()); - diversion_len += track->get_type().get_path_length(path); + unsigned path = diversion.get_path(*track1); + diversion_len += track1->get_type().get_path_length(path); - const Track *next = track->get_link(track->traverse(ep, path)); - ep = next->get_endpoint_by_link(*track); - track = next; + track1 = track1.next(path); - if(track==&from) + if(track1.looped()) return false; } list::iterator route = routes.begin(); - if(!advance_route(route, from)) + if(!advance_route(route, *from)) return false; - set visited; float route_len = 0; - track = &from; - ep = from_ep; + TrackLoopIter track2 = from; while(1) { - unsigned path = 0; - if(track->get_turnout_id()) - path = route->route->get_turnout(track->get_turnout_id()); - route_len += track->get_type().get_path_length(path); + unsigned path = route->route->get_path(*track2); + route_len += track2->get_type().get_path_length(path); + + bool ok = (track2!=from && diversion.has_track(*track2)); + + track2 = track2.next(path); - if(track!=&from && diversion.get_tracks().count(track)) + if(ok) break; - if(visited.count(track)) + if(track2.looped()) return false; - visited.insert(track); - - const Track *next = track->get_link(track->traverse(ep, path)); - ep = next->get_endpoint_by_link(*track); - track = next; - if(!advance_route(route, *track)) + if(!advance_route(route, *track2)) return false; } - return diversion_lenget_endpoints()[block->traverse(entry)].link; - if(!blk) - throw InvalidState("At end of line"); - - int ep = blk->get_endpoint_by_link(*block); - if(ep<0) - throw LogicError("Block links are inconsistent"); + // Must end up at the same place through both routes + if(track2!=track1) + return false; - return BlockRef(blk, ep); + return diversion_lenget_endpoints()[blkref.entry]; + TrackIter track = obj.blocks.front().track_iter(); float offset = 2*obj.layout.get_catalogue().get_scale(); - obj.vehicles.back()->place(*bep.track, bep.track_ep, offset, Vehicle::BACK_BUFFER); + obj.vehicles.back()->place(*track, track.entry(), offset, Vehicle::BACK_BUFFER); obj.set_status("Stopped"); } @@ -1457,7 +1362,7 @@ void Train::Loader::block(unsigned id) entry = 0; blk->reserve(&obj); - obj.cur_blocks.push_back(BlockRef(blk, entry)); + obj.blocks.push_back(BlockIter(blk, entry)); if(blk->get_sensor_id()) obj.layout.get_driver().set_sensor(blk->get_sensor_id(), true); @@ -1484,6 +1389,8 @@ void Train::Loader::name(const string &n) void Train::Loader::real_speed(unsigned i, float speed, float weight) { + if(i>=obj.real_speed.size()) + return; obj.real_speed[i].speed = speed; obj.real_speed[i].weight = weight; } @@ -1502,9 +1409,9 @@ void Train::Loader::timetable() load_sub(*obj.timetable); } -void Train::Loader::vehicle(unsigned n) +void Train::Loader::vehicle(ArticleNumber art_nr) { - const VehicleType &vtype = obj.layout.get_catalogue().get_vehicle(n); + const VehicleType &vtype = obj.layout.get_catalogue().get_vehicle(art_nr); Vehicle *veh = new Vehicle(obj.layout, vtype); obj.vehicles.back()->attach_back(*veh); obj.vehicles.push_back(veh);