X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Fcatalogue.h;h=d1829afae8b47a3510188523bffbb423ffaf38b5;hb=ab64a17394440c3cb18d0b30d75fb4ed6297a6b4;hp=73ad4e8e5694949d2de20fd597df1b7cb7320fb4;hpb=d15ac13f2e170f155b4bbd124df48400c339b644;p=r2c2.git

diff --git a/source/libr2c2/catalogue.h b/source/libr2c2/catalogue.h
index 73ad4e8..d1829af 100644
--- a/source/libr2c2/catalogue.h
+++ b/source/libr2c2/catalogue.h
@@ -2,20 +2,21 @@
 #define LIBR2C2_CATALOGUE_H_
 
 #include <map>
-#include <msp/datafile/loader.h>
+#include <msp/datafile/objectloader.h>
 #include "articlenumber.h"
 #include "layout.h"
 #include "profile.h"
 
 namespace R2C2 {
 
+class SignalType;
 class TrackType;
 class VehicleType;
 
 class Catalogue
 {
 public:
-	class Loader: public Msp::DataFile::BasicLoader<Catalogue>
+	class Loader: public Msp::DataFile::ObjectLoader<Catalogue>
 	{
 	public:
 		Loader(Catalogue &);
@@ -25,6 +26,7 @@ public:
 		void layout();
 		void rail_profile();
 		void scale(float, float);
+		void signal(ArticleNumber);
 		void track(unsigned);
 		void track(ArticleNumber);
 		void vehicle(unsigned);
@@ -33,9 +35,11 @@ public:
 
 	typedef std::map<ArticleNumber, TrackType *> TrackMap;
 	typedef std::map<ArticleNumber, VehicleType *> VehicleMap;
+	typedef std::map<ArticleNumber, SignalType *> SignalMap;
 
 	sigc::signal<void, const TrackType &> signal_track_added;
 	sigc::signal<void, const VehicleType &> signal_vehicle_added;
+	sigc::signal<void, const SignalType &> signal_signal_added;
 
 private:
 	float scale;
@@ -46,6 +50,7 @@ private:
 	std::string track_technique;
 	TrackMap tracks;
 	VehicleMap vehicles;
+	SignalMap signals;
 	Layout layout;
 
 public:
@@ -68,6 +73,10 @@ public:
 	const VehicleType &get_vehicle(const ArticleNumber &) const;
 	const VehicleMap &get_vehicles() const { return vehicles; }
 
+	void add_signal(SignalType &);
+	const SignalType &get_signal(const ArticleNumber &) const;
+	const SignalMap &get_signals() const { return signals; }
+
 	Layout &get_layout() { return layout; }
 };