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         Object(l),
  18         type(t),
  19         address(0),
  20         track(0),
  21         block(0),
  22         entry(0),
  23         train(0),
  24         check_allocated_blocks(false),
  25         passing(false)
  26 {
  27         layout.add(*this);
  28
  29         layout.signal_block_reserved.connect(sigc::mem_fun(this, &Signal::block_reserved));
  30 }
  31
  32 Signal::~Signal()
  33 {
  34         layout.remove(*this);
  35 }
  36
  37 Signal *Signal::clone(Layout *to_layout) const
  38 {
  39         Signal *sig = new Signal((to_layout ? *to_layout : layout), type);
  40         sig->set_position(position);
  41         sig->set_rotation(rotation);
  42         return sig;
  43 }
  44
  45 void Signal::set_address(unsigned a)
  46 {
  47         address = a;
  48
  49         if(layout.has_driver() && address)
  50                 layout.get_driver().add_signal(address, type);
  51 }
  52
  53 void Signal::set_position(const Vector &p)
  54 {
  55         position = p;
  56
  57         update_location();
  58         signal_moved.emit();
  59 }
  60
  61 void Signal::update_location()
  62 {
  63         const set<Track *> &tracks = layout.get_all<Track>();
  64         float limit = layout.get_catalogue().get_gauge()*2;
  65         float dist = -1;
  66         for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
  67                 if(!(*i)->get_type().is_turnout())
  68                 {
  69                         Snap sn;
  70                         sn.position = position;
  71                         sn.rotation = rotation;
  72                         (*i)->snap(sn, limit, SNAP_SEGMENT);
  73                         float d = distance(position, sn.position);
  74                         if(d<dist || dist<0)
  75                         {
  76                                 track = *i;
  77                                 dist = d;
  78                         }
  79                 }
  80
  81         block = 0;
  82
  83         if(!track)
  84                 return;
  85
  86         unsigned n_endpoints = track->get_type().get_endpoints().size();
  87         for(unsigned j=0; j<n_endpoints; ++j)
  88         {
  89                 Angle a = wrap_with_base(track->get_snap_node(j).rotation-rotation, -Angle::quarter_turn());
  90                 if(a>=Angle::quarter_turn())
  91                 {
  92                         BlockIter biter = TrackIter(track, j).block_iter();
  93                         if(biter)
  94                         {
  95                                 block = &track->get_block();
  96                                 entry = biter.entry();
  97                         }
  98                 }
  99         }
 100 }
 101
 102 void Signal::set_rotation(const Angle &r)
 103 {
 104         rotation = r;
 105
 106         update_location();
 107         signal_moved.emit();
 108 }
 109
 110 unsigned Signal::get_n_snap_nodes() const
 111 {
 112         return 1;
 113 }
 114
 115 Snap Signal::get_snap_node(unsigned i) const
 116 {
 117         if(i>=1)
 118                 throw out_of_range("Signal::get_snap_node");
 119
 120         Snap sn;
 121         sn.position = position;
 122         sn.rotation = rotation;
 123         return sn;
 124 }
 125
 126 SnapType Signal::get_default_snap_type_to(const Object &other) const
 127 {
 128         if(dynamic_cast<const Track *>(&other))
 129                 return SNAP_SEGMENT;
 130
 131         return NO_SNAP;
 132 }
 133
 134 void Signal::tick(const Time::TimeDelta &)
 135 {
 136         if(check_allocated_blocks)
 137         {
 138                 unsigned n_blocks = 0;
 139                 BlockIter iter(block, entry);
 140                 iter = iter.next();
 141                 while(iter && iter->get_train()==train)
 142                 {
 143                         if(iter->get_sensor_id())
 144                                 ++n_blocks;
 145                         iter=iter.next();
 146                 }
 147                 check_allocated_blocks = false;
 148
 149                 const list<SignalType::Indication> &indications = type.get_indications();
 150                 unsigned aspect = indications.back().aspect;
 151                 for(list<SignalType::Indication>::const_iterator i=indications.begin(); i!=indications.end(); ++i)
 152                         if(n_blocks>=i->free_blocks)
 153                         {
 154                                 aspect = i->aspect;
 155                                 break;
 156                         }
 157
 158                 layout.get_driver().set_signal(address, aspect);
 159         }
 160 }
 161
 162 void Signal::block_reserved(const Block &b, Train *t)
 163 {
 164         if(&b==block)
 165         {
 166                 if(t)
 167                 {
 168                         const BlockIter &b_iter = t->get_block_allocator().iter_for(b);
 169                         if(b_iter && b_iter.entry()==entry)
 170                         {
 171                                 if(train_conn)
 172                                         train_conn.disconnect();
 173                                 train = t;
 174                                 passing = false;
 175                                 train_conn = train->signal_advanced.connect(sigc::mem_fun(this, &Signal::train_advanced));
 176                                 check_allocated_blocks = true;
 177                         }
 178                 }
 179                 else
 180                 {
 181                         layout.get_driver().set_signal(address, type.get_indications().back().aspect);
 182                         reset();
 183                 }
 184         }
 185         else if(train && t==train)
 186                 check_allocated_blocks = true;
 187 }
 188
 189 void Signal::train_advanced(Block &b)
 190 {
 191         if(&b==block)
 192                 passing = true;
 193         else if(passing && b.get_sensor_id())
 194         {
 195                 layout.get_driver().set_signal(address, type.get_indications().back().aspect);
 196                 reset();
 197         }
 198 }
 199
 200 void Signal::reset()
 201 {
 202         train = 0;
 203         if(train_conn)
 204                 train_conn.disconnect();
 205         check_allocated_blocks = false;
 206 }
 207
 208 void Signal::save(list<DataFile::Statement> &st) const
 209 {
 210         st.push_back((DataFile::Statement("position"), position.x, position.y, position.z));
 211         st.push_back((DataFile::Statement("rotation"), rotation.radians()));
 212         if(address)
 213                 st.push_back((DataFile::Statement("address"), address));
 214 }
 215
 216
 217 Signal::Loader::Loader(Signal &s):
 218         DataFile::ObjectLoader<Signal>(s)
 219 {
 220         add("address",  &Loader::address);
 221         add("position", &Loader::position);
 222         add("rotation", &Loader::rotation);
 223 }
 224
 225 void Signal::Loader::address(unsigned a)
 226 {
 227         obj.set_address(a);
 228 }
 229
 230 void Signal::Loader::position(float x, float y, float z)
 231 {
 232         obj.set_position(Vector(x, y, z));
 233 }
 234
 235 void Signal::Loader::rotation(float d)
 236 {
 237         obj.set_rotation(Angle::from_radians(d));
 238 }
 239
 240 } // namespace R2C2