1 #include "resources.qh"
4 /// \brief Source file that contains implementation of the resource system.
6 /// \copyright GNU GPLv2 or any later version.
8 #include "autocvars.qh"
9 #include "miscfunctions.qh"
11 float GetResourceLimit(entity e, int resource_type)
14 switch (resource_type)
18 limit = autocvar_g_balance_health_limit;
23 limit = autocvar_g_balance_armor_limit;
28 limit = g_pickup_shells_max;
31 case RESOURCE_BULLETS:
33 limit = g_pickup_nails_max;
36 case RESOURCE_ROCKETS:
38 limit = g_pickup_rockets_max;
43 limit = g_pickup_cells_max;
48 limit = g_pickup_plasma_max;
53 limit = autocvar_g_balance_fuel_limit;
58 error("GetResourceLimit: Invalid resource type.");
62 MUTATOR_CALLHOOK(GetResourceLimit, e, resource_type, limit);
63 limit = M_ARGV(2, float);
64 if (limit > RESOURCE_AMOUNT_HARD_LIMIT)
66 limit = RESOURCE_AMOUNT_HARD_LIMIT;
71 float GetResourceAmount(entity e, int resource_type)
73 .float resource_field = GetResourceField(resource_type);
74 return e.(resource_field);
77 void SetResourceAmount(entity e, int resource_type, float amount)
79 bool forbid = MUTATOR_CALLHOOK(SetResourceAmount, e, resource_type, amount);
84 resource_type = M_ARGV(1, int);
85 amount = M_ARGV(2, float);
86 .float resource_field = GetResourceField(resource_type);
87 if (e.(resource_field) == amount)
91 float max_amount = GetResourceLimit(e, resource_type);
92 if (amount > max_amount)
96 e.(resource_field) = amount;
99 void GiveResource(entity receiver, int resource_type, float amount)
105 bool forbid = MUTATOR_CALLHOOK(GiveResource, receiver, resource_type,
111 resource_type = M_ARGV(1, int);
112 amount = M_ARGV(2, float);
117 SetResourceAmount(receiver, resource_type,
118 GetResourceAmount(receiver, resource_type) + amount);
119 switch (resource_type)
121 case RESOURCE_HEALTH:
123 receiver.pauserothealth_finished =
124 max(receiver.pauserothealth_finished, time +
125 autocvar_g_balance_pause_health_rot);
130 receiver.pauserotarmor_finished =
131 max(receiver.pauserotarmor_finished, time +
132 autocvar_g_balance_pause_armor_rot);
137 receiver.pauserotfuel_finished = max(receiver.pauserotfuel_finished,
138 time + autocvar_g_balance_pause_fuel_rot);
144 void GiveResourceWithLimit(entity receiver, int resource_type, float amount,
151 bool forbid = MUTATOR_CALLHOOK(GiveResourceWithLimit, receiver,
152 resource_type, amount, limit);
157 resource_type = M_ARGV(1, int);
158 amount = M_ARGV(2, float);
159 limit = M_ARGV(3, float);
164 float current_amount = GetResourceAmount(receiver, resource_type);
165 if (current_amount + amount > limit)
167 amount = limit - current_amount;
169 GiveResource(receiver, resource_type, amount);
172 int GetResourceType(.float resource_field)
174 switch (resource_field)
176 case health: { return RESOURCE_HEALTH; }
177 case armorvalue: { return RESOURCE_ARMOR; }
178 case ammo_shells: { return RESOURCE_SHELLS; }
179 case ammo_nails: { return RESOURCE_BULLETS; }
180 case ammo_rockets: { return RESOURCE_ROCKETS; }
181 case ammo_cells: { return RESOURCE_CELLS; }
182 case ammo_plasma: { return RESOURCE_PLASMA; }
183 case ammo_fuel: { return RESOURCE_FUEL; }
185 error("GetResourceType: Invalid field.");
189 .float GetResourceField(int resource_type)
191 switch (resource_type)
193 case RESOURCE_HEALTH: { return health; }
194 case RESOURCE_ARMOR: { return armorvalue; }
195 case RESOURCE_SHELLS: { return ammo_shells; }
196 case RESOURCE_BULLETS: { return ammo_nails; }
197 case RESOURCE_ROCKETS: { return ammo_rockets; }
198 case RESOURCE_CELLS: { return ammo_cells; }
199 case RESOURCE_PLASMA: { return ammo_plasma; }
200 case RESOURCE_FUEL: { return ammo_fuel; }
202 error("GetResourceField: Invalid resource type.");