# HG changeset patch
# User peter1138
# Date 1149665636 0
# Node ID 08b438ca34290341d7c8427f185b322173345da8
# Parent  9d3cd1ed6ac0e28bf3e7e973caa2b1e4d583d1e1
(svn r5147) - NewGRF: Use refit capacity callback when refitting an aircraft (mart3p)

diff -r 9d3cd1ed6ac0 -r 08b438ca3429 aircraft_cmd.c
--- a/aircraft_cmd.c	Wed Jun 07 07:20:28 2006 +0000
+++ b/aircraft_cmd.c	Wed Jun 07 07:33:56 2006 +0000
@@ -544,6 +544,7 @@
 	CargoID new_cid = GB(p2, 0, 8);
 	byte new_subtype = GB(p2, 8, 8);
 	const AircraftVehicleInfo *avi;
+	uint16 callback = CALLBACK_FAILED;
 
 	if (!IsVehicleIndex(p1)) return CMD_ERROR;
 
@@ -559,21 +560,24 @@
 
 	SET_EXPENSES_TYPE(EXPENSES_AIRCRAFT_RUN);
 
-	switch (new_cid) {
-		case CT_PASSENGERS:
-			pass = avi->passenger_capacity;
-			break;
-		case CT_MAIL:
-			pass = avi->passenger_capacity + avi->mail_capacity;
-			break;
-		case CT_GOODS:
-			pass = avi->passenger_capacity + avi->mail_capacity;
-			pass /= 2;
-			break;
-		default:
-			pass = avi->passenger_capacity + avi->mail_capacity;
-			pass /= 4;
-			break;
+	/* Check the refit capacity callback */
+	if (HASBIT(EngInfo(v->engine_type)->callbackmask, CBM_REFIT_CAPACITY)) {
+		/* Back up the existing cargo type */
+		CargoID temp_cid = v->cargo_type;
+		v->cargo_type = new_cid;
+
+		callback = GetVehicleCallback(CBID_VEHICLE_REFIT_CAPACITY, 0, 0, v->engine_type, v);
+
+		/* Restore the cargo type */
+		v->cargo_type = temp_cid;
+	}
+
+	if (callback == CALLBACK_FAILED) {
+		/* If the callback failed, or wasn't executed, use the aircraft's
+		 * default cargo capacity */
+		pass = AircraftDefaultCargoCapacity(new_cid, v->engine_type);
+	} else {
+		pass = callback;
 	}
 	_returned_refit_capacity = pass;