ai_tilelist.cpp

Go to the documentation of this file.
00001 /* $Id$ */
00002 
00003 /*
00004  * This file is part of OpenTTD.
00005  * 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.
00006  * 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.
00007  * 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/>.
00008  */
00009 
00012 #include "ai_tilelist.hpp"
00013 #include "ai_industry.hpp"
00014 #include "../../industry.h"
00015 #include "../../station_base.h"
00016 
00017 void AITileList::FixRectangleSpan(TileIndex &t1, TileIndex &t2)
00018 {
00019   uint x1 = ::TileX(t1);
00020   uint x2 = ::TileX(t2);
00021 
00022   uint y1 = ::TileY(t1);
00023   uint y2 = ::TileY(t2);
00024 
00025   if (x1 >= x2) ::Swap(x1, x2);
00026   if (y1 >= y2) ::Swap(y1, y2);
00027 
00028   t1 = ::TileXY(x1, y1);
00029   t2 = ::TileXY(x2, y2);
00030 }
00031 
00032 void AITileList::AddRectangle(TileIndex t1, TileIndex t2)
00033 {
00034   if (!::IsValidTile(t1)) return;
00035   if (!::IsValidTile(t2)) return;
00036 
00037   this->FixRectangleSpan(t1, t2);
00038 
00039   uint w = TileX(t2) - TileX(t1) + 1;
00040   uint h = TileY(t2) - TileY(t1) + 1;
00041 
00042   TILE_LOOP(t, w, h, t1) this->AddItem(t);
00043 }
00044 
00045 void AITileList::AddTile(TileIndex tile)
00046 {
00047   if (!::IsValidTile(tile)) return;
00048 
00049   this->AddItem(tile);
00050 }
00051 
00052 void AITileList::RemoveRectangle(TileIndex t1, TileIndex t2)
00053 {
00054   if (!::IsValidTile(t1)) return;
00055   if (!::IsValidTile(t2)) return;
00056 
00057   this->FixRectangleSpan(t1, t2);
00058 
00059   uint w = TileX(t2) - TileX(t1) + 1;
00060   uint h = TileY(t2) - TileY(t1) + 1;
00061 
00062   TILE_LOOP(t, w, h, t1) this->RemoveItem(t);
00063 }
00064 
00065 void AITileList::RemoveTile(TileIndex tile)
00066 {
00067   if (!::IsValidTile(tile)) return;
00068 
00069   this->RemoveItem(tile);
00070 }
00071 
00072 AITileList_IndustryAccepting::AITileList_IndustryAccepting(IndustryID industry_id, int radius)
00073 {
00074   if (!AIIndustry::IsValidIndustry(industry_id) || radius <= 0) return;
00075 
00076   const Industry *i = ::Industry::Get(industry_id);
00077 
00078   /* Check if this industry accepts anything */
00079   {
00080     bool cargo_accepts = false;
00081     for (byte j = 0; j < lengthof(i->accepts_cargo); j++) {
00082       if (i->accepts_cargo[j] != CT_INVALID) cargo_accepts = true;
00083     }
00084     if (!cargo_accepts) return;
00085   }
00086 
00087   if (!_settings_game.station.modified_catchment) radius = CA_UNMODIFIED;
00088 
00089   TILE_LOOP(cur_tile, i->width + radius * 2, i->height + radius * 2, i->xy - ::TileDiffXY(radius, radius)) {
00090     if (!::IsValidTile(cur_tile)) continue;
00091     /* Exclude all tiles that belong to this industry */
00092     if (::IsTileType(cur_tile, MP_INDUSTRY) && ::GetIndustryIndex(cur_tile) == industry_id) continue;
00093 
00094     /* Only add the tile if it accepts the cargo (sometimes just 1 tile of an
00095      *  industry triggers the acceptance). */
00096     CargoArray acceptance = ::GetAcceptanceAroundTiles(cur_tile, 1, 1, radius);
00097     {
00098       bool cargo_accepts = false;
00099       for (byte j = 0; j < lengthof(i->accepts_cargo); j++) {
00100         if (i->accepts_cargo[j] != CT_INVALID && acceptance[i->accepts_cargo[j]] != 0) cargo_accepts = true;
00101       }
00102       if (!cargo_accepts) continue;
00103     }
00104 
00105     this->AddTile(cur_tile);
00106   }
00107 }
00108 
00109 AITileList_IndustryProducing::AITileList_IndustryProducing(IndustryID industry_id, int radius)
00110 {
00111   if (!AIIndustry::IsValidIndustry(industry_id) || radius <= 0) return;
00112 
00113   const Industry *i = ::Industry::Get(industry_id);
00114 
00115   /* Check if this industry produces anything */
00116   {
00117     bool cargo_produces = false;
00118     for (byte j = 0; j < lengthof(i->produced_cargo); j++) {
00119       if (i->produced_cargo[j] != CT_INVALID) cargo_produces = true;
00120     }
00121     if (!cargo_produces) return;
00122   }
00123 
00124   if (!_settings_game.station.modified_catchment) radius = CA_UNMODIFIED;
00125 
00126   TILE_LOOP(cur_tile, i->width + radius * 2, i->height + radius * 2, i->xy - ::TileDiffXY(radius, radius)) {
00127     if (!::IsValidTile(cur_tile)) continue;
00128     /* Exclude all tiles that belong to this industry */
00129     if (::IsTileType(cur_tile, MP_INDUSTRY) && ::GetIndustryIndex(cur_tile) == industry_id) continue;
00130 
00131     /* Only add the tile if it produces the cargo (a bug in OpenTTD makes this
00132      *  inconsitance). */
00133     CargoArray produced = ::GetProductionAroundTiles(cur_tile, 1, 1, radius);
00134     {
00135       bool cargo_produces = false;
00136       for (byte j = 0; j < lengthof(i->produced_cargo); j++) {
00137         if (i->produced_cargo[j] != CT_INVALID && produced[i->produced_cargo[j]] != 0) cargo_produces = true;
00138       }
00139       if (!cargo_produces) continue;
00140     }
00141 
00142     this->AddTile(cur_tile);
00143   }
00144 }
00145 
00146 AITileList_StationType::AITileList_StationType(StationID station_id, AIStation::StationType station_type)
00147 {
00148   if (!AIStation::IsValidStation(station_id)) return;
00149 
00150   const StationRect *rect = &::Station::Get(station_id)->rect;
00151 
00152   uint station_type_value = 0;
00153   /* Convert AIStation::StationType to ::StationType, but do it in a
00154    *  bitmask, so we can scan for multiple entries at the same time. */
00155   if ((station_type & AIStation::STATION_TRAIN) != 0)      station_type_value |= (1 << ::STATION_RAIL);
00156   if ((station_type & AIStation::STATION_TRUCK_STOP) != 0) station_type_value |= (1 << ::STATION_TRUCK);
00157   if ((station_type & AIStation::STATION_BUS_STOP) != 0)   station_type_value |= (1 << ::STATION_BUS);
00158   if ((station_type & AIStation::STATION_AIRPORT) != 0)    station_type_value |= (1 << ::STATION_AIRPORT) | (1 << ::STATION_OILRIG);
00159   if ((station_type & AIStation::STATION_DOCK) != 0)       station_type_value |= (1 << ::STATION_DOCK)    | (1 << ::STATION_OILRIG);
00160 
00161   TILE_LOOP(cur_tile, rect->right - rect->left + 1, rect->bottom - rect->top + 1, ::TileXY(rect->left, rect->top)) {
00162     if (!::IsTileType(cur_tile, MP_STATION)) continue;
00163     if (::GetStationIndex(cur_tile) != station_id) continue;
00164     if (!HasBit(station_type_value, ::GetStationType(cur_tile))) continue;
00165     this->AddTile(cur_tile);
00166   }
00167 }

Generated on Wed Dec 30 20:40:00 2009 for OpenTTD by  doxygen 1.5.6