X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2F3d%2Fcatalogue.h;h=d826a780381990b6a471c0e47f7b56053e062d71;hb=e5cd9e4fbc577036a0385c985b6b65df8218d0a2;hp=6cc5bede4fcdc27f2faa845c1e96d723888f9eb0;hpb=48dff17ed3144d944965e7cb534e1f8fb92bd620;p=r2c2.git

diff --git a/source/3d/catalogue.h b/source/3d/catalogue.h
index 6cc5bed..d826a78 100644
--- a/source/3d/catalogue.h
+++ b/source/3d/catalogue.h
@@ -15,12 +15,14 @@ Distributed under the GPL
 namespace Marklin {
 
 class TrackType3D;
+class VehicleType3D;
 
 class Catalogue3D
 {
 private:
 	Catalogue &catalogue;
 	std::map<const TrackType *, TrackType3D *> tracks;
+	std::map<const VehicleType *, VehicleType3D *> vehicles;
 	Msp::GL::Material ballast_material;
 	Msp::GL::Material rail_material;
 	Msp::GL::Mesh endpoint_mesh;
@@ -31,11 +33,13 @@ public:
 
 	const Catalogue &get_catalogue() const { return catalogue; }
 	const TrackType3D &get_track(const TrackType &) const;
+	const VehicleType3D &get_vehicle(const VehicleType &) const;
 	const Msp::GL::Material &get_ballast_material() const { return ballast_material; }
 	const Msp::GL::Material &get_rail_material() const { return rail_material; }
 	const Msp::GL::Mesh &get_endpoint_mesh() const { return endpoint_mesh; }
 private:
 	void track_added(const TrackType &);
+	void loco_added(const LocoType &);
 	void build_endpoint_mesh();
 };