X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Fcatalogue.h;h=12ee006f45f34a1fcb5e7e399f39356e470c3236;hb=806bc53176e8ed216ba5db95bde5878832332393;hp=c0c6674c2a028e9ae44adc76ffdbba92761306b8;hpb=64340dad429ba4040538fc06b6882aabdb489925;p=r2c2.git

diff --git a/source/libr2c2/catalogue.h b/source/libr2c2/catalogue.h
index c0c6674..12ee006 100644
--- a/source/libr2c2/catalogue.h
+++ b/source/libr2c2/catalogue.h
@@ -9,6 +9,7 @@
 
 namespace R2C2 {
 
+class SignalType;
 class TrackType;
 class VehicleType;
 
@@ -25,17 +26,18 @@ public:
 		void layout();
 		void rail_profile();
 		void scale(float, float);
-		void track(unsigned);
+		void signal(ArticleNumber);
 		void track(ArticleNumber);
-		void vehicle(unsigned);
 		void vehicle(ArticleNumber);
 	};
 
 	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 +48,7 @@ private:
 	std::string track_technique;
 	TrackMap tracks;
 	VehicleMap vehicles;
+	SignalMap signals;
 	Layout layout;
 
 public:
@@ -68,6 +71,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; }
 };