]> git.tdb.fi Git - r2c2.git/blobdiff - source/libmarklin/train.cpp
Emit various signals from Train when it's loaded
[r2c2.git] / source / libmarklin / train.cpp
index 5524a5d77c6a2740e24d4894a811dcac75441b44..56832c837542d9447acb3022a8d655e1a8786939 100644 (file)
@@ -11,6 +11,7 @@ Distributed under the GPL
 #include <msp/time/utils.h>
 #include "control.h"
 #include "except.h"
+#include "layout.h"
 #include "route.h"
 #include "tracktype.h"
 #include "trafficmanager.h"
@@ -48,6 +49,8 @@ Train::Train(TrafficManager &tm, Locomotive &l):
                i->second->signal_path_changing.connect(sigc::bind(sigc::mem_fun(this, &Train::turnout_path_changing), i->second));
                i->second->signal_path_changed.connect(sigc::bind(sigc::mem_fun(this, &Train::turnout_path_changed), i->second));
        }
+
+       trfc_mgr.signal_block_reserved.connect(sigc::mem_fun(this, &Train::block_reserved));
 }
 
 void Train::set_name(const string &n)
@@ -66,8 +69,11 @@ void Train::set_speed(unsigned speed)
        target_speed = speed;
        if(!target_speed)
        {
+               pending_block = 0;
                trfc_mgr.get_control().set_timer(3*Time::sec).signal_timeout.connect(
-                       sigc::bind_return(sigc::bind(sigc::mem_fun(this, &Train::release_blocks), sigc::ref(rsv_blocks)), false));
+                       sigc::bind_return(sigc::bind(
+                               sigc::mem_fun(this, static_cast<void (Train::*)(list<BlockRef> &)>(&Train::release_blocks)),
+                               sigc::ref(rsv_blocks)), false));
        }
        else
                reserve_more();
@@ -118,9 +124,7 @@ bool Train::free_block(Block &block)
                {
                        if(nsens<1)
                                return false;
-                       for(list<BlockRef>::iterator j=i; j!=rsv_blocks.end(); ++j)
-                               j->block->reserve(0);
-                       rsv_blocks.erase(i, rsv_blocks.end());
+                       release_blocks(rsv_blocks, i, rsv_blocks.end());
                        update_speed();
                        return true;
                }
@@ -142,11 +146,8 @@ int Train::get_entry_to_block(Block &block) const
        return -1;
 }
 
-void Train::tick(const Time::TimeStamp &t, const Time::TimeDelta &dt)
+void Train::tick(const Time::TimeStamp &, const Time::TimeDelta &dt)
 {
-       if(try_reserve && t>try_reserve)
-               reserve_more();
-
        if(cur_track)
        {
                unsigned path = 0;
@@ -186,16 +187,27 @@ void Train::save(list<DataFile::Statement> &st) const
        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(route)
+               st.push_back((DataFile::Statement("route"), route->get_name()));
+
+       if(!cur_blocks.empty())
+       {
+               list<BlockRef> blocks = cur_blocks;
+               if(loco.get_reverse())
+                       reverse_blocks(blocks);
+
+               Block *prev = blocks.front().block->get_endpoints()[blocks.front().entry].link;
+               st.push_back((DataFile::Statement("block_hint"), prev->get_id()));
+
+               for(list<BlockRef>::const_iterator i=blocks.begin(); i!=blocks.end(); ++i)
+                       st.push_back((DataFile::Statement("block"), i->block->get_id()));
+       }
 }
 
 void Train::locomotive_reverse_changed(bool)
 {
-       for(list<BlockRef>::iterator i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i)
-               i->block->reserve(0);
-       rsv_blocks.clear();
-       cur_blocks.reverse();
-       for(list<BlockRef>::iterator i=cur_blocks.begin(); i!=cur_blocks.end(); ++i)
-               i->entry = i->block->traverse(i->entry);
+       release_blocks(rsv_blocks);
+       reverse_blocks(cur_blocks);
        reserve_more();
 
        if(cur_track)
@@ -214,6 +226,7 @@ void Train::sensor_event(bool state, Sensor *sensor)
 
        if(state)
        {
+               // Find the first sensor block from our reserved blocks that isn't this sensor
                list<BlockRef>::iterator i;
                for(i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i)
                        if(i->block->get_sensor_id() && i->block->get_sensor_id()!=addr)
@@ -221,6 +234,7 @@ void Train::sensor_event(bool state, Sensor *sensor)
 
                if(i!=rsv_blocks.begin())
                {
+                       // Compute speed and update related state
                        float travel_time_secs = (Time::now()-last_entry_time)/Time::sec;
                        travel_speed = static_cast<int>(round(travel_dist/travel_time_secs*87*3.6/5))*5;
 
@@ -236,37 +250,40 @@ void Train::sensor_event(bool state, Sensor *sensor)
                        {
                                j->block->traverse(j->entry, &block_len);
                                travel_dist += block_len;
+
+                               if(j->block->get_sensor_id()==addr)
+                                       set_position(j->block->get_endpoints()[j->entry]);
                        }
                        last_entry_time = Time::now();
                        pure_speed = true;
 
+                       // Move blocks up to the next sensor to our current blocks
                        cur_blocks.splice(cur_blocks.end(), rsv_blocks, rsv_blocks.begin(), i);
-               }
-
-               for(i=cur_blocks.begin(); i!=cur_blocks.end(); ++i)
-                       if(i->block->get_sensor_id()==addr)
-                               set_position(i->block->get_endpoints()[i->entry]);
 
-               if(target_speed && reserve_more()<2)
-                       update_speed();
+                       // Try to get more blocks if we're moving
+                       if(target_speed && reserve_more()<2)
+                               update_speed();
+               }
        }
        else
        {
+               // Find the first sensor in our current blocks that's still active
+               list<BlockRef>::iterator end = cur_blocks.begin();
                for(list<BlockRef>::iterator i=cur_blocks.begin(); i!=cur_blocks.end(); ++i)
-                       if(unsigned b_addr = i->block->get_sensor_id())
+                       if(i->block->get_sensor_id())
                        {
-                               if(b_addr==addr)
-                               {
-                                       ++i;
-                                       for(list<BlockRef>::iterator j=cur_blocks.begin(); j!=i; ++j)
-                                               j->block->reserve(0);
-                                       cur_blocks.erase(cur_blocks.begin(), i);
-                               }
-                               break;
+                               if(trfc_mgr.get_control().get_sensor(i->block->get_sensor_id()).get_state())
+                                       break;
+                               else
+                                       end = i;
                        }
-
-               if(target_speed && pending_block && addr==pending_block->get_sensor_id())
-                       reserve_more();
+               
+               if(end!=cur_blocks.begin())
+               {
+                       // Free blocks up to the last inactive sensor
+                       ++end;
+                       release_blocks(cur_blocks, cur_blocks.begin(), end);
+               }
        }
 }
 
@@ -297,11 +314,7 @@ void Train::turnout_path_changed(unsigned, Turnout *turnout)
        for(list<BlockRef>::iterator i=rsv_blocks.begin(); i!=rsv_blocks.end(); ++i)
                if(i->block->get_turnout_id()==tid)
                {
-                       while(i!=rsv_blocks.end())
-                       {
-                               i->block->reserve(0);
-                               i = rsv_blocks.erase(i);
-                       }
+                       release_blocks(rsv_blocks, i, rsv_blocks.end());
                        reserve_more();
                        return;
                }
@@ -310,6 +323,12 @@ void Train::turnout_path_changed(unsigned, Turnout *turnout)
                reserve_more();
 }
 
+void Train::block_reserved(const Block &block, const Train *train)
+{
+       if(&block==pending_block && !train)
+               reserve_more();
+}
+
 unsigned Train::reserve_more()
 {
        BlockRef *last = 0;
@@ -340,11 +359,15 @@ unsigned Train::reserve_more()
                        break;
 
                int entry = link->get_endpoint_by_link(*last->block);
+               if(entry<0)
+                       throw LogicError("Block links are inconsistent!");
                if(!link->reserve(this))
                {
-                       // If we found another train going in the same direction as us, we can keep the blocks we got
+                       // 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==entry || link->traverse(entry)==link->traverse(other_entry))
+                       if(other_entry<0)
+                               throw LogicError("Block reservation inconsistency");
+                       if(static_cast<unsigned>(entry)!=link->traverse(other_entry))
                        {
                                good = last;
                                good_sens = nsens;
@@ -358,12 +381,9 @@ unsigned Train::reserve_more()
                        const Block::Endpoint &ep = link->get_endpoints()[entry];
                        const Endpoint &track_ep = ep.track->get_type().get_endpoints()[ep.track_ep];
 
-                       if(track_ep.paths&(track_ep.paths-1))
-                       {
-                               // We're facing the points - keep the blocks reserved so far
-                               good = last;
-                               good_sens = nsens;
-                       }
+                       // Keep the blocks reserved so far, as either us or the other train can diverge
+                       good = last;
+                       good_sens = nsens;
 
                        Turnout &turnout = trfc_mgr.get_control().get_turnout(link->get_turnout_id());
 
@@ -419,7 +439,6 @@ void Train::update_speed()
        if(!target_speed)
        {
                loco.set_speed(0);
-               try_reserve = Time::TimeStamp();
                set_status("Stopped");
        }
        else
@@ -434,20 +453,17 @@ void Train::update_speed()
                {
                        loco.set_speed(0);
                        pure_speed = false;
-                       try_reserve = Time::now()+2*Time::sec;
                        set_status("Blocked");
                }
                else if(nsens==1 && target_speed>slow_speed)
                {
                        loco.set_speed(slow_speed);
                        pure_speed = false;
-                       try_reserve = Time::now()+2*Time::sec;
                        set_status("Slow");
                }
                else
                {
                        loco.set_speed(target_speed);
-                       try_reserve = Time::TimeStamp();
                        set_status(format("Traveling %d kmh", travel_speed));
                }
        }
@@ -525,9 +541,24 @@ void Train::set_position(const Block::Endpoint &bep)
 
 void Train::release_blocks(list<BlockRef> &blocks)
 {
+       release_blocks(blocks, blocks.begin(), blocks.end());
+}
+
+void Train::release_blocks(list<BlockRef> &blocks, list<BlockRef>::iterator begin, list<BlockRef>::iterator end)
+{
+       while(begin!=end)
+       {
+               Block *block = begin->block;
+               blocks.erase(begin++);
+               block->reserve(0);
+       }
+}
+
+void Train::reverse_blocks(list<BlockRef> &blocks) const
+{
+       blocks.reverse();
        for(list<BlockRef>::iterator i=blocks.begin(); i!=blocks.end(); ++i)
-               i->block->reserve(0);
-       blocks.clear();
+               i->entry = i->block->traverse(i->entry);
 }
 
 
@@ -544,10 +575,41 @@ void Train::RealSpeed::add(float s, float w)
 
 
 Train::Loader::Loader(Train &t):
-       DataFile::BasicLoader<Train>(t)
+       DataFile::BasicLoader<Train>(t),
+       prev_block(0)
 {
-       add("name",        &Train::name);
+       add("block",       &Loader::block);
+       add("block_hint",  &Loader::block_hint);
+       add("name",        &Loader::name);
        add("real_speed",  &Loader::real_speed);
+       add("route",       &Loader::route);
+}
+
+void Train::Loader::block(unsigned id)
+{
+       Block &blk = obj.trfc_mgr.get_block(id);
+       int entry = -1;
+       if(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");
+       obj.set_position(blk.get_endpoints()[entry]);
+
+       prev_block = &blk;
+}
+
+void Train::Loader::block_hint(unsigned id)
+{
+       prev_block = &obj.trfc_mgr.get_block(id);
+}
+
+void Train::Loader::name(const string &n)
+{
+       obj.set_name(n);
 }
 
 void Train::Loader::real_speed(unsigned i, float speed, float weight)
@@ -556,4 +618,9 @@ void Train::Loader::real_speed(unsigned i, float speed, float weight)
        obj.real_speed[i].weight = weight;
 }
 
+void Train::Loader::route(const string &n)
+{
+       obj.set_route(&obj.trfc_mgr.get_layout().get_route(n));
+}
+
 } // namespace Marklin