X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Ftrafficmanager.h;h=293783f112d8f914dcc027c903c16c4b8d1ae841;hb=3e9c210ddc036cd015228504cc0803c909e27f84;hp=83b03e2ec8c4684af75ab4adf5ca49d2d80321cd;hpb=06c100aacb559fbbe7380e15981c4772092c269b;p=r2c2.git

diff --git a/source/libmarklin/trafficmanager.h b/source/libmarklin/trafficmanager.h
index 83b03e2..293783f 100644
--- a/source/libmarklin/trafficmanager.h
+++ b/source/libmarklin/trafficmanager.h
@@ -1,3 +1,10 @@
+/* $Id$
+
+This file is part of the MSP Märklin suite
+Copyright © 2006-2008 Mikkosoft Productions, Mikko Rasa
+Distributed under the GPL
+*/
+
 #ifndef LIBMARKLIN_TRAFFICMANAGER_H_
 #define LIBMARKLIN_TRAFFICMANAGER_H_
 
@@ -12,6 +19,12 @@ class Turnout;
 
 class TrafficManager
 {
+private:
+	Control &control;
+	Layout &layout;
+	std::list<Block *> blocks;
+	std::list<Train *> trains;
+
 public:
 	sigc::signal<void, const Block &, const Train *> signal_block_reserved;
 
@@ -19,15 +32,12 @@ public:
 	~TrafficManager();
 
 	Control &get_control() const { return control; }
-	Block *get_block_by_track(const Track *) const;
-	const TrainSeq &get_trains() const { return trains; }
+	const std::list<Block *> &get_blocks() const { return blocks; }
+	Block &get_block_by_track(const Track &) const;
+	const std::list<Train *> &get_trains() const { return trains; }
 	void add_train(Train *);
+	void tick();
 private:
-	Control &control;
-	Layout &layout;
-	BlockSeq blocks;
-	TrainSeq trains;
-
 	void turnout_route_changed(unsigned, Turnout *);
 };