X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Ftrain.cpp;h=7030f3814f69d53e6d62faf2bc9333aef0ea2c34;hb=9b536a7197c424c3fa3809063c56da324b56dab2;hp=6d963396e3baa81bb5cdffa26b8b0546adfad139;hpb=a277834b140eb5987419e2c335032c534a7184cd;p=r2c2.git diff --git a/source/libmarklin/train.cpp b/source/libmarklin/train.cpp index 6d96339..7030f38 100644 --- a/source/libmarklin/train.cpp +++ b/source/libmarklin/train.cpp @@ -13,39 +13,62 @@ Distributed under the GPL #include "catalogue.h" #include "driver.h" #include "layout.h" -#include "locotype.h" #include "route.h" -#include "simplephysics.h" +#include "simplecontroller.h" #include "timetable.h" +#include "trackiter.h" #include "tracktype.h" #include "train.h" #include "vehicle.h" +#include "vehicletype.h" using namespace std; using namespace Msp; +namespace { + +struct SetFlag +{ + bool &flag; + + SetFlag(bool &f): flag(f) { flag = true; } + ~SetFlag() { flag = false; } +}; + +} + + namespace Marklin { -Train::Train(Layout &l, const LocoType &t, unsigned a): +Train::Train(Layout &l, const VehicleType &t, unsigned a): layout(l), loco_type(t), address(a), + priority(0), + yielding_to(0), + cur_blocks_end(blocks.end()), + clear_blocks_end(blocks.end()), pending_block(0), - control(new AIControl(*this, new SimplePhysics)), + reserving(false), + advancing(false), + controller(new AIControl(*this, new SimpleController)), timetable(0), active(false), current_speed(0), speed_changing(false), reverse(false), functions(0), - route(0), - next_route(0), end_of_route(false), status("Unplaced"), travel_dist(0), pure_speed(false), - real_speed(15) + real_speed(15), + accurate_position(false), + overshoot_dist(false) { + if(!loco_type.is_locomotive()) + throw InvalidParameterValue("Initial vehicle must be a locomotive"); + vehicles.push_back(new Vehicle(layout, loco_type)); layout.add_train(*this); @@ -56,14 +79,20 @@ Train::Train(Layout &l, const LocoType &t, unsigned a): 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)); - control->signal_control_changed.connect(signal_control_changed); + 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)); } Train::~Train() { - delete control; + delete controller; delete timetable; for(vector::iterator i=vehicles.begin(); i!=vehicles.end(); ++i) delete *i; @@ -77,6 +106,40 @@ void Train::set_name(const string &n) signal_name_changed.emit(name); } +void Train::set_priority(int p) +{ + priority = p; +} + +void Train::yield_to(const Train &t) +{ + yielding_to = &t; +} + +void Train::add_vehicle(const VehicleType &vt) +{ + Vehicle *veh = new Vehicle(layout, vt); + vehicles.back()->attach_back(*veh); + vehicles.push_back(veh); +} + +void Train::remove_vehicle(unsigned i) +{ + if(i>=vehicles.size()) + throw InvalidParameterValue("Vehicle index out of range"); + if(i==0) + throw InvalidParameterValue("Can't remove the locomotive"); + delete vehicles[i]; + vehicles.erase(vehicles.begin()+i); + if(iattach_back(*vehicles[i]); +} + +unsigned Train::get_n_vehicles() const +{ + return vehicles.size(); +} + Vehicle &Train::get_vehicle(unsigned i) { if(i>=vehicles.size()) @@ -93,14 +156,14 @@ const Vehicle &Train::get_vehicle(unsigned i) const void Train::set_control(const string &n, float v) { - control->set_control(n, v); + controller->set_control(n, v); } void Train::set_active(bool a) { if(a==active) return; - if(!a && control->get_speed()) + if(!a && controller->get_speed()) throw InvalidState("Can't deactivate while moving"); active = a; @@ -126,80 +189,193 @@ void Train::set_function(unsigned func, bool state) layout.get_driver().set_loco_function(address+1, func-4, state); } +float Train::get_control(const string &ctrl) const +{ + return controller->get_control(ctrl).value; +} + +float Train::get_speed() const +{ + return controller->get_speed(); +} + bool Train::get_function(unsigned func) const { return (functions>>func)&1; } -void Train::set_route(const Route *r) +void Train::set_timetable(Timetable *tt) { - if(!rsv_blocks.empty()) - { - for(list::iterator i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i) - if(i->block->get_sensor_id()) - { - release_blocks(rsv_blocks, ++i, rsv_blocks.end()); - break; - } - } + delete timetable; + timetable = tt; +} - route = r; - next_route = 0; - end_of_route = false; +bool Train::set_route(const Route *r) +{ + free_noncritical_blocks(); - if(route) + Route *lead = 0; + if(r && !blocks.empty()) { - 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]; - if(!route->get_tracks().count(ep.track)) + TrackIter first = blocks.front().track_iter(); + TrackIter next = blocks.back().next().track_iter(); + if(!r->has_track(*next)) { - next_route = route; - route = Route::find(*ep.track, ep.track_ep, *next_route); + lead = Route::find(next, *r); + if(!lead) + return false; + create_lead_route(lead, lead); + routes.push_front(lead); } + else if(!r->has_track(*first)) + lead = create_lead_route(0, r); } - if(active) - reserve_more(); + routes.clear(); + if(lead) + routes.push_back(lead); + if(r) + routes.push_back(r); + end_of_route = false; - signal_route_changed.emit(route); + 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(); + + TrackIter next = blocks.back().next().track_iter(); + + Route *route = Route::find(next, to); + if(!route) + return false; + create_lead_route(route, route); + return set_route(route); +} + +bool Train::divert(Track &from) +{ + if(!from.get_turnout_id()) + throw InvalidParameterValue("Can't divert from a non-turnout"); + if(routes.empty()) + return false; + + unsigned path = 0; + unsigned entry = 0; + list::iterator route = routes.begin(); + + // Follow our routes to find out where we're entering the turnout + for(TrackLoopIter track = blocks.front().track_iter();;) + { + if(!advance_route(route, *track)) + return false; + + if(&*track==&from) + { + Block &block = track->get_block(); + if(block.get_train()==this && !free_block(block)) + return false; + + int route_path = route->route->get_turnout(from.get_turnout_id()); + + // Check that more than one path is available + unsigned ep_paths = track->get_type().get_endpoints()[track.entry()].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<route->get_path(*track)); + + if(!track || track.looped()) + return false; + } + + TrackIter track = TrackIter(&from, entry).next(path); + if(!track) + return false; + + set tracks; + for(list::iterator i=routes.begin(); i!=routes.end(); ++i) + tracks.insert(i->route->get_tracks().begin(), i->route->get_tracks().end()); + 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, TrackIter(&from, entry))) + return false; + + // Follow the diversion route until we get back to the original route + list::iterator end = routes.end(); + while(1) + { + for(list::iterator i=route; (end==routes.end() && i!=routes.end()); ++i) + if(i->route->has_track(*track)) + end = i; + + if(end!=routes.end()) + break; + else if(!diversion->has_track(*track)) + throw LogicError("Pathfinder returned a bad route"); + + track = track.next(diversion->get_path(*track)); + } + + if(route==end) + // We are rejoining the same route we diverted from, duplicate it + routes.insert(end, *route); else { - for(list::iterator i=rsv_blocks.begin(); (i!=rsv_blocks.end() && !last); ++i) - if(i->block->get_sensor_id()) - last = &*i; + ++route; + routes.erase(route, end); } + routes.insert(end, RouteRef(diversion.release(), from.get_turnout_id())); - BlockRef next = last->next(); - const Block::Endpoint &ep = next.block->get_endpoints()[next.entry]; + return true; +} - set_route(Route::find(*ep.track, ep.track_ep, to)); +const Route *Train::get_route() const +{ + if(routes.empty()) + return 0; + return routes.front().route; } void Train::place(Block &block, unsigned entry) { - if(control->get_speed()) + 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; if(!block.reserve(this)) { @@ -207,123 +383,162 @@ 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]; - vehicles.front()->place(bep.track, bep.track_ep, 0, Vehicle::BACK_BUFFER); + vehicles.back()->place(*bep.track, bep.track_ep, 0, Vehicle::BACK_BUFFER); } } +void Train::unplace() +{ + if(controller->get_speed()) + throw InvalidState("Must be stopped before unplacing"); + + release_blocks(); + + set_active(false); + accurate_position = false; + + for(vector::iterator i=vehicles.begin(); i!=vehicles.end(); ++i) + (*i)->unplace(); + + set_status("Unplaced"); +} + bool Train::free_block(Block &block) { + float margin = 10*layout.get_catalogue().get_scale(); + if(get_reserved_distance_until(&block, false)get_braking_distance()*1.3+margin) + 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; } return false; } -int Train::get_entry_to_block(Block &block) const +void Train::free_noncritical_blocks() { - 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; - return -1; -} + if(blocks.empty()) + return; -float Train::get_reserved_distance() const -{ - Vehicle &veh = *(reverse ? vehicles.back() : vehicles.front()); - const VehicleType &vtype = veh.get_type(); + if(controller->get_speed()==0) + { + release_blocks(cur_blocks_end, blocks.end()); + return; + } - Track *track = veh.get_track(); - unsigned entry = veh.get_entry(); + float margin = 10*layout.get_catalogue().get_scale(); + float min_dist = controller->get_braking_distance()*1.3+margin; - float result = -vtype.get_length()/2; - if(reverse) + Vehicle &veh = *(reverse ? vehicles.back() : vehicles.front()); + + TrackIter track(veh.get_track(), veh.get_entry()); + BlockList::iterator block = blocks.begin(); + bool in_rsv = false; + while(block!=blocks.end() && !(*block)->has_track(*track)) { - entry = track->traverse(entry); - result += veh.get_offset(); + ++block; + if(block==cur_blocks_end) + in_rsv = true; } + + float dist = veh.get_offset(); + if(reverse) + track.reverse(); else - result -= veh.get_offset(); + dist = track->get_type().get_path_length(track->get_active_path())-dist; + dist -= veh.get_type().get_length()/2; - bool first = true; - list::const_iterator block = cur_blocks.begin(); + bool nsens = 0; while(1) { - if(!first || !reverse) - result += track->get_type().get_path_length(track->get_active_path()); - first = false; + track = track.next(); - unsigned exit = track->traverse(entry); - Track *next = track->get_link(exit); - - while(!block->block->get_tracks().count(next)) + if(!(*block)->has_track(*track)) { ++block; - if(block==cur_blocks.end()) - block = rsv_blocks.begin(); - if(block==rsv_blocks.end()) - return result; + if(block==cur_blocks_end) + in_rsv = true; + if(block==blocks.end()) + return; + + if(dist>min_dist && nsens>0) + { + release_blocks(block, blocks.end()); + return; + } + + if(in_rsv && (*block)->get_sensor_id()) + ++nsens; } - entry = next->get_endpoint_by_link(*track); - track = next; + dist += track->get_type().get_path_length(track->get_active_path()); } } +int Train::get_entry_to_block(Block &block) const +{ + for(BlockList::const_iterator i=blocks.begin(); i!=blocks.end(); ++i) + if(i->block()==&block) + return i->entry(); + return -1; +} + +float Train::get_reserved_distance() const +{ + return get_reserved_distance_until(0, false); +} + void Train::tick(const Time::TimeStamp &t, const Time::TimeDelta &dt) { if(!active && stop_timeout && t>=stop_timeout) { - release_blocks(rsv_blocks); + release_blocks(cur_blocks_end, blocks.end()); end_of_route = false; stop_timeout = Time::TimeStamp(); } + Driver &driver = layout.get_driver(); + if(timetable) timetable->tick(t); - control->tick(dt); - float speed = control->get_speed(); - unsigned speed_notch = find_speed(abs(speed)); + controller->tick(dt); + float speed = controller->get_speed(); + unsigned speed_notch = find_speed(speed); - if(speed && (speed<0)!=reverse) + if(controller->get_reverse()!=reverse) { - layout.get_driver().set_loco_reverse(address, speed<0); - reverse = speed<0; + 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) + if(speed_notch!=current_speed && !speed_changing && !driver.is_halted() && driver.get_power()) { speed_changing = true; - layout.get_driver().set_loco_speed(address, speed_notch); + driver.set_loco_speed(address, speed_notch); pure_speed = false; @@ -342,41 +557,78 @@ 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); if(ok) { - float d = get_real_speed(current_speed)*(dt/Time::sec); + SetFlag setf(advancing); vehicle.advance(reverse ? -d : d); } + else if(accurate_position) + { + overshoot_dist += d; + if(overshoot_dist>40*layout.get_catalogue().get_scale()) + { + layout.emergency(name+" has not arrived at sensor"); + accurate_position = false; + } + } } - else if(end_of_route) + else if(end_of_route && cur_blocks_end==blocks.end()) + { + set_active(false); + signal_arrived.emit(); set_route(0); + } + + if(!blocks.empty() && !blocks.front()->get_sensor_id()) + { + float dist = get_reserved_distance_until(&*blocks.front(), true); + + if(dist>10*layout.get_catalogue().get_scale()) + { + blocks.front()->reserve(0); + blocks.pop_front(); + } + } } void Train::save(list &st) const { st.push_back((DataFile::Statement("name"), name)); + + st.push_back((DataFile::Statement("priority"), priority)); + + for(vector::const_iterator i=vehicles.begin(); i!=vehicles.end(); ++i) + if(i!=vehicles.begin()) + st.push_back((DataFile::Statement("vehicle"), (*i)->get_type().get_article_number())); + for(unsigned i=0; i<=14; ++i) if(real_speed[i].weight) st.push_back((DataFile::Statement("real_speed"), i, real_speed[i].speed, real_speed[i].weight)); - if(!cur_blocks.empty()) + if(!blocks.empty() && cur_blocks_end!=blocks.begin()) { - list 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(route) - st.push_back((DataFile::Statement("route"), route->get_name())); + if(!routes.empty()) + { + list::const_iterator i = routes.begin(); + for(; (i!=routes.end() && i->route->is_temporary()); ++i) ; + if(i!=routes.end()) + st.push_back((DataFile::Statement("route"), i->route->get_name())); + } if(timetable) { @@ -386,12 +638,18 @@ void Train::save(list &st) const } } +void Train::control_changed(const Controller::Control &ctrl) +{ + signal_control_changed.emit(ctrl.name, ctrl.value); +} + void Train::loco_speed_event(unsigned addr, unsigned speed, bool) { if(addr==address) { current_speed = speed; speed_changing = false; + pure_speed = false; } } @@ -415,244 +673,460 @@ 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; - for(i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i) - if(i->block->get_sensor_id() && i->block->get_sensor_id()!=addr) - break; + BlockList::iterator end; + unsigned result = 0; + for(end=cur_blocks_end; end!=blocks.end(); ++end) + if((*end)->get_sensor_id()) + { + if((*end)->get_sensor_id()!=addr) + { + if(result==0) + result = 2; + else if(result==1) + break; + } + else if(result==0) + result = 1; + else if(result==2) + result = 3; + } - if(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) { - RealSpeed &rs = real_speed[current_speed]; - rs.add(travel_dist/travel_time_secs, travel_time_secs); + if(current_speed) + { + RealSpeed &rs = real_speed[current_speed]; + 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) + 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(); pure_speed = true; + accurate_position = true; + overshoot_dist = 0; // Check if we've reached the next route - if(next_route) + if(routes.size()>1) { - const set &rtracks = next_route->get_tracks(); - for(list::iterator j=rsv_blocks.begin(); j!=i; ++j) - if(rtracks.count(j->block->get_endpoints()[j->entry].track)) + const set &rtracks = (++routes.begin())->route->get_tracks(); + for(BlockList::iterator j=cur_blocks_end; j!=end; ++j) + if(rtracks.count((*j)->get_endpoints()[j->entry()].track)) { - route = next_route; - next_route = 0; + routes.pop_front(); // XXX Exceptions? - signal_route_changed.emit(route); + signal_route_changed.emit(routes.front().route); break; } } // 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) - { - unsigned nsens = reserve_more(); - if(!nsens && end_of_route) - signal_arrived.emit(); - } + reserve_more(); } + else if(result==3) + layout.emergency("Sensor for "+name+" triggered out of order"); } else { + 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) - if(i->block->get_sensor_id()) + BlockList::iterator end = blocks.begin(); + for(BlockList::iterator i=blocks.begin(); i!=cur_blocks_end; ++i) + { + if((*i)->has_track(*veh.get_track())) + break; + 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 + { end = i; + ++end; + } } + } - if(end!=cur_blocks.begin()) - { + if(end!=blocks.begin() && end!=cur_blocks_end) // Free blocks up to the last inactive sensor - ++end; - 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) - { - 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)) - reserve_more(); - } + for(list::iterator i=blocks.begin(); i!=blocks.end(); ++i) + if((*i)->get_turnout_id()==track.get_turnout_id()) + { + if(!reserving) + check_turnout_paths(false); + else if(i==clear_blocks_end) + ++clear_blocks_end; + } +} + +void Train::halt_event(bool h) +{ + if(h) + accurate_position = false; } void Train::block_reserved(const Block &block, const Train *train) { - if(&block==pending_block && !train) + if(&block==pending_block && !train && !reserving) reserve_more(); } unsigned Train::reserve_more() { - BlockRef *last = 0; - if(!rsv_blocks.empty()) - last = &rsv_blocks.back(); - else if(!cur_blocks.empty()) - last = &cur_blocks.back(); - if(!last) + if(!active || blocks.empty()) return 0; + BlockIter start = blocks.back(); + pending_block = 0; - // See how many sensor blocks we already have + // See how many sensor blocks and how much track we already have unsigned nsens = 0; - for(list::const_iterator i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i) - if(i->block->get_sensor_id()) + float dist = 0; + for(BlockList::const_iterator i=cur_blocks_end; i!=blocks.end(); ++i) + { + if((*i)->get_sensor_id()) ++nsens; + if(nsens>0) + dist += (*i)->get_path_length(i->entry()); + } if(end_of_route) return nsens; - const Route *cur_route = 0; - if(route) - { - const set &tracks = last->block->get_tracks(); - for(set::const_iterator i=tracks.begin(); (cur_route!=route && i!=tracks.end()); ++i) - { - if(route->get_tracks().count(*i)) - cur_route = route; - else if(next_route && next_route->get_tracks().count(*i)) - cur_route = next_route; - } - } + list::iterator cur_route = routes.begin(); + advance_route(cur_route, *start.track_iter()); - bool got_more = false; - BlockRef *good = last; + float approach_margin = 50*layout.get_catalogue().get_scale(); + float min_dist = controller->get_braking_distance()*1.3+approach_margin*2; + + BlockIter block = start; + list::iterator good_end = blocks.end(); + Track *divert_track = 0; + bool try_divert = false; unsigned good_sens = nsens; - while(good_sens<3) + float good_dist = dist; + Train *blocking_train = 0; + BlockList contested_blocks; + + SetFlag setf(reserving); + + while(good_sens<3 || good_distblock->traverse(last->entry); - Block *link = last->block->get_link(exit); - if(!link) + BlockIter last = block; + block = block.next(cur_route!=routes.end() ? cur_route->route : 0); + if(!block) break; - int entry = link->get_endpoint_by_link(*last->block); - if(entry<0) - throw LogicError("Block links are inconsistent!"); + TrackIter track = block.track_iter(); - const Block::Endpoint &entry_ep = link->get_endpoints()[entry]; - - if(cur_route) + if(cur_route!=routes.end()) { - if(cur_route!=next_route && next_route && next_route->get_tracks().count(entry_ep.track)) - cur_route = next_route; - else if(!cur_route->get_tracks().count(entry_ep.track)) + if(!advance_route(cur_route, *track)) { // Keep the blocks if we arrived at the end of the route - good = last; - good_sens = nsens; - end_of_route = true; + if(!blocking_train) + { + good_end = blocks.end(); + good_sens = nsens; + good_dist = dist; + end_of_route = true; + } break; } } - else if(route && route->get_tracks().count(entry_ep.track)) - cur_route = route; + else if(!routes.empty() && routes.front().route->has_track(*track)) + cur_route = routes.begin(); - if(!link->reserve(this)) + if(block->get_endpoints().size()<2) { - // If we found another train and it's not headed straight for us, we can keep the blocks we got - int other_entry = link->get_train()->get_entry_to_block(*link); - if(other_entry<0) - throw LogicError("Block reservation inconsistency"); - if(static_cast(entry)!=link->traverse(other_entry)) + if(!blocking_train) { - good = last; + good_end = blocks.end(); good_sens = nsens; + good_dist = dist; } - pending_block = link; break; } - if(link->get_turnout_id()) + if(blocking_train) { - const Endpoint &track_ep = entry_ep.track->get_type().get_endpoints()[entry_ep.track_ep]; + if(block->get_train()!=blocking_train) + { + if(blocking_train->free_block(*contested_blocks.back())) + { + // Roll back and start actually reserving the blocks + block = blocks.back(); + cur_route = routes.begin(); + advance_route(cur_route, *block.track_iter().track()); + if(blocking_train->get_priority()==priority) + blocking_train->yield_to(*this); + blocking_train = 0; + continue; + } + else + { + yield_to(*blocking_train); + pending_block = contested_blocks.front().block(); + try_divert = divert_track; + break; + } + } + else + { + contested_blocks.push_back(block); + continue; + } + } - // Keep the blocks reserved so far, as either us or the other train can diverge - good = last; - good_sens = nsens; + 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 = block->get_train(); + int other_entry = other_train->get_entry_to_block(*block); + if(other_entry<0) + throw LogicError("Block reservation inconsistency"); - // Figure out what path we'd like to take on the turnout - int path = -1; - if(cur_route) - path = cur_route->get_turnout(link->get_turnout_id()); - if(path<0) - path = entry_ep.track->get_active_path(); - if(!((track_ep.paths>>path)&1)) + 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()) { - for(unsigned i=0; track_ep.paths>>i; ++i) - if((track_ep.paths>>i)&1) - path = i; + /* The other train is not coming to the blocks we're holding, so we + can keep them. */ + good_end = blocks.end(); + good_sens = nsens; + good_dist = dist; } - if(path!=static_cast(entry_ep.track->get_active_path())) + int other_prio = other_train->get_priority(); + + if(!entry_conflict && !exit_conflict && other_priofree_block(*block)) + reserved = block->reserve(this); + } + else if(other_train!=yielding_to && (other_prioreserve(0); - pending_block = link; - entry_ep.track->set_active_path(path); + /* A lesser priority train is coming at us, we must ask it to free + enough blocks to get clear of it to avoid a potential deadlock */ + blocking_train = other_train; + contested_blocks.clear(); + contested_blocks.push_back(block); + continue; + } + else if(divert_track && (entry_conflict || exit_conflict)) + // We are blocked, but there's a diversion possibility + try_divert = true; + + if(!reserved) + { + pending_block = &*block; break; } } - rsv_blocks.push_back(BlockRef(link, entry)); - last = &rsv_blocks.back(); - if(last->block->get_sensor_id()) + if(block->get_turnout_id()) { - ++nsens; - got_more = true; + const TrackType::Endpoint &track_ep = track->get_type().get_endpoints()[track.entry()]; + bool multiple_paths = (track_ep.paths&(track_ep.paths-1)); + + if(multiple_paths || !last->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_end = blocks.end(); + good_sens = nsens; + good_dist = dist; + } + + 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 = &*track; } + + if(!contested_blocks.empty() && contested_blocks.front()==block) + contested_blocks.pop_front(); + + 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 += block->get_path_length(block.entry()); } // Unreserve blocks that were not good - while(!rsv_blocks.empty() && last!=good) + release_blocks(good_end, blocks.end()); + + 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 + while(cur_blocks_end!=blocks.end() && !(*cur_blocks_end)->get_sensor_id()) + ++cur_blocks_end; + + if(try_divert && divert(*divert_track)) + return reserve_more(); + + return good_sens; +} + +void Train::check_turnout_paths(bool set) +{ + if(clear_blocks_end==blocks.end()) + return; + + list::iterator route = routes.begin(); + + for(list::iterator i=clear_blocks_end; i!=blocks.end(); ++i) { - last->block->reserve(0); - rsv_blocks.erase(--rsv_blocks.end()); - if(!rsv_blocks.empty()) - last = &rsv_blocks.back(); + advance_route(route, *i->track_iter()); + + if((*i)->get_turnout_id()) + { + TrackIter track = i->track_iter(); + const TrackType::Endpoint &track_ep = track->get_type().get_endpoints()[track.entry()]; + + int path = -1; + for(list::iterator j=route; (path<0 && j!=routes.end()); ++j) + path = j->route->get_turnout((*i)->get_turnout_id()); + if(path<0) + path = track->get_active_path(); + if(!(track_ep.paths&(1<::iterator j = i; + if(++j!=blocks.end()) + { + TrackIter rev = j->track_iter().flip(); + unsigned mask = rev->get_type().get_endpoints()[rev.entry()].paths&track_ep.paths; + for(path=0; mask>1; mask>>=1, ++path) ; + } + else + return; + } + + if(path!=static_cast(track->get_active_path())) + { + if(set) + track->set_active_path(path); + + continue; + } + } + + if(i==clear_blocks_end) + ++clear_blocks_end; } +} - return good_sens; +float Train::get_reserved_distance_until(const Block *until_block, bool back) const +{ + if(blocks.empty()) + return 0; + + Vehicle &veh = *(reverse!=back ? vehicles.back() : vehicles.front()); + const VehicleType &vtype = veh.get_type(); + + TrackIter track(veh.get_track(), veh.get_entry()); + if(!track) // XXX Probably unnecessary + return 0; + + BlockList::const_iterator block = blocks.begin(); + while(block!=clear_blocks_end && !(*block)->has_track(*track)) + ++block; + if(block==clear_blocks_end || &**block==until_block) + return 0; + + float result = veh.get_offset(); + if(reverse!=back) + track = track.reverse(); + else + result = track->get_type().get_path_length(track->get_active_path())-result; + result -= vtype.get_length()/2; + + while(1) + { + track = track.next(); + if(!track) + break; + + if(!(*block)->has_track(*track)) + { + if(back) + { + if(block==blocks.begin()) + break; + --block; + } + else + { + ++block; + if(block==clear_blocks_end) + break; + } + + if(&**block==until_block) + break; + } + + result += track->get_type().get_path_length(track->get_active_path()); + } + + return result; } float Train::get_real_speed(unsigned i) const @@ -692,9 +1166,11 @@ unsigned Train::find_speed(float real) const unsigned low = 0; unsigned high = 0; + unsigned last = 0; for(unsigned i=0; (!high && i<=14); ++i) if(real_speed[i].weight) { + last = i; if(real_speed[i].speed(low*real/real_speed[low].speed), 14U); + { + if(real) + return 3; + else + return 0; + } + return min(min(static_cast(low*real/real_speed[low].speed), 14U), last+3); } float f = (real-real_speed[low].speed)/(real_speed[high].speed-real_speed[low].speed); @@ -724,48 +1205,125 @@ 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); } } -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, Track &track) +{ + while(iter!=routes.end() && !iter->route->has_track(track)) + ++iter; + if(iter==routes.end()) + return false; -Train::BlockRef::BlockRef(Block *b, unsigned e): - block(b), - entry(e) -{ } + list::iterator next = iter; + ++next; + if(next!=routes.end() && next->diversion && next->route->has_track(track)) + iter = next; + + return true; +} + +Route *Train::create_lead_route(Route *lead, const Route *target) +{ + if(!lead) + { + lead = new Route(layout); + lead->set_name("Lead"); + lead->set_temporary(true); + } + + set tracks; + for(BlockList::iterator i=blocks.begin(); i!=blocks.end(); ++i) + { + const set &btracks = (*i)->get_tracks(); + for(set::const_iterator j=btracks.begin(); j!=btracks.end(); ++j) + if(!target || !target->has_track(**j)) + tracks.insert(*j); + } + + lead->add_tracks(tracks); -Train::BlockRef Train::BlockRef::next() const + return lead; +} + +bool Train::is_valid_diversion(const Route &diversion, const TrackIter &from) { - Block *blk = block->get_endpoints()[block->traverse(entry)].link; - if(!blk) - throw InvalidState("At end of line"); + float diversion_len = 0; + TrackLoopIter track1 = from; + while(diversion.has_track(*track1)) + { + unsigned path = diversion.get_path(*track1); + diversion_len += track1->get_type().get_path_length(path); + + track1 = track1.next(path); + + if(track1.looped()) + return false; + } + + list::iterator route = routes.begin(); + if(!advance_route(route, *from)) + return false; + + float route_len = 0; + TrackLoopIter track2 = from; + while(1) + { + 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(ok) + break; + + if(track2.looped()) + return false; + + if(!advance_route(route, *track2)) + return false; + } - 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_len(t), - prev_block(0) + prev_block(0), + blocks_valid(true) { add("block", &Loader::block); add("block_hint", &Loader::block_hint); add("name", &Loader::name); + add("priority", &Train::priority); add("real_speed", &Loader::real_speed); add("route", &Loader::route); add("timetable", &Loader::timetable); add("vehicle", &Loader::vehicle); } +void Train::Loader::finish() +{ + if(!obj.blocks.empty()) + { + TrackIter track = obj.blocks.front().track_iter(); + float offset = 2*obj.layout.get_catalogue().get_scale(); + obj.vehicles.back()->place(*track, track.entry(), offset, Vehicle::BACK_BUFFER); + + obj.set_status("Stopped"); + } +} + void Train::Loader::block(unsigned id) { - Block &blk = obj.layout.get_block(id); + if(!blocks_valid) + return; + + Block *blk; + try + { + blk = &obj.layout.get_block(id); + } + catch(const KeyError &) + { + blocks_valid = false; + return; + } + int entry = -1; if(prev_block) - entry = blk.get_endpoint_by_link(*prev_block); + entry = blk->get_endpoint_by_link(*prev_block); if(entry<0) entry = 0; - blk.reserve(&obj); - obj.cur_blocks.push_back(BlockRef(&blk, entry)); - obj.set_status("Stopped"); - const Block::Endpoint &bep = blk.get_endpoints()[entry]; - obj.vehicles.back()->place(bep.track, bep.track_ep, 0, Vehicle::BACK_BUFFER); + blk->reserve(&obj); + obj.blocks.push_back(BlockIter(blk, entry)); - if(blk.get_sensor_id()) - obj.layout.get_driver().set_sensor(blk.get_sensor_id(), true); + if(blk->get_sensor_id()) + obj.layout.get_driver().set_sensor(blk->get_sensor_id(), true); - prev_block = &blk; + prev_block = blk; } void Train::Loader::block_hint(unsigned id) { - prev_block = &obj.layout.get_block(id); + try + { + prev_block = &obj.layout.get_block(id); + } + catch(const KeyError &) + { + blocks_valid = false; + } } void Train::Loader::name(const string &n) @@ -842,9 +1431,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);