00001
00002
00005 #include "ai_rail.hpp"
00006 #include "ai_map.hpp"
00007 #include "ai_station.hpp"
00008 #include "../../debug.h"
00009 #include "../../station_map.h"
00010 #include "../../company_func.h"
00011 #include "../../waypoint.h"
00012 #include "../../newgrf_generic.h"
00013 #include "../../newgrf_station.h"
00014
00015 bool AIRail::IsRailTile(TileIndex tile)
00016 {
00017 if (!::IsValidTile(tile)) return false;
00018
00019 return (::IsTileType(tile, MP_RAILWAY) && !::IsRailDepot(tile)) ||
00020 (::IsRailwayStationTile(tile) && !::IsStationTileBlocked(tile)) || ::IsLevelCrossingTile(tile);
00021 }
00022
00023 bool AIRail::IsLevelCrossingTile(TileIndex tile)
00024 {
00025 if (!::IsValidTile(tile)) return false;
00026
00027 return ::IsLevelCrossingTile(tile);
00028 }
00029
00030 bool AIRail::IsRailDepotTile(TileIndex tile)
00031 {
00032 if (!::IsValidTile(tile)) return false;
00033
00034 return ::IsRailDepotTile(tile);
00035 }
00036
00037 bool AIRail::IsRailStationTile(TileIndex tile)
00038 {
00039 if (!::IsValidTile(tile)) return false;
00040
00041 return ::IsRailwayStationTile(tile);
00042 }
00043
00044 bool AIRail::IsRailWaypointTile(TileIndex tile)
00045 {
00046 if (!::IsValidTile(tile)) return false;
00047
00048 return ::IsTileType(tile, MP_RAILWAY) && ::IsRailWaypointTile(tile);
00049 }
00050
00051 bool AIRail::IsRailTypeAvailable(RailType rail_type)
00052 {
00053 if ((::RailType)rail_type < RAILTYPE_BEGIN || (::RailType)rail_type >= RAILTYPE_END) return false;
00054
00055 return ::HasRailtypeAvail(_current_company, (::RailType)rail_type);
00056 }
00057
00058 AIRail::RailType AIRail::GetCurrentRailType()
00059 {
00060 return (RailType)AIObject::GetRailType();
00061 }
00062
00063 void AIRail::SetCurrentRailType(RailType rail_type)
00064 {
00065 if (!IsRailTypeAvailable(rail_type)) return;
00066
00067 AIObject::SetRailType((::RailType)rail_type);
00068 }
00069
00070 bool AIRail::TrainCanRunOnRail(AIRail::RailType engine_rail_type, AIRail::RailType track_rail_type)
00071 {
00072 if (!AIRail::IsRailTypeAvailable(engine_rail_type)) return false;
00073 if (!AIRail::IsRailTypeAvailable(track_rail_type)) return false;
00074
00075 return ::IsCompatibleRail((::RailType)engine_rail_type, (::RailType)track_rail_type);
00076 }
00077
00078 bool AIRail::TrainHasPowerOnRail(AIRail::RailType engine_rail_type, AIRail::RailType track_rail_type)
00079 {\
00080 if (!AIRail::IsRailTypeAvailable(engine_rail_type)) return false;
00081 if (!AIRail::IsRailTypeAvailable(track_rail_type)) return false;
00082
00083 return ::HasPowerOnRail((::RailType)engine_rail_type, (::RailType)track_rail_type);
00084 }
00085
00086 AIRail::RailType AIRail::GetRailType(TileIndex tile)
00087 {
00088 if (!AITile::HasTransportType(tile, AITile::TRANSPORT_RAIL)) return RAILTYPE_INVALID;
00089
00090 return (RailType)::GetRailType(tile);
00091 }
00092
00093 bool AIRail::ConvertRailType(TileIndex start_tile, TileIndex end_tile, AIRail::RailType convert_to)
00094 {
00095 EnforcePrecondition(false, ::IsValidTile(start_tile));
00096 EnforcePrecondition(false, ::IsValidTile(end_tile));
00097 EnforcePrecondition(false, IsRailTypeAvailable(convert_to));
00098
00099 return AIObject::DoCommand(start_tile, end_tile, convert_to, CMD_CONVERT_RAIL);
00100 }
00101
00102 TileIndex AIRail::GetRailDepotFrontTile(TileIndex depot)
00103 {
00104 if (!IsRailDepotTile(depot)) return INVALID_TILE;
00105
00106 return depot + ::TileOffsByDiagDir(::GetRailDepotDirection(depot));
00107 }
00108
00109 AIRail::RailTrack AIRail::GetRailStationDirection(TileIndex tile)
00110 {
00111 if (!IsRailStationTile(tile)) return RAILTRACK_INVALID;
00112
00113 return (RailTrack)::GetRailStationTrack(tile);
00114 }
00115
00116 bool AIRail::BuildRailDepot(TileIndex tile, TileIndex front)
00117 {
00118 EnforcePrecondition(false, tile != front);
00119 EnforcePrecondition(false, ::IsValidTile(tile));
00120 EnforcePrecondition(false, ::IsValidTile(front));
00121 EnforcePrecondition(false, ::TileX(tile) == ::TileX(front) || ::TileY(tile) == ::TileY(front));
00122 EnforcePrecondition(false, IsRailTypeAvailable(GetCurrentRailType()));
00123
00124 uint entrance_dir = (::TileX(tile) == ::TileX(front)) ? (::TileY(tile) < ::TileY(front) ? 1 : 3) : (::TileX(tile) < ::TileX(front) ? 2 : 0);
00125
00126 return AIObject::DoCommand(tile, AIObject::GetRailType(), entrance_dir, CMD_BUILD_TRAIN_DEPOT);
00127 }
00128
00129 bool AIRail::BuildRailStation(TileIndex tile, RailTrack direction, uint num_platforms, uint platform_length, StationID station_id)
00130 {
00131 EnforcePrecondition(false, ::IsValidTile(tile));
00132 EnforcePrecondition(false, direction == RAILTRACK_NW_SE || direction == RAILTRACK_NE_SW);
00133 EnforcePrecondition(false, num_platforms > 0 && num_platforms <= 0xFF);
00134 EnforcePrecondition(false, platform_length > 0 && platform_length <= 0xFF);
00135 EnforcePrecondition(false, IsRailTypeAvailable(GetCurrentRailType()));
00136 EnforcePrecondition(false, station_id == AIStation::STATION_NEW || station_id == AIStation::STATION_JOIN_ADJACENT || AIStation::IsValidStation(station_id));
00137
00138 uint32 p1 = GetCurrentRailType() | (platform_length << 16) | (num_platforms << 8);
00139 if (direction == RAILTRACK_NW_SE) p1 |= (1 << 4);
00140 if (station_id != AIStation::STATION_JOIN_ADJACENT) p1 |= (1 << 24);
00141 return AIObject::DoCommand(tile, p1, (AIStation::IsValidStation(station_id) ? station_id : INVALID_STATION) << 16, CMD_BUILD_RAILROAD_STATION);
00142 }
00143
00144 bool AIRail::BuildNewGRFRailStation(TileIndex tile, RailTrack direction, uint num_platforms, uint platform_length, StationID station_id, CargoID cargo_id, IndustryType source_industry, IndustryType goal_industry, int distance, bool source_station)
00145 {
00146 EnforcePrecondition(false, ::IsValidTile(tile));
00147 EnforcePrecondition(false, direction == RAILTRACK_NW_SE || direction == RAILTRACK_NE_SW);
00148 EnforcePrecondition(false, num_platforms > 0 && num_platforms <= 0xFF);
00149 EnforcePrecondition(false, platform_length > 0 && platform_length <= 0xFF);
00150 EnforcePrecondition(false, IsRailTypeAvailable(GetCurrentRailType()));
00151 EnforcePrecondition(false, station_id == AIStation::STATION_NEW || station_id == AIStation::STATION_JOIN_ADJACENT || AIStation::IsValidStation(station_id));
00152
00153 uint32 p1 = GetCurrentRailType() | (platform_length << 16) | (num_platforms << 8);
00154 if (direction == RAILTRACK_NW_SE) p1 |= 1 << 4;
00155 if (station_id != AIStation::STATION_JOIN_ADJACENT) p1 |= (1 << 24);
00156
00157 const GRFFile *file;
00158 uint16 res = GetAiPurchaseCallbackResult(GSF_STATION, cargo_id, 0, source_industry, goal_industry, min(255, distance / 2), AICE_STATION_GET_STATION_ID, source_station ? 0 : 1, min(15, num_platforms) << 4 | min(15, platform_length), &file);
00159 uint32 p2 = (AIStation::IsValidStation(station_id) ? station_id : INVALID_STATION) << 16;
00160 if (res != CALLBACK_FAILED) {
00161 int index = 0;
00162 const StationSpec *spec = GetCustomStationSpecByGrf(file->grfid, res, &index);
00163 if (spec == NULL) {
00164 DEBUG(grf, 1, "%s returned an invalid station ID for 'AI construction/purchase selection (18)' callback", file->filename);
00165 } else {
00166 p2 |= spec->sclass | index << 8;
00167 }
00168
00169 }
00170 return AIObject::DoCommand(tile, p1, p2, CMD_BUILD_RAILROAD_STATION);
00171 }
00172
00173 bool AIRail::RemoveRailStationTileRect(TileIndex tile, TileIndex tile2)
00174 {
00175 EnforcePrecondition(false, ::IsValidTile(tile));
00176 EnforcePrecondition(false, ::IsValidTile(tile2));
00177
00178 return AIObject::DoCommand(tile, tile2, 0, CMD_REMOVE_FROM_RAILROAD_STATION);
00179 }
00180
00181 uint AIRail::GetRailTracks(TileIndex tile)
00182 {
00183 if (!IsRailTile(tile)) return RAILTRACK_INVALID;
00184
00185 if (IsRailWaypointTile(tile)) return ::GetRailWaypointBits(tile);
00186 if (IsRailStationTile(tile)) return ::TrackToTrackBits(::GetRailStationTrack(tile));
00187 if (IsLevelCrossingTile(tile)) return ::GetCrossingRailBits(tile);
00188 return ::GetTrackBits(tile);
00189 }
00190
00191 bool AIRail::BuildRailTrack(TileIndex tile, RailTrack rail_track)
00192 {
00193 EnforcePrecondition(false, ::IsValidTile(tile));
00194 EnforcePrecondition(false, rail_track != 0);
00195 EnforcePrecondition(false, (rail_track & ~::TRACK_BIT_ALL) == 0);
00196 EnforcePrecondition(false, KillFirstBit((uint)rail_track) == 0);
00197 EnforcePrecondition(false, IsRailTypeAvailable(GetCurrentRailType()));
00198
00199 return AIObject::DoCommand(tile, tile, GetCurrentRailType() | (FindFirstTrack((::TrackBits)rail_track) << 4), CMD_BUILD_RAILROAD_TRACK);
00200 }
00201
00202 bool AIRail::RemoveRailTrack(TileIndex tile, RailTrack rail_track)
00203 {
00204 EnforcePrecondition(false, ::IsValidTile(tile));
00205 EnforcePrecondition(false, ::IsTileType(tile, MP_RAILWAY) && ::IsPlainRailTile(tile));
00206 EnforcePrecondition(false, GetRailTracks(tile) & rail_track);
00207 EnforcePrecondition(false, KillFirstBit((uint)rail_track) == 0);
00208
00209 return AIObject::DoCommand(tile, tile, GetCurrentRailType() | (FindFirstTrack((::TrackBits)rail_track) << 4), CMD_REMOVE_RAILROAD_TRACK);
00210 }
00211
00212 bool AIRail::AreTilesConnected(TileIndex from, TileIndex tile, TileIndex to)
00213 {
00214 if (!IsRailTile(tile)) return false;
00215 if (from == to || AIMap::DistanceManhattan(from, tile) != 1 || AIMap::DistanceManhattan(tile, to) != 1) return false;
00216
00217 if (to < from) ::Swap(from, to);
00218
00219 if (tile - from == 1) {
00220 if (to - tile == 1) return (GetRailTracks(tile) & RAILTRACK_NE_SW) != 0;
00221 if (to - tile == ::MapSizeX()) return (GetRailTracks(tile) & RAILTRACK_NE_SE) != 0;
00222 } else if (tile - from == ::MapSizeX()) {
00223 if (tile - to == 1) return (GetRailTracks(tile) & RAILTRACK_NW_NE) != 0;
00224 if (to - tile == 1) return (GetRailTracks(tile) & RAILTRACK_NW_SW) != 0;
00225 if (to - tile == ::MapSizeX()) return (GetRailTracks(tile) & RAILTRACK_NW_SE) != 0;
00226 } else {
00227 return (GetRailTracks(tile) & RAILTRACK_SW_SE) != 0;
00228 }
00229
00230 NOT_REACHED();
00231 }
00232
00237 static uint32 SimulateDrag(TileIndex from, TileIndex tile, TileIndex *to)
00238 {
00239 int diag_offset = abs(abs((int)::TileX(*to) - (int)::TileX(tile)) - abs((int)::TileY(*to) - (int)::TileY(tile)));
00240 uint32 p2 = AIRail::GetCurrentRailType();
00241 if (::TileY(from) == ::TileY(*to)) {
00242 p2 |= (TRACK_X << 4);
00243 *to -= Clamp((int)::TileX(*to) - (int)::TileX(tile), -1, 1);
00244 } else if (::TileX(from) == ::TileX(*to)) {
00245 p2 |= (TRACK_Y << 4);
00246 *to -= ::MapSizeX() * Clamp((int)::TileY(*to) - (int)::TileY(tile), -1, 1);
00247 } else if (::TileY(from) < ::TileY(tile)) {
00248 if (::TileX(*to) < ::TileX(tile)) {
00249 p2 |= (TRACK_UPPER << 4);
00250 } else {
00251 p2 |= (TRACK_LEFT << 4);
00252 }
00253 if (diag_offset) {
00254 *to -= Clamp((int)::TileX(*to) - (int)::TileX(tile), -1, 1);
00255 } else {
00256 *to -= ::MapSizeX() * Clamp((int)::TileY(*to) - (int)::TileY(tile), -1, 1);
00257 }
00258 } else if (::TileY(from) > ::TileY(tile)) {
00259 if (::TileX(*to) < ::TileX(tile)) {
00260 p2 |= (TRACK_RIGHT << 4);
00261 } else {
00262 p2 |= (TRACK_LOWER << 4);
00263 }
00264 if (diag_offset) {
00265 *to -= Clamp((int)::TileX(*to) - (int)::TileX(tile), -1, 1);
00266 } else {
00267 *to -= ::MapSizeX() * Clamp((int)::TileY(*to) - (int)::TileY(tile), -1, 1);
00268 }
00269 } else if (::TileX(from) < ::TileX(tile)) {
00270 if (::TileY(*to) < ::TileY(tile)) {
00271 p2 |= (TRACK_UPPER << 4);
00272 } else {
00273 p2 |= (TRACK_RIGHT << 4);
00274 }
00275 if (!diag_offset) {
00276 *to -= Clamp((int)::TileX(*to) - (int)::TileX(tile), -1, 1);
00277 } else {
00278 *to -= ::MapSizeX() * Clamp((int)::TileY(*to) - (int)::TileY(tile), -1, 1);
00279 }
00280 } else if (::TileX(from) > ::TileX(tile)) {
00281 if (::TileY(*to) < ::TileY(tile)) {
00282 p2 |= (TRACK_LEFT << 4);
00283 } else {
00284 p2 |= (TRACK_LOWER << 4);
00285 }
00286 if (!diag_offset) {
00287 *to -= Clamp((int)::TileX(*to) - (int)::TileX(tile), -1, 1);
00288 } else {
00289 *to -= ::MapSizeX() * Clamp((int)::TileY(*to) - (int)::TileY(tile), -1, 1);
00290 }
00291 }
00292 return p2;
00293 }
00294
00295 bool AIRail::BuildRail(TileIndex from, TileIndex tile, TileIndex to)
00296 {
00297 EnforcePrecondition(false, ::IsValidTile(from));
00298 EnforcePrecondition(false, ::IsValidTile(tile));
00299 EnforcePrecondition(false, ::IsValidTile(to));
00300 EnforcePrecondition(false, ::DistanceManhattan(from, tile) == 1);
00301 EnforcePrecondition(false, ::DistanceManhattan(tile,to) >= 1);
00302 EnforcePrecondition(false, IsRailTypeAvailable(GetCurrentRailType()));
00303 int diag_offset = abs(abs((int)::TileX(to) - (int)::TileX(tile)) - abs((int)::TileY(to) - (int)::TileY(tile)));
00304 EnforcePrecondition(false, diag_offset <= 1 ||
00305 (::TileX(from) == ::TileX(tile) && ::TileX(tile) == ::TileX(to)) ||
00306 (::TileY(from) == ::TileY(tile) && ::TileY(tile) == ::TileY(to)));
00307
00308 uint32 p2 = SimulateDrag(from, tile, &to);
00309 return AIObject::DoCommand(tile, to, p2, CMD_BUILD_RAILROAD_TRACK);
00310 }
00311
00312 bool AIRail::RemoveRail(TileIndex from, TileIndex tile, TileIndex to)
00313 {
00314 EnforcePrecondition(false, ::IsValidTile(from));
00315 EnforcePrecondition(false, ::IsValidTile(tile));
00316 EnforcePrecondition(false, ::IsValidTile(to));
00317 EnforcePrecondition(false, ::DistanceManhattan(from, tile) == 1);
00318 EnforcePrecondition(false, ::DistanceManhattan(tile,to) >= 1);
00319 int diag_offset = abs(abs((int)::TileX(to) - (int)::TileX(tile)) - abs((int)::TileY(to) - (int)::TileY(tile)));
00320 EnforcePrecondition(false, diag_offset <= 1 ||
00321 (::TileX(from) == ::TileX(tile) && ::TileX(tile) == ::TileX(to)) ||
00322 (::TileY(from) == ::TileY(tile) && ::TileY(tile) == ::TileY(to)));
00323
00324 if (!IsRailTypeAvailable(GetCurrentRailType())) SetCurrentRailType(GetRailType(tile));
00325 uint32 p2 = SimulateDrag(from, tile, &to);
00326 return AIObject::DoCommand(tile, to, p2, CMD_REMOVE_RAILROAD_TRACK);
00327 }
00328
00333 struct AIRailSignalData {
00334 Track track;
00335 Trackdir trackdir;
00336 uint signal_cycles;
00337 };
00338
00339 static const int NUM_TRACK_DIRECTIONS = 3;
00340
00347 static const AIRailSignalData _possible_trackdirs[5][NUM_TRACK_DIRECTIONS] = {
00348 {{TRACK_UPPER, TRACKDIR_UPPER_E, 0}, {TRACK_Y, TRACKDIR_Y_SE, 0}, {TRACK_LEFT, TRACKDIR_LEFT_S, 1}},
00349 {{TRACK_RIGHT, TRACKDIR_RIGHT_S, 1}, {TRACK_X, TRACKDIR_X_SW, 1}, {TRACK_UPPER, TRACKDIR_UPPER_W, 1}},
00350 {{INVALID_TRACK, INVALID_TRACKDIR, 0}, {INVALID_TRACK, INVALID_TRACKDIR, 0}, {INVALID_TRACK, INVALID_TRACKDIR, 0}},
00351 {{TRACK_LOWER, TRACKDIR_LOWER_E, 0}, {TRACK_X, TRACKDIR_X_NE, 0}, {TRACK_LEFT, TRACKDIR_LEFT_N, 0}},
00352 {{TRACK_RIGHT, TRACKDIR_RIGHT_N, 0}, {TRACK_Y, TRACKDIR_Y_NW, 1}, {TRACK_LOWER, TRACKDIR_LOWER_W, 1}}
00353 };
00354
00355 AIRail::SignalType AIRail::GetSignalType(TileIndex tile, TileIndex front)
00356 {
00357 if (AIMap::DistanceManhattan(tile, front) != 1) return SIGNALTYPE_NONE;
00358 if (!::IsTileType(tile, MP_RAILWAY) || !::HasSignals(tile)) return SIGNALTYPE_NONE;
00359
00360 int data_index = 2 + (::TileX(front) - ::TileX(tile)) + 2 * (::TileY(front) - ::TileY(tile));
00361
00362 for (int i = 0; i < NUM_TRACK_DIRECTIONS; i++) {
00363 const Track &track = _possible_trackdirs[data_index][i].track;
00364 if (!(::TrackToTrackBits(track) & GetRailTracks(tile))) continue;
00365 if (!HasSignalOnTrack(tile, track)) continue;
00366 if (!HasSignalOnTrackdir(tile, _possible_trackdirs[data_index][i].trackdir)) continue;
00367 SignalType st = (SignalType)::GetSignalType(tile, track);
00368 if (HasSignalOnTrackdir(tile, ::ReverseTrackdir(_possible_trackdirs[data_index][i].trackdir))) st = (SignalType)(st | SIGNALTYPE_TWOWAY);
00369 return st;
00370 }
00371
00372 return SIGNALTYPE_NONE;
00373 }
00374
00378 static bool IsValidSignalType(int signal_type)
00379 {
00380 if (signal_type < AIRail::SIGNALTYPE_NORMAL || signal_type > AIRail::SIGNALTYPE_COMBO_TWOWAY) return false;
00381 if (signal_type > AIRail::SIGNALTYPE_PBS_ONEWAY && signal_type < AIRail::SIGNALTYPE_NORMAL_TWOWAY) return false;
00382 return true;
00383 }
00384
00385 bool AIRail::BuildSignal(TileIndex tile, TileIndex front, SignalType signal)
00386 {
00387 EnforcePrecondition(false, AIMap::DistanceManhattan(tile, front) == 1)
00388 EnforcePrecondition(false, ::IsTileType(tile, MP_RAILWAY) && ::IsPlainRailTile(tile));
00389 EnforcePrecondition(false, ::IsValidSignalType(signal));
00390
00391 Track track = INVALID_TRACK;
00392 uint signal_cycles;
00393
00394 int data_index = 2 + (::TileX(front) - ::TileX(tile)) + 2 * (::TileY(front) - ::TileY(tile));
00395 for (int i = 0; i < NUM_TRACK_DIRECTIONS; i++) {
00396 const Track &t = _possible_trackdirs[data_index][i].track;
00397 if (!(::TrackToTrackBits(t) & GetRailTracks(tile))) continue;
00398 track = t;
00399 signal_cycles = _possible_trackdirs[data_index][i].signal_cycles;
00400 break;
00401 }
00402 EnforcePrecondition(false, track != INVALID_TRACK);
00403
00404 uint p1 = track;
00405 if (signal < SIGNALTYPE_TWOWAY) {
00406 if (signal != SIGNALTYPE_PBS && signal != SIGNALTYPE_PBS_ONEWAY) signal_cycles++;
00407 p1 |= (signal_cycles << 15);
00408 }
00409 p1 |= ((signal >= SIGNALTYPE_TWOWAY ? signal ^ SIGNALTYPE_TWOWAY : signal) << 5);
00410
00411 return AIObject::DoCommand(tile, p1, 0, CMD_BUILD_SIGNALS);
00412 }
00413
00414 bool AIRail::RemoveSignal(TileIndex tile, TileIndex front)
00415 {
00416 EnforcePrecondition(false, AIMap::DistanceManhattan(tile, front) == 1)
00417 EnforcePrecondition(false, GetSignalType(tile, front) != SIGNALTYPE_NONE);
00418
00419 Track track = INVALID_TRACK;
00420 int data_index = 2 + (::TileX(front) - ::TileX(tile)) + 2 * (::TileY(front) - ::TileY(tile));
00421 for (int i = 0; i < NUM_TRACK_DIRECTIONS; i++) {
00422 const Track &t = _possible_trackdirs[data_index][i].track;
00423 if (!(::TrackToTrackBits(t) & GetRailTracks(tile))) continue;
00424 track = t;
00425 break;
00426 }
00427 EnforcePrecondition(false, track != INVALID_TRACK);
00428
00429 return AIObject::DoCommand(tile, track, 0, CMD_REMOVE_SIGNALS);
00430 }