source/libmarklin/catalogue.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 <msp/core/refptr.h>
   9 #include <msp/datafile/parser.h>
  10 #include "catalogue.h"
  11 #include "locotype.h"
  12 #include "tracktype.h"
  13
  14 using namespace std;
  15 using namespace Msp;
  16
  17 namespace Marklin {
  18
  19 Catalogue::~Catalogue()
  20 {
  21         for(map<unsigned, TrackType *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
  22                 delete i->second;
  23 }
  24
  25 TrackType &Catalogue::get_track(unsigned art_nr) const
  26 {
  27         map<unsigned, TrackType *>::const_iterator i=tracks.find(art_nr);
  28         if(i==tracks.end())
  29                 throw KeyError("Unknown track type");
  30
  31         return *i->second;
  32 }
  33
  34 LocoType &Catalogue::get_locomotive(unsigned art_nr) const
  35 {
  36         map<unsigned, LocoType *>::const_iterator i=locos.find(art_nr);
  37         if(i==locos.end())
  38                 throw KeyError("Unknown locomotive type");
  39
  40         return *i->second;
  41 }
  42
  43
  44 Catalogue::Loader::Loader(Catalogue &c):
  45         DataFile::BasicLoader<Catalogue>(c)
  46 {
  47         add("locomotive", &Loader::locomotive);
  48         add("track", &Loader::track);
  49 }
  50
  51 void Catalogue::Loader::locomotive(unsigned art_nr)
  52 {
  53         if(obj.locos.count(art_nr))
  54                 throw Exception("Duplicate locomotive number");
  55
  56         RefPtr<LocoType> loco=new LocoType(art_nr);
  57         load_sub(*loco);
  58         obj.locos[art_nr]=loco.release();
  59 }
  60
  61 void Catalogue::Loader::track(unsigned art_nr)
  62 {
  63         if(obj.tracks.count(art_nr))
  64                 throw Exception("Duplicate track number");
  65
  66         RefPtr<TrackType> trk=new TrackType(art_nr);
  67         load_sub(*trk);
  68         obj.tracks[art_nr]=trk.release();
  69 }
  70
  71 } // namespace Marklin