Forgot to add the new files

[r2c2.git] / source / libmarklin / trafficmanager.h

diff --git a/source/libmarklin/trafficmanager.h b/source/libmarklin/trafficmanager.h
index ac3f395ce45a50bd1f3ad248c2dc246a0bdffa2f..266d2ff65ccf037eafcd7fdce468fddbd8bb9ad3 100644 (file)
--- a/source/libmarklin/trafficmanager.h
+++ b/source/libmarklin/trafficmanager.h
@@ -19,6 +19,18 @@ class Turnout;
 
 class TrafficManager
 {
+public:
+       class Loader: public Msp::DataFile::BasicLoader<TrafficManager>
+       {
+       public:
+               Loader(TrafficManager &);
+       private:
+               void train(unsigned, unsigned);
+       };
+
+       sigc::signal<void, Train &> signal_train_added;
+       sigc::signal<void, const Block &, const Train *> signal_block_reserved;
+
 private:
        Control &control;
        Layout &layout;
@@ -27,19 +39,21 @@ private:
        Msp::Time::TimeStamp last_tick;
 
 public:
-       sigc::signal<void, const Block &, const Train *> signal_block_reserved;
-
        TrafficManager(Control &, Layout &);
        ~TrafficManager();
 
        Control &get_control() const { return control; }
+       Layout &get_layout() const { return layout; }
        const std::list<Block *> &get_blocks() const { return blocks; }
+       Block &get_block(unsigned) const;
        Block &get_block_by_track(const Track &) const;
        const std::list<Train *> &get_trains() const { return trains; }
+       Train &get_train_by_locomotive(const Locomotive &) const;
        void add_train(Train *);
        void tick();
+       void save(const std::string &) const;
 private:
-       void turnout_route_changed(unsigned, Turnout *);
+       void turnout_path_changed(unsigned, Turnout *);
 };
 
 } // namespace Marklin