1 /*
2  * This file is part of OpenTTD.
3  * 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.
4  * 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.
5  * 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/>.
6  */
7 
8 /** @file script_subsidy.cpp Implementation of ScriptSubsidy. */
9 
10 #include "../../stdafx.h"
11 #include "script_subsidy.hpp"
12 #include "script_date.hpp"
13 #include "script_industry.hpp"
14 #include "script_town.hpp"
15 #include "script_error.hpp"
16 #include "../../subsidy_base.h"
17 #include "../../station_base.h"
18 
19 #include "../../safeguards.h"
20 
IsValidSubsidy(SubsidyID subsidy_id)21 /* static */ bool ScriptSubsidy::IsValidSubsidy(SubsidyID subsidy_id)
22 {
23 	return ::Subsidy::IsValidID(subsidy_id);
24 }
25 
IsAwarded(SubsidyID subsidy_id)26 /* static */ bool ScriptSubsidy::IsAwarded(SubsidyID subsidy_id)
27 {
28 	if (!IsValidSubsidy(subsidy_id)) return false;
29 
30 	return ::Subsidy::Get(subsidy_id)->IsAwarded();
31 }
32 
Create(CargoID cargo_type,SubsidyParticipantType from_type,uint16 from_id,SubsidyParticipantType to_type,uint16 to_id)33 /* static */ bool ScriptSubsidy::Create(CargoID cargo_type, SubsidyParticipantType from_type, uint16 from_id, SubsidyParticipantType to_type, uint16 to_id)
34 {
35 	EnforcePrecondition(false, ScriptCargo::IsValidCargo(cargo_type));
36 	EnforcePrecondition(false, from_type == SPT_INDUSTRY || from_type == SPT_TOWN);
37 	EnforcePrecondition(false, to_type == SPT_INDUSTRY || to_type == SPT_TOWN);
38 	EnforcePrecondition(false, (from_type == SPT_INDUSTRY && ScriptIndustry::IsValidIndustry(from_id)) || (from_type == SPT_TOWN && ScriptTown::IsValidTown(from_id)));
39 	EnforcePrecondition(false, (to_type == SPT_INDUSTRY && ScriptIndustry::IsValidIndustry(to_id)) || (to_type == SPT_TOWN && ScriptTown::IsValidTown(to_id)));
40 
41 	return ScriptObject::DoCommand(0, from_type | (from_id << 8) | (cargo_type << 24), to_type | (to_id << 8), CMD_CREATE_SUBSIDY);
42 }
43 
GetAwardedTo(SubsidyID subsidy_id)44 /* static */ ScriptCompany::CompanyID ScriptSubsidy::GetAwardedTo(SubsidyID subsidy_id)
45 {
46 	if (!IsAwarded(subsidy_id)) return ScriptCompany::COMPANY_INVALID;
47 
48 	return (ScriptCompany::CompanyID)((byte)::Subsidy::Get(subsidy_id)->awarded);
49 }
50 
GetExpireDate(SubsidyID subsidy_id)51 /* static */ ScriptDate::Date ScriptSubsidy::GetExpireDate(SubsidyID subsidy_id)
52 {
53 	if (!IsValidSubsidy(subsidy_id)) return ScriptDate::DATE_INVALID;
54 
55 	int year = ScriptDate::GetYear(ScriptDate::GetCurrentDate());
56 	int month = ScriptDate::GetMonth(ScriptDate::GetCurrentDate());
57 
58 	month += ::Subsidy::Get(subsidy_id)->remaining;
59 
60 	year += (month - 1) / 12;
61 	month = ((month - 1) % 12) + 1;
62 
63 	return ScriptDate::GetDate(year, month, 1);
64 }
65 
GetCargoType(SubsidyID subsidy_id)66 /* static */ CargoID ScriptSubsidy::GetCargoType(SubsidyID subsidy_id)
67 {
68 	if (!IsValidSubsidy(subsidy_id)) return CT_INVALID;
69 
70 	return ::Subsidy::Get(subsidy_id)->cargo_type;
71 }
72 
GetSourceType(SubsidyID subsidy_id)73 /* static */ ScriptSubsidy::SubsidyParticipantType ScriptSubsidy::GetSourceType(SubsidyID subsidy_id)
74 {
75 	if (!IsValidSubsidy(subsidy_id)) return SPT_INVALID;
76 
77 	return (SubsidyParticipantType)(uint)::Subsidy::Get(subsidy_id)->src_type;
78 }
79 
GetSourceIndex(SubsidyID subsidy_id)80 /* static */ int32 ScriptSubsidy::GetSourceIndex(SubsidyID subsidy_id)
81 {
82 	if (!IsValidSubsidy(subsidy_id)) return INVALID_STATION;
83 
84 	return ::Subsidy::Get(subsidy_id)->src;
85 }
86 
GetDestinationType(SubsidyID subsidy_id)87 /* static */ ScriptSubsidy::SubsidyParticipantType ScriptSubsidy::GetDestinationType(SubsidyID subsidy_id)
88 {
89 	if (!IsValidSubsidy(subsidy_id)) return SPT_INVALID;
90 
91 	return (SubsidyParticipantType)(uint)::Subsidy::Get(subsidy_id)->dst_type;
92 }
93 
GetDestinationIndex(SubsidyID subsidy_id)94 /* static */ int32 ScriptSubsidy::GetDestinationIndex(SubsidyID subsidy_id)
95 {
96 	if (!IsValidSubsidy(subsidy_id)) return INVALID_STATION;
97 
98 	return ::Subsidy::Get(subsidy_id)->dst;
99 }
100