source/libmarklin/layout.cpp

   1 #include <fstream>
   2 #include <msp/core/refptr.h>
   3 #include <msp/datafile/parser.h>
   4 #include "catalogue.h"
   5 #include "layout.h"
   6 #include "tracktype.h"
   7
   8 using namespace std;
   9 using namespace Msp;
  10
  11 namespace Marklin {
  12
  13 Layout::Layout(Catalogue &c):
  14         catalogue(c)
  15 { }
  16
  17 Layout::~Layout()
  18 {
  19         for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
  20                 delete *i;
  21 }
  22
  23 void Layout::add_track(Track &t)
  24 {
  25         if(tracks.insert(&t).second)
  26                 signal_track_added.emit(t);
  27 }
  28
  29 void Layout::remove_track(Track &t)
  30 {
  31         if(tracks.count(&t))
  32         {
  33                 tracks.erase(&t);
  34                 signal_track_removed.emit(t);
  35         }
  36 }
  37
  38 void Layout::check_links()
  39 {
  40         for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
  41                 (*i)->break_links();
  42
  43         for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
  44                 for(set<Track *>::iterator j=i; j!=tracks.end(); ++j)
  45                         if(j!=i)
  46                                 (*i)->snap_to(**j, true);
  47 }
  48
  49 void Layout::load(const string &fn)
  50 {
  51         IO::File in(fn);
  52         IO::Buffered inb(in);
  53
  54         filename=fn;
  55         DataFile::Parser parser(inb, fn);
  56         Loader loader(*this);
  57         loader.load(parser);
  58
  59         check_links();
  60
  61         for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
  62                 (*i)->check_slope();
  63 }
  64
  65 int Layout::save(const string &fn)
  66 {
  67         ofstream out(fn.c_str());
  68         if(!out) return -1;
  69
  70         filename=fn;
  71
  72         if(!base.empty())
  73                 out<<"base \""<<base<<"\";\n";
  74         for(set<Track *>::iterator i=tracks.begin(); i!=tracks.end(); ++i)
  75         {
  76                 out<<"track "<<(*i)->get_type().get_article_number()<<"\n{\n";
  77                 const Point &p=(*i)->get_position();
  78                 out<<"\tposition "<<p.x<<' '<<p.y<<' '<<p.z<<";\n";
  79                 out<<"\trotation "<<(*i)->get_rotation()<<";\n";
  80                 out<<"\tslope "<<(*i)->get_slope()<<";\n";
  81
  82                 unsigned id=(*i)->get_turnout_id();
  83                 if(id)
  84                         out<<"\tturnout_id "<<id<<";\n";
  85
  86                 id=(*i)->get_sensor_id();
  87                 if(id)
  88                         out<<"\tsensor_id "<<id<<";\n";
  89
  90                 if((*i)->get_flex())
  91                         out<<"\tflex true;\n";
  92
  93                 out<<"};\n";
  94         }
  95
  96         return 0;
  97 }
  98
  99
 100 Layout::Loader::Loader(Layout &l):
 101         layout(l)
 102 {
 103         add("base",  &Layout::base);
 104         add("track", &Loader::track);
 105 }
 106
 107 void Layout::Loader::track(unsigned art_nr)
 108 {
 109         TrackType &type=layout.catalogue.get_track(art_nr);
 110
 111         RefPtr<Track> trk=new Track(type);
 112         load_sub(*trk);
 113         layout.add_track(*trk.release());
 114 }
 115
 116 } // namespace Marklin