X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Fvehicle.h;h=616ad366d2de5ce2973a48848e2e33fb2d169719;hb=ab64a17394440c3cb18d0b30d75fb4ed6297a6b4;hp=e7427b1e92ef6bb61b95ae48b9241664997b7413;hpb=e214fd389b9819eac0379cfb78e0f446e267b839;p=r2c2.git

diff --git a/source/libr2c2/vehicle.h b/source/libr2c2/vehicle.h
index e7427b1..616ad36 100644
--- a/source/libr2c2/vehicle.h
+++ b/source/libr2c2/vehicle.h
@@ -9,6 +9,7 @@
 namespace R2C2 {
 
 class Layout;
+class Train;
 
 class attachment_error: public std::logic_error
 {
@@ -69,6 +70,7 @@ private:
 	};
 
 	const VehicleType &type;
+	Train *train;
 	Vehicle *next;
 	Vehicle *prev;
 	TrackPosition track_pos;
@@ -85,6 +87,8 @@ public:
 	virtual Vehicle *clone(Layout * = 0) const;
 	virtual const VehicleType &get_type() const { return type; }
 
+	void set_train(Train *);
+	Train *get_train() const { return train; }
 	void attach_back(Vehicle &);
 	void attach_front(Vehicle &);
 	void detach_back();