From 36672afd8f20859189e783c693e9dc770bae1cac Mon Sep 17 00:00:00 2001 From: peter1138 Date: Wed, 7 Jun 2006 07:33:56 +0000 Subject: (svn r5147) - NewGRF: Use refit capacity callback when refitting an aircraft (mart3p) --- aircraft_cmd.c | 34 +++++++++++++++++++--------------- 1 file changed, 19 insertions(+), 15 deletions(-) diff --git a/aircraft_cmd.c b/aircraft_cmd.c index 2ef5b0c6f..1729dcdbe 100644 --- a/aircraft_cmd.c +++ b/aircraft_cmd.c @@ -544,6 +544,7 @@ int32 CmdRefitAircraft(TileIndex tile, uint32 flags, uint32 p1, uint32 p2) 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 @@ int32 CmdRefitAircraft(TileIndex tile, uint32 flags, uint32 p1, uint32 p2) 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; -- cgit v1.2.3-70-g09d2