Merge branch 'packer/dmg-and-ca-scoring' into 'master'
[xonotic/xonotic-data.pk3dir.git] / qcsrc / common / gamemodes / gamemode / clanarena / sv_clanarena.qc
1 #include "sv_clanarena.qh"
2
3 float autocvar_g_ca_damage2score_multiplier;
4 bool autocvar_g_ca_spectate_enemies;
5
6 void CA_count_alive_players()
7 {
8         total_players = 0;
9         for (int i = 1; i <= NUM_TEAMS; ++i)
10         {
11                 Team_SetNumberOfAlivePlayers(Team_GetTeamFromIndex(i), 0);
12         }
13         FOREACH_CLIENT(IS_PLAYER(it) && Entity_HasValidTeam(it),
14         {
15                 ++total_players;
16                 if (IS_DEAD(it))
17                 {
18                         continue;
19                 }
20                 entity team_ = Entity_GetTeam(it);
21                 int num_alive = Team_GetNumberOfAlivePlayers(team_);
22                 ++num_alive;
23                 Team_SetNumberOfAlivePlayers(team_, num_alive);
24         });
25         FOREACH_CLIENT(IS_REAL_CLIENT(it),
26         {
27                 STAT(REDALIVE, it) = Team_GetNumberOfAlivePlayers(Team_GetTeamFromIndex(
28                         1));
29                 STAT(BLUEALIVE, it) = Team_GetNumberOfAlivePlayers(
30                         Team_GetTeamFromIndex(2));
31                 STAT(YELLOWALIVE, it) = Team_GetNumberOfAlivePlayers(
32                         Team_GetTeamFromIndex(3));
33                 STAT(PINKALIVE, it) = Team_GetNumberOfAlivePlayers(
34                         Team_GetTeamFromIndex(4));
35         });
36 }
37
38 int CA_GetWinnerTeam()
39 {
40         int winner_team = 0;
41         if (Team_GetNumberOfAlivePlayers(Team_GetTeamFromIndex(1)) >= 1)
42         {
43                 winner_team = NUM_TEAM_1;
44         }
45         for (int i = 2; i <= NUM_TEAMS; ++i)
46         {
47                 if (Team_GetNumberOfAlivePlayers(Team_GetTeamFromIndex(i)) >= 1)
48                 {
49                         if (winner_team != 0)
50                         {
51                                 return 0;
52                         }
53                         winner_team = Team_IndexToTeam(i);
54                 }
55         }
56         if (winner_team)
57         {
58                 return winner_team;
59         }
60         return -1; // no player left
61 }
62
63 void nades_Clear(entity player);
64
65 #define CA_ALIVE_TEAMS_OK() (Team_GetNumberOfAliveTeams() == NumTeams(ca_teams))
66 float CA_CheckWinner()
67 {
68         if(round_handler_GetEndTime() > 0 && round_handler_GetEndTime() - time <= 0)
69         {
70                 Send_Notification(NOTIF_ALL, NULL, MSG_CENTER, CENTER_ROUND_OVER);
71                 Send_Notification(NOTIF_ALL, NULL, MSG_INFO, INFO_ROUND_OVER);
72                 FOREACH_CLIENT(IS_PLAYER(it), { nades_Clear(it); });
73
74                 allowed_to_spawn = false;
75                 game_stopped = true;
76                 round_handler_Init(5, autocvar_g_ca_warmup, autocvar_g_ca_round_timelimit);
77                 return 1;
78         }
79
80         CA_count_alive_players();
81         if (Team_GetNumberOfAliveTeams() > 1)
82         {
83                 return 0;
84         }
85
86         int winner_team = CA_GetWinnerTeam();
87         if(winner_team > 0)
88         {
89                 Send_Notification(NOTIF_ALL, NULL, MSG_CENTER, APP_TEAM_NUM(winner_team, CENTER_ROUND_TEAM_WIN));
90                 Send_Notification(NOTIF_ALL, NULL, MSG_INFO, APP_TEAM_NUM(winner_team, INFO_ROUND_TEAM_WIN));
91                 TeamScore_AddToTeam(winner_team, ST_CA_ROUNDS, +1);
92         }
93         else if(winner_team == -1)
94         {
95                 Send_Notification(NOTIF_ALL, NULL, MSG_CENTER, CENTER_ROUND_TIED);
96                 Send_Notification(NOTIF_ALL, NULL, MSG_INFO, INFO_ROUND_TIED);
97         }
98
99         allowed_to_spawn = false;
100         game_stopped = true;
101         round_handler_Init(5, autocvar_g_ca_warmup, autocvar_g_ca_round_timelimit);
102
103         FOREACH_CLIENT(IS_PLAYER(it), { nades_Clear(it); });
104
105         return 1;
106 }
107
108 void CA_RoundStart()
109 {
110         allowed_to_spawn = boolean(warmup_stage);
111 }
112
113 bool CA_CheckTeams()
114 {
115         static int prev_missing_teams_mask;
116         allowed_to_spawn = true;
117         CA_count_alive_players();
118         if(CA_ALIVE_TEAMS_OK())
119         {
120                 if(prev_missing_teams_mask > 0)
121                         Kill_Notification(NOTIF_ALL, NULL, MSG_CENTER, CPID_MISSING_TEAMS);
122                 prev_missing_teams_mask = -1;
123                 return true;
124         }
125         if(total_players == 0)
126         {
127                 if(prev_missing_teams_mask > 0)
128                         Kill_Notification(NOTIF_ALL, NULL, MSG_CENTER, CPID_MISSING_TEAMS);
129                 prev_missing_teams_mask = -1;
130                 return false;
131         }
132         int missing_teams_mask = 0;
133         for (int i = 1; i <= NUM_TEAMS; ++i)
134         {
135                 if ((ca_teams & Team_IndexToBit(i)) &&
136                         (Team_GetNumberOfAlivePlayers(Team_GetTeamFromIndex(i)) == 0))
137                 {
138                         missing_teams_mask |= Team_IndexToBit(i);
139                 }
140         }
141         if(prev_missing_teams_mask != missing_teams_mask)
142         {
143                 Send_Notification(NOTIF_ALL, NULL, MSG_CENTER, CENTER_MISSING_TEAMS, missing_teams_mask);
144                 prev_missing_teams_mask = missing_teams_mask;
145         }
146         return false;
147 }
148
149 bool ca_isEliminated(entity e)
150 {
151         if(e.caplayer == 1 && (IS_DEAD(e) || e.frags == FRAGS_LMS_LOSER))
152                 return true;
153         if(e.caplayer == 0.5)
154                 return true;
155         return false;
156 }
157
158 /** Returns next available player to spectate if g_ca_spectate_enemies == 0 */
159 entity CA_SpectateNext(entity player, entity start)
160 {
161         if (SAME_TEAM(start, player)) return start;
162         // continue from current player
163         for (entity e = start; (e = find(e, classname, STR_PLAYER)); )
164         {
165                 if (SAME_TEAM(player, e)) return e;
166         }
167         // restart from begining
168         for (entity e = NULL; (e = find(e, classname, STR_PLAYER)); )
169         {
170                 if (SAME_TEAM(player, e)) return e;
171         }
172         return start;
173 }
174
175
176 MUTATOR_HOOKFUNCTION(ca, PlayerSpawn)
177 {
178         entity player = M_ARGV(0, entity);
179
180         player.caplayer = 1;
181         if (!warmup_stage)
182                 eliminatedPlayers.SendFlags |= 1;
183 }
184
185 MUTATOR_HOOKFUNCTION(ca, ForbidSpawn)
186 {
187         entity player = M_ARGV(0, entity);
188
189         // spectators / observers that weren't playing can join; they are
190         // immediately forced to observe in the PutClientInServer hook
191         // this way they are put in a team and can play in the next round
192         if (!allowed_to_spawn && player.caplayer)
193                 return true;
194         return false;
195 }
196
197 MUTATOR_HOOKFUNCTION(ca, PutClientInServer)
198 {
199         entity player = M_ARGV(0, entity);
200
201         if (!allowed_to_spawn && IS_PLAYER(player)) // this is true even when player is trying to join
202         {
203                 TRANSMUTE(Observer, player);
204                 if (CS(player).jointime != time && !player.caplayer) // not when connecting
205                 {
206                         player.caplayer = 0.5;
207                         Send_Notification(NOTIF_ONE_ONLY, player, MSG_INFO, INFO_CA_JOIN_LATE);
208                 }
209         }
210 }
211
212 MUTATOR_HOOKFUNCTION(ca, reset_map_players)
213 {
214         FOREACH_CLIENT(true, {
215                 CS(it).killcount = 0;
216                 if (!it.caplayer && IS_BOT_CLIENT(it))
217                 {
218                         it.team = -1;
219                         it.caplayer = 1;
220                 }
221                 if (it.caplayer)
222                 {
223                         TRANSMUTE(Player, it);
224                         it.caplayer = 1;
225                         PutClientInServer(it);
226                 }
227         });
228         bot_relinkplayerlist();
229         return true;
230 }
231
232 MUTATOR_HOOKFUNCTION(ca, ClientConnect)
233 {
234         entity player = M_ARGV(0, entity);
235
236         TRANSMUTE(Observer, player);
237         return true;
238 }
239
240 MUTATOR_HOOKFUNCTION(ca, reset_map_global)
241 {
242         allowed_to_spawn = true;
243         return true;
244 }
245
246 MUTATOR_HOOKFUNCTION(ca, TeamBalance_CheckAllowedTeams, CBC_ORDER_EXCLUSIVE)
247 {
248         M_ARGV(0, float) = ca_teams;
249         return true;
250 }
251
252 entity ca_LastPlayerForTeam(entity this)
253 {
254         entity last_pl = NULL;
255         FOREACH_CLIENT(IS_PLAYER(it) && it != this, {
256                 if (!IS_DEAD(it))
257                 if (SAME_TEAM(this, it))
258                 if (!last_pl)
259                         last_pl = it;
260                 else
261                         return NULL;
262         });
263         return last_pl;
264 }
265
266 void ca_LastPlayerForTeam_Notify(entity this)
267 {
268         if (!warmup_stage && round_handler_IsActive() && round_handler_IsRoundStarted())
269         {
270                 entity pl = ca_LastPlayerForTeam(this);
271                 if (pl)
272                         Send_Notification(NOTIF_ONE, pl, MSG_CENTER, CENTER_ALONE);
273         }
274 }
275
276 MUTATOR_HOOKFUNCTION(ca, PlayerDies)
277 {
278         entity frag_target = M_ARGV(2, entity);
279
280         ca_LastPlayerForTeam_Notify(frag_target);
281         if (!allowed_to_spawn)
282         {
283                 frag_target.respawn_flags = RESPAWN_SILENT;
284                 // prevent unwanted sudden rejoin as spectator and movement of spectator camera
285                 frag_target.respawn_time = time + 2;
286         }
287         frag_target.respawn_flags |= RESPAWN_FORCE;
288         if (!warmup_stage)
289         {
290                 eliminatedPlayers.SendFlags |= 1;
291                 if (IS_BOT_CLIENT(frag_target))
292                         bot_clear(frag_target);
293         }
294         return true;
295 }
296
297 MUTATOR_HOOKFUNCTION(ca, ClientDisconnect)
298 {
299         entity player = M_ARGV(0, entity);
300
301         if (player.caplayer == 1)
302                 ca_LastPlayerForTeam_Notify(player);
303         return true;
304 }
305
306 MUTATOR_HOOKFUNCTION(ca, MakePlayerObserver)
307 {
308         entity player = M_ARGV(0, entity);
309
310         if (!IS_DEAD(player))
311                 ca_LastPlayerForTeam_Notify(player);
312         if (player.killindicator_teamchange == -2) // player wants to spectate
313                 player.caplayer = 0;
314         if (player.caplayer)
315                 player.frags = FRAGS_LMS_LOSER;
316         if (!warmup_stage)
317                 eliminatedPlayers.SendFlags |= 1;
318         if (!player.caplayer)
319                 return false;  // allow team reset
320         return true;  // prevent team reset
321 }
322
323 MUTATOR_HOOKFUNCTION(ca, ForbidThrowCurrentWeapon)
324 {
325         return true;
326 }
327
328 MUTATOR_HOOKFUNCTION(ca, GiveFragsForKill, CBC_ORDER_FIRST)
329 {
330         M_ARGV(2, float) = 0; // score will be given to the winner team when the round ends
331         return true;
332 }
333
334 MUTATOR_HOOKFUNCTION(ca, SetStartItems)
335 {
336         start_items       &= ~IT_UNLIMITED_AMMO;
337         start_health       = warmup_start_health       = cvar("g_lms_start_health");
338         start_armorvalue   = warmup_start_armorvalue   = cvar("g_lms_start_armor");
339         start_ammo_shells  = warmup_start_ammo_shells  = cvar("g_lms_start_ammo_shells");
340         start_ammo_nails   = warmup_start_ammo_nails   = cvar("g_lms_start_ammo_nails");
341         start_ammo_rockets = warmup_start_ammo_rockets = cvar("g_lms_start_ammo_rockets");
342         start_ammo_cells   = warmup_start_ammo_cells   = cvar("g_lms_start_ammo_cells");
343         start_ammo_plasma  = warmup_start_ammo_plasma  = cvar("g_lms_start_ammo_plasma");
344         start_ammo_fuel    = warmup_start_ammo_fuel    = cvar("g_lms_start_ammo_fuel");
345 }
346
347 MUTATOR_HOOKFUNCTION(ca, Damage_Calculate)
348 {
349         entity frag_attacker = M_ARGV(1, entity);
350         entity frag_target = M_ARGV(2, entity);
351         float frag_deathtype = M_ARGV(3, float);
352         float frag_damage = M_ARGV(4, float);
353         float frag_mirrordamage = M_ARGV(5, float);
354
355         if (IS_PLAYER(frag_target))
356         if (!IS_DEAD(frag_target))
357         if (frag_target == frag_attacker || SAME_TEAM(frag_target, frag_attacker) || frag_deathtype == DEATH_FALL.m_id)
358                 frag_damage = 0;
359
360         frag_mirrordamage = 0;
361
362         M_ARGV(4, float) = frag_damage;
363         M_ARGV(5, float) = frag_mirrordamage;
364 }
365
366 MUTATOR_HOOKFUNCTION(ca, FilterItem)
367 {
368         entity item = M_ARGV(0, entity);
369
370         if (autocvar_g_powerups <= 0)
371         if (item.flags & FL_POWERUP)
372                 return true;
373
374         if (autocvar_g_pickup_items <= 0)
375                 return true;
376 }
377
378 MUTATOR_HOOKFUNCTION(ca, PlayerDamage_SplitHealthArmor)
379 {
380         entity frag_attacker = M_ARGV(1, entity);
381         entity frag_target = M_ARGV(2, entity);
382         float frag_damage = M_ARGV(7, float);
383         float damage_take = bound(0, M_ARGV(4, float), GetResourceAmount(frag_target, RESOURCE_HEALTH));
384         float damage_save = bound(0, M_ARGV(5, float), GetResourceAmount(frag_target, RESOURCE_ARMOR));
385
386         float excess = max(0, frag_damage - damage_take - damage_save);
387
388         if (frag_target != frag_attacker && IS_PLAYER(frag_attacker) && !SAME_TEAM(frag_target, frag_attacker))
389                 GameRules_scoring_add_team(frag_attacker, SCORE, (frag_damage - excess) * autocvar_g_ca_damage2score_multiplier);
390 }
391
392 MUTATOR_HOOKFUNCTION(ca, CalculateRespawnTime)
393 {
394         // no respawn calculations needed, player is forced to spectate anyway
395         return true;
396 }
397
398 MUTATOR_HOOKFUNCTION(ca, PlayerRegen)
399 {
400         // no regeneration in CA
401         return true;
402 }
403
404 MUTATOR_HOOKFUNCTION(ca, Scores_CountFragsRemaining)
405 {
406         // announce remaining frags
407         return true;
408 }
409
410 MUTATOR_HOOKFUNCTION(ca, SpectateSet)
411 {
412         entity client = M_ARGV(0, entity);
413         entity targ = M_ARGV(1, entity);
414
415         if (!autocvar_g_ca_spectate_enemies && client.caplayer)
416         if (DIFF_TEAM(targ, client))
417                 return true;
418 }
419
420 MUTATOR_HOOKFUNCTION(ca, SpectateNext)
421 {
422         entity client = M_ARGV(0, entity);
423
424         if (!autocvar_g_ca_spectate_enemies && client.caplayer)
425         {
426                 entity targ = M_ARGV(1, entity);
427                 M_ARGV(1, entity) = CA_SpectateNext(client, targ);
428                 return true;
429         }
430 }
431
432 MUTATOR_HOOKFUNCTION(ca, SpectatePrev)
433 {
434         entity client = M_ARGV(0, entity);
435         entity targ = M_ARGV(1, entity);
436         entity first = M_ARGV(2, entity);
437
438         if (!autocvar_g_ca_spectate_enemies && client.caplayer)
439         {
440                 do { targ = targ.chain; }
441                 while(targ && DIFF_TEAM(targ, client));
442
443                 if (!targ)
444                 {
445                         for (targ = first; targ && DIFF_TEAM(targ, client); targ = targ.chain);
446
447                         if (targ == client.enemy)
448                                 return MUT_SPECPREV_RETURN;
449                 }
450         }
451
452         M_ARGV(1, entity) = targ;
453
454         return MUT_SPECPREV_FOUND;
455 }
456
457 MUTATOR_HOOKFUNCTION(ca, Bot_FixCount, CBC_ORDER_EXCLUSIVE)
458 {
459         FOREACH_CLIENT(IS_REAL_CLIENT(it), {
460                 if (IS_PLAYER(it) || it.caplayer == 1)
461                         ++M_ARGV(0, int);
462                 ++M_ARGV(1, int);
463         });
464         return true;
465 }
466
467 MUTATOR_HOOKFUNCTION(ca, ClientCommand_Spectate)
468 {
469         entity player = M_ARGV(0, entity);
470
471         if (player.caplayer)
472         {
473                 // they're going to spec, we can do other checks
474                 if (autocvar_sv_spectate && (IS_SPEC(player) || IS_OBSERVER(player)))
475                         Send_Notification(NOTIF_ONE_ONLY, player, MSG_INFO, INFO_CA_LEAVE);
476                 return MUT_SPECCMD_FORCE;
477         }
478
479         return MUT_SPECCMD_CONTINUE;
480 }
481
482 MUTATOR_HOOKFUNCTION(ca, WantWeapon)
483 {
484         M_ARGV(2, bool) = true; // all weapons
485 }
486
487 MUTATOR_HOOKFUNCTION(ca, HideTeamNagger)
488 {
489         return true; // doesn't work well with the whole spectator as player thing
490 }
491
492 MUTATOR_HOOKFUNCTION(ca, GetPlayerStatus)
493 {
494         entity player = M_ARGV(0, entity);
495
496         return player.caplayer == 1;
497 }
498
499 MUTATOR_HOOKFUNCTION(ca, SetWeaponArena)
500 {
501         // most weapons arena
502         if (M_ARGV(0, string) == "0" || M_ARGV(0, string) == "") M_ARGV(0, string) = "most";
503 }