mirror of https://github.com/OpenTTD/OpenTTD.git
182 lines
5.1 KiB
C++
182 lines
5.1 KiB
C++
/* $Id$ */
|
|
|
|
/** @file ai_industry.cpp Implementation of AIIndustry. */
|
|
|
|
#include "ai_industry.hpp"
|
|
#include "ai_cargo.hpp"
|
|
#include "ai_map.hpp"
|
|
#include "../../tile_type.h"
|
|
#include "../../industry.h"
|
|
#include "../../strings_func.h"
|
|
#include "../../station_map.h"
|
|
#include "table/strings.h"
|
|
|
|
/* static */ int32 AIIndustry::GetIndustryCount()
|
|
{
|
|
return ::GetNumIndustries();
|
|
}
|
|
|
|
/* static */ bool AIIndustry::IsValidIndustry(IndustryID industry_id)
|
|
{
|
|
return ::IsValidIndustryID(industry_id);
|
|
}
|
|
|
|
/* static */ char *AIIndustry::GetName(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return NULL;
|
|
static const int len = 64;
|
|
char *industry_name = MallocT<char>(len);
|
|
|
|
::SetDParam(0, industry_id);
|
|
::GetString(industry_name, STR_INDUSTRY, &industry_name[len - 1]);
|
|
|
|
return industry_name;
|
|
}
|
|
|
|
/* static */ bool AIIndustry::IsCargoAccepted(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return false;
|
|
if (!AICargo::IsValidCargo(cargo_id)) return false;
|
|
|
|
const Industry *i = ::Industry::Get(industry_id);
|
|
|
|
for (byte j = 0; j < lengthof(i->accepts_cargo); j++) {
|
|
if (i->accepts_cargo[j] == cargo_id) return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/* static */ int32 AIIndustry::GetStockpiledCargo(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
if (!AICargo::IsValidCargo(cargo_id)) return -1;
|
|
|
|
Industry *ind = ::Industry::Get(industry_id);
|
|
for (uint i = 0; i < lengthof(ind->accepts_cargo); i++) {
|
|
CargoID cid = ind->accepts_cargo[i];
|
|
if (cid == cargo_id) {
|
|
return ind->incoming_cargo_waiting[i];
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
/* static */ int32 AIIndustry::GetLastMonthProduction(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
if (!AICargo::IsValidCargo(cargo_id)) return -1;
|
|
|
|
const Industry *i = ::Industry::Get(industry_id);
|
|
|
|
for (byte j = 0; j < lengthof(i->produced_cargo); j++) {
|
|
if (i->produced_cargo[j] == cargo_id) return i->last_month_production[j];
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
/* static */ int32 AIIndustry::GetLastMonthTransported(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
if (!AICargo::IsValidCargo(cargo_id)) return -1;
|
|
|
|
const Industry *i = ::Industry::Get(industry_id);
|
|
|
|
for (byte j = 0; j < lengthof(i->produced_cargo); j++) {
|
|
if (i->produced_cargo[j] == cargo_id) return i->last_month_transported[j];
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
/* static */ TileIndex AIIndustry::GetLocation(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return INVALID_TILE;
|
|
|
|
return ::Industry::Get(industry_id)->xy;
|
|
}
|
|
|
|
/* static */ int32 AIIndustry::GetAmountOfStationsAround(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
|
|
Industry *ind = ::Industry::Get(industry_id);
|
|
StationList stations;
|
|
::FindStationsAroundTiles(ind->xy, ind->width, ind->height, &stations);
|
|
return (int32)stations.Length();
|
|
}
|
|
|
|
/* static */ int32 AIIndustry::GetDistanceManhattanToTile(IndustryID industry_id, TileIndex tile)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
|
|
return AIMap::DistanceManhattan(tile, GetLocation(industry_id));
|
|
}
|
|
|
|
/* static */ int32 AIIndustry::GetDistanceSquareToTile(IndustryID industry_id, TileIndex tile)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
|
|
return AIMap::DistanceSquare(tile, GetLocation(industry_id));
|
|
}
|
|
|
|
/* static */ bool AIIndustry::IsBuiltOnWater(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return false;
|
|
|
|
return (::GetIndustrySpec(::Industry::Get(industry_id)->type)->behaviour & INDUSTRYBEH_BUILT_ONWATER) != 0;
|
|
}
|
|
|
|
/* static */ bool AIIndustry::HasHeliport(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return false;
|
|
|
|
return (::GetIndustrySpec(::Industry::Get(industry_id)->type)->behaviour & INDUSTRYBEH_AI_AIRSHIP_ROUTES) != 0;
|
|
}
|
|
|
|
/* static */ TileIndex AIIndustry::GetHeliportLocation(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return INVALID_TILE;
|
|
if (!HasHeliport(industry_id)) return INVALID_TILE;
|
|
|
|
const Industry *ind = ::Industry::Get(industry_id);
|
|
BEGIN_TILE_LOOP(tile_cur, ind->width, ind->height, ind->xy);
|
|
if (IsTileType(tile_cur, MP_STATION) && IsOilRig(tile_cur)) {
|
|
return tile_cur;
|
|
}
|
|
END_TILE_LOOP(tile_cur, ind->width, ind->height, ind->xy);
|
|
|
|
return INVALID_TILE;
|
|
}
|
|
|
|
/* static */ bool AIIndustry::HasDock(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return false;
|
|
|
|
return (::GetIndustrySpec(::Industry::Get(industry_id)->type)->behaviour & INDUSTRYBEH_AI_AIRSHIP_ROUTES) != 0;
|
|
}
|
|
|
|
/* static */ TileIndex AIIndustry::GetDockLocation(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return INVALID_TILE;
|
|
if (!HasDock(industry_id)) return INVALID_TILE;
|
|
|
|
const Industry *ind = ::Industry::Get(industry_id);
|
|
BEGIN_TILE_LOOP(tile_cur, ind->width, ind->height, ind->xy);
|
|
if (IsTileType(tile_cur, MP_STATION) && IsOilRig(tile_cur)) {
|
|
return tile_cur;
|
|
}
|
|
END_TILE_LOOP(tile_cur, ind->width, ind->height, ind->xy);
|
|
|
|
return INVALID_TILE;
|
|
}
|
|
|
|
/* static */ IndustryType AIIndustry::GetIndustryType(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return INVALID_INDUSTRYTYPE;
|
|
|
|
return ::Industry::Get(industry_id)->type;
|
|
}
|