mirror of https://github.com/OpenTTD/OpenTTD.git
290 lines
9.5 KiB
C++
290 lines
9.5 KiB
C++
/*
|
|
* This file is part of OpenTTD.
|
|
* 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.
|
|
* 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.
|
|
* 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/>.
|
|
*/
|
|
|
|
/** @file script_industry.cpp Implementation of ScriptIndustry. */
|
|
|
|
#include "../../stdafx.h"
|
|
#include "script_industry.hpp"
|
|
#include "script_cargo.hpp"
|
|
#include "script_company.hpp"
|
|
#include "script_error.hpp"
|
|
#include "script_map.hpp"
|
|
#include "../../company_base.h"
|
|
#include "../../industry.h"
|
|
#include "../../string_func.h"
|
|
#include "../../strings_func.h"
|
|
#include "../../station_base.h"
|
|
#include "../../newgrf_industries.h"
|
|
#include "../../industry_cmd.h"
|
|
#include "../../timer/timer_game_calendar.h"
|
|
#include "table/strings.h"
|
|
|
|
#include "../../safeguards.h"
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetIndustryCount()
|
|
{
|
|
return ::Industry::GetNumItems();
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::IsValidIndustry(IndustryID industry_id)
|
|
{
|
|
return ::Industry::IsValidID(industry_id);
|
|
}
|
|
|
|
/* static */ IndustryID ScriptIndustry::GetIndustryID(TileIndex tile)
|
|
{
|
|
if (!::IsValidTile(tile) || !::IsTileType(tile, MP_INDUSTRY)) return INVALID_INDUSTRY;
|
|
return ::GetIndustryIndex(tile);
|
|
}
|
|
|
|
/* static */ std::optional<std::string> ScriptIndustry::GetName(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return std::nullopt;
|
|
|
|
::SetDParam(0, industry_id);
|
|
return GetString(STR_INDUSTRY_NAME);
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::SetText(IndustryID industry_id, Text *text)
|
|
{
|
|
CCountedPtr<Text> counter(text);
|
|
|
|
EnforceDeityMode(false);
|
|
EnforcePrecondition(false, IsValidIndustry(industry_id));
|
|
|
|
return ScriptObject::Command<CMD_INDUSTRY_SET_TEXT>::Do(industry_id, text != nullptr ? text->GetEncodedText() : std::string{});
|
|
}
|
|
|
|
/* static */ ScriptIndustry::CargoAcceptState ScriptIndustry::IsCargoAccepted(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return CAS_NOT_ACCEPTED;
|
|
if (!ScriptCargo::IsValidCargo(cargo_id)) return CAS_NOT_ACCEPTED;
|
|
|
|
Industry *i = ::Industry::Get(industry_id);
|
|
|
|
if (!i->IsCargoAccepted(cargo_id)) return CAS_NOT_ACCEPTED;
|
|
if (IndustryTemporarilyRefusesCargo(i, cargo_id)) return CAS_TEMP_REFUSED;
|
|
|
|
return CAS_ACCEPTED;
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetStockpiledCargo(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
if (!ScriptCargo::IsValidCargo(cargo_id)) return -1;
|
|
|
|
Industry *i = ::Industry::Get(industry_id);
|
|
|
|
auto it = i->GetCargoAccepted(cargo_id);
|
|
if (it == std::end(i->accepted)) return -1;
|
|
|
|
return it->waiting;
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetLastMonthProduction(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
if (!ScriptCargo::IsValidCargo(cargo_id)) return -1;
|
|
|
|
Industry *i = ::Industry::Get(industry_id);
|
|
|
|
auto it = i->GetCargoProduced(cargo_id);
|
|
if (it == std::end(i->produced)) return -1;
|
|
|
|
return it->history[LAST_MONTH].production;
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetLastMonthTransported(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
if (!ScriptCargo::IsValidCargo(cargo_id)) return -1;
|
|
|
|
Industry *i = ::Industry::Get(industry_id);
|
|
|
|
auto it = i->GetCargoProduced(cargo_id);
|
|
if (it == std::end(i->produced)) return -1;
|
|
|
|
return it->history[LAST_MONTH].transported;
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetLastMonthTransportedPercentage(IndustryID industry_id, CargoID cargo_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
if (!ScriptCargo::IsValidCargo(cargo_id)) return -1;
|
|
|
|
Industry *i = ::Industry::Get(industry_id);
|
|
|
|
auto it = i->GetCargoProduced(cargo_id);
|
|
if (it == std::end(i->produced)) return -1;
|
|
|
|
return ::ToPercent8(it->history[LAST_MONTH].PctTransported());
|
|
}
|
|
|
|
/* static */ TileIndex ScriptIndustry::GetLocation(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return INVALID_TILE;
|
|
|
|
return ::Industry::Get(industry_id)->location.tile;
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetAmountOfStationsAround(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
|
|
Industry *ind = ::Industry::Get(industry_id);
|
|
return ind->stations_near.size();
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetDistanceManhattanToTile(IndustryID industry_id, TileIndex tile)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
|
|
return ScriptMap::DistanceManhattan(tile, GetLocation(industry_id));
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetDistanceSquareToTile(IndustryID industry_id, TileIndex tile)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return -1;
|
|
|
|
return ScriptMap::DistanceSquare(tile, GetLocation(industry_id));
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::IsBuiltOnWater(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return false;
|
|
|
|
return (::GetIndustrySpec(::Industry::Get(industry_id)->type)->behaviour & INDUSTRYBEH_BUILT_ONWATER) != 0;
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::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 ScriptIndustry::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);
|
|
for (TileIndex tile_cur : ind->location) {
|
|
if (IsTileType(tile_cur, MP_STATION) && IsOilRig(tile_cur)) {
|
|
return tile_cur;
|
|
}
|
|
}
|
|
|
|
return INVALID_TILE;
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::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 ScriptIndustry::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);
|
|
for (TileIndex tile_cur : ind->location) {
|
|
if (IsTileType(tile_cur, MP_STATION) && IsOilRig(tile_cur)) {
|
|
return tile_cur;
|
|
}
|
|
}
|
|
|
|
return INVALID_TILE;
|
|
}
|
|
|
|
/* static */ IndustryType ScriptIndustry::GetIndustryType(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return INVALID_INDUSTRYTYPE;
|
|
|
|
return ::Industry::Get(industry_id)->type;
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetLastProductionYear(IndustryID industry_id)
|
|
{
|
|
Industry *i = Industry::GetIfValid(industry_id);
|
|
if (i == nullptr) return 0;
|
|
return i->last_prod_year;
|
|
}
|
|
|
|
/* static */ ScriptDate::Date ScriptIndustry::GetCargoLastAcceptedDate(IndustryID industry_id, CargoID cargo_type)
|
|
{
|
|
Industry *i = Industry::GetIfValid(industry_id);
|
|
if (i == nullptr) return ScriptDate::DATE_INVALID;
|
|
|
|
if (!::IsValidCargoID(cargo_type)) {
|
|
auto it = std::max_element(std::begin(i->accepted), std::end(i->accepted), [](const auto &a, const auto &b) { return a.last_accepted < b.last_accepted; });
|
|
return (ScriptDate::Date)it->last_accepted;
|
|
} else {
|
|
auto it = i->GetCargoAccepted(cargo_type);
|
|
if (it == std::end(i->accepted)) return ScriptDate::DATE_INVALID;
|
|
return (ScriptDate::Date)it->last_accepted;
|
|
}
|
|
}
|
|
|
|
/* static */ SQInteger ScriptIndustry::GetControlFlags(IndustryID industry_id)
|
|
{
|
|
Industry *i = Industry::GetIfValid(industry_id);
|
|
if (i == nullptr) return 0;
|
|
return i->ctlflags;
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::SetControlFlags(IndustryID industry_id, SQInteger control_flags)
|
|
{
|
|
EnforceDeityMode(false);
|
|
if (!IsValidIndustry(industry_id)) return false;
|
|
|
|
return ScriptObject::Command<CMD_INDUSTRY_SET_FLAGS>::Do(industry_id, (::IndustryControlFlags)control_flags & ::INDCTL_MASK);
|
|
}
|
|
|
|
/* static */ ScriptCompany::CompanyID ScriptIndustry::GetExclusiveSupplier(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return ScriptCompany::COMPANY_INVALID;
|
|
|
|
auto company_id = ::Industry::Get(industry_id)->exclusive_supplier;
|
|
if (!::Company::IsValidID(company_id)) return ScriptCompany::COMPANY_INVALID;
|
|
|
|
return (ScriptCompany::CompanyID)((byte)company_id);
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::SetExclusiveSupplier(IndustryID industry_id, ScriptCompany::CompanyID company_id)
|
|
{
|
|
EnforceDeityMode(false);
|
|
EnforcePrecondition(false, IsValidIndustry(industry_id));
|
|
|
|
auto company = ScriptCompany::ResolveCompanyID(company_id);
|
|
::Owner owner = (company == ScriptCompany::COMPANY_INVALID ? ::INVALID_OWNER : (::Owner)company);
|
|
return ScriptObject::Command<CMD_INDUSTRY_SET_EXCLUSIVITY>::Do(industry_id, owner, false);
|
|
}
|
|
|
|
/* static */ ScriptCompany::CompanyID ScriptIndustry::GetExclusiveConsumer(IndustryID industry_id)
|
|
{
|
|
if (!IsValidIndustry(industry_id)) return ScriptCompany::COMPANY_INVALID;
|
|
|
|
auto company_id = ::Industry::Get(industry_id)->exclusive_consumer;
|
|
if (!::Company::IsValidID(company_id)) return ScriptCompany::COMPANY_INVALID;
|
|
|
|
return (ScriptCompany::CompanyID)((byte)company_id);
|
|
}
|
|
|
|
/* static */ bool ScriptIndustry::SetExclusiveConsumer(IndustryID industry_id, ScriptCompany::CompanyID company_id)
|
|
{
|
|
EnforceDeityMode(false);
|
|
EnforcePrecondition(false, IsValidIndustry(industry_id));
|
|
|
|
auto company = ScriptCompany::ResolveCompanyID(company_id);
|
|
::Owner owner = (company == ScriptCompany::COMPANY_INVALID ? ::INVALID_OWNER : (::Owner)company);
|
|
return ScriptObject::Command<CMD_INDUSTRY_SET_EXCLUSIVITY>::Do(industry_id, owner, true);
|
|
}
|