(svn r27756) -Codechange: Add StringTab enum
[openttd.git] / src / newgrf_station.cpp
blobcd5dad7b47a0b27c04d832cf9fe8d8614be6e536
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 newgrf_station.cpp Functions for dealing with station classes and custom stations. */
12 #include "stdafx.h"
13 #include "debug.h"
14 #include "station_base.h"
15 #include "waypoint_base.h"
16 #include "roadstop_base.h"
17 #include "newgrf_cargo.h"
18 #include "newgrf_station.h"
19 #include "newgrf_spritegroup.h"
20 #include "newgrf_sound.h"
21 #include "newgrf_railtype.h"
22 #include "town.h"
23 #include "newgrf_town.h"
24 #include "company_func.h"
25 #include "tunnelbridge_map.h"
26 #include "newgrf_animation_base.h"
27 #include "newgrf_class_func.h"
29 #include "safeguards.h"
32 template <typename Tspec, typename Tid, Tid Tmax>
33 /* static */ void NewGRFClass<Tspec, Tid, Tmax>::InsertDefaults()
35 /* Set up initial data */
36 classes[0].global_id = 'DFLT';
37 classes[0].name = STR_STATION_CLASS_DFLT;
38 classes[0].Insert(NULL);
40 classes[1].global_id = 'WAYP';
41 classes[1].name = STR_STATION_CLASS_WAYP;
42 classes[1].Insert(NULL);
45 template <typename Tspec, typename Tid, Tid Tmax>
46 bool NewGRFClass<Tspec, Tid, Tmax>::IsUIAvailable(uint index) const
48 return true;
51 INSTANTIATE_NEWGRF_CLASS_METHODS(StationClass, StationSpec, StationClassID, STAT_CLASS_MAX)
53 static const uint NUM_STATIONSSPECS_PER_STATION = 255; ///< Maximum number of parts per station.
55 enum TriggerArea {
56 TA_TILE,
57 TA_PLATFORM,
58 TA_WHOLE,
61 struct ETileArea : TileArea {
62 ETileArea(const BaseStation *st, TileIndex tile, TriggerArea ta)
64 switch (ta) {
65 default: NOT_REACHED();
67 case TA_TILE:
68 this->tile = tile;
69 this->w = 1;
70 this->h = 1;
71 break;
73 case TA_PLATFORM: {
74 TileIndex start, end;
75 Axis axis = GetRailStationAxis(tile);
76 TileIndexDiff delta = TileOffsByDiagDir(AxisToDiagDir(axis));
78 for (end = tile; IsRailStationTile(end + delta) && IsCompatibleTrainStationTile(end + delta, tile); end += delta) { /* Nothing */ }
79 for (start = tile; IsRailStationTile(start - delta) && IsCompatibleTrainStationTile(start - delta, tile); start -= delta) { /* Nothing */ }
81 this->tile = start;
82 this->w = TileX(end) - TileX(start) + 1;
83 this->h = TileY(end) - TileY(start) + 1;
84 break;
87 case TA_WHOLE:
88 st->GetTileArea(this, Station::IsExpected(st) ? STATION_RAIL : STATION_WAYPOINT);
89 break;
95 /**
96 * Evaluate a tile's position within a station, and return the result in a bit-stuffed format.
97 * if not centered: .TNLcCpP, if centered: .TNL..CP
98 * - T = Tile layout number (#GetStationGfx)
99 * - N = Number of platforms
100 * - L = Length of platforms
101 * - C = Current platform number from start, c = from end
102 * - P = Position along platform from start, p = from end
104 * if centered, C/P start from the centre and c/p are not available.
105 * @return Platform information in bit-stuffed format.
107 uint32 GetPlatformInfo(Axis axis, byte tile, int platforms, int length, int x, int y, bool centred)
109 uint32 retval = 0;
111 if (axis == AXIS_X) {
112 Swap(platforms, length);
113 Swap(x, y);
116 if (centred) {
117 x -= platforms / 2;
118 y -= length / 2;
119 x = Clamp(x, -8, 7);
120 y = Clamp(y, -8, 7);
121 SB(retval, 0, 4, y & 0xF);
122 SB(retval, 4, 4, x & 0xF);
123 } else {
124 SB(retval, 0, 4, min(15, y));
125 SB(retval, 4, 4, min(15, length - y - 1));
126 SB(retval, 8, 4, min(15, x));
127 SB(retval, 12, 4, min(15, platforms - x - 1));
129 SB(retval, 16, 4, min(15, length));
130 SB(retval, 20, 4, min(15, platforms));
131 SB(retval, 24, 4, tile);
133 return retval;
138 * Find the end of a railway station, from the \a tile, in the direction of \a delta.
139 * @param tile Start tile.
140 * @param delta Movement direction.
141 * @param check_type Stop when the custom station type changes.
142 * @param check_axis Stop when the station direction changes.
143 * @return Found end of the railway station.
145 static TileIndex FindRailStationEnd(TileIndex tile, TileIndexDiff delta, bool check_type, bool check_axis)
147 byte orig_type = 0;
148 Axis orig_axis = AXIS_X;
149 StationID sid = GetStationIndex(tile);
151 if (check_type) orig_type = GetCustomStationSpecIndex(tile);
152 if (check_axis) orig_axis = GetRailStationAxis(tile);
154 for (;;) {
155 TileIndex new_tile = TILE_ADD(tile, delta);
157 if (!IsTileType(new_tile, MP_STATION) || GetStationIndex(new_tile) != sid) break;
158 if (!HasStationRail(new_tile)) break;
159 if (check_type && GetCustomStationSpecIndex(new_tile) != orig_type) break;
160 if (check_axis && GetRailStationAxis(new_tile) != orig_axis) break;
162 tile = new_tile;
164 return tile;
168 static uint32 GetPlatformInfoHelper(TileIndex tile, bool check_type, bool check_axis, bool centred)
170 int tx = TileX(tile);
171 int ty = TileY(tile);
172 int sx = TileX(FindRailStationEnd(tile, TileDiffXY(-1, 0), check_type, check_axis));
173 int sy = TileY(FindRailStationEnd(tile, TileDiffXY( 0, -1), check_type, check_axis));
174 int ex = TileX(FindRailStationEnd(tile, TileDiffXY( 1, 0), check_type, check_axis)) + 1;
175 int ey = TileY(FindRailStationEnd(tile, TileDiffXY( 0, 1), check_type, check_axis)) + 1;
177 tx -= sx; ex -= sx;
178 ty -= sy; ey -= sy;
180 return GetPlatformInfo(GetRailStationAxis(tile), GetStationGfx(tile), ex, ey, tx, ty, centred);
184 static uint32 GetRailContinuationInfo(TileIndex tile)
186 /* Tile offsets and exit dirs for X axis */
187 static const Direction x_dir[8] = { DIR_SW, DIR_NE, DIR_SE, DIR_NW, DIR_S, DIR_E, DIR_W, DIR_N };
188 static const DiagDirection x_exits[8] = { DIAGDIR_SW, DIAGDIR_NE, DIAGDIR_SE, DIAGDIR_NW, DIAGDIR_SW, DIAGDIR_NE, DIAGDIR_SW, DIAGDIR_NE };
190 /* Tile offsets and exit dirs for Y axis */
191 static const Direction y_dir[8] = { DIR_SE, DIR_NW, DIR_SW, DIR_NE, DIR_S, DIR_W, DIR_E, DIR_N };
192 static const DiagDirection y_exits[8] = { DIAGDIR_SE, DIAGDIR_NW, DIAGDIR_SW, DIAGDIR_NE, DIAGDIR_SE, DIAGDIR_NW, DIAGDIR_SE, DIAGDIR_NW };
194 Axis axis = GetRailStationAxis(tile);
196 /* Choose appropriate lookup table to use */
197 const Direction *dir = axis == AXIS_X ? x_dir : y_dir;
198 const DiagDirection *diagdir = axis == AXIS_X ? x_exits : y_exits;
200 uint32 res = 0;
201 uint i;
203 for (i = 0; i < lengthof(x_dir); i++, dir++, diagdir++) {
204 TileIndex neighbour_tile = tile + TileOffsByDir(*dir);
205 TrackBits trackbits = TrackStatusToTrackBits(GetTileTrackStatus(neighbour_tile, TRANSPORT_RAIL, 0));
206 if (trackbits != TRACK_BIT_NONE) {
207 /* If there is any track on the tile, set the bit in the second byte */
208 SetBit(res, i + 8);
210 /* With tunnels and bridges the tile has tracks, but they are not necessarily connected
211 * with the next tile because the ramp is not going in the right direction. */
212 if (IsTileType(neighbour_tile, MP_TUNNELBRIDGE) && GetTunnelBridgeDirection(neighbour_tile) != *diagdir) {
213 continue;
216 /* If any track reaches our exit direction, set the bit in the lower byte */
217 if (trackbits & DiagdirReachesTracks(*diagdir)) SetBit(res, i);
221 return res;
225 /* Station Resolver Functions */
226 /* virtual */ uint32 StationScopeResolver::GetRandomBits() const
228 return (this->st == NULL ? 0 : this->st->random_bits) | (this->tile == INVALID_TILE ? 0 : GetStationTileRandomBits(this->tile) << 16);
232 /* virtual */ uint32 StationScopeResolver::GetTriggers() const
234 return this->st == NULL ? 0 : this->st->waiting_triggers;
238 /* virtual */ void StationScopeResolver::SetTriggers(int triggers) const
240 BaseStation *st = const_cast<BaseStation *>(this->st);
241 assert(st != NULL);
242 st->waiting_triggers = triggers;
246 * Station variable cache
247 * This caches 'expensive' station variable lookups which iterate over
248 * several tiles that may be called multiple times per Resolve().
250 static struct {
251 uint32 v40;
252 uint32 v41;
253 uint32 v45;
254 uint32 v46;
255 uint32 v47;
256 uint32 v49;
257 uint8 valid; ///< Bits indicating what variable is valid (for each bit, \c 0 is invalid, \c 1 is valid).
258 } _svc;
261 * Get the town scope associated with a station, if it exists.
262 * On the first call, the town scope is created (if possible).
263 * @return Town scope, if available.
265 TownScopeResolver *StationResolverObject::GetTown()
267 if (this->town_scope == NULL) {
268 Town *t = NULL;
269 if (this->station_scope.st != NULL) {
270 t = this->station_scope.st->town;
271 } else if (this->station_scope.tile != INVALID_TILE) {
272 t = ClosestTownFromTile(this->station_scope.tile, UINT_MAX);
274 if (t == NULL) return NULL;
275 this->town_scope = new TownScopeResolver(*this, t, this->station_scope.st == NULL);
277 return this->town_scope;
280 /* virtual */ uint32 StationScopeResolver::GetVariable(byte variable, uint32 parameter, bool *available) const
282 if (this->st == NULL) {
283 /* Station does not exist, so we're in a purchase list or the land slope check callback. */
284 switch (variable) {
285 case 0x40:
286 case 0x41:
287 case 0x46:
288 case 0x47:
289 case 0x49: return 0x2110000; // Platforms, tracks & position
290 case 0x42: return 0; // Rail type (XXX Get current type from GUI?)
291 case 0x43: return GetCompanyInfo(_current_company); // Station owner
292 case 0x44: return 2; // PBS status
293 case 0x67: // Land info of nearby tile
294 if (this->axis != INVALID_AXIS && this->tile != INVALID_TILE) {
295 TileIndex tile = this->tile;
296 if (parameter != 0) tile = GetNearbyTile(parameter, tile, true, this->axis); // only perform if it is required
298 Slope tileh = GetTileSlope(tile);
299 bool swap = (this->axis == AXIS_Y && HasBit(tileh, CORNER_W) != HasBit(tileh, CORNER_E));
301 return GetNearbyTileInformation(tile, this->ro.grffile->grf_version >= 8) ^ (swap ? SLOPE_EW : 0);
303 break;
305 case 0xFA: return Clamp(_date - DAYS_TILL_ORIGINAL_BASE_YEAR, 0, 65535); // Build date, clamped to a 16 bit value
308 *available = false;
309 return UINT_MAX;
312 switch (variable) {
313 /* Calculated station variables */
314 case 0x40:
315 if (!HasBit(_svc.valid, 0)) { _svc.v40 = GetPlatformInfoHelper(this->tile, false, false, false); SetBit(_svc.valid, 0); }
316 return _svc.v40;
318 case 0x41:
319 if (!HasBit(_svc.valid, 1)) { _svc.v41 = GetPlatformInfoHelper(this->tile, true, false, false); SetBit(_svc.valid, 1); }
320 return _svc.v41;
322 case 0x42: return GetTerrainType(this->tile) | (GetReverseRailTypeTranslation(GetRailType(this->tile), this->statspec->grf_prop.grffile) << 8);
323 case 0x43: return GetCompanyInfo(this->st->owner); // Station owner
324 case 0x44: return HasStationReservation(this->tile) ? 7 : 4; // PBS status
325 case 0x45:
326 if (!HasBit(_svc.valid, 2)) { _svc.v45 = GetRailContinuationInfo(this->tile); SetBit(_svc.valid, 2); }
327 return _svc.v45;
329 case 0x46:
330 if (!HasBit(_svc.valid, 3)) { _svc.v46 = GetPlatformInfoHelper(this->tile, false, false, true); SetBit(_svc.valid, 3); }
331 return _svc.v46;
333 case 0x47:
334 if (!HasBit(_svc.valid, 4)) { _svc.v47 = GetPlatformInfoHelper(this->tile, true, false, true); SetBit(_svc.valid, 4); }
335 return _svc.v47;
337 case 0x49:
338 if (!HasBit(_svc.valid, 5)) { _svc.v49 = GetPlatformInfoHelper(this->tile, false, true, false); SetBit(_svc.valid, 5); }
339 return _svc.v49;
341 case 0x4A: // Animation frame of tile
342 return GetAnimationFrame(this->tile);
344 /* Variables which use the parameter */
345 /* Variables 0x60 to 0x65 and 0x69 are handled separately below */
346 case 0x66: { // Animation frame of nearby tile
347 TileIndex tile = this->tile;
348 if (parameter != 0) tile = GetNearbyTile(parameter, tile);
349 return this->st->TileBelongsToRailStation(tile) ? GetAnimationFrame(tile) : UINT_MAX;
352 case 0x67: { // Land info of nearby tile
353 Axis axis = GetRailStationAxis(this->tile);
354 TileIndex tile = this->tile;
355 if (parameter != 0) tile = GetNearbyTile(parameter, tile); // only perform if it is required
357 Slope tileh = GetTileSlope(tile);
358 bool swap = (axis == AXIS_Y && HasBit(tileh, CORNER_W) != HasBit(tileh, CORNER_E));
360 return GetNearbyTileInformation(tile, this->ro.grffile->grf_version >= 8) ^ (swap ? SLOPE_EW : 0);
363 case 0x68: { // Station info of nearby tiles
364 TileIndex nearby_tile = GetNearbyTile(parameter, this->tile);
366 if (!HasStationTileRail(nearby_tile)) return 0xFFFFFFFF;
368 uint32 grfid = this->st->speclist[GetCustomStationSpecIndex(this->tile)].grfid;
369 bool perpendicular = GetRailStationAxis(this->tile) != GetRailStationAxis(nearby_tile);
370 bool same_station = this->st->TileBelongsToRailStation(nearby_tile);
371 uint32 res = GB(GetStationGfx(nearby_tile), 1, 2) << 12 | !!perpendicular << 11 | !!same_station << 10;
373 if (IsCustomStationSpecIndex(nearby_tile)) {
374 const StationSpecList ssl = BaseStation::GetByTile(nearby_tile)->speclist[GetCustomStationSpecIndex(nearby_tile)];
375 res |= 1 << (ssl.grfid != grfid ? 9 : 8) | ssl.localidx;
377 return res;
380 /* General station variables */
381 case 0x82: return 50;
382 case 0x84: return this->st->string_id;
383 case 0x86: return 0;
384 case 0xF0: return this->st->facilities;
385 case 0xFA: return Clamp(this->st->build_date - DAYS_TILL_ORIGINAL_BASE_YEAR, 0, 65535);
388 return this->st->GetNewGRFVariable(this->ro, variable, parameter, available);
391 uint32 Station::GetNewGRFVariable(const ResolverObject &object, byte variable, byte parameter, bool *available) const
393 switch (variable) {
394 case 0x48: { // Accepted cargo types
395 CargoID cargo_type;
396 uint32 value = 0;
398 for (cargo_type = 0; cargo_type < NUM_CARGO; cargo_type++) {
399 if (HasBit(this->goods[cargo_type].status, GoodsEntry::GES_ACCEPTANCE)) SetBit(value, cargo_type);
401 return value;
404 case 0x8A: return this->had_vehicle_of_type;
405 case 0xF1: return (this->airport.tile != INVALID_TILE) ? this->airport.GetSpec()->ttd_airport_type : ATP_TTDP_LARGE;
406 case 0xF2: return (this->truck_stops != NULL) ? this->truck_stops->status : 0;
407 case 0xF3: return (this->bus_stops != NULL) ? this->bus_stops->status : 0;
408 case 0xF6: return this->airport.flags;
409 case 0xF7: return GB(this->airport.flags, 8, 8);
412 /* Handle cargo variables with parameter, 0x60 to 0x65 and 0x69 */
413 if ((variable >= 0x60 && variable <= 0x65) || variable == 0x69) {
414 CargoID c = GetCargoTranslation(parameter, object.grffile);
416 if (c == CT_INVALID) {
417 switch (variable) {
418 case 0x62: return 0xFFFFFFFF;
419 case 0x64: return 0xFF00;
420 default: return 0;
423 const GoodsEntry *ge = &this->goods[c];
425 switch (variable) {
426 case 0x60: return min(ge->cargo.TotalCount(), 4095);
427 case 0x61: return ge->HasVehicleEverTriedLoading() ? ge->time_since_pickup : 0;
428 case 0x62: return ge->HasRating() ? ge->rating : 0xFFFFFFFF;
429 case 0x63: return ge->cargo.DaysInTransit();
430 case 0x64: return ge->HasVehicleEverTriedLoading() ? ge->last_speed | (ge->last_age << 8) : 0xFF00;
431 case 0x65: return GB(ge->status, GoodsEntry::GES_ACCEPTANCE, 1) << 3;
432 case 0x69: {
433 assert_compile((int)GoodsEntry::GES_EVER_ACCEPTED + 1 == (int)GoodsEntry::GES_LAST_MONTH);
434 assert_compile((int)GoodsEntry::GES_EVER_ACCEPTED + 2 == (int)GoodsEntry::GES_CURRENT_MONTH);
435 assert_compile((int)GoodsEntry::GES_EVER_ACCEPTED + 3 == (int)GoodsEntry::GES_ACCEPTED_BIGTICK);
436 return GB(ge->status, GoodsEntry::GES_EVER_ACCEPTED, 4);
441 /* Handle cargo variables (deprecated) */
442 if (variable >= 0x8C && variable <= 0xEC) {
443 const GoodsEntry *g = &this->goods[GB(variable - 0x8C, 3, 4)];
444 switch (GB(variable - 0x8C, 0, 3)) {
445 case 0: return g->cargo.TotalCount();
446 case 1: return GB(min(g->cargo.TotalCount(), 4095), 0, 4) | (GB(g->status, GoodsEntry::GES_ACCEPTANCE, 1) << 7);
447 case 2: return g->time_since_pickup;
448 case 3: return g->rating;
449 case 4: return g->cargo.Source();
450 case 5: return g->cargo.DaysInTransit();
451 case 6: return g->last_speed;
452 case 7: return g->last_age;
456 DEBUG(grf, 1, "Unhandled station variable 0x%X", variable);
458 *available = false;
459 return UINT_MAX;
462 uint32 Waypoint::GetNewGRFVariable(const ResolverObject &object, byte variable, byte parameter, bool *available) const
464 switch (variable) {
465 case 0x48: return 0; // Accepted cargo types
466 case 0x8A: return HVOT_WAYPOINT;
467 case 0xF1: return 0; // airport type
468 case 0xF2: return 0; // truck stop status
469 case 0xF3: return 0; // bus stop status
470 case 0xF6: return 0; // airport flags
471 case 0xF7: return 0; // airport flags cont.
474 /* Handle cargo variables with parameter, 0x60 to 0x65 */
475 if (variable >= 0x60 && variable <= 0x65) {
476 return 0;
479 /* Handle cargo variables (deprecated) */
480 if (variable >= 0x8C && variable <= 0xEC) {
481 switch (GB(variable - 0x8C, 0, 3)) {
482 case 3: return INITIAL_STATION_RATING;
483 case 4: return INVALID_STATION;
484 default: return 0;
488 DEBUG(grf, 1, "Unhandled station variable 0x%X", variable);
490 *available = false;
491 return UINT_MAX;
494 /* virtual */ const SpriteGroup *StationResolverObject::ResolveReal(const RealSpriteGroup *group) const
496 if (this->station_scope.st == NULL || this->station_scope.statspec->cls_id == STAT_CLASS_WAYP) {
497 return group->loading[0];
500 uint cargo = 0;
501 const Station *st = Station::From(this->station_scope.st);
503 switch (this->station_scope.cargo_type) {
504 case CT_INVALID:
505 case CT_DEFAULT_NA:
506 case CT_PURCHASE:
507 cargo = 0;
508 break;
510 case CT_DEFAULT:
511 for (CargoID cargo_type = 0; cargo_type < NUM_CARGO; cargo_type++) {
512 cargo += st->goods[cargo_type].cargo.TotalCount();
514 break;
516 default:
517 cargo = st->goods[this->station_scope.cargo_type].cargo.TotalCount();
518 break;
521 if (HasBit(this->station_scope.statspec->flags, SSF_DIV_BY_STATION_SIZE)) cargo /= (st->train_station.w + st->train_station.h);
522 cargo = min(0xfff, cargo);
524 if (cargo > this->station_scope.statspec->cargo_threshold) {
525 if (group->num_loading > 0) {
526 uint set = ((cargo - this->station_scope.statspec->cargo_threshold) * group->num_loading) / (4096 - this->station_scope.statspec->cargo_threshold);
527 return group->loading[set];
529 } else {
530 if (group->num_loaded > 0) {
531 uint set = (cargo * group->num_loaded) / (this->station_scope.statspec->cargo_threshold + 1);
532 return group->loaded[set];
536 return group->loading[0];
540 * Resolver for stations.
541 * @param statspec Station (type) specification.
542 * @param st Instance of the station.
543 * @param tile %Tile of the station.
544 * @param callback Callback ID.
545 * @param callback_param1 First parameter (var 10) of the callback.
546 * @param callback_param2 Second parameter (var 18) of the callback.
548 StationResolverObject::StationResolverObject(const StationSpec *statspec, BaseStation *st, TileIndex tile,
549 CallbackID callback, uint32 callback_param1, uint32 callback_param2)
550 : ResolverObject(statspec->grf_prop.grffile, callback, callback_param1, callback_param2),
551 station_scope(*this, statspec, st, tile), town_scope(NULL)
553 /* Invalidate all cached vars */
554 _svc.valid = 0;
556 CargoID ctype = CT_DEFAULT_NA;
558 if (this->station_scope.st == NULL) {
559 /* No station, so we are in a purchase list */
560 ctype = CT_PURCHASE;
561 } else if (Station::IsExpected(this->station_scope.st)) {
562 const Station *st = Station::From(this->station_scope.st);
563 /* Pick the first cargo that we have waiting */
564 const CargoSpec *cs;
565 FOR_ALL_CARGOSPECS(cs) {
566 if (this->station_scope.statspec->grf_prop.spritegroup[cs->Index()] != NULL &&
567 st->goods[cs->Index()].cargo.TotalCount() > 0) {
568 ctype = cs->Index();
569 break;
574 if (this->station_scope.statspec->grf_prop.spritegroup[ctype] == NULL) {
575 ctype = CT_DEFAULT;
578 /* Remember the cargo type we've picked */
579 this->station_scope.cargo_type = ctype;
580 this->root_spritegroup = this->station_scope.statspec->grf_prop.spritegroup[this->station_scope.cargo_type];
583 StationResolverObject::~StationResolverObject()
585 delete this->town_scope;
589 * Constructor for station scopes.
590 * @param ro Surrounding resolver.
591 * @param statspec Station (type) specification.
592 * @param st Instance of the station.
593 * @param tile %Tile of the station.
595 StationScopeResolver::StationScopeResolver(ResolverObject &ro, const StationSpec *statspec, BaseStation *st, TileIndex tile)
596 : ScopeResolver(ro)
598 this->tile = tile;
599 this->st = st;
600 this->statspec = statspec;
601 this->cargo_type = CT_INVALID;
602 this->axis = INVALID_AXIS;
606 * Resolve sprites for drawing a station tile.
607 * @param statspec Station spec
608 * @param st Station (NULL in GUI)
609 * @param tile Station tile being drawn (INVALID_TILE in GUI)
610 * @param var10 Value to put in variable 10; normally 0; 1 when resolving the groundsprite and SSF_SEPARATE_GROUND is set.
611 * @return First sprite of the Action 1 spriteset to use, minus an offset of 0x42D to accommodate for weird NewGRF specs.
613 SpriteID GetCustomStationRelocation(const StationSpec *statspec, BaseStation *st, TileIndex tile, uint32 var10)
615 StationResolverObject object(statspec, st, tile, CBID_NO_CALLBACK, var10);
616 const SpriteGroup *group = object.Resolve();
617 if (group == NULL || group->type != SGT_RESULT) return 0;
618 return group->GetResult() - 0x42D;
622 * Resolve the sprites for custom station foundations.
623 * @param statspec Station spec
624 * @param st Station
625 * @param tile Station tile being drawn
626 * @param layout Spritelayout as returned by previous callback
627 * @param edge_info Information about northern tile edges; whether they need foundations or merge into adjacent tile's foundations.
628 * @return First sprite of a set of foundation sprites for various slopes, or 0 if default foundations shall be drawn.
630 SpriteID GetCustomStationFoundationRelocation(const StationSpec *statspec, BaseStation *st, TileIndex tile, uint layout, uint edge_info)
632 /* callback_param1 == 2 means we are resolving the foundation sprites. */
633 StationResolverObject object(statspec, st, tile, CBID_NO_CALLBACK, 2, layout | (edge_info << 16));
635 const SpriteGroup *group = object.Resolve();
636 if (group == NULL || group->type != SGT_RESULT) return 0;
638 /* Note: SpriteGroup::Resolve zeroes all registers, so register 0x100 is initialised to 0. (compatibility) */
639 return group->GetResult() + GetRegister(0x100);
643 uint16 GetStationCallback(CallbackID callback, uint32 param1, uint32 param2, const StationSpec *statspec, BaseStation *st, TileIndex tile)
645 StationResolverObject object(statspec, st, tile, callback, param1, param2);
646 return object.ResolveCallback();
650 * Check the slope of a tile of a new station.
651 * @param north_tile Norther tile of the station rect.
652 * @param cur_tile Tile to check.
653 * @param statspec Station spec.
654 * @param axis Axis of the new station.
655 * @param plat_len Platform length.
656 * @param numtracks Number of platforms.
657 * @return Succeeded or failed command.
659 CommandCost PerformStationTileSlopeCheck(TileIndex north_tile, TileIndex cur_tile, const StationSpec *statspec, Axis axis, byte plat_len, byte numtracks)
661 TileIndexDiff diff = cur_tile - north_tile;
662 Slope slope = GetTileSlope(cur_tile);
664 StationResolverObject object(statspec, NULL, cur_tile, CBID_STATION_LAND_SLOPE_CHECK,
665 (slope << 4) | (slope ^ (axis == AXIS_Y && HasBit(slope, CORNER_W) != HasBit(slope, CORNER_E) ? SLOPE_EW : 0)),
666 (numtracks << 24) | (plat_len << 16) | (axis == AXIS_Y ? TileX(diff) << 8 | TileY(diff) : TileY(diff) << 8 | TileX(diff)));
667 object.station_scope.axis = axis;
669 uint16 cb_res = object.ResolveCallback();
671 /* Failed callback means success. */
672 if (cb_res == CALLBACK_FAILED) return CommandCost();
674 /* The meaning of bit 10 is inverted for a grf version < 8. */
675 if (statspec->grf_prop.grffile->grf_version < 8) ToggleBit(cb_res, 10);
676 return GetErrorMessageFromLocationCallbackResult(cb_res, statspec->grf_prop.grffile, STR_ERROR_LAND_SLOPED_IN_WRONG_DIRECTION);
681 * Allocate a StationSpec to a Station. This is called once per build operation.
682 * @param statspec StationSpec to allocate.
683 * @param st Station to allocate it to.
684 * @param exec Whether to actually allocate the spec.
685 * @return Index within the Station's spec list, or -1 if the allocation failed.
687 int AllocateSpecToStation(const StationSpec *statspec, BaseStation *st, bool exec)
689 uint i;
691 if (statspec == NULL || st == NULL) return 0;
693 for (i = 1; i < st->num_specs && i < NUM_STATIONSSPECS_PER_STATION; i++) {
694 if (st->speclist[i].spec == NULL && st->speclist[i].grfid == 0) break;
697 if (i == NUM_STATIONSSPECS_PER_STATION) {
698 /* As final effort when the spec list is already full...
699 * try to find the same spec and return that one. This might
700 * result in slightly "wrong" (as per specs) looking stations,
701 * but it's fairly unlikely that one reaches the limit anyways.
703 for (i = 1; i < st->num_specs && i < NUM_STATIONSSPECS_PER_STATION; i++) {
704 if (st->speclist[i].spec == statspec) return i;
707 return -1;
710 if (exec) {
711 if (i >= st->num_specs) {
712 st->num_specs = i + 1;
713 st->speclist = ReallocT(st->speclist, st->num_specs);
715 if (st->num_specs == 2) {
716 /* Initial allocation */
717 st->speclist[0].spec = NULL;
718 st->speclist[0].grfid = 0;
719 st->speclist[0].localidx = 0;
723 st->speclist[i].spec = statspec;
724 st->speclist[i].grfid = statspec->grf_prop.grffile->grfid;
725 st->speclist[i].localidx = statspec->grf_prop.local_id;
727 StationUpdateCachedTriggers(st);
730 return i;
735 * Deallocate a StationSpec from a Station. Called when removing a single station tile.
736 * @param st Station to work with.
737 * @param specindex Index of the custom station within the Station's spec list.
738 * @return Indicates whether the StationSpec was deallocated.
740 void DeallocateSpecFromStation(BaseStation *st, byte specindex)
742 /* specindex of 0 (default) is never freeable */
743 if (specindex == 0) return;
745 ETileArea area = ETileArea(st, INVALID_TILE, TA_WHOLE);
746 /* Check all tiles over the station to check if the specindex is still in use */
747 TILE_AREA_LOOP(tile, area) {
748 if (st->TileBelongsToRailStation(tile) && GetCustomStationSpecIndex(tile) == specindex) {
749 return;
753 /* This specindex is no longer in use, so deallocate it */
754 st->speclist[specindex].spec = NULL;
755 st->speclist[specindex].grfid = 0;
756 st->speclist[specindex].localidx = 0;
758 /* If this was the highest spec index, reallocate */
759 if (specindex == st->num_specs - 1) {
760 for (; st->speclist[st->num_specs - 1].grfid == 0 && st->num_specs > 1; st->num_specs--) {}
762 if (st->num_specs > 1) {
763 st->speclist = ReallocT(st->speclist, st->num_specs);
764 } else {
765 free(st->speclist);
766 st->num_specs = 0;
767 st->speclist = NULL;
768 st->cached_anim_triggers = 0;
769 st->cached_cargo_triggers = 0;
770 return;
774 StationUpdateCachedTriggers(st);
778 * Draw representation of a station tile for GUI purposes.
779 * @param x Position x of image.
780 * @param y Position y of image.
781 * @param axis Axis.
782 * @param railtype Rail type.
783 * @param sclass, station Type of station.
784 * @param station station ID
785 * @return True if the tile was drawn (allows for fallback to default graphic)
787 bool DrawStationTile(int x, int y, RailType railtype, Axis axis, StationClassID sclass, uint station)
789 const DrawTileSprites *sprites = NULL;
790 const RailtypeInfo *rti = GetRailTypeInfo(railtype);
791 PaletteID palette = COMPANY_SPRITE_COLOUR(_local_company);
792 uint tile = 2;
794 const StationSpec *statspec = StationClass::Get(sclass)->GetSpec(station);
795 if (statspec == NULL) return false;
797 if (HasBit(statspec->callback_mask, CBM_STATION_SPRITE_LAYOUT)) {
798 uint16 callback = GetStationCallback(CBID_STATION_SPRITE_LAYOUT, 0x2110000, 0, statspec, NULL, INVALID_TILE);
799 if (callback != CALLBACK_FAILED) tile = callback;
802 uint32 total_offset = rti->GetRailtypeSpriteOffset();
803 uint32 relocation = 0;
804 uint32 ground_relocation = 0;
805 const NewGRFSpriteLayout *layout = NULL;
806 DrawTileSprites tmp_rail_layout;
808 if (statspec->renderdata == NULL) {
809 sprites = GetStationTileLayout(STATION_RAIL, tile + axis);
810 } else {
811 layout = &statspec->renderdata[(tile < statspec->tiles) ? tile + axis : (uint)axis];
812 if (!layout->NeedsPreprocessing()) {
813 sprites = layout;
814 layout = NULL;
818 if (layout != NULL) {
819 /* Sprite layout which needs preprocessing */
820 bool separate_ground = HasBit(statspec->flags, SSF_SEPARATE_GROUND);
821 uint32 var10_values = layout->PrepareLayout(total_offset, rti->fallback_railtype, 0, 0, separate_ground);
822 uint8 var10;
823 FOR_EACH_SET_BIT(var10, var10_values) {
824 uint32 var10_relocation = GetCustomStationRelocation(statspec, NULL, INVALID_TILE, var10);
825 layout->ProcessRegisters(var10, var10_relocation, separate_ground);
828 tmp_rail_layout.seq = layout->GetLayout(&tmp_rail_layout.ground);
829 sprites = &tmp_rail_layout;
830 total_offset = 0;
831 } else {
832 /* Simple sprite layout */
833 ground_relocation = relocation = GetCustomStationRelocation(statspec, NULL, INVALID_TILE, 0);
834 if (HasBit(sprites->ground.sprite, SPRITE_MODIFIER_CUSTOM_SPRITE)) {
835 ground_relocation = GetCustomStationRelocation(statspec, NULL, INVALID_TILE, 1);
837 ground_relocation += rti->fallback_railtype;
840 SpriteID image = sprites->ground.sprite;
841 PaletteID pal = sprites->ground.pal;
842 RailTrackOffset overlay_offset;
843 if (rti->UsesOverlay() && SplitGroundSpriteForOverlay(NULL, &image, &overlay_offset)) {
844 SpriteID ground = GetCustomRailSprite(rti, INVALID_TILE, RTSG_GROUND);
845 DrawSprite(image, PAL_NONE, x, y);
846 DrawSprite(ground + overlay_offset, PAL_NONE, x, y);
847 } else {
848 image += HasBit(image, SPRITE_MODIFIER_CUSTOM_SPRITE) ? ground_relocation : total_offset;
849 if (HasBit(pal, SPRITE_MODIFIER_CUSTOM_SPRITE)) pal += ground_relocation;
850 DrawSprite(image, GroundSpritePaletteTransform(image, pal, palette), x, y);
853 DrawRailTileSeqInGUI(x, y, sprites, total_offset, relocation, palette);
855 return true;
859 const StationSpec *GetStationSpec(TileIndex t)
861 if (!IsCustomStationSpecIndex(t)) return NULL;
863 const BaseStation *st = BaseStation::GetByTile(t);
864 uint specindex = GetCustomStationSpecIndex(t);
865 return specindex < st->num_specs ? st->speclist[specindex].spec : NULL;
870 * Check whether a rail station tile is NOT traversable.
871 * @param tile %Tile to test.
872 * @return Station tile is blocked.
873 * @note This could be cached (during build) in the map array to save on all the dereferencing.
875 bool IsStationTileBlocked(TileIndex tile)
877 const StationSpec *statspec = GetStationSpec(tile);
879 return statspec != NULL && HasBit(statspec->blocked, GetStationGfx(tile));
883 * Check if a rail station tile shall have pylons when electrified.
884 * @param tile %Tile to test.
885 * @return Tile shall have pylons.
886 * @note This could be cached (during build) in the map array to save on all the dereferencing.
888 bool CanStationTileHavePylons(TileIndex tile)
890 const StationSpec *statspec = GetStationSpec(tile);
891 uint gfx = GetStationGfx(tile);
892 /* Default stations do not draw pylons under roofs (gfx >= 4) */
893 return statspec != NULL ? HasBit(statspec->pylons, gfx) : gfx < 4;
897 * Check if a rail station tile shall have wires when electrified.
898 * @param tile %Tile to test.
899 * @return Tile shall have wires.
900 * @note This could be cached (during build) in the map array to save on all the dereferencing.
902 bool CanStationTileHaveWires(TileIndex tile)
904 const StationSpec *statspec = GetStationSpec(tile);
905 return statspec == NULL || !HasBit(statspec->wires, GetStationGfx(tile));
908 /** Wrapper for animation control, see #GetStationCallback. */
909 uint16 GetAnimStationCallback(CallbackID callback, uint32 param1, uint32 param2, const StationSpec *statspec, BaseStation *st, TileIndex tile, int extra_data)
911 return GetStationCallback(callback, param1, param2, statspec, st, tile);
914 /** Helper class for animation control. */
915 struct StationAnimationBase : public AnimationBase<StationAnimationBase, StationSpec, BaseStation, int, GetAnimStationCallback> {
916 static const CallbackID cb_animation_speed = CBID_STATION_ANIMATION_SPEED;
917 static const CallbackID cb_animation_next_frame = CBID_STATION_ANIM_NEXT_FRAME;
919 static const StationCallbackMask cbm_animation_speed = CBM_STATION_ANIMATION_SPEED;
920 static const StationCallbackMask cbm_animation_next_frame = CBM_STATION_ANIMATION_NEXT_FRAME;
923 void AnimateStationTile(TileIndex tile)
925 const StationSpec *ss = GetStationSpec(tile);
926 if (ss == NULL) return;
928 StationAnimationBase::AnimateTile(ss, BaseStation::GetByTile(tile), tile, HasBit(ss->flags, SSF_CB141_RANDOM_BITS));
931 void TriggerStationAnimation(BaseStation *st, TileIndex tile, StationAnimationTrigger trigger, CargoID cargo_type)
933 /* List of coverage areas for each animation trigger */
934 static const TriggerArea tas[] = {
935 TA_TILE, TA_WHOLE, TA_WHOLE, TA_PLATFORM, TA_PLATFORM, TA_PLATFORM, TA_WHOLE
938 /* Get Station if it wasn't supplied */
939 if (st == NULL) st = BaseStation::GetByTile(tile);
941 /* Check the cached animation trigger bitmask to see if we need
942 * to bother with any further processing. */
943 if (!HasBit(st->cached_anim_triggers, trigger)) return;
945 uint16 random_bits = Random();
946 ETileArea area = ETileArea(st, tile, tas[trigger]);
948 /* Check all tiles over the station to check if the specindex is still in use */
949 TILE_AREA_LOOP(tile, area) {
950 if (st->TileBelongsToRailStation(tile)) {
951 const StationSpec *ss = GetStationSpec(tile);
952 if (ss != NULL && HasBit(ss->animation.triggers, trigger)) {
953 CargoID cargo;
954 if (cargo_type == CT_INVALID) {
955 cargo = CT_INVALID;
956 } else {
957 cargo = ss->grf_prop.grffile->cargo_map[cargo_type];
959 StationAnimationBase::ChangeAnimationFrame(CBID_STATION_ANIM_START_STOP, ss, st, tile, (random_bits << 16) | Random(), (uint8)trigger | (cargo << 8));
966 * Trigger station randomisation
967 * @param st station being triggered
968 * @param tile specific tile of platform to trigger
969 * @param trigger trigger type
970 * @param cargo_type cargo type causing trigger
972 void TriggerStationRandomisation(Station *st, TileIndex tile, StationRandomTrigger trigger, CargoID cargo_type)
974 /* List of coverage areas for each animation trigger */
975 static const TriggerArea tas[] = {
976 TA_WHOLE, TA_WHOLE, TA_PLATFORM, TA_PLATFORM, TA_PLATFORM, TA_PLATFORM
979 /* Get Station if it wasn't supplied */
980 if (st == NULL) st = Station::GetByTile(tile);
982 /* Check the cached cargo trigger bitmask to see if we need
983 * to bother with any further processing. */
984 if (st->cached_cargo_triggers == 0) return;
985 if (cargo_type != CT_INVALID && !HasBit(st->cached_cargo_triggers, cargo_type)) return;
987 uint32 whole_reseed = 0;
988 ETileArea area = ETileArea(st, tile, tas[trigger]);
990 uint32 empty_mask = 0;
991 if (trigger == SRT_CARGO_TAKEN) {
992 /* Create a bitmask of completely empty cargo types to be matched */
993 for (CargoID i = 0; i < NUM_CARGO; i++) {
994 if (st->goods[i].cargo.TotalCount() == 0) {
995 SetBit(empty_mask, i);
1000 /* Convert trigger to bit */
1001 uint8 trigger_bit = 1 << trigger;
1003 /* Check all tiles over the station to check if the specindex is still in use */
1004 TILE_AREA_LOOP(tile, area) {
1005 if (st->TileBelongsToRailStation(tile)) {
1006 const StationSpec *ss = GetStationSpec(tile);
1007 if (ss == NULL) continue;
1009 /* Cargo taken "will only be triggered if all of those
1010 * cargo types have no more cargo waiting." */
1011 if (trigger == SRT_CARGO_TAKEN) {
1012 if ((ss->cargo_triggers & ~empty_mask) != 0) continue;
1015 if (cargo_type == CT_INVALID || HasBit(ss->cargo_triggers, cargo_type)) {
1016 StationResolverObject object(ss, st, tile, CBID_RANDOM_TRIGGER, 0);
1017 object.trigger = trigger_bit;
1019 const SpriteGroup *group = object.Resolve();
1020 if (group == NULL) continue;
1022 uint32 reseed = object.GetReseedSum();
1023 if (reseed != 0) {
1024 whole_reseed |= reseed;
1025 reseed >>= 16;
1027 /* Set individual tile random bits */
1028 uint8 random_bits = GetStationTileRandomBits(tile);
1029 random_bits &= ~reseed;
1030 random_bits |= Random() & reseed;
1031 SetStationTileRandomBits(tile, random_bits);
1033 MarkTileDirtyByTile(tile);
1039 /* Update whole station random bits */
1040 if ((whole_reseed & 0xFFFF) != 0) {
1041 st->random_bits &= ~whole_reseed;
1042 st->random_bits |= Random() & whole_reseed;
1047 * Update the cached animation trigger bitmask for a station.
1048 * @param st Station to update.
1050 void StationUpdateCachedTriggers(BaseStation *st)
1052 st->cached_anim_triggers = 0;
1053 st->cached_cargo_triggers = 0;
1055 /* Combine animation trigger bitmask for all station specs
1056 * of this station. */
1057 for (uint i = 0; i < st->num_specs; i++) {
1058 const StationSpec *ss = st->speclist[i].spec;
1059 if (ss != NULL) {
1060 st->cached_anim_triggers |= ss->animation.triggers;
1061 st->cached_cargo_triggers |= ss->cargo_triggers;