]> de.git.xonotic.org Git - xonotic/xonotic-data.pk3dir.git/blob - qcsrc/server/mutators/gamemode_keyhunt.qc
9b33d5d7524ea464a894ed1e1414acbe0e2dd4b0
[xonotic/xonotic-data.pk3dir.git] / qcsrc / server / mutators / gamemode_keyhunt.qc
1 #include "gamemode_keyhunt.qh"
2
3 #include "gamemode.qh"
4
5 float autocvar_g_balance_keyhunt_damageforcescale;
6 float autocvar_g_balance_keyhunt_delay_collect;
7 float autocvar_g_balance_keyhunt_delay_return;
8 float autocvar_g_balance_keyhunt_delay_round;
9 float autocvar_g_balance_keyhunt_delay_tracking;
10 float autocvar_g_balance_keyhunt_dropvelocity;
11 float autocvar_g_balance_keyhunt_maxdist;
12 float autocvar_g_balance_keyhunt_protecttime;
13
14 int autocvar_g_balance_keyhunt_score_capture;
15 int autocvar_g_balance_keyhunt_score_carrierfrag;
16 int autocvar_g_balance_keyhunt_score_collect;
17 int autocvar_g_balance_keyhunt_score_destroyed;
18 int autocvar_g_balance_keyhunt_score_destroyed_ownfactor;
19 int autocvar_g_balance_keyhunt_score_push;
20 float autocvar_g_balance_keyhunt_throwvelocity;
21
22 int autocvar_g_keyhunt_point_leadlimit;
23 bool autocvar_g_keyhunt_team_spawns;
24 #define autocvar_g_keyhunt_point_limit cvar("g_keyhunt_point_limit")
25 int autocvar_g_keyhunt_teams;
26 int autocvar_g_keyhunt_teams_override;
27
28 // #define KH_PLAYER_USE_ATTACHMENT
29 // #define KH_PLAYER_USE_CARRIEDMODEL
30
31 #ifdef KH_PLAYER_USE_ATTACHMENT
32 const vector KH_PLAYER_ATTACHMENT_DIST_ROTATED = '0 -4 0';
33 const vector KH_PLAYER_ATTACHMENT_DIST = '4 0 0';
34 const vector KH_PLAYER_ATTACHMENT = '0 0 0';
35 const vector KH_PLAYER_ATTACHMENT_ANGLES = '0 0 0';
36 const string KH_PLAYER_ATTACHMENT_BONE = "";
37 #else
38 const float KH_KEY_ZSHIFT = 22;
39 const float KH_KEY_XYDIST = 24;
40 const float KH_KEY_XYSPEED = 45;
41 #endif
42 const float KH_KEY_WP_ZSHIFT = 20;
43
44 const vector KH_KEY_MIN = '-10 -10 -46';
45 const vector KH_KEY_MAX = '10 10 3';
46 const float KH_KEY_BRIGHTNESS = 2;
47
48 float kh_no_radar_circles;
49
50 // kh_state
51 //     bits  0- 4: team of key 1, or 0 for no such key, or 30 for dropped, or 31 for self
52 //     bits  5- 9: team of key 2, or 0 for no such key, or 30 for dropped, or 31 for self
53 //     bits 10-14: team of key 3, or 0 for no such key, or 30 for dropped, or 31 for self
54 //     bits 15-19: team of key 4, or 0 for no such key, or 30 for dropped, or 31 for self
55 .float kh_state;
56 .float siren_time;  //  time delay the siren
57 //.float stuff_time;  //  time delay to stuffcmd a cvar
58
59 float kh_keystatus[17];
60 //kh_keystatus[0] = status of dropped keys, kh_keystatus[1 - 16] = player #
61 //replace 17 with cvar("maxplayers") or similar !!!!!!!!!
62 //for(i = 0; i < maxplayers; ++i)
63 //      kh_keystatus[i] = "0";
64
65 float kh_Team_ByID(float t)
66 {
67         if(t == 0) return NUM_TEAM_1;
68         if(t == 1) return NUM_TEAM_2;
69         if(t == 2) return NUM_TEAM_3;
70         if(t == 3) return NUM_TEAM_4;
71         return 0;
72 }
73
74 //entity kh_worldkeylist;
75 .entity kh_worldkeynext;
76 entity kh_controller;
77 //float kh_tracking_enabled;
78 float kh_teams;
79 float kh_interferemsg_time, kh_interferemsg_team;
80 .entity kh_next, kh_prev; // linked list
81 .float kh_droptime;
82 .float kh_dropperteam;
83 .entity kh_previous_owner;
84 .float kh_previous_owner_playerid;
85 .float kh_cp_duration;
86
87 float kh_key_dropped, kh_key_carried;
88
89 const float ST_KH_CAPS = 1;
90 const float SP_KH_CAPS = 4;
91 const float SP_KH_PUSHES = 5;
92 const float SP_KH_DESTROYS = 6;
93 const float SP_KH_PICKUPS = 7;
94 const float SP_KH_KCKILLS = 8;
95 const float SP_KH_LOSSES = 9;
96 void kh_ScoreRules(float teams)
97 {
98         ScoreRules_basics(teams, SFL_SORT_PRIO_PRIMARY, SFL_SORT_PRIO_PRIMARY, true);
99         ScoreInfo_SetLabel_TeamScore(  ST_KH_CAPS,      "caps",      SFL_SORT_PRIO_SECONDARY);
100         ScoreInfo_SetLabel_PlayerScore(SP_KH_CAPS,      "caps",      SFL_SORT_PRIO_SECONDARY);
101         ScoreInfo_SetLabel_PlayerScore(SP_KH_PUSHES,    "pushes",    0);
102         ScoreInfo_SetLabel_PlayerScore(SP_KH_DESTROYS,  "destroyed", SFL_LOWER_IS_BETTER);
103         ScoreInfo_SetLabel_PlayerScore(SP_KH_PICKUPS,   "pickups",   0);
104         ScoreInfo_SetLabel_PlayerScore(SP_KH_KCKILLS,   "kckills",   0);
105         ScoreInfo_SetLabel_PlayerScore(SP_KH_LOSSES,    "losses",    SFL_LOWER_IS_BETTER);
106         ScoreRules_basics_end();
107 }
108
109 float kh_KeyCarrier_waypointsprite_visible_for_player(entity e)  // runs all the time
110 {SELFPARAM();
111         if(!IS_PLAYER(e) || self.team != e.team)
112                 if(!kh_tracking_enabled)
113                         return false;
114
115         return true;
116 }
117
118 float kh_Key_waypointsprite_visible_for_player(entity e) // ??
119 {SELFPARAM();
120         if(!kh_tracking_enabled)
121                 return false;
122         if(!self.owner)
123                 return true;
124         if(!self.owner.owner)
125                 return true;
126         return false;  // draw only when key is not owned
127 }
128
129 void kh_update_state()
130 {
131         entity player;
132         entity key;
133         float s;
134         float f;
135
136         s = 0;
137         FOR_EACH_KH_KEY(key)
138         {
139                 if(key.owner)
140                         f = key.team;
141                 else
142                         f = 30;
143                 s |= pow(32, key.count) * f;
144         }
145
146         FOR_EACH_CLIENT(player)
147         {
148                 player.kh_state = s;
149         }
150
151         FOR_EACH_KH_KEY(key)
152         {
153                 if(key.owner)
154                         key.owner.kh_state |= pow(32, key.count) * 31;
155         }
156         //print(ftos((nextent(world)).kh_state), "\n");
157 }
158
159
160
161
162 var kh_Think_t kh_Controller_Thinkfunc;
163 void kh_Controller_SetThink(float t, kh_Think_t func)  // runs occasionaly
164 {
165         kh_Controller_Thinkfunc = func;
166         kh_controller.cnt = ceil(t);
167         if(t == 0)
168                 kh_controller.nextthink = time; // force
169 }
170 void kh_WaitForPlayers();
171 void kh_Controller_Think()  // called a lot
172 {SELFPARAM();
173         if(intermission_running)
174                 return;
175         if(self.cnt > 0)
176         { if(self.think != kh_WaitForPlayers) { self.cnt -= 1; } }
177         else if(self.cnt == 0)
178         {
179                 self.cnt -= 1;
180                 kh_Controller_Thinkfunc();
181         }
182         self.nextthink = time + 1;
183 }
184
185 // frags f: take from cvar * f
186 // frags 0: no frags
187 void kh_Scores_Event(entity player, entity key, string what, float frags_player, float frags_owner)  // update the score when a key is captured
188 {
189         string s;
190         if(intermission_running)
191                 return;
192
193         if(frags_player)
194                 UpdateFrags(player, frags_player);
195
196         if(key && key.owner && frags_owner)
197                 UpdateFrags(key.owner, frags_owner);
198
199         if(!autocvar_sv_eventlog)  //output extra info to the console or text file
200                 return;
201
202         s = strcat(":keyhunt:", what, ":", ftos(player.playerid), ":", ftos(frags_player));
203
204         if(key && key.owner)
205                 s = strcat(s, ":", ftos(key.owner.playerid));
206         else
207                 s = strcat(s, ":0");
208
209         s = strcat(s, ":", ftos(frags_owner), ":");
210
211         if(key)
212                 s = strcat(s, key.netname);
213
214         GameLogEcho(s);
215 }
216
217 vector kh_AttachedOrigin(entity e)  // runs when a team captures the flag, it can run 2 or 3 times.
218 {
219         if(e.tag_entity)
220         {
221                 makevectors(e.tag_entity.angles);
222                 return e.tag_entity.origin + e.origin.x * v_forward - e.origin.y * v_right + e.origin.z * v_up;
223         }
224         else
225                 return e.origin;
226 }
227
228 void kh_Key_Attach(entity key)  // runs when a player picks up a key and several times when a key is assigned to a player at the start of a round
229 {
230 #ifdef KH_PLAYER_USE_ATTACHMENT
231         entity first;
232         first = key.owner.kh_next;
233         if(key == first)
234         {
235                 setattachment(key, key.owner, KH_PLAYER_ATTACHMENT_BONE);
236                 if(key.kh_next)
237                 {
238                         setattachment(key.kh_next, key, "");
239                         setorigin(key, key.kh_next.origin - 0.5 * KH_PLAYER_ATTACHMENT_DIST);
240                         setorigin(key.kh_next, KH_PLAYER_ATTACHMENT_DIST_ROTATED);
241                         key.kh_next.angles = '0 0 0';
242                 }
243                 else
244                         setorigin(key, KH_PLAYER_ATTACHMENT);
245                 key.angles = KH_PLAYER_ATTACHMENT_ANGLES;
246         }
247         else
248         {
249                 setattachment(key, key.kh_prev, "");
250                 if(key.kh_next)
251                         setattachment(key.kh_next, key, "");
252                 setorigin(key, KH_PLAYER_ATTACHMENT_DIST_ROTATED);
253                 setorigin(first, first.origin - 0.5 * KH_PLAYER_ATTACHMENT_DIST);
254                 key.angles = '0 0 0';
255         }
256 #else
257         setattachment(key, key.owner, "");
258         setorigin(key, '0 0 1' * KH_KEY_ZSHIFT);  // fixing x, y in think
259         key.angles_y -= key.owner.angles.y;
260 #endif
261         key.flags = 0;
262         key.solid = SOLID_NOT;
263         key.movetype = MOVETYPE_NONE;
264         key.team = key.owner.team;
265         key.nextthink = time;
266         key.damageforcescale = 0;
267         key.takedamage = DAMAGE_NO;
268         key.modelindex = kh_key_carried;
269 }
270
271 void kh_Key_Detach(entity key) // runs every time a key is dropped or lost. Runs several times times when all the keys are captured
272 {
273 #ifdef KH_PLAYER_USE_ATTACHMENT
274         entity first;
275         first = key.owner.kh_next;
276         if(key == first)
277         {
278                 if(key.kh_next)
279                 {
280                         setattachment(key.kh_next, key.owner, KH_PLAYER_ATTACHMENT_BONE);
281                         setorigin(key.kh_next, key.origin + 0.5 * KH_PLAYER_ATTACHMENT_DIST);
282                         key.kh_next.angles = KH_PLAYER_ATTACHMENT_ANGLES;
283                 }
284         }
285         else
286         {
287                 if(key.kh_next)
288                         setattachment(key.kh_next, key.kh_prev, "");
289                 setorigin(first, first.origin + 0.5 * KH_PLAYER_ATTACHMENT_DIST);
290         }
291         // in any case:
292         setattachment(key, world, "");
293         setorigin(key, key.owner.origin + '0 0 1' * (PL_MIN.z - KH_KEY_MIN_z));
294         key.angles = key.owner.angles;
295 #else
296         setorigin(key, key.owner.origin + key.origin.z * '0 0 1');
297         setattachment(key, world, "");
298         key.angles_y += key.owner.angles.y;
299 #endif
300         key.flags = FL_ITEM;
301         key.solid = SOLID_TRIGGER;
302         key.movetype = MOVETYPE_TOSS;
303         key.pain_finished = time + autocvar_g_balance_keyhunt_delay_return;
304         key.damageforcescale = autocvar_g_balance_keyhunt_damageforcescale;
305         key.takedamage = DAMAGE_YES;
306         // let key.team stay
307         key.modelindex = kh_key_dropped;
308         key.kh_previous_owner = key.owner;
309         key.kh_previous_owner_playerid = key.owner.playerid;
310 }
311
312 void kh_Key_AssignTo(entity key, entity player)  // runs every time a key is picked up or assigned. Runs prior to kh_key_attach
313 {
314         entity k;
315         float ownerteam0, ownerteam;
316         if(key.owner == player)
317                 return;
318
319         ownerteam0 = kh_Key_AllOwnedByWhichTeam();
320
321         if(key.owner)
322         {
323                 kh_Key_Detach(key);
324
325                 // remove from linked list
326                 if(key.kh_next)
327                         key.kh_next.kh_prev = key.kh_prev;
328                 key.kh_prev.kh_next = key.kh_next;
329                 key.kh_next = world;
330                 key.kh_prev = world;
331
332                 if(key.owner.kh_next == world)
333                 {
334                         // No longer a key carrier
335                         if(!kh_no_radar_circles)
336                                 WaypointSprite_Ping(key.owner.waypointsprite_attachedforcarrier);
337                         WaypointSprite_DetachCarrier(key.owner);
338                 }
339         }
340
341         key.owner = player;
342
343         if(player)
344         {
345                 // insert into linked list
346                 key.kh_next = player.kh_next;
347                 key.kh_prev = player;
348                 player.kh_next = key;
349                 if(key.kh_next)
350                         key.kh_next.kh_prev = key;
351
352                 float i;
353                 i = kh_keystatus[key.owner.playerid];
354                         if(key.netname == "^1red key")
355                                 i += 1;
356                         if(key.netname == "^4blue key")
357                                 i += 2;
358                         if(key.netname == "^3yellow key")
359                                 i += 4;
360                         if(key.netname == "^6pink key")
361                                 i += 8;
362                 kh_keystatus[key.owner.playerid] = i;
363
364                 kh_Key_Attach(key);
365
366                 if(key.kh_next == world)
367                 {
368                         // player is now a key carrier
369                         entity wp = WaypointSprite_AttachCarrier(WP_Null, player, RADARICON_FLAGCARRIER);
370                         wp.colormod = colormapPaletteColor(player.team - 1, 0);
371                         player.waypointsprite_attachedforcarrier.waypointsprite_visible_for_player = kh_KeyCarrier_waypointsprite_visible_for_player;
372                         WaypointSprite_UpdateRule(player.waypointsprite_attachedforcarrier, player.team, SPRITERULE_TEAMPLAY);
373                         if(player.team == NUM_TEAM_1)
374                                 WaypointSprite_UpdateSprites(player.waypointsprite_attachedforcarrier, WP_KeyCarrierRed, WP_KeyCarrierFriend, WP_KeyCarrierRed);
375                         else if(player.team == NUM_TEAM_2)
376                                 WaypointSprite_UpdateSprites(player.waypointsprite_attachedforcarrier, WP_KeyCarrierBlue, WP_KeyCarrierFriend, WP_KeyCarrierBlue);
377                         else if(player.team == NUM_TEAM_3)
378                                 WaypointSprite_UpdateSprites(player.waypointsprite_attachedforcarrier, WP_KeyCarrierYellow, WP_KeyCarrierFriend, WP_KeyCarrierYellow);
379                         else if(player.team == NUM_TEAM_4)
380                                 WaypointSprite_UpdateSprites(player.waypointsprite_attachedforcarrier, WP_KeyCarrierPink, WP_KeyCarrierFriend, WP_KeyCarrierPink);
381                         if(!kh_no_radar_circles)
382                                 WaypointSprite_Ping(player.waypointsprite_attachedforcarrier);
383                 }
384         }
385
386         // moved that here, also update if there's no player
387         kh_update_state();
388
389         key.pusher = world;
390
391         ownerteam = kh_Key_AllOwnedByWhichTeam();
392         if(ownerteam != ownerteam0)
393         {
394                 if(ownerteam != -1)
395                 {
396                         kh_interferemsg_time = time + 0.2;
397                         kh_interferemsg_team = player.team;
398
399                         // audit all key carrier sprites, update them to RUN HERE
400                         FOR_EACH_KH_KEY(k)
401                         {
402                                 if (!k.owner) continue;
403                                 entity first = WP_Null;
404                                 FOREACH(Waypoints, it.netname == k.owner.waypointsprite_attachedforcarrier.model1, LAMBDA(first = it; break));
405                                 entity third = WP_Null;
406                                 FOREACH(Waypoints, it.netname == k.owner.waypointsprite_attachedforcarrier.model3, LAMBDA(third = it; break));
407                                 WaypointSprite_UpdateSprites(k.owner.waypointsprite_attachedforcarrier, first, WP_KeyCarrierFinish, third);
408                         }
409                 }
410                 else
411                 {
412                         kh_interferemsg_time = 0;
413
414                         // audit all key carrier sprites, update them to RUN HERE
415                         FOR_EACH_KH_KEY(k)
416                         {
417                                 if (!k.owner) continue;
418                                 entity first = WP_Null;
419                                 FOREACH(Waypoints, it.netname == k.owner.waypointsprite_attachedforcarrier.model1, LAMBDA(first = it; break));
420                                 entity third = WP_Null;
421                                 FOREACH(Waypoints, it.netname == k.owner.waypointsprite_attachedforcarrier.model3, LAMBDA(third = it; break));
422                                 WaypointSprite_UpdateSprites(k.owner.waypointsprite_attachedforcarrier, first, WP_KeyCarrierFriend, third);
423                         }
424                 }
425         }
426 }
427
428 void kh_Key_Damage(entity inflictor, entity attacker, float damage, int deathtype, vector hitloc, vector force)
429 {SELFPARAM();
430         if(self.owner)
431                 return;
432         if(ITEM_DAMAGE_NEEDKILL(deathtype))
433         {
434                 // touching lava, or hurt trigger
435                 // what shall we do?
436                 // immediately return is bad
437                 // maybe start a shorter countdown?
438         }
439         if(vlen(force) <= 0)
440                 return;
441         if(time > self.pushltime)
442                 if(IS_PLAYER(attacker))
443                         self.team = attacker.team;
444 }
445
446 void kh_Key_Collect(entity key, entity player)  //a player picks up a dropped key
447 {
448         sound(player, CH_TRIGGER, SND_KH_COLLECT, VOL_BASE, ATTEN_NORM);
449
450         if(key.kh_dropperteam != player.team)
451         {
452                 kh_Scores_Event(player, key, "collect", autocvar_g_balance_keyhunt_score_collect, 0);
453                 PlayerScore_Add(player, SP_KH_PICKUPS, 1);
454         }
455         key.kh_dropperteam = 0;
456         Send_Notification(NOTIF_ALL, world, MSG_INFO, APP_TEAM_ENT_4(key, INFO_KEYHUNT_PICKUP_), player.netname);
457
458         kh_Key_AssignTo(key, player); // this also updates .kh_state
459 }
460
461 void kh_Key_Touch()  // runs many, many times when a key has been dropped and can be picked up
462 {SELFPARAM();
463         if(intermission_running)
464                 return;
465
466         if(self.owner) // already carried
467                 return;
468
469         if(ITEM_TOUCH_NEEDKILL())
470         {
471                 // touching sky, or nodrop
472                 // what shall we do?
473                 // immediately return is bad
474                 // maybe start a shorter countdown?
475         }
476
477         if (!IS_PLAYER(other))
478                 return;
479         if(other.deadflag != DEAD_NO)
480                 return;
481         if(other == self.enemy)
482                 if(time < self.kh_droptime + autocvar_g_balance_keyhunt_delay_collect)
483                         return;  // you just dropped it!
484         kh_Key_Collect(self, other);
485 }
486
487 void kh_Key_Remove(entity key)  // runs after when all the keys have been collected or when a key has been dropped for more than X seconds
488 {
489         entity o;
490         o = key.owner;
491         kh_Key_AssignTo(key, world);
492         if(o) // it was attached
493                 WaypointSprite_Kill(key.waypointsprite_attachedforcarrier);
494         else // it was dropped
495                 WaypointSprite_DetachCarrier(key);
496
497         // remove key from key list
498         if (kh_worldkeylist == key)
499                 kh_worldkeylist = kh_worldkeylist.kh_worldkeynext;
500         else
501         {
502                 o = kh_worldkeylist;
503                 while (o)
504                 {
505                         if (o.kh_worldkeynext == key)
506                         {
507                                 o.kh_worldkeynext = o.kh_worldkeynext.kh_worldkeynext;
508                                 break;
509                         }
510                         o = o.kh_worldkeynext;
511                 }
512         }
513
514         remove(key);
515
516         kh_update_state();
517 }
518
519 void kh_FinishRound()  // runs when a team captures the keys
520 {
521         // prepare next round
522         kh_interferemsg_time = 0;
523         entity key;
524
525         kh_no_radar_circles = true;
526         FOR_EACH_KH_KEY(key)
527                 kh_Key_Remove(key);
528         kh_no_radar_circles = false;
529
530         Send_Notification(NOTIF_ALL, world, MSG_CENTER, CENTER_KEYHUNT_ROUNDSTART, autocvar_g_balance_keyhunt_delay_round);
531         kh_Controller_SetThink(autocvar_g_balance_keyhunt_delay_round, kh_StartRound);
532 }
533
534 void kh_WinnerTeam(float teem)  // runs when a team wins // Samual: Teem?.... TEEM?!?! what the fuck is wrong with you people
535 {
536         // all key carriers get some points
537         vector firstorigin, lastorigin, midpoint;
538         float first;
539         entity key;
540         float score;
541         score = (kh_teams - 1) * autocvar_g_balance_keyhunt_score_capture;
542         DistributeEvenly_Init(score, kh_teams);
543         // twice the score for 3 team games, three times the score for 4 team games!
544         // note: for a win by destroying the key, this should NOT be applied
545         FOR_EACH_KH_KEY(key)
546         {
547                 float f;
548                 f = DistributeEvenly_Get(1);
549                 kh_Scores_Event(key.owner, key, "capture", f, 0);
550                 PlayerTeamScore_Add(key.owner, SP_KH_CAPS, ST_KH_CAPS, 1);
551                 nades_GiveBonus(key.owner, autocvar_g_nades_bonus_score_high);
552         }
553
554         first = true;
555         string keyowner = "";
556         FOR_EACH_KH_KEY(key)
557                 if(key.owner.kh_next == key)
558                 {
559                         if(!first)
560                                 keyowner = strcat(keyowner, ", ");
561                         keyowner = key.owner.netname;
562                         first = false;
563                 }
564
565         Send_Notification(NOTIF_ALL, world, MSG_INFO, APP_TEAM_NUM_4(teem, INFO_KEYHUNT_CAPTURE_), keyowner);
566
567         first = true;
568         midpoint = '0 0 0';
569         firstorigin = '0 0 0';
570         lastorigin = '0 0 0';
571         FOR_EACH_KH_KEY(key)
572         {
573                 vector thisorigin;
574
575                 thisorigin = kh_AttachedOrigin(key);
576                 //dprint("Key origin: ", vtos(thisorigin), "\n");
577                 midpoint += thisorigin;
578
579                 if(!first)
580                         te_lightning2(world, lastorigin, thisorigin);
581                 lastorigin = thisorigin;
582                 if(first)
583                         firstorigin = thisorigin;
584                 first = false;
585         }
586         if(kh_teams > 2)
587         {
588                 te_lightning2(world, lastorigin, firstorigin);
589         }
590         midpoint = midpoint * (1 / kh_teams);
591         te_customflash(midpoint, 1000, 1, Team_ColorRGB(teem) * 0.5 + '0.5 0.5 0.5');  // make the color >=0.5 in each component
592
593         play2all(SND(KH_CAPTURE));
594         kh_FinishRound();
595 }
596
597 void kh_LoserTeam(float teem, entity lostkey)  // runs when a player pushes a flag carrier off the map
598 {
599         entity player, key, attacker;
600         float players;
601         float keys;
602         float f;
603
604         attacker = world;
605         if(lostkey.pusher)
606                 if(lostkey.pusher.team != teem)
607                         if(IS_PLAYER(lostkey.pusher))
608                                 attacker = lostkey.pusher;
609
610         players = keys = 0;
611
612         if(attacker)
613         {
614                 if(lostkey.kh_previous_owner)
615                         kh_Scores_Event(lostkey.kh_previous_owner, world, "pushed", 0, -autocvar_g_balance_keyhunt_score_push);
616                         // don't actually GIVE him the -nn points, just log
617                 kh_Scores_Event(attacker, world, "push", autocvar_g_balance_keyhunt_score_push, 0);
618                 PlayerScore_Add(attacker, SP_KH_PUSHES, 1);
619                 //centerprint(attacker, "Your push is the best!"); // does this really need to exist?
620         }
621         else
622         {
623                 float of, fragsleft, i, j, thisteam;
624                 of = autocvar_g_balance_keyhunt_score_destroyed_ownfactor;
625
626                 FOR_EACH_PLAYER(player)
627                         if(player.team != teem)
628                                 ++players;
629
630                 FOR_EACH_KH_KEY(key)
631                         if(key.owner && key.team != teem)
632                                 ++keys;
633
634                 if(lostkey.kh_previous_owner)
635                         kh_Scores_Event(lostkey.kh_previous_owner, world, "destroyed", 0, -autocvar_g_balance_keyhunt_score_destroyed);
636                         // don't actually GIVE him the -nn points, just log
637
638                 if(lostkey.kh_previous_owner.playerid == lostkey.kh_previous_owner_playerid)
639                         PlayerScore_Add(lostkey.kh_previous_owner, SP_KH_DESTROYS, 1);
640
641                 DistributeEvenly_Init(autocvar_g_balance_keyhunt_score_destroyed, keys * of + players);
642
643                 FOR_EACH_KH_KEY(key)
644                         if(key.owner && key.team != teem)
645                         {
646                                 f = DistributeEvenly_Get(of);
647                                 kh_Scores_Event(key.owner, world, "destroyed_holdingkey", f, 0);
648                         }
649
650                 fragsleft = DistributeEvenly_Get(players);
651
652                 // Now distribute these among all other teams...
653                 j = kh_teams - 1;
654                 for(i = 0; i < kh_teams; ++i)
655                 {
656                         thisteam = kh_Team_ByID(i);
657                         if(thisteam == teem) // bad boy, no cookie - this WILL happen
658                                 continue;
659
660                         players = 0;
661                         FOR_EACH_PLAYER(player)
662                                 if(player.team == thisteam)
663                                         ++players;
664
665                         DistributeEvenly_Init(fragsleft, j);
666                         fragsleft = DistributeEvenly_Get(j - 1);
667                         DistributeEvenly_Init(DistributeEvenly_Get(1), players);
668
669                         FOR_EACH_PLAYER(player)
670                                 if(player.team == thisteam)
671                                 {
672                                         f = DistributeEvenly_Get(1);
673                                         kh_Scores_Event(player, world, "destroyed", f, 0);
674                                 }
675
676                         --j;
677                 }
678         }
679
680         Send_Notification(NOTIF_ALL, world, MSG_INFO, APP_TEAM_ENT_4(lostkey, INFO_KEYHUNT_LOST_), lostkey.kh_previous_owner.netname);
681
682         play2all(SND(KH_DESTROY));
683         te_tarexplosion(lostkey.origin);
684
685         kh_FinishRound();
686 }
687
688 void kh_Key_Think()  // runs all the time
689 {SELFPARAM();
690         entity head;
691         //entity player;  // needed by FOR_EACH_PLAYER
692
693         if(intermission_running)
694                 return;
695
696         if(self.owner)
697         {
698 #ifndef KH_PLAYER_USE_ATTACHMENT
699                 makevectors('0 1 0' * (self.cnt + (time % 360) * KH_KEY_XYSPEED));
700                 setorigin(self, v_forward * KH_KEY_XYDIST + '0 0 1' * self.origin.z);
701 #endif
702         }
703
704         // if in nodrop or time over, end the round
705         if(!self.owner)
706                 if(time > self.pain_finished)
707                         kh_LoserTeam(self.team, self);
708
709         if(self.owner)
710         if(kh_Key_AllOwnedByWhichTeam() != -1)
711         {
712                 if(self.siren_time < time)
713                 {
714                         sound(self.owner, CH_TRIGGER, SND_KH_ALARM, VOL_BASE, ATTEN_NORM);  // play a simple alarm
715                         self.siren_time = time + 2.5;  // repeat every 2.5 seconds
716                 }
717
718                 entity key;
719                 vector p;
720                 p = self.owner.origin;
721                 FOR_EACH_KH_KEY(key)
722                         if(vlen(key.owner.origin - p) > autocvar_g_balance_keyhunt_maxdist)
723                                 goto not_winning;
724                 kh_WinnerTeam(self.team);
725 :not_winning
726         }
727
728         if(kh_interferemsg_time && time > kh_interferemsg_time)
729         {
730                 kh_interferemsg_time = 0;
731                 FOR_EACH_PLAYER(head)
732                 {
733                         if(head.team == kh_interferemsg_team)
734                                 if(head.kh_next)
735                                         Send_Notification(NOTIF_ONE, head, MSG_CENTER, CENTER_KEYHUNT_MEET);
736                                 else
737                                         Send_Notification(NOTIF_ONE, head, MSG_CENTER, CENTER_KEYHUNT_HELP);
738                         else
739                                 Send_Notification(NOTIF_ONE, head, MSG_CENTER, APP_TEAM_NUM_4(kh_interferemsg_team, CENTER_KEYHUNT_INTERFERE_));
740                 }
741         }
742
743         self.nextthink = time + 0.05;
744 }
745
746 void key_reset()
747 {SELFPARAM();
748         kh_Key_AssignTo(self, world);
749         kh_Key_Remove(self);
750 }
751
752 const string STR_ITEM_KH_KEY = "item_kh_key";
753 void kh_Key_Spawn(entity initial_owner, float angle, float i)  // runs every time a new flag is created, ie after all the keys have been collected
754 {
755         entity key;
756         key = spawn();
757         key.count = i;
758         key.classname = STR_ITEM_KH_KEY;
759         key.touch = kh_Key_Touch;
760         key.think = kh_Key_Think;
761         key.nextthink = time;
762         key.items = IT_KEY1 | IT_KEY2;
763         key.cnt = angle;
764         key.angles = '0 360 0' * random();
765         key.event_damage = kh_Key_Damage;
766         key.takedamage = DAMAGE_YES;
767         key.modelindex = kh_key_dropped;
768         key.model = "key";
769         key.kh_dropperteam = 0;
770         key.dphitcontentsmask = DPCONTENTS_SOLID | DPCONTENTS_BODY | DPCONTENTS_PLAYERCLIP | DPCONTENTS_BOTCLIP;
771         setsize(key, KH_KEY_MIN, KH_KEY_MAX);
772         key.colormod = Team_ColorRGB(initial_owner.team) * KH_KEY_BRIGHTNESS;
773         key.reset = key_reset;
774
775         switch(initial_owner.team)
776         {
777                 case NUM_TEAM_1:
778                         key.netname = "^1red key";
779                         break;
780                 case NUM_TEAM_2:
781                         key.netname = "^4blue key";
782                         break;
783                 case NUM_TEAM_3:
784                         key.netname = "^3yellow key";
785                         break;
786                 case NUM_TEAM_4:
787                         key.netname = "^6pink key";
788                         break;
789                 default:
790                         key.netname = "NETGIER key";
791                         break;
792         }
793
794         // link into key list
795         key.kh_worldkeynext = kh_worldkeylist;
796         kh_worldkeylist = key;
797
798         Send_Notification(NOTIF_ONE, initial_owner, MSG_CENTER, APP_TEAM_NUM_4(initial_owner.team, CENTER_KEYHUNT_START_));
799
800         WaypointSprite_Spawn(WP_KeyDropped, 0, 0, key, '0 0 1' * KH_KEY_WP_ZSHIFT, world, key.team, key, waypointsprite_attachedforcarrier, false, RADARICON_FLAG);
801         key.waypointsprite_attachedforcarrier.waypointsprite_visible_for_player = kh_Key_waypointsprite_visible_for_player;
802
803         kh_Key_AssignTo(key, initial_owner);
804 }
805
806 // -1 when no team completely owns all keys yet
807 float kh_Key_AllOwnedByWhichTeam()  // constantly called. check to see if all the keys are owned by the same team
808 {
809         entity key;
810         float teem;
811         float keys;
812
813         teem = -1;
814         keys = kh_teams;
815         FOR_EACH_KH_KEY(key)
816         {
817                 if(!key.owner)
818                         return -1;
819                 if(teem == -1)
820                         teem = key.team;
821                 else if(teem != key.team)
822                         return -1;
823                 --keys;
824         }
825         if(keys != 0)
826                 return -1;
827         return teem;
828 }
829
830 void kh_Key_DropOne(entity key)
831 {
832         // prevent collecting this one for some time
833         entity player;
834         player = key.owner;
835
836         key.kh_droptime = time;
837         key.enemy = player;
838
839         kh_Scores_Event(player, key, "dropkey", 0, 0);
840         PlayerScore_Add(player, SP_KH_LOSSES, 1);
841         Send_Notification(NOTIF_ALL, world, MSG_INFO, APP_TEAM_ENT_4(key, INFO_KEYHUNT_DROP_), player.netname);
842
843         kh_Key_AssignTo(key, world);
844         makevectors(player.v_angle);
845         key.velocity = W_CalculateProjectileVelocity(player.velocity, autocvar_g_balance_keyhunt_throwvelocity * v_forward, false);
846         key.pusher = world;
847         key.pushltime = time + autocvar_g_balance_keyhunt_protecttime;
848         key.kh_dropperteam = key.team;
849
850         sound(player, CH_TRIGGER, SND_KH_DROP, VOL_BASE, ATTEN_NORM);
851 }
852
853 void kh_Key_DropAll(entity player, float suicide) // runs whenever a player dies
854 {
855         entity key;
856         entity mypusher;
857         if(player.kh_next)
858         {
859                 mypusher = world;
860                 if(player.pusher)
861                         if(time < player.pushltime)
862                                 mypusher = player.pusher;
863                 while((key = player.kh_next))
864                 {
865                         kh_Scores_Event(player, key, "losekey", 0, 0);
866                         PlayerScore_Add(player, SP_KH_LOSSES, 1);
867                         Send_Notification(NOTIF_ALL, world, MSG_INFO, APP_TEAM_ENT_4(key, INFO_KEYHUNT_LOST_), player.netname);
868                         kh_Key_AssignTo(key, world);
869                         makevectors('-1 0 0' * (45 + 45 * random()) + '0 360 0' * random());
870                         key.velocity = W_CalculateProjectileVelocity(player.velocity, autocvar_g_balance_keyhunt_dropvelocity * v_forward, false);
871                         key.pusher = mypusher;
872                         key.pushltime = time + autocvar_g_balance_keyhunt_protecttime;
873                         if(suicide)
874                                 key.kh_dropperteam = player.team;
875                 }
876                 sound(player, CH_TRIGGER, SND_KH_DROP, VOL_BASE, ATTEN_NORM);
877         }
878 }
879
880 float kh_CheckPlayers(float num)
881 {
882         if(num < kh_teams)
883         {
884                 float t_team = kh_Team_ByID(num);
885                 float players = 0;
886                 entity tmp_player;
887                 FOR_EACH_PLAYER(tmp_player)
888                         if(tmp_player.deadflag == DEAD_NO)
889                                 if(!tmp_player.BUTTON_CHAT)
890                                         if(tmp_player.team == t_team)
891                                                 ++players;
892
893                 if (!players) { return t_team; }
894         }
895         return 0;
896 }
897
898 #define KH_READY_TEAMS() (!p1 + !p2 + ((kh_teams >= 3) ? !p3 : p3) + ((kh_teams >= 4) ? !p4 : p4))
899 #define KH_READY_TEAMS_OK() (KH_READY_TEAMS() == kh_teams)
900 void kh_WaitForPlayers()  // delay start of the round until enough players are present
901 {
902         if(time < game_starttime)
903         {
904                 kh_Controller_SetThink(game_starttime - time + 0.1, kh_WaitForPlayers);
905                 return;
906         }
907
908         static float prev_missing_teams_mask;
909         float p1 = kh_CheckPlayers(0), p2 = kh_CheckPlayers(1), p3 = kh_CheckPlayers(2), p4 = kh_CheckPlayers(3);
910         if(KH_READY_TEAMS_OK())
911         {
912                 if(prev_missing_teams_mask > 0)
913                         Kill_Notification(NOTIF_ALL, world, MSG_CENTER_CPID, CPID_MISSING_TEAMS);
914                 prev_missing_teams_mask = -1;
915                 Send_Notification(NOTIF_ALL, world, MSG_CENTER, CENTER_KEYHUNT_ROUNDSTART, autocvar_g_balance_keyhunt_delay_round);
916                 kh_Controller_SetThink(autocvar_g_balance_keyhunt_delay_round, kh_StartRound);
917         }
918         else
919         {
920                 if(player_count == 0)
921                 {
922                         if(prev_missing_teams_mask > 0)
923                                 Kill_Notification(NOTIF_ALL, world, MSG_CENTER_CPID, CPID_MISSING_TEAMS);
924                         prev_missing_teams_mask = -1;
925                 }
926                 else
927                 {
928                         float missing_teams_mask = (!!p1) + (!!p2) * 2;
929                         if(kh_teams >= 3) missing_teams_mask += (!!p3) * 4;
930                         if(kh_teams >= 4) missing_teams_mask += (!!p4) * 8;
931                         if(prev_missing_teams_mask != missing_teams_mask)
932                         {
933                                 Send_Notification(NOTIF_ALL, world, MSG_CENTER, CENTER_MISSING_TEAMS, missing_teams_mask);
934                                 prev_missing_teams_mask = missing_teams_mask;
935                         }
936                 }
937                 kh_Controller_SetThink(1, kh_WaitForPlayers);
938         }
939 }
940
941 void kh_EnableTrackingDevice()  // runs after each round
942 {
943         Kill_Notification(NOTIF_ALL, world, MSG_CENTER_CPID, CPID_KEYHUNT);
944         Kill_Notification(NOTIF_ALL, world, MSG_CENTER_CPID, CPID_KEYHUNT_OTHER);
945
946         kh_tracking_enabled = true;
947 }
948
949 void kh_StartRound()  // runs at the start of each round
950 {
951         float i, players, teem;
952         entity player;
953
954         if(time < game_starttime)
955         {
956                 kh_Controller_SetThink(game_starttime - time + 0.1, kh_WaitForPlayers);
957                 return;
958         }
959
960         float p1 = kh_CheckPlayers(0), p2 = kh_CheckPlayers(1), p3 = kh_CheckPlayers(2), p4 = kh_CheckPlayers(3);
961         if(!KH_READY_TEAMS_OK())
962         {
963                 kh_Controller_SetThink(1, kh_WaitForPlayers);
964                 return;
965         }
966
967         Kill_Notification(NOTIF_ALL, world, MSG_CENTER_CPID, CPID_KEYHUNT);
968         Kill_Notification(NOTIF_ALL, world, MSG_CENTER_CPID, CPID_KEYHUNT_OTHER);
969
970         for(i = 0; i < kh_teams; ++i)
971         {
972                 teem = kh_Team_ByID(i);
973                 players = 0;
974                 entity my_player = world;
975                 FOR_EACH_PLAYER(player)
976                         if(player.deadflag == DEAD_NO)
977                                 if(!player.BUTTON_CHAT)
978                                         if(player.team == teem)
979                                         {
980                                                 ++players;
981                                                 if(random() * players <= 1)
982                                                         my_player = player;
983                                         }
984                 kh_Key_Spawn(my_player, 360 * i / kh_teams, i);
985         }
986
987         kh_tracking_enabled = false;
988         Send_Notification(NOTIF_ALL, world, MSG_CENTER, CENTER_KEYHUNT_SCAN, autocvar_g_balance_keyhunt_delay_tracking);
989         kh_Controller_SetThink(autocvar_g_balance_keyhunt_delay_tracking, kh_EnableTrackingDevice);
990 }
991
992 float kh_HandleFrags(entity attacker, entity targ, float f)  // adds to the player score
993 {
994         if(attacker == targ)
995                 return f;
996
997         if(targ.kh_next)
998         {
999                 if(attacker.team == targ.team)
1000                 {
1001                         entity k;
1002                         float nk;
1003                         nk = 0;
1004                         for(k = targ.kh_next; k != world; k = k.kh_next)
1005                                 ++nk;
1006                         kh_Scores_Event(attacker, targ.kh_next, "carrierfrag", -nk * autocvar_g_balance_keyhunt_score_collect, 0);
1007                 }
1008                 else
1009                 {
1010                         kh_Scores_Event(attacker, targ.kh_next, "carrierfrag", autocvar_g_balance_keyhunt_score_carrierfrag-1, 0);
1011                         PlayerScore_Add(attacker, SP_KH_KCKILLS, 1);
1012                         // the frag gets added later
1013                 }
1014         }
1015
1016         return f;
1017 }
1018
1019 void kh_Initialize()  // sets up th KH environment
1020 {
1021         // setup variables
1022         kh_teams = autocvar_g_keyhunt_teams_override;
1023         if(kh_teams < 2)
1024                 kh_teams = autocvar_g_keyhunt_teams;
1025         kh_teams = bound(2, kh_teams, 4);
1026
1027         // make a KH entity for controlling the game
1028         kh_controller = spawn();
1029         kh_controller.think = kh_Controller_Think;
1030         kh_Controller_SetThink(0, kh_WaitForPlayers);
1031
1032         setmodel(kh_controller, MDL_KH_KEY);
1033         kh_key_dropped = kh_controller.modelindex;
1034         /*
1035         dprint(vtos(kh_controller.mins));
1036         dprint(vtos(kh_controller.maxs));
1037         dprint("\n");
1038         */
1039 #ifdef KH_PLAYER_USE_CARRIEDMODEL
1040         setmodel(kh_controller, MDL_KH_KEY_CARRIED);
1041         kh_key_carried = kh_controller.modelindex;
1042 #else
1043         kh_key_carried = kh_key_dropped;
1044 #endif
1045
1046         kh_controller.model = "";
1047         kh_controller.modelindex = 0;
1048
1049         addstat(STAT_KH_KEYS, AS_INT, kh_state);
1050
1051         kh_ScoreRules(kh_teams);
1052 }
1053
1054 void kh_finalize()
1055 {
1056         // to be called before intermission
1057         kh_FinishRound();
1058         remove(kh_controller);
1059         kh_controller = world;
1060 }
1061
1062 // legacy bot role
1063
1064 void() havocbot_role_kh_carrier;
1065 void() havocbot_role_kh_defense;
1066 void() havocbot_role_kh_offense;
1067 void() havocbot_role_kh_freelancer;
1068
1069
1070 void havocbot_goalrating_kh(float ratingscale_team, float ratingscale_dropped, float ratingscale_enemy)
1071 {SELFPARAM();
1072         entity head;
1073         for (head = kh_worldkeylist; head; head = head.kh_worldkeynext)
1074         {
1075                 if(head.owner == self)
1076                         continue;
1077                 if(!kh_tracking_enabled)
1078                 {
1079                         // if it's carried by our team we know about it
1080                         // otherwise we have to see it to know about it
1081                         if(!head.owner || head.team != self.team)
1082                         {
1083                                 traceline(self.origin + self.view_ofs, head.origin, MOVE_NOMONSTERS, self);
1084                                 if (trace_fraction < 1 && trace_ent != head)
1085                                         continue; // skip what I can't see
1086                         }
1087                 }
1088                 if(!head.owner)
1089                         navigation_routerating(head, ratingscale_dropped * BOT_PICKUP_RATING_HIGH, 100000);
1090                 else if(head.team == self.team)
1091                         navigation_routerating(head.owner, ratingscale_team * BOT_PICKUP_RATING_HIGH, 100000);
1092                 else
1093                         navigation_routerating(head.owner, ratingscale_enemy * BOT_PICKUP_RATING_HIGH, 100000);
1094         }
1095
1096         havocbot_goalrating_items(1, self.origin, 10000);
1097 }
1098
1099 void havocbot_role_kh_carrier()
1100 {SELFPARAM();
1101         if(self.deadflag != DEAD_NO)
1102                 return;
1103
1104         if (!(self.kh_next))
1105         {
1106                 LOG_TRACE("changing role to freelancer\n");
1107                 self.havocbot_role = havocbot_role_kh_freelancer;
1108                 self.havocbot_role_timeout = 0;
1109                 return;
1110         }
1111
1112         if (self.bot_strategytime < time)
1113         {
1114                 self.bot_strategytime = time + autocvar_bot_ai_strategyinterval;
1115                 navigation_goalrating_start();
1116
1117                 if(kh_Key_AllOwnedByWhichTeam() == self.team)
1118                         havocbot_goalrating_kh(10, 0.1, 0.1); // bring home
1119                 else
1120                         havocbot_goalrating_kh(4, 4, 1); // play defensively
1121
1122                 navigation_goalrating_end();
1123         }
1124 }
1125
1126 void havocbot_role_kh_defense()
1127 {SELFPARAM();
1128         if(self.deadflag != DEAD_NO)
1129                 return;
1130
1131         if (self.kh_next)
1132         {
1133                 LOG_TRACE("changing role to carrier\n");
1134                 self.havocbot_role = havocbot_role_kh_carrier;
1135                 self.havocbot_role_timeout = 0;
1136                 return;
1137         }
1138
1139         if (!self.havocbot_role_timeout)
1140                 self.havocbot_role_timeout = time + random() * 10 + 20;
1141         if (time > self.havocbot_role_timeout)
1142         {
1143                 LOG_TRACE("changing role to freelancer\n");
1144                 self.havocbot_role = havocbot_role_kh_freelancer;
1145                 self.havocbot_role_timeout = 0;
1146                 return;
1147         }
1148
1149         if (self.bot_strategytime < time)
1150         {
1151                 float key_owner_team;
1152                 self.bot_strategytime = time + autocvar_bot_ai_strategyinterval;
1153                 navigation_goalrating_start();
1154
1155                 key_owner_team = kh_Key_AllOwnedByWhichTeam();
1156                 if(key_owner_team == self.team)
1157                         havocbot_goalrating_kh(10, 0.1, 0.1); // defend key carriers
1158                 else if(key_owner_team == -1)
1159                         havocbot_goalrating_kh(4, 1, 0.1); // play defensively
1160                 else
1161                         havocbot_goalrating_kh(0.1, 0.1, 10); // ATTACK ANYWAY
1162
1163                 navigation_goalrating_end();
1164         }
1165 }
1166
1167 void havocbot_role_kh_offense()
1168 {SELFPARAM();
1169         if(self.deadflag != DEAD_NO)
1170                 return;
1171
1172         if (self.kh_next)
1173         {
1174                 LOG_TRACE("changing role to carrier\n");
1175                 self.havocbot_role = havocbot_role_kh_carrier;
1176                 self.havocbot_role_timeout = 0;
1177                 return;
1178         }
1179
1180         if (!self.havocbot_role_timeout)
1181                 self.havocbot_role_timeout = time + random() * 10 + 20;
1182         if (time > self.havocbot_role_timeout)
1183         {
1184                 LOG_TRACE("changing role to freelancer\n");
1185                 self.havocbot_role = havocbot_role_kh_freelancer;
1186                 self.havocbot_role_timeout = 0;
1187                 return;
1188         }
1189
1190         if (self.bot_strategytime < time)
1191         {
1192                 float key_owner_team;
1193
1194                 self.bot_strategytime = time + autocvar_bot_ai_strategyinterval;
1195                 navigation_goalrating_start();
1196
1197                 key_owner_team = kh_Key_AllOwnedByWhichTeam();
1198                 if(key_owner_team == self.team)
1199                         havocbot_goalrating_kh(10, 0.1, 0.1); // defend anyway
1200                 else if(key_owner_team == -1)
1201                         havocbot_goalrating_kh(0.1, 1, 4); // play offensively
1202                 else
1203                         havocbot_goalrating_kh(0.1, 0.1, 10); // ATTACK! EMERGENCY!
1204
1205                 navigation_goalrating_end();
1206         }
1207 }
1208
1209 void havocbot_role_kh_freelancer()
1210 {SELFPARAM();
1211         if(self.deadflag != DEAD_NO)
1212                 return;
1213
1214         if (self.kh_next)
1215         {
1216                 LOG_TRACE("changing role to carrier\n");
1217                 self.havocbot_role = havocbot_role_kh_carrier;
1218                 self.havocbot_role_timeout = 0;
1219                 return;
1220         }
1221
1222         if (!self.havocbot_role_timeout)
1223                 self.havocbot_role_timeout = time + random() * 10 + 10;
1224         if (time > self.havocbot_role_timeout)
1225         {
1226                 if (random() < 0.5)
1227                 {
1228                         LOG_TRACE("changing role to offense\n");
1229                         self.havocbot_role = havocbot_role_kh_offense;
1230                 }
1231                 else
1232                 {
1233                         LOG_TRACE("changing role to defense\n");
1234                         self.havocbot_role = havocbot_role_kh_defense;
1235                 }
1236                 self.havocbot_role_timeout = 0;
1237                 return;
1238         }
1239
1240         if (self.bot_strategytime < time)
1241         {
1242                 float key_owner_team;
1243
1244                 self.bot_strategytime = time + autocvar_bot_ai_strategyinterval;
1245                 navigation_goalrating_start();
1246
1247                 key_owner_team = kh_Key_AllOwnedByWhichTeam();
1248                 if(key_owner_team == self.team)
1249                         havocbot_goalrating_kh(10, 0.1, 0.1); // defend anyway
1250                 else if(key_owner_team == -1)
1251                         havocbot_goalrating_kh(1, 10, 4); // prefer dropped keys
1252                 else
1253                         havocbot_goalrating_kh(0.1, 0.1, 10); // ATTACK ANYWAY
1254
1255                 navigation_goalrating_end();
1256         }
1257 }
1258
1259
1260 // register this as a mutator
1261
1262 MUTATOR_HOOKFUNCTION(kh, ClientDisconnect)
1263 {SELFPARAM();
1264         kh_Key_DropAll(self, true);
1265         return 0;
1266 }
1267
1268 MUTATOR_HOOKFUNCTION(kh, MakePlayerObserver)
1269 {SELFPARAM();
1270         kh_Key_DropAll(self, true);
1271         return 0;
1272 }
1273
1274 MUTATOR_HOOKFUNCTION(kh, PlayerDies)
1275 {SELFPARAM();
1276         if(self == other)
1277                 kh_Key_DropAll(self, true);
1278         else if(IS_PLAYER(other))
1279                 kh_Key_DropAll(self, false);
1280         else
1281                 kh_Key_DropAll(self, true);
1282         return 0;
1283 }
1284
1285 MUTATOR_HOOKFUNCTION(kh, GiveFragsForKill, CBC_ORDER_FIRST)
1286 {
1287         frag_score = kh_HandleFrags(frag_attacker, frag_target, frag_score);
1288         return 0;
1289 }
1290
1291 MUTATOR_HOOKFUNCTION(kh, MatchEnd)
1292 {
1293         kh_finalize();
1294         return 0;
1295 }
1296
1297 MUTATOR_HOOKFUNCTION(kh, GetTeamCount, CBC_ORDER_EXCLUSIVE)
1298 {
1299         ret_float = kh_teams;
1300         return false;
1301 }
1302
1303 MUTATOR_HOOKFUNCTION(kh, SpectateCopy)
1304 {SELFPARAM();
1305         self.kh_state = other.kh_state;
1306         return 0;
1307 }
1308
1309 MUTATOR_HOOKFUNCTION(kh, PlayerUseKey)
1310 {SELFPARAM();
1311         if(MUTATOR_RETURNVALUE == 0)
1312         {
1313                 entity k;
1314                 k = self.kh_next;
1315                 if(k)
1316                 {
1317                         kh_Key_DropOne(k);
1318                         return 1;
1319                 }
1320         }
1321         return 0;
1322 }
1323
1324 MUTATOR_HOOKFUNCTION(kh, HavocBot_ChooseRole)
1325 {
1326         if(self.deadflag != DEAD_NO)
1327                 return true;
1328
1329         float r = random() * 3;
1330         if (r < 1)
1331                 self.havocbot_role = havocbot_role_kh_offense;
1332         else if (r < 2)
1333                 self.havocbot_role = havocbot_role_kh_defense;
1334         else
1335                 self.havocbot_role = havocbot_role_kh_freelancer;
1336
1337         return true;
1338 }
1339
1340 MUTATOR_HOOKFUNCTION(kh, DropSpecialItems)
1341 {
1342         kh_Key_DropAll(frag_target, false);
1343         return false;
1344 }
1345
1346 MUTATOR_HOOKFUNCTION(kh, reset_map_global)
1347 {
1348         kh_Controller_SetThink(autocvar_g_balance_keyhunt_delay_round + (game_starttime - time), kh_StartRound);
1349         return false;
1350 }
1351
1352 REGISTER_MUTATOR(kh, g_keyhunt)
1353 {
1354         ActivateTeamplay();
1355         SetLimits(autocvar_g_keyhunt_point_limit, autocvar_g_keyhunt_point_leadlimit, -1, -1);
1356         if(autocvar_g_keyhunt_team_spawns)
1357                 have_team_spawns = -1; // request team spawns
1358
1359         MUTATOR_ONADD
1360         {
1361                 if(time > 1) // game loads at time 1
1362                         error("This is a game type and it cannot be added at runtime.");
1363                 kh_Initialize();
1364         }
1365
1366         MUTATOR_ONROLLBACK_OR_REMOVE
1367         {
1368                 // we actually cannot roll back kh_Initialize here
1369                 // BUT: we don't need to! If this gets called, adding always
1370                 // succeeds.
1371         }
1372
1373         MUTATOR_ONREMOVE
1374         {
1375                 LOG_INFO("This is a game type and it cannot be removed at runtime.");
1376                 return -1;
1377         }
1378
1379         return 0;
1380 }