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                 unsigned path = (route ? route->get_path(*t_iter) : t_iter->get_active_path());
  52                 TrackIter t_exit = t_iter.reverse(path);
  53
  54                 for(unsigned i=0; i<eps.size(); ++i)
  55                         if(eps[i].track==t_exit.track() && eps[i].track_ep==t_exit.entry())
  56                                 return i;
  57
  58                 t_iter = t_exit.flip();
  59         }
  60
  61         return -1;
  62 }
  63
  64 BlockIter BlockIter::next(const Route *route) const
  65 {
  66         if(!_block)
  67                 return BlockIter();
  68
  69         int exit = get_exit(route);
  70         if(exit<0)
  71                 return BlockIter();
  72
  73         BlockIter result;
  74         result._block = _block->get_link(exit);
  75         result._entry = (result._block ? result._block->get_endpoint_by_link(*_block) : 0);
  76
  77         return result;
  78 }
  79
  80 BlockIter BlockIter::reverse(const Route *route) const
  81 {
  82         if(!_block)
  83                 return BlockIter();
  84
  85         int exit = get_exit(route);
  86         if(exit<0)
  87                 return BlockIter();
  88
  89         return BlockIter(_block, exit);
  90 }
  91
  92 BlockIter BlockIter::flip() const
  93 {
  94         if(!_block)
  95                 return BlockIter();
  96
  97         BlockIter result;
  98         result._block = _block->get_link(_entry);
  99         result._entry = (result._block ? result._block->get_endpoint_by_link(*_block) : 0);
 100
 101         return result;
 102 }
 103
 104 Block &BlockIter::operator*() const
 105 {
 106         if(!_block)
 107                 throw InvalidState("BlockIter is null");
 108
 109         return *_block;
 110 }
 111
 112 bool BlockIter::operator==(const BlockIter &other) const
 113 {
 114         return _block==other._block && _entry==other._entry;
 115 }
 116
 117 } // namespace Marklin