X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibmarklin%2Fcatalogue.h;h=a436994ecea6ae700268bd878d8c4a25135831be;hb=8e3009751ee81e0e5312cae53ee844bdd9e49628;hp=524d3f2b56a6593cb82c859af044dfbf027838d9;hpb=6968273080fa2a1cbcfc506610d5f249299611e9;p=r2c2.git

diff --git a/source/libmarklin/catalogue.h b/source/libmarklin/catalogue.h
index 524d3f2..a436994 100644
--- a/source/libmarklin/catalogue.h
+++ b/source/libmarklin/catalogue.h
@@ -10,12 +10,12 @@ Distributed under the GPL
 
 #include <map>
 #include <msp/datafile/loader.h>
+#include "articlenumber.h"
 #include "layout.h"
 #include "profile.h"
 
 namespace Marklin {
 
-class LocoType;
 class TrackType;
 class VehicleType;
 
@@ -30,13 +30,17 @@ public:
 		void ballast_profile();
 		void gauge(float);
 		void layout();
-		void locomotive(unsigned);
 		void rail_profile();
 		void scale(float, float);
 		void track(unsigned);
+		void track(ArticleNumber);
 		void vehicle(unsigned);
+		void vehicle(ArticleNumber);
 	};
 
+	typedef std::map<ArticleNumber, TrackType *> TrackMap;
+	typedef std::map<ArticleNumber, VehicleType *> VehicleMap;
+
 	sigc::signal<void, const TrackType &> signal_track_added;
 	sigc::signal<void, const VehicleType &> signal_vehicle_added;
 
@@ -46,8 +50,8 @@ private:
 	Profile rail_profile;
 	Profile ballast_profile;
 	Profile path_profile;
-	std::map<unsigned, TrackType *> tracks;
-	std::map<unsigned, VehicleType *> vehicles;
+	TrackMap tracks;
+	VehicleMap vehicles;
 	Layout layout;
 
 public:
@@ -56,18 +60,18 @@ public:
 
 	float get_scale() const { return scale; }
 	float get_gauge() const { return gauge; }
+	float get_rail_elevation() const;
 	const Profile &get_rail_profile() const { return rail_profile; }
 	const Profile &get_ballast_profile() const { return ballast_profile; }
 	const Profile &get_path_profile() const { return path_profile; }
 
 	void add_track(TrackType &);
-	const TrackType &get_track(unsigned) const;
-	const std::map<unsigned, TrackType *> &get_tracks() const { return tracks; }
+	const TrackType &get_track(const ArticleNumber &) const;
+	const TrackMap &get_tracks() const { return tracks; }
 
 	void add_vehicle(VehicleType &);
-	const VehicleType &get_vehicle(unsigned) const;
-	const LocoType &get_locomotive(unsigned) const;
-	const std::map<unsigned, VehicleType *> &get_vehicles() const { return vehicles; }
+	const VehicleType &get_vehicle(const ArticleNumber &) const;
+	const VehicleMap &get_vehicles() const { return vehicles; }
 
 	Layout &get_layout() { return layout; }
 };