source/libr2c2/signal.cpp

   1 #include "blockiter.h"
   2 #include "catalogue.h"
   3 #include "driver.h"
   4 #include "layout.h"
   5 #include "signal.h"
   6 #include "signaltype.h"
   7 #include "trackiter.h"
   8 #include "tracktype.h"
   9 #include "train.h"
  10
  11 using namespace std;
  12 using namespace Msp;
  13
  14 namespace R2C2 {
  15
  16 Signal::Signal(Layout &l, const SignalType &t):
  17         TrackAttachment(l),
  18         type(t),
  19         address(0),
  20         train(0),
  21         check_allocated_blocks(false),
  22         passing(false)
  23 {
  24         layout.add(*this);
  25
  26         layout.signal_block_reserved.connect(sigc::mem_fun(this, &Signal::block_reserved));
  27 }
  28
  29 Signal::~Signal()
  30 {
  31         layout.remove(*this);
  32 }
  33
  34 Signal *Signal::clone(Layout *to_layout) const
  35 {
  36         Signal *sig = new Signal((to_layout ? *to_layout : layout), type);
  37         sig->set_position(position);
  38         sig->set_rotation(rotation);
  39         return sig;
  40 }
  41
  42 void Signal::set_address(unsigned a)
  43 {
  44         address = a;
  45
  46         if(layout.has_driver() && address)
  47                 layout.get_driver().add_signal(address, type);
  48 }
  49
  50 void Signal::set_position(const Vector &p)
  51 {
  52         position = p;
  53
  54         update_attachment();
  55         signal_moved.emit();
  56 }
  57
  58 void Signal::set_rotation(const Angle &r)
  59 {
  60         rotation = r;
  61
  62         update_attachment();
  63         signal_moved.emit();
  64 }
  65
  66 void Signal::update_attachment()
  67 {
  68         attach_to_closest(layout.get_catalogue().get_gauge()*2);
  69
  70         if(track)
  71                 block = track.block_iter();
  72         else
  73                 block = BlockIter();
  74 }
  75
  76 unsigned Signal::get_n_snap_nodes() const
  77 {
  78         return 1;
  79 }
  80
  81 Snap Signal::get_snap_node(unsigned i) const
  82 {
  83         if(i>=1)
  84                 throw out_of_range("Signal::get_snap_node");
  85
  86         Snap sn;
  87         sn.position = position;
  88         sn.rotation = rotation;
  89         return sn;
  90 }
  91
  92 SnapType Signal::get_default_snap_type_to(const Object &other) const
  93 {
  94         if(dynamic_cast<const Track *>(&other))
  95                 return SNAP_SEGMENT;
  96
  97         return NO_SNAP;
  98 }
  99
 100 void Signal::tick(const Time::TimeDelta &)
 101 {
 102         if(check_allocated_blocks)
 103         {
 104                 unsigned n_blocks = 0;
 105                 BlockIter iter = block.next();
 106                 while(iter && iter->get_train()==train)
 107                 {
 108                         if(iter->get_sensor_id())
 109                                 ++n_blocks;
 110                         iter=iter.next();
 111                 }
 112                 check_allocated_blocks = false;
 113
 114                 const list<SignalType::Indication> &indications = type.get_indications();
 115                 unsigned aspect = indications.back().aspect;
 116                 for(list<SignalType::Indication>::const_iterator i=indications.begin(); i!=indications.end(); ++i)
 117                         if(n_blocks>=i->free_blocks)
 118                         {
 119                                 aspect = i->aspect;
 120                                 break;
 121                         }
 122
 123                 layout.get_driver().set_signal(address, aspect);
 124         }
 125 }
 126
 127 void Signal::block_reserved(const Block &b, Train *t)
 128 {
 129         if(&b==block.block())
 130         {
 131                 if(t)
 132                 {
 133                         const BlockIter &b_iter = t->get_block_allocator().iter_for(b);
 134                         if(b_iter && b_iter.entry()==block.entry())
 135                         {
 136                                 if(train_conn)
 137                                         train_conn.disconnect();
 138                                 train = t;
 139                                 passing = false;
 140                                 train_conn = train->signal_advanced.connect(sigc::mem_fun(this, &Signal::train_advanced));
 141                                 check_allocated_blocks = true;
 142                         }
 143                 }
 144                 else
 145                 {
 146                         layout.get_driver().set_signal(address, type.get_indications().back().aspect);
 147                         reset();
 148                 }
 149         }
 150         else if(train && t==train)
 151                 check_allocated_blocks = true;
 152 }
 153
 154 void Signal::train_advanced(Block &b)
 155 {
 156         if(&b==block.block())
 157                 passing = true;
 158         else if(passing && b.get_sensor_id())
 159         {
 160                 layout.get_driver().set_signal(address, type.get_indications().back().aspect);
 161                 reset();
 162         }
 163 }
 164
 165 void Signal::reset()
 166 {
 167         train = 0;
 168         if(train_conn)
 169                 train_conn.disconnect();
 170         check_allocated_blocks = false;
 171 }
 172
 173 void Signal::save(list<DataFile::Statement> &st) const
 174 {
 175         st.push_back((DataFile::Statement("position"), position.x, position.y, position.z));
 176         st.push_back((DataFile::Statement("rotation"), rotation.radians()));
 177         if(address)
 178                 st.push_back((DataFile::Statement("address"), address));
 179 }
 180
 181
 182 Signal::Loader::Loader(Signal &s):
 183         DataFile::ObjectLoader<Signal>(s)
 184 {
 185         add("address",  &Loader::address);
 186         add("position", &Loader::position);
 187         add("rotation", &Loader::rotation);
 188 }
 189
 190 void Signal::Loader::address(unsigned a)
 191 {
 192         obj.set_address(a);
 193 }
 194
 195 void Signal::Loader::position(float x, float y, float z)
 196 {
 197         obj.set_position(Vector(x, y, z));
 198 }
 199
 200 void Signal::Loader::rotation(float d)
 201 {
 202         obj.set_rotation(Angle::from_radians(d));
 203 }
 204
 205 } // namespace R2C2