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