Add: Draw network password indicator lock in company drop down list. (#7079)
[openttd-github.git] / src / vehicle_cmd.cpp
blob9231ad689dc5559ef200078bfc5d4f24c80ebadf
1 /* $Id$ */
3 /*
4 * This file is part of OpenTTD.
5 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
6 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
7 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
8 */
10 /** @file vehicle_cmd.cpp Commands for vehicles. */
12 #include "stdafx.h"
13 #include "roadveh.h"
14 #include "news_func.h"
15 #include "airport.h"
16 #include "cmd_helper.h"
17 #include "command_func.h"
18 #include "company_func.h"
19 #include "train.h"
20 #include "aircraft.h"
21 #include "newgrf_text.h"
22 #include "vehicle_func.h"
23 #include "string_func.h"
24 #include "depot_map.h"
25 #include "vehiclelist.h"
26 #include "engine_func.h"
27 #include "articulated_vehicles.h"
28 #include "autoreplace_gui.h"
29 #include "group.h"
30 #include "order_backup.h"
31 #include "ship.h"
32 #include "newgrf.h"
33 #include "company_base.h"
35 #include "table/strings.h"
37 #include "safeguards.h"
39 /* Tables used in vehicle.h to find the right command for a certain vehicle type */
40 const uint32 _veh_build_proc_table[] = {
41 CMD_BUILD_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_BUY_TRAIN),
42 CMD_BUILD_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_BUY_ROAD_VEHICLE),
43 CMD_BUILD_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_BUY_SHIP),
44 CMD_BUILD_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_BUY_AIRCRAFT),
47 const uint32 _veh_sell_proc_table[] = {
48 CMD_SELL_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_SELL_TRAIN),
49 CMD_SELL_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_SELL_ROAD_VEHICLE),
50 CMD_SELL_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_SELL_SHIP),
51 CMD_SELL_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_SELL_AIRCRAFT),
54 const uint32 _veh_refit_proc_table[] = {
55 CMD_REFIT_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_REFIT_TRAIN),
56 CMD_REFIT_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_REFIT_ROAD_VEHICLE),
57 CMD_REFIT_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_REFIT_SHIP),
58 CMD_REFIT_VEHICLE | CMD_MSG(STR_ERROR_CAN_T_REFIT_AIRCRAFT),
61 const uint32 _send_to_depot_proc_table[] = {
62 CMD_SEND_VEHICLE_TO_DEPOT | CMD_MSG(STR_ERROR_CAN_T_SEND_TRAIN_TO_DEPOT),
63 CMD_SEND_VEHICLE_TO_DEPOT | CMD_MSG(STR_ERROR_CAN_T_SEND_ROAD_VEHICLE_TO_DEPOT),
64 CMD_SEND_VEHICLE_TO_DEPOT | CMD_MSG(STR_ERROR_CAN_T_SEND_SHIP_TO_DEPOT),
65 CMD_SEND_VEHICLE_TO_DEPOT | CMD_MSG(STR_ERROR_CAN_T_SEND_AIRCRAFT_TO_HANGAR),
69 CommandCost CmdBuildRailVehicle(TileIndex tile, DoCommandFlag flags, const Engine *e, uint16 data, Vehicle **v);
70 CommandCost CmdBuildRoadVehicle(TileIndex tile, DoCommandFlag flags, const Engine *e, uint16 data, Vehicle **v);
71 CommandCost CmdBuildShip (TileIndex tile, DoCommandFlag flags, const Engine *e, uint16 data, Vehicle **v);
72 CommandCost CmdBuildAircraft (TileIndex tile, DoCommandFlag flags, const Engine *e, uint16 data, Vehicle **v);
74 /**
75 * Build a vehicle.
76 * @param tile tile of depot where the vehicle is built
77 * @param flags for command
78 * @param p1 various bitstuffed data
79 * bits 0-15: vehicle type being built.
80 * bits 16-31: vehicle type specific bits passed on to the vehicle build functions.
81 * @param p2 User
82 * @param text unused
83 * @return the cost of this operation or an error
85 CommandCost CmdBuildVehicle(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
87 /* Elementary check for valid location. */
88 if (!IsDepotTile(tile) || !IsTileOwner(tile, _current_company)) return CMD_ERROR;
90 VehicleType type = GetDepotVehicleType(tile);
92 /* Validate the engine type. */
93 EngineID eid = GB(p1, 0, 16);
94 if (!IsEngineBuildable(eid, type, _current_company)) return_cmd_error(STR_ERROR_RAIL_VEHICLE_NOT_AVAILABLE + type);
96 const Engine *e = Engine::Get(eid);
97 CommandCost value(EXPENSES_NEW_VEHICLES, e->GetCost());
99 /* Engines without valid cargo should not be available */
100 if (e->GetDefaultCargoType() == CT_INVALID) return CMD_ERROR;
102 /* Check whether the number of vehicles we need to build can be built according to pool space. */
103 uint num_vehicles;
104 switch (type) {
105 case VEH_TRAIN: num_vehicles = (e->u.rail.railveh_type == RAILVEH_MULTIHEAD ? 2 : 1) + CountArticulatedParts(eid, false); break;
106 case VEH_ROAD: num_vehicles = 1 + CountArticulatedParts(eid, false); break;
107 case VEH_SHIP: num_vehicles = 1; break;
108 case VEH_AIRCRAFT: num_vehicles = e->u.air.subtype & AIR_CTOL ? 2 : 3; break;
109 default: NOT_REACHED(); // Safe due to IsDepotTile()
111 if (!Vehicle::CanAllocateItem(num_vehicles)) return_cmd_error(STR_ERROR_TOO_MANY_VEHICLES_IN_GAME);
113 /* Check whether we can allocate a unit number. Autoreplace does not allocate
114 * an unit number as it will (always) reuse the one of the replaced vehicle
115 * and (train) wagons don't have an unit number in any scenario. */
116 UnitID unit_num = (flags & DC_AUTOREPLACE || (type == VEH_TRAIN && e->u.rail.railveh_type == RAILVEH_WAGON)) ? 0 : GetFreeUnitNumber(type);
117 if (unit_num == UINT16_MAX) return_cmd_error(STR_ERROR_TOO_MANY_VEHICLES_IN_GAME);
119 Vehicle *v;
120 switch (type) {
121 case VEH_TRAIN: value.AddCost(CmdBuildRailVehicle(tile, flags, e, GB(p1, 16, 16), &v)); break;
122 case VEH_ROAD: value.AddCost(CmdBuildRoadVehicle(tile, flags, e, GB(p1, 16, 16), &v)); break;
123 case VEH_SHIP: value.AddCost(CmdBuildShip (tile, flags, e, GB(p1, 16, 16), &v)); break;
124 case VEH_AIRCRAFT: value.AddCost(CmdBuildAircraft (tile, flags, e, GB(p1, 16, 16), &v)); break;
125 default: NOT_REACHED(); // Safe due to IsDepotTile()
128 if (value.Succeeded() && flags & DC_EXEC) {
129 v->unitnumber = unit_num;
130 v->value = value.GetCost();
132 InvalidateWindowData(WC_VEHICLE_DEPOT, v->tile);
133 InvalidateWindowClassesData(GetWindowClassForVehicleType(type), 0);
134 SetWindowDirty(WC_COMPANY, _current_company);
135 if (IsLocalCompany()) {
136 InvalidateAutoreplaceWindow(v->engine_type, v->group_id); // updates the auto replace window (must be called before incrementing num_engines)
139 GroupStatistics::CountEngine(v, 1);
140 GroupStatistics::UpdateAutoreplace(_current_company);
142 if (v->IsPrimaryVehicle()) {
143 GroupStatistics::CountVehicle(v, 1);
144 OrderBackup::Restore(v, p2);
148 return value;
151 CommandCost CmdSellRailWagon(DoCommandFlag flags, Vehicle *v, uint16 data, uint32 user);
154 * Sell a vehicle.
155 * @param tile unused.
156 * @param flags for command.
157 * @param p1 various bitstuffed data.
158 * bits 0-19: vehicle ID being sold.
159 * bits 20-30: vehicle type specific bits passed on to the vehicle build functions.
160 * bit 31: make a backup of the vehicle's order (if an engine).
161 * @param p2 User.
162 * @param text unused.
163 * @return the cost of this operation or an error.
165 CommandCost CmdSellVehicle(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
167 Vehicle *v = Vehicle::GetIfValid(GB(p1, 0, 20));
168 if (v == NULL) return CMD_ERROR;
170 Vehicle *front = v->First();
172 CommandCost ret = CheckOwnership(front->owner);
173 if (ret.Failed()) return ret;
175 if (front->vehstatus & VS_CRASHED) return_cmd_error(STR_ERROR_VEHICLE_IS_DESTROYED);
177 if (!front->IsStoppedInDepot()) return_cmd_error(STR_ERROR_TRAIN_MUST_BE_STOPPED_INSIDE_DEPOT + front->type);
179 /* Can we actually make the order backup, i.e. are there enough orders? */
180 if (p1 & MAKE_ORDER_BACKUP_FLAG &&
181 front->orders.list != NULL &&
182 !front->orders.list->IsShared() &&
183 !Order::CanAllocateItem(front->orders.list->GetNumOrders())) {
184 /* Only happens in exceptional cases when there aren't enough orders anyhow.
185 * Thus it should be safe to just drop the orders in that case. */
186 p1 &= ~MAKE_ORDER_BACKUP_FLAG;
189 if (v->type == VEH_TRAIN) {
190 ret = CmdSellRailWagon(flags, v, GB(p1, 20, 12), p2);
191 } else {
192 ret = CommandCost(EXPENSES_NEW_VEHICLES, -front->value);
194 if (flags & DC_EXEC) {
195 if (front->IsPrimaryVehicle() && p1 & MAKE_ORDER_BACKUP_FLAG) OrderBackup::Backup(front, p2);
196 delete front;
200 return ret;
204 * Helper to run the refit cost callback.
205 * @param v The vehicle we are refitting, can be NULL.
206 * @param engine_type Which engine to refit
207 * @param new_cid Cargo type we are refitting to.
208 * @param new_subtype New cargo subtype.
209 * @param[out] auto_refit_allowed The refit is allowed as an auto-refit.
210 * @return Price for refitting
212 static int GetRefitCostFactor(const Vehicle *v, EngineID engine_type, CargoID new_cid, byte new_subtype, bool *auto_refit_allowed)
214 /* Prepare callback param with info about the new cargo type. */
215 const Engine *e = Engine::Get(engine_type);
217 /* Is this vehicle a NewGRF vehicle? */
218 if (e->GetGRF() != NULL) {
219 const CargoSpec *cs = CargoSpec::Get(new_cid);
220 uint32 param1 = (cs->classes << 16) | (new_subtype << 8) | e->GetGRF()->cargo_map[new_cid];
222 uint16 cb_res = GetVehicleCallback(CBID_VEHICLE_REFIT_COST, param1, 0, engine_type, v);
223 if (cb_res != CALLBACK_FAILED) {
224 *auto_refit_allowed = HasBit(cb_res, 14);
225 int factor = GB(cb_res, 0, 14);
226 if (factor >= 0x2000) factor -= 0x4000; // Treat as signed integer.
227 return factor;
231 *auto_refit_allowed = e->info.refit_cost == 0;
232 return (v == NULL || v->cargo_type != new_cid) ? e->info.refit_cost : 0;
236 * Learn the price of refitting a certain engine
237 * @param v The vehicle we are refitting, can be NULL.
238 * @param engine_type Which engine to refit
239 * @param new_cid Cargo type we are refitting to.
240 * @param new_subtype New cargo subtype.
241 * @param[out] auto_refit_allowed The refit is allowed as an auto-refit.
242 * @return Price for refitting
244 static CommandCost GetRefitCost(const Vehicle *v, EngineID engine_type, CargoID new_cid, byte new_subtype, bool *auto_refit_allowed)
246 ExpensesType expense_type;
247 const Engine *e = Engine::Get(engine_type);
248 Price base_price;
249 int cost_factor = GetRefitCostFactor(v, engine_type, new_cid, new_subtype, auto_refit_allowed);
250 switch (e->type) {
251 case VEH_SHIP:
252 base_price = PR_BUILD_VEHICLE_SHIP;
253 expense_type = EXPENSES_SHIP_RUN;
254 break;
256 case VEH_ROAD:
257 base_price = PR_BUILD_VEHICLE_ROAD;
258 expense_type = EXPENSES_ROADVEH_RUN;
259 break;
261 case VEH_AIRCRAFT:
262 base_price = PR_BUILD_VEHICLE_AIRCRAFT;
263 expense_type = EXPENSES_AIRCRAFT_RUN;
264 break;
266 case VEH_TRAIN:
267 base_price = (e->u.rail.railveh_type == RAILVEH_WAGON) ? PR_BUILD_VEHICLE_WAGON : PR_BUILD_VEHICLE_TRAIN;
268 cost_factor <<= 1;
269 expense_type = EXPENSES_TRAIN_RUN;
270 break;
272 default: NOT_REACHED();
274 if (cost_factor < 0) {
275 return CommandCost(expense_type, -GetPrice(base_price, -cost_factor, e->GetGRF(), -10));
276 } else {
277 return CommandCost(expense_type, GetPrice(base_price, cost_factor, e->GetGRF(), -10));
281 /** Helper structure for RefitVehicle() */
282 struct RefitResult {
283 Vehicle *v; ///< Vehicle to refit
284 uint capacity; ///< New capacity of vehicle
285 uint mail_capacity; ///< New mail capacity of aircraft
286 byte subtype; ///< cargo subtype to refit to
290 * Refits a vehicle (chain).
291 * This is the vehicle-type independent part of the CmdRefitXXX functions.
292 * @param v The vehicle to refit.
293 * @param only_this Whether to only refit this vehicle, or to check the rest of them.
294 * @param num_vehicles Number of vehicles to refit (not counting articulated parts). Zero means the whole chain.
295 * @param new_cid Cargotype to refit to
296 * @param new_subtype Cargo subtype to refit to. 0xFF means to try keeping the same subtype according to GetBestFittingSubType().
297 * @param flags Command flags
298 * @param auto_refit Refitting is done as automatic refitting outside a depot.
299 * @return Refit cost.
301 static CommandCost RefitVehicle(Vehicle *v, bool only_this, uint8 num_vehicles, CargoID new_cid, byte new_subtype, DoCommandFlag flags, bool auto_refit)
303 CommandCost cost(v->GetExpenseType(false));
304 uint total_capacity = 0;
305 uint total_mail_capacity = 0;
306 num_vehicles = num_vehicles == 0 ? UINT8_MAX : num_vehicles;
308 VehicleSet vehicles_to_refit;
309 if (!only_this) {
310 GetVehicleSet(vehicles_to_refit, v, num_vehicles);
311 /* In this case, we need to check the whole chain. */
312 v = v->First();
315 static SmallVector<RefitResult, 16> refit_result;
316 refit_result.Clear();
318 v->InvalidateNewGRFCacheOfChain();
319 byte actual_subtype = new_subtype;
320 for (; v != NULL; v = (only_this ? NULL : v->Next())) {
321 /* Reset actual_subtype for every new vehicle */
322 if (!v->IsArticulatedPart()) actual_subtype = new_subtype;
324 if (v->type == VEH_TRAIN && !vehicles_to_refit.Contains(v->index) && !only_this) continue;
326 const Engine *e = v->GetEngine();
327 if (!e->CanCarryCargo()) continue;
329 /* If the vehicle is not refittable, or does not allow automatic refitting,
330 * count its capacity nevertheless if the cargo matches */
331 bool refittable = HasBit(e->info.refit_mask, new_cid) && (!auto_refit || HasBit(e->info.misc_flags, EF_AUTO_REFIT));
332 if (!refittable && v->cargo_type != new_cid) continue;
334 /* Determine best fitting subtype if requested */
335 if (actual_subtype == 0xFF) {
336 actual_subtype = GetBestFittingSubType(v, v, new_cid);
339 /* Back up the vehicle's cargo type */
340 CargoID temp_cid = v->cargo_type;
341 byte temp_subtype = v->cargo_subtype;
342 if (refittable) {
343 v->cargo_type = new_cid;
344 v->cargo_subtype = actual_subtype;
347 uint16 mail_capacity = 0;
348 uint amount = e->DetermineCapacity(v, &mail_capacity);
349 total_capacity += amount;
350 /* mail_capacity will always be zero if the vehicle is not an aircraft. */
351 total_mail_capacity += mail_capacity;
353 if (!refittable) continue;
355 /* Restore the original cargo type */
356 v->cargo_type = temp_cid;
357 v->cargo_subtype = temp_subtype;
359 bool auto_refit_allowed;
360 CommandCost refit_cost = GetRefitCost(v, v->engine_type, new_cid, actual_subtype, &auto_refit_allowed);
361 if (auto_refit && (flags & DC_QUERY_COST) == 0 && !auto_refit_allowed) {
362 /* Sorry, auto-refitting not allowed, subtract the cargo amount again from the total.
363 * When querrying cost/capacity (for example in order refit GUI), we always assume 'allowed'.
364 * It is not predictable. */
365 total_capacity -= amount;
366 total_mail_capacity -= mail_capacity;
368 if (v->cargo_type == new_cid) {
369 /* Add the old capacity nevertheless, if the cargo matches */
370 total_capacity += v->cargo_cap;
371 if (v->type == VEH_AIRCRAFT) total_mail_capacity += v->Next()->cargo_cap;
373 continue;
375 cost.AddCost(refit_cost);
377 /* Record the refitting.
378 * Do not execute the refitting immediately, so DetermineCapacity and GetRefitCost do the same in test and exec run.
379 * (weird NewGRFs)
380 * Note:
381 * - If the capacity of vehicles depends on other vehicles in the chain, the actual capacity is
382 * set after RefitVehicle() via ConsistChanged() and friends. The estimation via _returned_refit_capacity will be wrong.
383 * - We have to call the refit cost callback with the pre-refit configuration of the chain because we want refit and
384 * autorefit to behave the same, and we need its result for auto_refit_allowed.
386 RefitResult *result = refit_result.Append();
387 result->v = v;
388 result->capacity = amount;
389 result->mail_capacity = mail_capacity;
390 result->subtype = actual_subtype;
393 if (flags & DC_EXEC) {
394 /* Store the result */
395 for (RefitResult *result = refit_result.Begin(); result != refit_result.End(); result++) {
396 Vehicle *u = result->v;
397 u->refit_cap = (u->cargo_type == new_cid) ? min(result->capacity, u->refit_cap) : 0;
398 if (u->cargo.TotalCount() > u->refit_cap) u->cargo.Truncate(u->cargo.TotalCount() - u->refit_cap);
399 u->cargo_type = new_cid;
400 u->cargo_cap = result->capacity;
401 u->cargo_subtype = result->subtype;
402 if (u->type == VEH_AIRCRAFT) {
403 Vehicle *w = u->Next();
404 w->refit_cap = min(w->refit_cap, result->mail_capacity);
405 w->cargo_cap = result->mail_capacity;
406 if (w->cargo.TotalCount() > w->refit_cap) w->cargo.Truncate(w->cargo.TotalCount() - w->refit_cap);
411 refit_result.Clear();
412 _returned_refit_capacity = total_capacity;
413 _returned_mail_refit_capacity = total_mail_capacity;
414 return cost;
418 * Refits a vehicle to the specified cargo type.
419 * @param tile unused
420 * @param flags type of operation
421 * @param p1 vehicle ID to refit
422 * @param p2 various bitstuffed elements
423 * - p2 = (bit 0-4) - New cargo type to refit to.
424 * - p2 = (bit 6) - Automatic refitting.
425 * - p2 = (bit 7) - Refit only this vehicle. Used only for cloning vehicles.
426 * - p2 = (bit 8-15) - New cargo subtype to refit to. 0xFF means to try keeping the same subtype according to GetBestFittingSubType().
427 * - p2 = (bit 16-23) - Number of vehicles to refit (not counting articulated parts). Zero means all vehicles.
428 * Only used if "refit only this vehicle" is false.
429 * @param text unused
430 * @return the cost of this operation or an error
432 CommandCost CmdRefitVehicle(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
434 Vehicle *v = Vehicle::GetIfValid(p1);
435 if (v == NULL) return CMD_ERROR;
437 /* Don't allow disasters and sparks and such to be refitted.
438 * We cannot check for IsPrimaryVehicle as autoreplace also refits in free wagon chains. */
439 if (!IsCompanyBuildableVehicleType(v->type)) return CMD_ERROR;
441 Vehicle *front = v->First();
443 CommandCost ret = CheckOwnership(front->owner);
444 if (ret.Failed()) return ret;
446 bool auto_refit = HasBit(p2, 6);
447 bool free_wagon = v->type == VEH_TRAIN && Train::From(front)->IsFreeWagon(); // used by autoreplace/renew
449 /* Don't allow shadows and such to be refitted. */
450 if (v != front && (v->type == VEH_SHIP || v->type == VEH_AIRCRAFT)) return CMD_ERROR;
452 /* Allow auto-refitting only during loading and normal refitting only in a depot. */
453 if ((flags & DC_QUERY_COST) == 0 && // used by the refit GUI, including the order refit GUI.
454 !free_wagon && // used by autoreplace/renew
455 (!auto_refit || !front->current_order.IsType(OT_LOADING)) && // refit inside stations
456 !front->IsStoppedInDepot()) { // refit inside depots
457 return_cmd_error(STR_ERROR_TRAIN_MUST_BE_STOPPED_INSIDE_DEPOT + front->type);
460 if (front->vehstatus & VS_CRASHED) return_cmd_error(STR_ERROR_VEHICLE_IS_DESTROYED);
462 /* Check cargo */
463 CargoID new_cid = GB(p2, 0, 5);
464 byte new_subtype = GB(p2, 8, 8);
465 if (new_cid >= NUM_CARGO) return CMD_ERROR;
467 /* For ships and aircraft there is always only one. */
468 bool only_this = HasBit(p2, 7) || front->type == VEH_SHIP || front->type == VEH_AIRCRAFT;
469 uint8 num_vehicles = GB(p2, 16, 8);
471 CommandCost cost = RefitVehicle(v, only_this, num_vehicles, new_cid, new_subtype, flags, auto_refit);
473 if (flags & DC_EXEC) {
474 /* Update the cached variables */
475 switch (v->type) {
476 case VEH_TRAIN:
477 Train::From(front)->ConsistChanged(auto_refit ? CCF_AUTOREFIT : CCF_REFIT);
478 break;
479 case VEH_ROAD:
480 RoadVehUpdateCache(RoadVehicle::From(front), auto_refit);
481 if (_settings_game.vehicle.roadveh_acceleration_model != AM_ORIGINAL) RoadVehicle::From(front)->CargoChanged();
482 break;
484 case VEH_SHIP:
485 v->InvalidateNewGRFCacheOfChain();
486 Ship::From(v)->UpdateCache();
487 break;
489 case VEH_AIRCRAFT:
490 v->InvalidateNewGRFCacheOfChain();
491 UpdateAircraftCache(Aircraft::From(v), true);
492 break;
494 default: NOT_REACHED();
496 front->MarkDirty();
498 if (!free_wagon) {
499 InvalidateWindowData(WC_VEHICLE_DETAILS, front->index);
500 InvalidateWindowClassesData(GetWindowClassForVehicleType(v->type), 0);
502 SetWindowDirty(WC_VEHICLE_DEPOT, front->tile);
503 } else {
504 /* Always invalidate the cache; querycost might have filled it. */
505 v->InvalidateNewGRFCacheOfChain();
508 return cost;
512 * Start/Stop a vehicle
513 * @param tile unused
514 * @param flags type of operation
515 * @param p1 vehicle to start/stop, don't forget to change CcStartStopVehicle if you modify this!
516 * @param p2 bit 0: Shall the start/stop newgrf callback be evaluated (only valid with DC_AUTOREPLACE for network safety)
517 * @param text unused
518 * @return the cost of this operation or an error
520 CommandCost CmdStartStopVehicle(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
522 /* Disable the effect of p2 bit 0, when DC_AUTOREPLACE is not set */
523 if ((flags & DC_AUTOREPLACE) == 0) SetBit(p2, 0);
525 Vehicle *v = Vehicle::GetIfValid(p1);
526 if (v == NULL || !v->IsPrimaryVehicle()) return CMD_ERROR;
528 CommandCost ret = CheckOwnership(v->owner);
529 if (ret.Failed()) return ret;
531 if (v->vehstatus & VS_CRASHED) return_cmd_error(STR_ERROR_VEHICLE_IS_DESTROYED);
533 switch (v->type) {
534 case VEH_TRAIN:
535 if ((v->vehstatus & VS_STOPPED) && Train::From(v)->gcache.cached_power == 0) return_cmd_error(STR_ERROR_TRAIN_START_NO_POWER);
536 break;
538 case VEH_SHIP:
539 case VEH_ROAD:
540 break;
542 case VEH_AIRCRAFT: {
543 Aircraft *a = Aircraft::From(v);
544 /* cannot stop airplane when in flight, or when taking off / landing */
545 if (a->state >= STARTTAKEOFF && a->state < TERM7) return_cmd_error(STR_ERROR_AIRCRAFT_IS_IN_FLIGHT);
546 if (HasBit(a->flags, VAF_HELI_DIRECT_DESCENT)) return_cmd_error(STR_ERROR_AIRCRAFT_IS_IN_FLIGHT);
547 break;
550 default: return CMD_ERROR;
553 if (HasBit(p2, 0)) {
554 /* Check if this vehicle can be started/stopped. Failure means 'allow'. */
555 uint16 callback = GetVehicleCallback(CBID_VEHICLE_START_STOP_CHECK, 0, 0, v->engine_type, v);
556 StringID error = STR_NULL;
557 if (callback != CALLBACK_FAILED) {
558 if (v->GetGRF()->grf_version < 8) {
559 /* 8 bit result 0xFF means 'allow' */
560 if (callback < 0x400 && GB(callback, 0, 8) != 0xFF) error = GetGRFStringID(v->GetGRFID(), 0xD000 + callback);
561 } else {
562 if (callback < 0x400) {
563 error = GetGRFStringID(v->GetGRFID(), 0xD000 + callback);
564 } else {
565 switch (callback) {
566 case 0x400: // allow
567 break;
569 default: // unknown reason -> disallow
570 error = STR_ERROR_INCOMPATIBLE_RAIL_TYPES;
571 break;
576 if (error != STR_NULL) return_cmd_error(error);
579 if (flags & DC_EXEC) {
580 if (v->IsStoppedInDepot() && (flags & DC_AUTOREPLACE) == 0) DeleteVehicleNews(p1, STR_NEWS_TRAIN_IS_WAITING + v->type);
582 v->vehstatus ^= VS_STOPPED;
583 if (v->type != VEH_TRAIN) v->cur_speed = 0; // trains can stop 'slowly'
584 v->MarkDirty();
585 SetWindowWidgetDirty(WC_VEHICLE_VIEW, v->index, WID_VV_START_STOP);
586 SetWindowDirty(WC_VEHICLE_DEPOT, v->tile);
587 SetWindowClassesDirty(GetWindowClassForVehicleType(v->type));
589 return CommandCost();
593 * Starts or stops a lot of vehicles
594 * @param tile Tile of the depot where the vehicles are started/stopped (only used for depots)
595 * @param flags type of operation
596 * @param p1 bitmask
597 * - bit 0 set = start vehicles, unset = stop vehicles
598 * - bit 1 if set, then it's a vehicle list window, not a depot and Tile is ignored in this case
599 * @param p2 packed VehicleListIdentifier
600 * @param text unused
601 * @return the cost of this operation or an error
603 CommandCost CmdMassStartStopVehicle(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
605 VehicleList list;
606 bool do_start = HasBit(p1, 0);
607 bool vehicle_list_window = HasBit(p1, 1);
609 VehicleListIdentifier vli;
610 if (!vli.UnpackIfValid(p2)) return CMD_ERROR;
611 if (!IsCompanyBuildableVehicleType(vli.vtype)) return CMD_ERROR;
613 if (vehicle_list_window) {
614 if (!GenerateVehicleSortList(&list, vli)) return CMD_ERROR;
615 } else {
616 /* Get the list of vehicles in the depot */
617 BuildDepotVehicleList(vli.vtype, tile, &list, NULL);
620 for (uint i = 0; i < list.Length(); i++) {
621 const Vehicle *v = list[i];
623 if (!!(v->vehstatus & VS_STOPPED) != do_start) continue;
625 if (!vehicle_list_window && !v->IsChainInDepot()) continue;
627 /* Just try and don't care if some vehicle's can't be stopped. */
628 DoCommand(tile, v->index, 0, flags, CMD_START_STOP_VEHICLE);
631 return CommandCost();
635 * Sells all vehicles in a depot
636 * @param tile Tile of the depot where the depot is
637 * @param flags type of operation
638 * @param p1 Vehicle type
639 * @param p2 unused
640 * @param text unused
641 * @return the cost of this operation or an error
643 CommandCost CmdDepotSellAllVehicles(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
645 VehicleList list;
647 CommandCost cost(EXPENSES_NEW_VEHICLES);
648 VehicleType vehicle_type = Extract<VehicleType, 0, 3>(p1);
650 if (!IsCompanyBuildableVehicleType(vehicle_type)) return CMD_ERROR;
652 uint sell_command = GetCmdSellVeh(vehicle_type);
654 /* Get the list of vehicles in the depot */
655 BuildDepotVehicleList(vehicle_type, tile, &list, &list);
657 CommandCost last_error = CMD_ERROR;
658 bool had_success = false;
659 for (uint i = 0; i < list.Length(); i++) {
660 CommandCost ret = DoCommand(tile, list[i]->index | (1 << 20), 0, flags, sell_command);
661 if (ret.Succeeded()) {
662 cost.AddCost(ret);
663 had_success = true;
664 } else {
665 last_error = ret;
669 return had_success ? cost : last_error;
673 * Autoreplace all vehicles in the depot
674 * @param tile Tile of the depot where the vehicles are
675 * @param flags type of operation
676 * @param p1 Type of vehicle
677 * @param p2 unused
678 * @param text unused
679 * @return the cost of this operation or an error
681 CommandCost CmdDepotMassAutoReplace(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
683 VehicleList list;
684 CommandCost cost = CommandCost(EXPENSES_NEW_VEHICLES);
685 VehicleType vehicle_type = Extract<VehicleType, 0, 3>(p1);
687 if (!IsCompanyBuildableVehicleType(vehicle_type)) return CMD_ERROR;
688 if (!IsDepotTile(tile) || !IsTileOwner(tile, _current_company)) return CMD_ERROR;
690 /* Get the list of vehicles in the depot */
691 BuildDepotVehicleList(vehicle_type, tile, &list, &list, true);
693 for (uint i = 0; i < list.Length(); i++) {
694 const Vehicle *v = list[i];
696 /* Ensure that the vehicle completely in the depot */
697 if (!v->IsChainInDepot()) continue;
699 CommandCost ret = DoCommand(0, v->index, 0, flags, CMD_AUTOREPLACE_VEHICLE);
701 if (ret.Succeeded()) cost.AddCost(ret);
703 return cost;
707 * Test if a name is unique among vehicle names.
708 * @param name Name to test.
709 * @return True ifffffff the name is unique.
711 static bool IsUniqueVehicleName(const char *name)
713 const Vehicle *v;
715 FOR_ALL_VEHICLES(v) {
716 if (v->name != NULL && strcmp(v->name, name) == 0) return false;
719 return true;
723 * Clone the custom name of a vehicle, adding or incrementing a number.
724 * @param src Source vehicle, with a custom name.
725 * @param dst Destination vehicle.
727 static void CloneVehicleName(const Vehicle *src, Vehicle *dst)
729 char buf[256];
731 /* Find the position of the first digit in the last group of digits. */
732 size_t number_position;
733 for (number_position = strlen(src->name); number_position > 0; number_position--) {
734 /* The design of UTF-8 lets this work simply without having to check
735 * for UTF-8 sequences. */
736 if (src->name[number_position - 1] < '0' || src->name[number_position - 1] > '9') break;
739 /* Format buffer and determine starting number. */
740 int num;
741 byte padding = 0;
742 if (number_position == strlen(src->name)) {
743 /* No digit at the end, so start at number 2. */
744 strecpy(buf, src->name, lastof(buf));
745 strecat(buf, " ", lastof(buf));
746 number_position = strlen(buf);
747 num = 2;
748 } else {
749 /* Found digits, parse them and start at the next number. */
750 strecpy(buf, src->name, lastof(buf));
751 buf[number_position] = '\0';
752 char *endptr;
753 num = strtol(&src->name[number_position], &endptr, 10) + 1;
754 padding = endptr - &src->name[number_position];
757 /* Check if this name is already taken. */
758 for (int max_iterations = 1000; max_iterations > 0; max_iterations--, num++) {
759 /* Attach the number to the temporary name. */
760 seprintf(&buf[number_position], lastof(buf), "%0*d", padding, num);
762 /* Check the name is unique. */
763 if (IsUniqueVehicleName(buf)) {
764 dst->name = stredup(buf);
765 break;
769 /* All done. If we didn't find a name, it'll just use its default. */
773 * Clone a vehicle. If it is a train, it will clone all the cars too
774 * @param tile tile of the depot where the cloned vehicle is build
775 * @param flags type of operation
776 * @param p1 the original vehicle's index
777 * @param p2 1 = shared orders, else copied orders
778 * @param text unused
779 * @return the cost of this operation or an error
781 CommandCost CmdCloneVehicle(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
783 CommandCost total_cost(EXPENSES_NEW_VEHICLES);
785 Vehicle *v = Vehicle::GetIfValid(p1);
786 if (v == NULL || !v->IsPrimaryVehicle()) return CMD_ERROR;
787 Vehicle *v_front = v;
788 Vehicle *w = NULL;
789 Vehicle *w_front = NULL;
790 Vehicle *w_rear = NULL;
793 * v_front is the front engine in the original vehicle
794 * v is the car/vehicle of the original vehicle that is currently being copied
795 * w_front is the front engine of the cloned vehicle
796 * w is the car/vehicle currently being cloned
797 * w_rear is the rear end of the cloned train. It's used to add more cars and is only used by trains
800 CommandCost ret = CheckOwnership(v->owner);
801 if (ret.Failed()) return ret;
803 if (v->type == VEH_TRAIN && (!v->IsFrontEngine() || Train::From(v)->crash_anim_pos >= 4400)) return CMD_ERROR;
805 /* check that we can allocate enough vehicles */
806 if (!(flags & DC_EXEC)) {
807 int veh_counter = 0;
808 do {
809 veh_counter++;
810 } while ((v = v->Next()) != NULL);
812 if (!Vehicle::CanAllocateItem(veh_counter)) {
813 return_cmd_error(STR_ERROR_TOO_MANY_VEHICLES_IN_GAME);
817 v = v_front;
819 do {
820 if (v->type == VEH_TRAIN && Train::From(v)->IsRearDualheaded()) {
821 /* we build the rear ends of multiheaded trains with the front ones */
822 continue;
825 /* In case we're building a multi headed vehicle and the maximum number of
826 * vehicles is almost reached (e.g. max trains - 1) not all vehicles would
827 * be cloned. When the non-primary engines were build they were seen as
828 * 'new' vehicles whereas they would immediately be joined with a primary
829 * engine. This caused the vehicle to be not build as 'the limit' had been
830 * reached, resulting in partially build vehicles and such. */
831 DoCommandFlag build_flags = flags;
832 if ((flags & DC_EXEC) && !v->IsPrimaryVehicle()) build_flags |= DC_AUTOREPLACE;
834 CommandCost cost = DoCommand(tile, v->engine_type | (1 << 16), 0, build_flags, GetCmdBuildVeh(v));
836 if (cost.Failed()) {
837 /* Can't build a part, then sell the stuff we already made; clear up the mess */
838 if (w_front != NULL) DoCommand(w_front->tile, w_front->index | (1 << 20), 0, flags, GetCmdSellVeh(w_front));
839 return cost;
842 total_cost.AddCost(cost);
844 if (flags & DC_EXEC) {
845 w = Vehicle::Get(_new_vehicle_id);
847 if (v->type == VEH_TRAIN && HasBit(Train::From(v)->flags, VRF_REVERSE_DIRECTION)) {
848 SetBit(Train::From(w)->flags, VRF_REVERSE_DIRECTION);
851 if (v->type == VEH_TRAIN && !v->IsFrontEngine()) {
852 /* this s a train car
853 * add this unit to the end of the train */
854 CommandCost result = DoCommand(0, w->index | 1 << 20, w_rear->index, flags, CMD_MOVE_RAIL_VEHICLE);
855 if (result.Failed()) {
856 /* The train can't be joined to make the same consist as the original.
857 * Sell what we already made (clean up) and return an error. */
858 DoCommand(w_front->tile, w_front->index | 1 << 20, 0, flags, GetCmdSellVeh(w_front));
859 DoCommand(w_front->tile, w->index | 1 << 20, 0, flags, GetCmdSellVeh(w));
860 return result; // return error and the message returned from CMD_MOVE_RAIL_VEHICLE
862 } else {
863 /* this is a front engine or not a train. */
864 w_front = w;
865 w->service_interval = v->service_interval;
866 w->SetServiceIntervalIsCustom(v->ServiceIntervalIsCustom());
867 w->SetServiceIntervalIsPercent(v->ServiceIntervalIsPercent());
869 w_rear = w; // trains needs to know the last car in the train, so they can add more in next loop
871 } while (v->type == VEH_TRAIN && (v = v->GetNextVehicle()) != NULL);
873 if ((flags & DC_EXEC) && v_front->type == VEH_TRAIN) {
874 /* for trains this needs to be the front engine due to the callback function */
875 _new_vehicle_id = w_front->index;
878 if (flags & DC_EXEC) {
879 /* Cloned vehicles belong to the same group */
880 DoCommand(0, v_front->group_id, w_front->index, flags, CMD_ADD_VEHICLE_GROUP);
884 /* Take care of refitting. */
885 w = w_front;
886 v = v_front;
888 /* Both building and refitting are influenced by newgrf callbacks, which
889 * makes it impossible to accurately estimate the cloning costs. In
890 * particular, it is possible for engines of the same type to be built with
891 * different numbers of articulated parts, so when refitting we have to
892 * loop over real vehicles first, and then the articulated parts of those
893 * vehicles in a different loop. */
894 do {
895 do {
896 if (flags & DC_EXEC) {
897 assert(w != NULL);
899 /* Find out what's the best sub type */
900 byte subtype = GetBestFittingSubType(v, w, v->cargo_type);
901 if (w->cargo_type != v->cargo_type || w->cargo_subtype != subtype) {
902 CommandCost cost = DoCommand(0, w->index, v->cargo_type | 1U << 7 | (subtype << 8), flags, GetCmdRefitVeh(v));
903 if (cost.Succeeded()) total_cost.AddCost(cost);
906 if (w->IsGroundVehicle() && w->HasArticulatedPart()) {
907 w = w->GetNextArticulatedPart();
908 } else {
909 break;
911 } else {
912 const Engine *e = v->GetEngine();
913 CargoID initial_cargo = (e->CanCarryCargo() ? e->GetDefaultCargoType() : (CargoID)CT_INVALID);
915 if (v->cargo_type != initial_cargo && initial_cargo != CT_INVALID) {
916 bool dummy;
917 total_cost.AddCost(GetRefitCost(NULL, v->engine_type, v->cargo_type, v->cargo_subtype, &dummy));
921 if (v->IsGroundVehicle() && v->HasArticulatedPart()) {
922 v = v->GetNextArticulatedPart();
923 } else {
924 break;
926 } while (v != NULL);
928 if ((flags & DC_EXEC) && v->type == VEH_TRAIN) w = w->GetNextVehicle();
929 } while (v->type == VEH_TRAIN && (v = v->GetNextVehicle()) != NULL);
931 if (flags & DC_EXEC) {
933 * Set the orders of the vehicle. Cannot do it earlier as we need
934 * the vehicle refitted before doing this, otherwise the moved
935 * cargo types might not match (passenger vs non-passenger)
937 DoCommand(0, w_front->index | (p2 & 1 ? CO_SHARE : CO_COPY) << 30, v_front->index, flags, CMD_CLONE_ORDER);
939 /* Now clone the vehicle's name, if it has one. */
940 if (v_front->name != NULL) CloneVehicleName(v_front, w_front);
943 /* Since we can't estimate the cost of cloning a vehicle accurately we must
944 * check whether the company has enough money manually. */
945 if (!CheckCompanyHasMoney(total_cost)) {
946 if (flags & DC_EXEC) {
947 /* The vehicle has already been bought, so now it must be sold again. */
948 DoCommand(w_front->tile, w_front->index | 1 << 20, 0, flags, GetCmdSellVeh(w_front));
950 return total_cost;
953 return total_cost;
957 * Send all vehicles of type to depots
958 * @param flags the flags used for DoCommand()
959 * @param service should the vehicles only get service in the depots
960 * @param vli identifier of the vehicle list
961 * @return 0 for success and CMD_ERROR if no vehicle is able to go to depot
963 static CommandCost SendAllVehiclesToDepot(DoCommandFlag flags, bool service, const VehicleListIdentifier &vli)
965 VehicleList list;
967 if (!GenerateVehicleSortList(&list, vli)) return CMD_ERROR;
969 /* Send all the vehicles to a depot */
970 bool had_success = false;
971 for (uint i = 0; i < list.Length(); i++) {
972 const Vehicle *v = list[i];
973 CommandCost ret = DoCommand(v->tile, v->index | (service ? DEPOT_SERVICE : 0U) | DEPOT_DONT_CANCEL, 0, flags, GetCmdSendToDepot(vli.vtype));
975 if (ret.Succeeded()) {
976 had_success = true;
978 /* Return 0 if DC_EXEC is not set this is a valid goto depot command)
979 * In this case we know that at least one vehicle can be sent to a depot
980 * and we will issue the command. We can now safely quit the loop, knowing
981 * it will succeed at least once. With DC_EXEC we really need to send them to the depot */
982 if (!(flags & DC_EXEC)) break;
986 return had_success ? CommandCost() : CMD_ERROR;
990 * Send a vehicle to the depot.
991 * @param tile unused
992 * @param flags for command type
993 * @param p1 bitmask
994 * - p1 0-20: bitvehicle ID to send to the depot
995 * - p1 bits 25-8 - DEPOT_ flags (see vehicle_type.h)
996 * @param p2 packed VehicleListIdentifier.
997 * @param text unused
998 * @return the cost of this operation or an error
1000 CommandCost CmdSendVehicleToDepot(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
1002 if (p1 & DEPOT_MASS_SEND) {
1003 /* Mass goto depot requested */
1004 VehicleListIdentifier vli;
1005 if (!vli.UnpackIfValid(p2)) return CMD_ERROR;
1006 return SendAllVehiclesToDepot(flags, (p1 & DEPOT_SERVICE) != 0, vli);
1009 Vehicle *v = Vehicle::GetIfValid(GB(p1, 0, 20));
1010 if (v == NULL) return CMD_ERROR;
1011 if (!v->IsPrimaryVehicle()) return CMD_ERROR;
1013 return v->SendToDepot(flags, (DepotCommand)(p1 & DEPOT_COMMAND_MASK));
1017 * Give a custom name to your vehicle
1018 * @param tile unused
1019 * @param flags type of operation
1020 * @param p1 vehicle ID to name
1021 * @param p2 unused
1022 * @param text the new name or an empty string when resetting to the default
1023 * @return the cost of this operation or an error
1025 CommandCost CmdRenameVehicle(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
1027 Vehicle *v = Vehicle::GetIfValid(p1);
1028 if (v == NULL || !v->IsPrimaryVehicle()) return CMD_ERROR;
1030 CommandCost ret = CheckOwnership(v->owner);
1031 if (ret.Failed()) return ret;
1033 bool reset = StrEmpty(text);
1035 if (!reset) {
1036 if (Utf8StringLength(text) >= MAX_LENGTH_VEHICLE_NAME_CHARS) return CMD_ERROR;
1037 if (!(flags & DC_AUTOREPLACE) && !IsUniqueVehicleName(text)) return_cmd_error(STR_ERROR_NAME_MUST_BE_UNIQUE);
1040 if (flags & DC_EXEC) {
1041 free(v->name);
1042 v->name = reset ? NULL : stredup(text);
1043 InvalidateWindowClassesData(GetWindowClassForVehicleType(v->type), 1);
1044 MarkWholeScreenDirty();
1047 return CommandCost();
1052 * Change the service interval of a vehicle
1053 * @param tile unused
1054 * @param flags type of operation
1055 * @param p1 vehicle ID that is being service-interval-changed
1056 * @param p2 bitmask
1057 * - p2 = (bit 0-15) - new service interval
1058 * - p2 = (bit 16) - service interval is custom flag
1059 * - p2 = (bit 17) - service interval is percentage flag
1060 * @param text unused
1061 * @return the cost of this operation or an error
1063 CommandCost CmdChangeServiceInt(TileIndex tile, DoCommandFlag flags, uint32 p1, uint32 p2, const char *text)
1065 Vehicle *v = Vehicle::GetIfValid(p1);
1066 if (v == NULL || !v->IsPrimaryVehicle()) return CMD_ERROR;
1068 CommandCost ret = CheckOwnership(v->owner);
1069 if (ret.Failed()) return ret;
1071 const Company *company = Company::Get(v->owner);
1072 bool iscustom = HasBit(p2, 16);
1073 bool ispercent = iscustom ? HasBit(p2, 17) : company->settings.vehicle.servint_ispercent;
1075 uint16 serv_int;
1076 if (iscustom) {
1077 serv_int = GB(p2, 0, 16);
1078 if (serv_int != GetServiceIntervalClamped(serv_int, ispercent)) return CMD_ERROR;
1079 } else {
1080 serv_int = CompanyServiceInterval(company, v->type);
1083 if (flags & DC_EXEC) {
1084 v->SetServiceInterval(serv_int);
1085 v->SetServiceIntervalIsCustom(iscustom);
1086 v->SetServiceIntervalIsPercent(ispercent);
1087 SetWindowDirty(WC_VEHICLE_DETAILS, v->index);
1090 return CommandCost();