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                         if(!control.get_sensors().count(sid))
  37                                 new Sensor(control, sid);
  38
  39                 if(used_tracks.count(*i)==0)
  40                 {
  41                         Block *block = new Block(*this, **i);
  42                         blocks.push_back(block);
  43                         used_tracks.insert(block->get_tracks().begin(), block->get_tracks().end());
  44                 }
  45         }
  46
  47         for(list<Block *>::iterator i=blocks.begin(); i!=blocks.end(); ++i)
  48                 for(list<Block *>::iterator j=i; j!=blocks.end(); ++j)
  49                         if(j!=i)
  50                                 (*i)->check_link(**j);
  51 }
  52
  53 TrafficManager::~TrafficManager()
  54 {
  55         for(list<Block *>::iterator i=blocks.begin(); i!=blocks.end(); ++i)
  56                 delete *i;
  57         for(list<Train *>::iterator i=trains.begin(); i!=trains.end(); ++i)
  58                 delete *i;
  59 }
  60
  61 Block &TrafficManager::get_block_by_track(const Track &t) const
  62 {
  63         for(list<Block *>::const_iterator i=blocks.begin(); i!=blocks.end(); ++i)
  64         {
  65                 const set<Track *> &tracks = (*i)->get_tracks();
  66                 if(tracks.count(const_cast<Track *>(&t)))
  67                         return **i;
  68         }
  69
  70         throw InvalidParameterValue("Unknown track");
  71 }
  72
  73 Train &TrafficManager::get_train_by_locomotive(const Locomotive &loco) const
  74 {
  75         for(list<Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
  76                 if(&(*i)->get_locomotive()==&loco)
  77                         return **i;
  78
  79         throw InvalidParameterValue("Unknown locomotive");
  80 }
  81
  82 void TrafficManager::add_train(Train *t)
  83 {
  84         if(find(trains.begin(), trains.end(), t)==trains.end())
  85         {
  86                 trains.push_back(t);
  87                 signal_train_added.emit(*t);
  88         }
  89 }
  90
  91 void TrafficManager::tick()
  92 {
  93         Time::TimeStamp t = Time::now();
  94         Time::TimeDelta dt;
  95         if(last_tick)
  96                 dt = t-last_tick;
  97         last_tick = t;
  98
  99         for(list<Train *>::iterator i=trains.begin(); i!=trains.end(); ++i)
 100                 (*i)->tick(t, dt);
 101 }
 102
 103 void TrafficManager::save(const string &fn) const
 104 {
 105         IO::BufferedFile out(fn, IO::M_WRITE);
 106         DataFile::Writer writer(out);
 107         for(list<Train *>::const_iterator i=trains.begin(); i!=trains.end(); ++i)
 108         {
 109                 const Locomotive &loco = (*i)->get_locomotive();
 110                 DataFile::Statement st("train");
 111                 st.append(loco.get_type().get_article_number());
 112                 st.append(loco.get_address());
 113                 (*i)->save(st.sub);
 114                 writer.write(st);
 115         }
 116 }
 117
 118 void TrafficManager::turnout_route_changed(unsigned, Turnout *)
 119 {
 120 }
 121
 122
 123 TrafficManager::Loader::Loader(TrafficManager &tm):
 124         DataFile::BasicLoader<TrafficManager>(tm)
 125 {
 126         add("train", &Loader::train);
 127 }
 128
 129 void TrafficManager::Loader::train(unsigned art_nr, unsigned addr)
 130 {
 131         Locomotive *loco = new Locomotive(obj.layout.get_catalogue().get_locomotive(art_nr), obj.control, addr);
 132         Train *trn = new Train(obj, *loco);
 133         load_sub(*trn);
 134 }
 135
 136 } // namespace Marklin