source/libmarklin/blockiter.cpp

   1 /* $Id$
   2
   3 This file is part of the MSP Märklin suite
   4 Copyright © 2010  Mikkosoft Productions, Mikko Rasa
   5 Distributed under the GPL
   6 */
   7
   8 #include <msp/core/except.h>
   9 #include "block.h"
  10 #include "blockiter.h"
  11 #include "route.h"
  12 #include "trackiter.h"
  13
  14 using namespace std;
  15 using namespace Msp;
  16
  17 namespace Marklin {
  18
  19 BlockIter::BlockIter():
  20         _block(0),
  21         _entry(0)
  22 { }
  23
  24 BlockIter::BlockIter(Block *b, unsigned e):
  25         _block(b),
  26         _entry(b ? e : 0)
  27 {
  28         if(_block && _entry>_block->get_endpoints().size())
  29                 throw InvalidParameterValue("Endpoint index not valid for block");
  30 }
  31
  32 TrackIter BlockIter::track_iter() const
  33 {
  34         if(!_block)
  35                 return TrackIter();
  36
  37         const Block::Endpoint &ep = _block->get_endpoints()[_entry];
  38         return TrackIter(ep.track, ep.track_ep);
  39 }
  40
  41 int BlockIter::get_exit(const Route *route) const
  42 {
  43         const vector<Block::Endpoint> &eps = _block->get_endpoints();
  44         TrackIter t_iter = track_iter();
  45
  46         while(t_iter)
  47         {
  48                 if(!_block->has_track(*t_iter))
  49                         throw LogicError("Block traversal strayed out of the block");
  50
  51                 int path = -1;
  52                 if(t_iter->get_turnout_id() && route)
  53                         path = route->get_turnout(t_iter->get_turnout_id());
  54                 if(path==-1)
  55                         path = t_iter->get_active_path();
  56
  57                 TrackIter t_exit = t_iter.reverse(path);
  58
  59                 for(unsigned i=0; i<eps.size(); ++i)
  60                         if(eps[i].track==t_exit.track() && eps[i].track_ep==t_exit.entry())
  61                                 return i;
  62
  63                 t_iter = t_iter.next(path);
  64         }
  65
  66         return -1;
  67 }
  68
  69 BlockIter BlockIter::next(const Route *route) const
  70 {
  71         if(!_block)
  72                 return BlockIter();
  73
  74         int exit = get_exit(route);
  75         if(exit<0)
  76                 return BlockIter();
  77
  78         BlockIter result;
  79         result._block = _block->get_link(exit);
  80         result._entry = (result._block ? result._block->get_endpoint_by_link(*_block) : 0);
  81
  82         return result;
  83 }
  84
  85 BlockIter BlockIter::reverse(const Route *route) const
  86 {
  87         if(!_block)
  88                 return BlockIter();
  89
  90         int exit = get_exit(route);
  91         if(exit<0)
  92                 return BlockIter();
  93
  94         return BlockIter(_block, exit);
  95 }
  96
  97 BlockIter BlockIter::flip() const
  98 {
  99         if(!_block)
 100                 return BlockIter();
 101
 102         BlockIter result;
 103         result._block = _block->get_link(_entry);
 104         result._entry = (result._block ? result._block->get_endpoint_by_link(*_block) : 0);
 105
 106         return result;
 107 }
 108
 109 Block &BlockIter::operator*() const
 110 {
 111         if(!_block)
 112                 throw InvalidState("BlockIter is null");
 113
 114         return *_block;
 115 }
 116
 117 bool BlockIter::operator==(const BlockIter &other) const
 118 {
 119         return _block==other._block && _entry==other._entry;
 120 }
 121
 122 } // namespace Marklin