rubidium@9599: /* $Id$ */
rubidium@9599: 
truebrain@9833: /** @file ai_tilelist.cpp Implementation of AITileList and friends. */
rubidium@9599: 
truelight@9592: #include "ai_tilelist.hpp"
truebrain@9757: #include "ai_industry.hpp"
truelight@9592: 
rubidium@9599: #include "../../landscape.h"
truebrain@9757: #include "../../settings_type.h"
rubidium@9837: #include "../../station_func.h"
rubidium@9723: #include "../../map_func.h"
truebrain@9757: #include "../../tile_map.h"
truebrain@9757: #include "../../industry_map.h"
rubidium@9599: 
truelight@9592: void AITileList::FixRectangleSpan(TileIndex &t1, TileIndex &t2)
truelight@9592: {
truelight@9592: 	uint x1 = ::TileX(t1);
truelight@9592: 	uint x2 = ::TileX(t2);
truelight@9592: 
truelight@9592: 	uint y1 = ::TileY(t1);
truelight@9592: 	uint y2 = ::TileY(t2);
truelight@9592: 
truelight@9592: 	if (x1 >= x2) ::Swap(x1, x2);
truelight@9592: 	if (y1 >= y2) ::Swap(y1, y2);
truelight@9592: 
truelight@9592: 	t1 = ::TileXY(x1, y1);
truelight@9592: 	t2 = ::TileXY(x2, y2);
truelight@9592: }
truelight@9592: 
truelight@9592: void AITileList::AddRectangle(TileIndex t1, TileIndex t2)
truelight@9592: {
truebrain@9801: 	if (!::IsValidTile(t1)) return;
truebrain@9801: 	if (!::IsValidTile(t2)) return;
truelight@9592: 
truelight@9592: 	this->FixRectangleSpan(t1, t2);
truelight@9592: 
truelight@9592: 	uint w = TileX(t2) - TileX(t1) + 1;
truelight@9592: 	uint h = TileY(t2) - TileY(t1) + 1;
truelight@9592: 
truelight@9592: 	BEGIN_TILE_LOOP(t, w, h, t1) {
truelight@9592: 		this->AddItem(t);
truelight@9592: 	} END_TILE_LOOP(t, w, h, t1)
truelight@9592: }
truelight@9592: 
truelight@9592: void AITileList::AddTile(TileIndex tile)
truelight@9592: {
truebrain@9801: 	if (!::IsValidTile(tile)) return;
truelight@9592: 
truelight@9592: 	this->AddItem(tile);
truelight@9592: }
truelight@9592: 
truelight@9592: void AITileList::RemoveRectangle(TileIndex t1, TileIndex t2)
truelight@9592: {
truebrain@9801: 	if (!::IsValidTile(t1)) return;
truebrain@9801: 	if (!::IsValidTile(t2)) return;
truelight@9592: 
truelight@9592: 	this->FixRectangleSpan(t1, t2);
truelight@9592: 
truelight@9592: 	uint w = TileX(t2) - TileX(t1) + 1;
truelight@9592: 	uint h = TileY(t2) - TileY(t1) + 1;
truelight@9592: 
truelight@9592: 	BEGIN_TILE_LOOP(t, w, h, t1) {
truelight@9592: 		this->RemoveItem(t);
truelight@9592: 	} END_TILE_LOOP(t, w, h, t1)
truelight@9592: }
truelight@9592: 
truelight@9592: void AITileList::RemoveTile(TileIndex tile)
truelight@9592: {
truebrain@9801: 	if (!::IsValidTile(tile)) return;
truelight@9592: 
truelight@9592: 	this->RemoveItem(tile);
truelight@9592: }
truebrain@9757: 
truebrain@9757: AITileList_IndustryAccepting::AITileList_IndustryAccepting(IndustryID industry_id, uint radius)
truebrain@9757: {
truebrain@9757: 	if (!AIIndustry::IsValidIndustry(industry_id)) return;
truebrain@9757: 
truebrain@9757: 	const Industry *i = ::GetIndustry(industry_id);
truebrain@9757: 	const IndustrySpec *indsp = ::GetIndustrySpec(i->type);
truebrain@9757: 
truebrain@9757: 	/* Check if this industry accepts anything */
truebrain@9765: 	{
truebrain@9765: 		bool cargo_accepts = false;
truebrain@9765: 		for (byte j = 0; j < lengthof(indsp->accepts_cargo); j++) {
truebrain@9765: 			if (indsp->accepts_cargo[j] != CT_INVALID) cargo_accepts = true;
truebrain@9765: 		}
truebrain@9765: 		if (!cargo_accepts) return;
truebrain@9765: 	}
truebrain@9757: 
truebrain@9757: 	if (!_patches.modified_catchment) radius = CA_UNMODIFIED;
truebrain@9757: 
truebrain@9757: 	BEGIN_TILE_LOOP(cur_tile, i->width + radius * 2, i->height + radius * 2, i->xy - ::TileDiffXY(radius, radius)) {
truebrain@9766: 		if (!::IsValidTile(cur_tile)) continue;
truebrain@9757: 		/* Exclude all tiles that belong to this industry */
truebrain@9757: 		if (::IsTileType(cur_tile, MP_INDUSTRY) && ::GetIndustryIndex(cur_tile) == industry_id) continue;
truebrain@9757: 
truebrain@9757: 		/* Only add the tile if it accepts the cargo (sometimes just 1 tile of an
truebrain@9757: 		 *  industry triggers the acceptance). */
truebrain@9757: 		AcceptedCargo accepts;
truebrain@9757: 		::GetAcceptanceAroundTiles(accepts, cur_tile, 1, 1, radius);
truebrain@9765: 		{
truebrain@9765: 			bool cargo_accepts = false;
truebrain@9765: 			for (byte j = 0; j < lengthof(indsp->accepts_cargo); j++) {
truebrain@9765: 				if (indsp->accepts_cargo[j] != CT_INVALID && accepts[indsp->accepts_cargo[j]] != 0) cargo_accepts = true;
truebrain@9765: 			}
truebrain@9765: 			if (!cargo_accepts) continue;
truebrain@9765: 		}
truebrain@9757: 
truebrain@9757: 		this->AddTile(cur_tile);
truebrain@9757: 	} END_TILE_LOOP(cur_tile, i->width + radius * 2, i->height + radius * 2, i->xy - ::TileDiffXY(radius, radius))
truebrain@9757: }
truebrain@9757: 
truebrain@9757: AITileList_IndustryProducing::AITileList_IndustryProducing(IndustryID industry_id, uint radius)
truebrain@9757: {
truebrain@9757: 	if (!AIIndustry::IsValidIndustry(industry_id)) return;
truebrain@9757: 
truebrain@9757: 	const Industry *i = ::GetIndustry(industry_id);
truebrain@9757: 	const IndustrySpec *indsp = ::GetIndustrySpec(i->type);
truebrain@9757: 
truebrain@9757: 	/* Check if this industry produces anything */
truebrain@9765: 	{
truebrain@9765: 		bool cargo_produces = false;
truebrain@9765: 		for (byte j = 0; j < lengthof(indsp->produced_cargo); j++) {
truebrain@9765: 			if (indsp->produced_cargo[j] != CT_INVALID) cargo_produces = true;
truebrain@9765: 		}
truebrain@9765: 		if (!cargo_produces) return;
truebrain@9765: 	}
truebrain@9757: 
truebrain@9757: 	if (!_patches.modified_catchment) radius = CA_UNMODIFIED;
truebrain@9757: 
truebrain@9757: 	BEGIN_TILE_LOOP(cur_tile, i->width + radius * 2, i->height + radius * 2, i->xy - ::TileDiffXY(radius, radius)) {
truebrain@9766: 		if (!::IsValidTile(cur_tile)) continue;
truebrain@9757: 		/* Exclude all tiles that belong to this industry */
truebrain@9757: 		if (::IsTileType(cur_tile, MP_INDUSTRY) && ::GetIndustryIndex(cur_tile) == industry_id) continue;
truebrain@9767: 
truebrain@9767: 		/* Only add the tile if it produces the cargo (a bug in OpenTTD makes this
truebrain@9767: 		 *  inconsitance). */
truebrain@9767: 		AcceptedCargo produces;
truebrain@9767: 		::GetProductionAroundTiles(produces, cur_tile, 1, 1, radius);
truebrain@9767: 		{
truebrain@9767: 			bool cargo_produces = false;
truebrain@9767: 			for (byte j = 0; j < lengthof(indsp->produced_cargo); j++) {
truebrain@9767: 				if (indsp->produced_cargo[j] != CT_INVALID && produces[indsp->produced_cargo[j]] != 0) cargo_produces = true;
truebrain@9767: 			}
truebrain@9767: 			if (!cargo_produces) continue;
truebrain@9767: 		}
truebrain@9767: 
truebrain@9757: 		this->AddTile(cur_tile);
truebrain@9757: 	} END_TILE_LOOP(cur_tile, i->width + radius * 2, i->height + radius * 2, i->xy - ::TileDiffXY(radius, radius))
truebrain@9757: }