1 #include "resources.qh"
3 #include "autocvars.qh"
4 #include "miscfunctions.qh"
6 float GetResourceLimit(entity e, int resource_type)
13 limit = autocvar_g_balance_health_limit;
18 limit = autocvar_g_balance_armor_limit;
23 limit = g_pickup_shells_max;
26 case RESOURCE_BULLETS:
28 limit = g_pickup_nails_max;
31 case RESOURCE_ROCKETS:
33 limit = g_pickup_rockets_max;
38 limit = g_pickup_cells_max;
43 limit = g_pickup_plasma_max;
48 limit = autocvar_g_balance_fuel_limit;
53 error("GetResourceLimit: Invalid resource type.");
57 MUTATOR_CALLHOOK(GetResourceLimit, e, resource_type, limit);
58 limit = M_ARGV(2, float);
59 if (limit > RESOURCE_AMOUNT_HARD_LIMIT)
61 limit = RESOURCE_AMOUNT_HARD_LIMIT;
66 float GetResourceAmount(entity e, int resource_type)
68 .float resource_field = GetResourceField(resource_type);
69 return e.(resource_field);
72 void SetResourceAmount(entity e, int resource_type, float amount)
74 bool forbid = MUTATOR_CALLHOOK(SetResourceAmount, e, resource_type, amount);
79 resource_type = M_ARGV(1, int);
80 amount = M_ARGV(2, float);
81 .float resource_field = GetResourceField(resource_type);
82 if (e.(resource_field) == amount)
86 float max_amount = GetResourceLimit(e, resource_type);
87 if (amount > max_amount)
91 e.(resource_field) = amount;
94 void GiveResource(entity receiver, int resource_type, float amount)
100 bool forbid = MUTATOR_CALLHOOK(GiveResource, receiver, resource_type,
106 resource_type = M_ARGV(1, int);
107 amount = M_ARGV(2, float);
112 SetResourceAmount(receiver, resource_type,
113 GetResourceAmount(receiver, resource_type) + amount);
114 switch (resource_type)
116 case RESOURCE_HEALTH:
118 receiver.pauserothealth_finished =
119 max(receiver.pauserothealth_finished, time +
120 autocvar_g_balance_pause_health_rot);
125 receiver.pauserotarmor_finished =
126 max(receiver.pauserotarmor_finished, time +
127 autocvar_g_balance_pause_armor_rot);
132 receiver.pauserotfuel_finished = max(receiver.pauserotfuel_finished,
133 time + autocvar_g_balance_pause_fuel_rot);
139 int GetResourceType(.float resource_field)
141 switch (resource_field)
143 case health: { return RESOURCE_HEALTH; }
144 case armorvalue: { return RESOURCE_ARMOR; }
145 case ammo_shells: { return RESOURCE_SHELLS; }
146 case ammo_nails: { return RESOURCE_BULLETS; }
147 case ammo_rockets: { return RESOURCE_ROCKETS; }
148 case ammo_cells: { return RESOURCE_CELLS; }
149 case ammo_plasma: { return RESOURCE_PLASMA; }
150 case ammo_fuel: { return RESOURCE_FUEL; }
152 error("GetResourceType: Invalid field.");
156 .float GetResourceField(int resource_type)
158 switch (resource_type)
160 case RESOURCE_HEALTH: { return health; }
161 case RESOURCE_ARMOR: { return armorvalue; }
162 case RESOURCE_SHELLS: { return ammo_shells; }
163 case RESOURCE_BULLETS: { return ammo_nails; }
164 case RESOURCE_ROCKETS: { return ammo_rockets; }
165 case RESOURCE_CELLS: { return ammo_cells; }
166 case RESOURCE_PLASMA: { return ammo_plasma; }
167 case RESOURCE_FUEL: { return ammo_fuel; }
169 error("GetResourceField: Invalid resource type.");