Add vehicles

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

diff --git a/source/libmarklin/layout.h b/source/libmarklin/layout.h
index 608ddece1eed1c2398fc36b101d2b0d6fb0c5063..385a3cfab760e17d7beab074501e6d4cd3bf30ce 100644 (file)
--- a/source/libmarklin/layout.h
+++ b/source/libmarklin/layout.h
@@ -21,6 +21,7 @@ class Driver;
 class Route;
 class Track;
 class Train;
+class Vehicle;
 
 class Layout
 {
@@ -46,6 +47,8 @@ public:
        sigc::signal<void, Route &> signal_route_removed;
        sigc::signal<void, Train &> signal_train_added;
        sigc::signal<void, Train &> signal_train_removed;
+       sigc::signal<void, Vehicle &> signal_vehicle_added;
+       sigc::signal<void, Vehicle &> signal_vehicle_removed;
        sigc::signal<void, Block &, Train *> signal_block_reserved;
 
 private:
@@ -56,6 +59,7 @@ private:
        std::map<std::string, Route *> routes;
        std::set<Block *> blocks;
        std::map<unsigned, Train *> trains;
+       std::set<Vehicle *> vehicles;
        Msp::Time::TimeStamp last_tick;
 
 public:
@@ -89,6 +93,9 @@ public:
        const std::map<unsigned, Train *> &get_trains() const { return trains; }
        void remove_train(Train &);
 
+       void add_vehicle(Vehicle &);
+       void remove_vehicle(Vehicle &);
+
        void tick();
 
        void save(const std::string &);