(svn r9754) -Codechange: make classes for all vehicle types, so we can make nicer/better maintainable code, i.e. virtual methods instead of switches.

1 files changed, 18 insertions, 0 deletions

diff --git a/src/aircraft.h b/src/aircraft.h
index eb11f755a..0876fee7f 100644
--- a/src/aircraft.h
+++ b/src/aircraft.h
@@ -115,4 +115,22 @@ void UpdateAirplanesOnNewStation(const Station *st);
  */
 void UpdateAircraftCache(Vehicle *v);
 
+/**
+ * This class 'wraps' Vehicle; you do not actually instantiate this class.
+ * You create a Vehicle using AllocateVehicle, so it is added to the pool
+ * and you reinitialize that to a Train using:
+ *   v = new (v) Aircraft();
+ *
+ * As side-effect the vehicle type is set correctly.
+ */
+struct Aircraft : public Vehicle {
+	/** Initializes the Vehicle to an aircraft */
+	Aircraft() { this->type = VEH_AIRCRAFT; }
+
+	/** We want to 'destruct' the right class. */
+	virtual ~Aircraft() {}
+
+	const char *GetTypeString() { return "aircraft"; }
+};
+
 #endif /* AIRCRAFT_H */