source/libmarklin/trafficmanager.cpp

   1 /* $Id$
   2
   3 This file is part of the MSP Märklin suite
   4 Copyright © 2006-2009 Mikkosoft Productions, Mikko Rasa
   5 Distributed under the GPL
   6 */
   7
   8 #include <algorithm>
   9 #include <msp/datafile/writer.h>
  10 #include <msp/time/utils.h>
  11 #include "catalogue.h"
  12 #include "control.h"
  13 #include "layout.h"
  14 #include "locotype.h"
  15 #include "tracktype.h"
  16 #include "trafficmanager.h"
  17 #include "turnout.h"
  18
  19 using namespace std;
  20 using namespace Msp;
  21
  22 namespace Marklin {
  23
  24 TrafficManager::TrafficManager(Control &c, Layout &l):
  25         control(c),
  26         layout(l)
  27 {
  28         const set<Track *> &tracks = layout.get_tracks();
  29
  30         set<Track *> used_tracks;
  31         for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
  32         {
  33                 if(unsigned tid=(*i)->get_turnout_id())
  34                         new Turnout(control, tid, (*i)->get_type().get_n_routes()>=3);
  35                 if(unsigned sid=(*i)->get_sensor_id())
  36                         new Sensor(control, sid);
  37
  38                 if(used_tracks.count(*i)==0)
  39                 {
  40                         Block *block = new Block(*this, **i);
  41                         blocks.push_back(block);
  42                         used_tracks.insert(block->get_tracks().begin(), block->get_tracks().end());
  43                 }
  44         }
  45
  46         for(list<Block *>::iterator i=blocks.begin(); i!=blocks.end(); ++i)
  47                 for(list<Block *>::iterator j=i; j!=blocks.end(); ++j)
  48                         if(j!=i)
  49                                 (*i)->check_link(**j);
  50 }
  51
  52 TrafficManager::~TrafficManager()
  53 {
  54         for(list<Block *>::iterator i=blocks.begin(); i!=blocks.end(); ++i)
  55                 delete *i;
  56         for(list<Train *>::iterator i=trains.begin(); i!=trains.end(); ++i)
  57                 delete *i;
  58 }
  59
  60 Block &TrafficManager::get_block_by_track(const Track &t) const
  61 {
  62         for(list<Block *>::const_iterator i=blocks.begin(); i!=blocks.end(); ++i)
  63         {
  64                 const set<Track *> &tracks = (*i)->get_tracks();
  65                 if(tracks.count(const_cast<Track *>(&t)))
  66                         return **i;
  67         }
  68
  69         throw InvalidParameterValue("Unknown track");
  70 }
  71
  72 Train &TrafficManager::get_train_by_locomotive(const Locomotive &loco) const
  73 {
  74         for(list<Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
  75                 if(&(*i)->get_locomotive()==&loco)
  76                         return **i;
  77
  78         throw InvalidParameterValue("Unknown locomotive");
  79 }
  80
  81 void TrafficManager::add_train(Train *t)
  82 {
  83         if(find(trains.begin(), trains.end(), t)==trains.end())
  84         {
  85                 trains.push_back(t);
  86                 signal_train_added.emit(*t);
  87         }
  88 }
  89
  90 void TrafficManager::tick()
  91 {
  92         Time::TimeStamp t = Time::now();
  93         Time::TimeDelta dt;
  94         if(last_tick)
  95                 dt = t-last_tick;
  96         last_tick = t;
  97
  98         for(list<Train *>::iterator i=trains.begin(); i!=trains.end(); ++i)
  99                 (*i)->tick(t, dt);
 100 }
 101
 102 void TrafficManager::save(const string &fn) const
 103 {
 104         IO::BufferedFile out(fn, IO::M_WRITE);
 105         DataFile::Writer writer(out);
 106         for(list<Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
 107         {
 108                 const Locomotive &loco = (*i)->get_locomotive();
 109                 DataFile::Statement st("train");
 110                 st.append(loco.get_type().get_article_number());
 111                 st.append(loco.get_address());
 112                 (*i)->save(st.sub);
 113                 writer.write(st);
 114         }
 115 }
 116
 117 void TrafficManager::turnout_route_changed(unsigned, Turnout *)
 118 {
 119 }
 120
 121
 122 TrafficManager::Loader::Loader(TrafficManager &tm):
 123         DataFile::BasicLoader<TrafficManager>(tm)
 124 {
 125         add("train", &Loader::train);
 126 }
 127
 128 void TrafficManager::Loader::train(unsigned art_nr, unsigned addr)
 129 {
 130         Locomotive *loco = new Locomotive(obj.layout.get_catalogue().get_locomotive(art_nr), obj.control, addr);
 131         Train *trn = new Train(obj, *loco);
 132         load_sub(*trn);
 133 }
 134
 135 } // namespace Marklin