data/qcsrc/server/teamplay.qc

   1 string cache_mutatormsg;\r
   2 string cache_lastmutatormsg;\r
   3 \r
   4 // client counts for each team\r
   5 float c1, c2, c3, c4;\r
   6 // # of bots on those teams\r
   7 float cb1, cb2, cb3, cb4;\r
   8 \r
   9 float audit_teams_time;\r
  10 \r
  11 float IsTeamBalanceForced()\r
  12 {\r
  13         if(intermission_running)\r
  14                 return 0; // no rebalancing whatsoever please\r
  15         if(!teams_matter)\r
  16                 return 0;\r
  17         if(cvar("g_campaign"))\r
  18                 return 0;\r
  19         if(!cvar("g_balance_teams_force"))\r
  20                 return -1;\r
  21         return 1;\r
  22 }\r
  23 \r
  24 void TeamchangeFrags(entity e)\r
  25 {\r
  26         PlayerScore_Clear(e);\r
  27 }\r
  28 \r
  29 vector TeamColor(float teem)\r
  30 {\r
  31         switch(teem)\r
  32         {\r
  33                 case COLOR_TEAM1:\r
  34                         return '1 0.0625 0.0625';\r
  35                 case COLOR_TEAM2:\r
  36                         return '0.0625 0.0625 1';\r
  37                 case COLOR_TEAM3:\r
  38                         return '1 1 0.0625';\r
  39                 case COLOR_TEAM4:\r
  40                         return '1 0.0625 1';\r
  41                 default:\r
  42                         return '1 1 1';\r
  43         }\r
  44 }\r
  45 \r
  46 string TeamName(float t)\r
  47 {\r
  48         return strcat(Team_ColorName(t), " Team");\r
  49 }\r
  50 string ColoredTeamName(float t)\r
  51 {\r
  52         return strcat(Team_ColorCode(t), Team_ColorName(t), " Team^7");\r
  53 }\r
  54 string TeamNoName(float t)\r
  55 {\r
  56         // fixme: Search for team entities and get their .netname's!\r
  57         if(t == 1)\r
  58                 return "Red Team";\r
  59         if(t == 2)\r
  60                 return "Blue Team";\r
  61         if(t == 3)\r
  62                 return "Yellow Team";\r
  63         if(t == 4)\r
  64                 return "Pink Team";\r
  65         return "Neutral Team";\r
  66 }\r
  67 \r
  68 void dom_init();\r
  69 void ctf_init();\r
  70 void tdm_init();\r
  71 void entcs_init();\r
  72 \r
  73 void LogTeamchange(float player_id, float team_number, float type)\r
  74 {\r
  75         if(!cvar("sv_eventlog"))\r
  76                 return;\r
  77 \r
  78         if(player_id < 1)\r
  79                 return;\r
  80 \r
  81         GameLogEcho(strcat(":team:", ftos(player_id), ":", ftos(team_number), ":", ftos(type)));\r
  82 }\r
  83 \r
  84 void WriteGameCvars()\r
  85 {\r
  86         cvar_set("g_dm", ftos(g_dm));\r
  87         cvar_set("g_tdm", ftos(g_tdm));\r
  88         cvar_set("g_domination", ftos(g_domination));\r
  89         cvar_set("g_ctf", ftos(g_ctf));\r
  90         cvar_set("g_lms", ftos(g_lms));\r
  91         cvar_set("g_arena", ftos(g_arena));\r
  92         cvar_set("g_ca", ftos(g_ca));\r
  93         cvar_set("g_keyhunt", ftos(g_keyhunt));\r
  94         cvar_set("g_assault", ftos(g_assault));\r
  95         cvar_set("g_onslaught", ftos(g_onslaught));\r
  96         cvar_set("g_race", ftos(g_race));\r
  97         cvar_set("g_cts", ftos(g_cts));\r
  98         cvar_set("g_rpg", ftos(g_rpg));\r
  99 }\r
 100 \r
 101 void ReadGameCvars()\r
 102 {\r
 103         float found;\r
 104         float prev;\r
 105         float i;\r
 106 \r
 107         found = 0;\r
 108         prev = cvar("gamecfg");\r
 109         for(i = 0; i < 2; ++i)\r
 110         {\r
 111                 found += (g_dm = (!found && (prev != GAME_DEATHMATCH) && cvar("g_dm")));\r
 112                 found += (g_tdm = (!found && (prev != GAME_TEAM_DEATHMATCH) && cvar("g_tdm")));\r
 113                 found += (g_domination = (!found && (prev != GAME_DOMINATION) && cvar("g_domination")));\r
 114                 found += (g_ctf = (!found && (prev != GAME_CTF) && cvar("g_ctf")));\r
 115                 found += (g_lms = (!found && (prev != GAME_LMS) && cvar("g_lms")));\r
 116                 found += (g_arena = (!found && (prev != GAME_ARENA) && cvar("g_arena")));\r
 117                 found += (g_ca = (!found && (prev != GAME_CA) && cvar("g_ca")));\r
 118                 found += (g_keyhunt = (!found && (prev != GAME_KEYHUNT) && cvar("g_keyhunt")));\r
 119                 found += (g_assault = (!found && (prev != GAME_ASSAULT) && cvar("g_assault")));\r
 120                 found += (g_onslaught = (!found && (prev != GAME_ONSLAUGHT) && cvar("g_onslaught")));\r
 121                 found += (g_race = (!found && (prev != GAME_RACE) && cvar("g_race")));\r
 122                 found += (g_cts = (!found && (prev != GAME_CTS) && cvar("g_cts")));\r
 123                 found += (g_rpg = (!found && (prev != GAME_RPG) && cvar("g_rpg")));\r
 124 \r
 125                 if(found)\r
 126                         break;\r
 127 \r
 128                 prev = -1; // second attempt takes place WITHOUT prev set\r
 129         }\r
 130 \r
 131         if(!found)\r
 132                 g_dm = 1;\r
 133 \r
 134         if(g_dm && cvar("deathmatch_force_teamplay"))\r
 135         {\r
 136                 g_dm = 0;\r
 137                 g_tdm = 1;\r
 138         }\r
 139 \r
 140         teams_matter = 0;\r
 141 }\r
 142 \r
 143 void default_delayedinit()\r
 144 {\r
 145         if(!scores_initialized)\r
 146                 ScoreRules_generic();\r
 147 }\r
 148 \r
 149 void ActivateTeamplay()\r
 150 {\r
 151         float teamplay_default;\r
 152         teamplay_default = cvar("teamplay_default");\r
 153 \r
 154         if(teamplay_default)\r
 155                 teamplay = teamplay_default;\r
 156         else\r
 157                 teamplay = 3;\r
 158         cvar_set("teamplay", ftos(teamplay));\r
 159 \r
 160         teams_matter = 1;\r
 161 }\r
 162 \r
 163 void InitGameplayMode()\r
 164 {\r
 165         float fraglimit_override, timelimit_override, leadlimit_override, qualifying_override;\r
 166 \r
 167         qualifying_override = -1;\r
 168 \r
 169         VoteReset();\r
 170 \r
 171         teams_matter = 0;\r
 172         cvar_set("teamplay", "0");\r
 173 \r
 174         // make sure only ONE type is selected\r
 175         ReadGameCvars();\r
 176         WriteGameCvars();\r
 177 \r
 178         // find out good world mins/maxs bounds, either the static bounds found by looking for solid, or the mapinfo specified bounds\r
 179         get_mi_min_max(1);\r
 180         world.mins = mi_min;\r
 181         world.maxs = mi_max;\r
 182 \r
 183         MapInfo_LoadMapSettings(mapname);\r
 184 \r
 185         if not(cvar_value_issafe(world.fog))\r
 186         {\r
 187                 print("The current map contains a potentially harmful fog setting, ignored\n");\r
 188                 world.fog = string_null;\r
 189         }\r
 190         if(MapInfo_Map_fog != "")\r
 191                 if(MapInfo_Map_fog == "none")\r
 192                         world.fog = string_null;\r
 193                 else\r
 194                         world.fog = strzone(MapInfo_Map_fog);\r
 195         clientstuff = strzone(MapInfo_Map_clientstuff);\r
 196 \r
 197         MapInfo_ClearTemps();\r
 198 \r
 199         // in case mapinfo switched the type\r
 200         ReadGameCvars();\r
 201 \r
 202         // set both here, gamemode can override it later\r
 203         timelimit_override = cvar("timelimit_override");\r
 204         fraglimit_override = cvar("fraglimit_override");\r
 205         leadlimit_override = cvar("leadlimit_override");\r
 206 \r
 207         if(g_dm)\r
 208         {\r
 209                 game = GAME_DEATHMATCH;\r
 210                 gamemode_name = "Deathmatch";\r
 211         }\r
 212 \r
 213         if(g_tdm)\r
 214         {\r
 215                 game = GAME_TEAM_DEATHMATCH;\r
 216                 gamemode_name = "Team Deathmatch";\r
 217                 ActivateTeamplay();\r
 218                 tdm_init();\r
 219                 if(cvar("g_tdm_team_spawns"))\r
 220                         have_team_spawns = -1; // request team spawns\r
 221         }\r
 222 \r
 223         if(g_domination)\r
 224         {\r
 225                 game = GAME_DOMINATION;\r
 226                 gamemode_name = "Domination";\r
 227                 ActivateTeamplay();\r
 228                 fraglimit_override = cvar("g_domination_point_limit");\r
 229                 leadlimit_override = cvar("g_domination_point_leadlimit");\r
 230                 dom_init();\r
 231                 have_team_spawns = -1; // request team spawns\r
 232         }\r
 233 \r
 234         if(g_ctf)\r
 235         {\r
 236                 game = GAME_CTF;\r
 237                 gamemode_name = "Capture the Flag";\r
 238                 ActivateTeamplay();\r
 239                 g_ctf_win_mode = cvar("g_ctf_win_mode");\r
 240                 g_ctf_ignore_frags = cvar("g_ctf_ignore_frags");\r
 241                 if(g_ctf_win_mode == 2)\r
 242                 {\r
 243                         fraglimit_override = cvar("g_ctf_capture_limit");\r
 244                         leadlimit_override = cvar("g_ctf_capture_leadlimit");\r
 245                 }\r
 246                 else\r
 247                 {\r
 248                         fraglimit_override = cvar("capturelimit_override");\r
 249                         leadlimit_override = cvar("captureleadlimit_override");\r
 250                 }\r
 251                 ctf_init();\r
 252                 have_team_spawns = -1; // request team spawns\r
 253         }\r
 254 \r
 255         if(g_lms)\r
 256         {\r
 257                 game = GAME_LMS;\r
 258                 gamemode_name = "Last Man Standing";\r
 259                 fraglimit_override = cvar("g_lms_lives_override");\r
 260                 leadlimit_override = 0; // not supported by LMS\r
 261                 if(fraglimit_override == 0)\r
 262                         fraglimit_override = -1;\r
 263                 lms_lowest_lives = 9999;\r
 264                 lms_next_place = 0;\r
 265                 ScoreRules_lms();\r
 266         }\r
 267 \r
 268         if(g_arena)\r
 269         {\r
 270                 game = GAME_ARENA;\r
 271                 gamemode_name = "Arena";\r
 272                 fraglimit_override = cvar("g_arena_point_limit");\r
 273                 leadlimit_override = cvar("g_arena_point_leadlimit");\r
 274                 maxspawned = cvar("g_arena_maxspawned");\r
 275                 if(maxspawned < 2)\r
 276                         maxspawned = 2;\r
 277                 arena_roundbased = cvar("g_arena_roundbased");\r
 278         }\r
 279 \r
 280         if(g_ca)\r
 281         {\r
 282                 game = GAME_CA;\r
 283                 gamemode_name = "Clan Arena";\r
 284                 ActivateTeamplay();\r
 285                 fraglimit_override = cvar("g_ca_point_limit");\r
 286                 leadlimit_override = cvar("g_ca_point_leadlimit");\r
 287                 precache_sound("ctf/red_capture.wav");\r
 288                 precache_sound("ctf/blue_capture.wav");\r
 289         }\r
 290         if(g_keyhunt)\r
 291         {\r
 292                 game = GAME_KEYHUNT;\r
 293                 gamemode_name = "Key Hunt";\r
 294                 ActivateTeamplay();\r
 295                 fraglimit_override = cvar("g_keyhunt_point_limit");\r
 296                 leadlimit_override = cvar("g_keyhunt_point_leadlimit");\r
 297                 kh_init();\r
 298         }\r
 299 \r
 300         if(g_assault)\r
 301         {\r
 302                 game = GAME_ASSAULT;\r
 303                 gamemode_name = "Assault";\r
 304                 ActivateTeamplay();\r
 305                 ScoreRules_assault();\r
 306                 have_team_spawns = -1; // request team spawns\r
 307         }\r
 308 \r
 309         if(g_onslaught)\r
 310         {\r
 311                 game = GAME_ONSLAUGHT;\r
 312                 gamemode_name = "Onslaught";\r
 313                 ActivateTeamplay();\r
 314                 have_team_spawns = -1; // request team spawns\r
 315         }\r
 316 \r
 317         if(g_race)\r
 318         {\r
 319                 game = GAME_RACE;\r
 320                 gamemode_name = "Race";\r
 321 \r
 322                 if(cvar("g_race_teams"))\r
 323                 {\r
 324                         ActivateTeamplay();\r
 325                         race_teams = bound(2, cvar("g_race_teams"), 4);\r
 326                         have_team_spawns = -1; // request team spawns\r
 327                 }\r
 328                 else\r
 329                         race_teams = 0;\r
 330 \r
 331                 qualifying_override = cvar("g_race_qualifying_timelimit_override");\r
 332                 fraglimit_override = cvar("g_race_laps_limit");\r
 333                 leadlimit_override = 0; // currently not supported by race\r
 334         }\r
 335 \r
 336         if(g_cts)\r
 337         {\r
 338                 game = GAME_CTS;\r
 339                 gamemode_name = "CTS";\r
 340                 g_race_qualifying = 1;\r
 341                 fraglimit_override = 0;\r
 342                 leadlimit_override = 0;\r
 343         }\r
 344 \r
 345         if(g_rpg)\r
 346         {\r
 347                 game = GAME_RPG;\r
 348                 gamemode_name = "Role Play";\r
 349                 fraglimit_override = 0;\r
 350                 leadlimit_override = 0;\r
 351                 ScoreRules_rpg();\r
 352         }\r
 353 \r
 354         if(teams_matter)\r
 355                 entcs_init();\r
 356 \r
 357         // save it (for the next startup)\r
 358         cvar_set("gamecfg", ftos(game));\r
 359 \r
 360         cache_mutatormsg = strzone("");\r
 361         cache_lastmutatormsg = strzone("");\r
 362 \r
 363         // enforce the server's universal frag/time limits\r
 364         if(!cvar("g_campaign"))\r
 365         {\r
 366                 if(fraglimit_override >= 0)\r
 367                         cvar_set("fraglimit", ftos(fraglimit_override));\r
 368                 if(timelimit_override >= 0)\r
 369                         cvar_set("timelimit", ftos(timelimit_override));\r
 370                 if(leadlimit_override >= 0)\r
 371                         cvar_set("leadlimit", ftos(leadlimit_override));\r
 372                 if(qualifying_override >= 0)\r
 373                         cvar_set("g_race_qualifying_timelimit", ftos(qualifying_override));\r
 374         }\r
 375 \r
 376         if(g_race)\r
 377         {\r
 378                 // we need to find out the correct value for g_race_qualifying\r
 379                 if(cvar("g_campaign"))\r
 380                 {\r
 381                         g_race_qualifying = 1;\r
 382                 }\r
 383                 else if(!cvar("g_campaign") && cvar("g_race_qualifying_timelimit") > 0)\r
 384                 {\r
 385                         g_race_qualifying = 2;\r
 386                         race_fraglimit = cvar("fraglimit");\r
 387                         race_leadlimit = cvar("leadlimit");\r
 388                         race_timelimit = cvar("timelimit");\r
 389                         cvar_set("fraglimit", "0");\r
 390                         cvar_set("leadlimit", "0");\r
 391                         cvar_set("timelimit", cvar_string("g_race_qualifying_timelimit"));\r
 392                 }\r
 393                 else\r
 394                         g_race_qualifying = 0;\r
 395         }\r
 396 \r
 397         if(g_race || g_cts)\r
 398         {\r
 399                 if(g_race_qualifying)\r
 400                         independent_players = 1;\r
 401 \r
 402                 ScoreRules_race();\r
 403         }\r
 404 \r
 405         InitializeEntity(world, default_delayedinit, INITPRIO_GAMETYPE_FALLBACK);\r
 406 }\r
 407 \r
 408 string GetClientVersionMessage() {\r
 409         local string versionmsg;\r
 410         if (self.version_mismatch) {\r
 411                 if(self.version < cvar("gameversion")) {\r
 412                         versionmsg = "^3Your client version is outdated.\n\n\n### YOU WON'T BE ABLE TO PLAY ON THIS SERVER ###\n\n\nPlease update!!!^8";\r
 413                 } else {\r
 414                         versionmsg = "^3This server is using an outdated Voretournament version.\n\n\n ### THIS SERVER IS INCOMPATIBLE AND THUS YOU CANNOT JOIN ###.^8";\r
 415                 }\r
 416         } else {\r
 417                 versionmsg = "^2client version and server version are compatible.^8";\r
 418         }\r
 419         return versionmsg;\r
 420 }\r
 421 \r
 422 \r
 423 void PrintWelcomeMessage(entity pl)\r
 424 {\r
 425         string s, modifications, motd;\r
 426 \r
 427         if(self.cvar_scr_centertime == 0) return;\r
 428 \r
 429         if(cvar("g_campaign"))\r
 430         {\r
 431                 if(self.classname == "player" && !self.BUTTON_INFO)\r
 432                         return;\r
 433         }\r
 434         else\r
 435         {\r
 436                 if((time - self.jointime) > cvar("welcome_message_time") && !self.BUTTON_INFO)\r
 437                         return;\r
 438         }\r
 439 \r
 440         if( !(timeoutStatus >= 1) ) { //really print the WelcomeMessage to the player every frame when timeout-seconds are shown or the game is restarted, to make sure that the shown number is accurate\r
 441                 if(self.welcomemessage_time > time) return;\r
 442                 self.welcomemessage_time = time + max(0.5, self.cvar_scr_centertime * 0.6);\r
 443         }\r
 444 \r
 445         if(cvar("g_campaign"))\r
 446         {\r
 447                 centerprint(pl, campaign_message);\r
 448                 return;\r
 449         }\r
 450 \r
 451 //TODO GreEn`mArine: make the timeout-messages clientside as well (just like the ready restart countdown)!\r
 452         if(!self.BUTTON_INFO)\r
 453         {\r
 454                 // TODO get rid of this too\r
 455                 local string specString;\r
 456                 specString = NEWLINES;\r
 457                 //if(time < game_starttime) //also show the countdown when being a spectator\r
 458                 //      specString = strcat(specString, "\n\n^1Game starts in ", ftos(ceil(game_starttime - time)), " seconds^7");\r
 459                 //else\r
 460                 if (timeoutStatus != 0)\r
 461                         specString = strcat(specString, "\n\n", getTimeoutText(1));\r
 462                 else\r
 463                 {\r
 464                         if(self.classname == "player")\r
 465                                 return;\r
 466                         goto normal;\r
 467                 }\r
 468                 return centerprint_atprio(self, CENTERPRIO_SPAM, specString);\r
 469         }\r
 470 \r
 471 :normal\r
 472         modifications = "";\r
 473         if(cvar("sv_gravity") < 800)\r
 474                 modifications = strcat(modifications, ", Low gravity");\r
 475         if(g_cloaked)\r
 476                 modifications = strcat(modifications, ", Cloaked");\r
 477         if(g_midair)\r
 478                 modifications = strcat(modifications, ", Midair");\r
 479         if(g_vampire)\r
 480                 modifications = strcat(modifications, ", Vampire");\r
 481         if(g_weapon_stay)\r
 482                 modifications = strcat(modifications, ", Weapons stay");\r
 483         if(g_bloodloss > 0)\r
 484                 modifications = strcat(modifications, ", Bloodloss");\r
 485         if(g_jetpack)\r
 486                 modifications = strcat(modifications, ", Jet pack");\r
 487         if(!cvar("g_start_weapon_grabber"))\r
 488                 modifications = strcat(modifications, ", No start weapon");\r
 489         if(!cvar("g_vore_digestion"))\r
 490                 modifications = strcat(modifications, ", Gentle Vore");\r
 491         if(cvar("g_balance_vore_digestion_damage") >= 1000)\r
 492                 modifications = strcat(modifications, ", InstaDigestion");\r
 493         if(cvar("g_balance_vore_weight_gravity") < 0)\r
 494                 modifications = strcat(modifications, ", Lighten");\r
 495         modifications = substring(modifications, 2, strlen(modifications) - 2);\r
 496 \r
 497         local string versionmessage;\r
 498         versionmessage = GetClientVersionMessage();\r
 499 \r
 500         s = strcat(s, NEWLINES, "This is Voretournament ", cvar_string("g_voretournamentversion"), "\n", versionmessage);\r
 501         s = strcat(s, "^8\n\ngame type is ^1", gamemode_name, "^8\n");\r
 502 \r
 503         if(modifications != "")\r
 504                 s = strcat(s, "^8\nactive modifications: ^3", modifications, "^8\n");\r
 505 \r
 506         if(timeoutStatus != 0)\r
 507                 s = strcat(s, "\n\n", getTimeoutText(1));\r
 508 \r
 509         if(cache_lastmutatormsg != cvar_string("g_mutatormsg"))\r
 510         {\r
 511                 if(cache_lastmutatormsg)\r
 512                         strunzone(cache_lastmutatormsg);\r
 513                 if(cache_mutatormsg)\r
 514                         strunzone(cache_mutatormsg);\r
 515                 cache_lastmutatormsg = strzone(cvar_string("g_mutatormsg"));\r
 516                 cache_mutatormsg = strzone(cache_lastmutatormsg);\r
 517         }\r
 518 \r
 519         if (cache_mutatormsg != "") {\r
 520                 s = strcat(s, "\n\n^8special gameplay tips: ^7", cache_mutatormsg);\r
 521         }\r
 522 \r
 523         motd = cvar_string("sv_motd");\r
 524         if (motd != "") {\r
 525                 s = strcat(s, "\n\n^8MOTD: ^7", strreplace("\\n", "\n", motd));\r
 526         }\r
 527         s = strcat(s, "\n");\r
 528 \r
 529         centerprint(pl, s);\r
 530 }\r
 531 \r
 532 \r
 533 void SetPlayerColors(entity pl, float _color)\r
 534 {\r
 535         /*string s;\r
 536         s = ftos(cl);\r
 537         stuffcmd(pl, strcat("color ", s, " ", s, "\n")  );\r
 538         pl.team = cl + 1;\r
 539         //pl.clientcolors = pl.clientcolors - (pl.clientcolors & 15) + cl;\r
 540         pl.clientcolors = 16*cl + cl;*/\r
 541 \r
 542         float pants, shirt;\r
 543         pants = _color & 0x0F;\r
 544         shirt = _color & 0xF0;\r
 545 \r
 546 \r
 547         if(teams_matter) {\r
 548                 setcolor(pl, 16*pants + pants);\r
 549         } else {\r
 550                 setcolor(pl, shirt + pants);\r
 551         }\r
 552 }\r
 553 \r
 554 void SetPlayerTeam(entity pl, float t, float s, float noprint)\r
 555 {\r
 556         float _color;\r
 557 \r
 558         if(t == 4)\r
 559                 _color = COLOR_TEAM4 - 1;\r
 560         else if(t == 3)\r
 561                 _color = COLOR_TEAM3 - 1;\r
 562         else if(t == 2)\r
 563                 _color = COLOR_TEAM2 - 1;\r
 564         else\r
 565                 _color = COLOR_TEAM1 - 1;\r
 566 \r
 567         SetPlayerColors(pl,_color);\r
 568 \r
 569         if(t != s) {\r
 570                 LogTeamchange(pl.playerid, pl.team, 3);  // log manual team join\r
 571 \r
 572                 if(!noprint)\r
 573                 bprint(pl.netname, "^7 has changed from ", TeamNoName(s), " to ", TeamNoName(t), "\n");\r
 574         }\r
 575 \r
 576 }\r
 577 \r
 578 // set c1...c4 to show what teams are allowed\r
 579 void CheckAllowedTeams (entity for_whom)\r
 580 {\r
 581         float dm;\r
 582         entity head;\r
 583         string teament_name;\r
 584 \r
 585         c1 = c2 = c3 = c4 = -1;\r
 586         cb1 = cb2 = cb3 = cb4 = 0;\r
 587 \r
 588         if(g_onslaught)\r
 589         {\r
 590                 // onslaught is special\r
 591                 head = findchain(classname, "onslaught_generator");\r
 592                 while (head)\r
 593                 {\r
 594                         if (head.team == COLOR_TEAM1) c1 = 0;\r
 595                         if (head.team == COLOR_TEAM2) c2 = 0;\r
 596                         if (head.team == COLOR_TEAM3) c3 = 0;\r
 597                         if (head.team == COLOR_TEAM4) c4 = 0;\r
 598                         head = head.chain;\r
 599                 }\r
 600         }\r
 601         else if(g_domination)\r
 602                 teament_name = "dom_team";\r
 603         else if(g_ctf)\r
 604                 teament_name = "ctf_team";\r
 605         else if(g_tdm)\r
 606                 teament_name = "tdm_team";\r
 607         else if(g_assault)\r
 608                 c1 = c2 = 0; // Assault always has 2 teams\r
 609         else\r
 610         {\r
 611                 // cover anything else by treating it like tdm with no teams spawned\r
 612                 if(g_keyhunt)\r
 613                         dm = kh_teams;\r
 614                 else if(g_race)\r
 615                         dm = race_teams;\r
 616                 else\r
 617                         dm = 2;\r
 618 \r
 619                 if(dm >= 4)\r
 620                         c1 = c2 = c3 = c4 = 0;\r
 621                 else if(dm >= 3)\r
 622                         c1 = c2 = c3 = 0;\r
 623                 else\r
 624                         c1 = c2 = 0;\r
 625         }\r
 626 \r
 627         // find out what teams are allowed if necessary\r
 628         if(teament_name)\r
 629         {\r
 630                 head = find(world, classname, teament_name);\r
 631                 while(head)\r
 632                 {\r
 633                         if(!(g_domination && head.netname == ""))\r
 634                         {\r
 635                                 if(head.team == COLOR_TEAM1)\r
 636                                         c1 = 0;\r
 637                                 else if(head.team == COLOR_TEAM2)\r
 638                                         c2 = 0;\r
 639                                 else if(head.team == COLOR_TEAM3)\r
 640                                         c3 = 0;\r
 641                                 else if(head.team == COLOR_TEAM4)\r
 642                                         c4 = 0;\r
 643                         }\r
 644                         head = find(head, classname, teament_name);\r
 645                 }\r
 646         }\r
 647 \r
 648         // TODO: Balance quantity of bots across > 2 teams when bot_vs_human is set (and remove next line)\r
 649         if(c3==-1&&c4==-1)\r
 650         if(cvar("bot_vs_human") && for_whom)\r
 651         {\r
 652                 if(cvar("bot_vs_human") > 0)\r
 653                 {\r
 654                         // bots are all blue\r
 655                         if(clienttype(for_whom) == CLIENTTYPE_BOT)\r
 656                                 c1 = c3 = c4 = -1;\r
 657                         else\r
 658                                 c2 = -1;\r
 659                 }\r
 660                 else\r
 661                 {\r
 662                         // bots are all red\r
 663                         if(clienttype(for_whom) == CLIENTTYPE_BOT)\r
 664                                 c2 = c3 = c4 = -1;\r
 665                         else\r
 666                                 c1 = -1;\r
 667                 }\r
 668         }\r
 669 }\r
 670 \r
 671 float PlayerValue(entity p)\r
 672 {\r
 673         if(IsTeamBalanceForced() == 1)\r
 674                 return 1;\r
 675         return 1;\r
 676 }\r
 677 \r
 678 // c1...c4 should be set to -1 (not allowed) or 0 (allowed).\r
 679 // teams that are allowed will now have their player counts stored in c1...c4\r
 680 void GetTeamCounts(entity ignore)\r
 681 {\r
 682         entity head;\r
 683         float value, bvalue;\r
 684         // now count how many players are on each team already\r
 685 \r
 686         // FIXME: also find and memorize the lowest-scoring bot on each team (in case players must be shuffled around)\r
 687         // also remember the lowest-scoring player\r
 688 \r
 689         FOR_EACH_PLAYER(head)\r
 690         {\r
 691                 if(head != ignore)// && head.netname != "")\r
 692                 {\r
 693                         value = PlayerValue(head);\r
 694                         if(clienttype(head) == CLIENTTYPE_BOT)\r
 695                                 bvalue = value;\r
 696                         else\r
 697                                 bvalue = 0;\r
 698                         if(head.team == COLOR_TEAM1)\r
 699                         {\r
 700                                 if(c1 >= 0)\r
 701                                 {\r
 702                                         c1 = c1 + value;\r
 703                                         cb1 = cb1 + bvalue;\r
 704                                 }\r
 705                         }\r
 706                         if(head.team == COLOR_TEAM2)\r
 707                         {\r
 708                                 if(c2 >= 0)\r
 709                                 {\r
 710                                         c2 = c2 + value;\r
 711                                         cb2 = cb2 + bvalue;\r
 712                                 }\r
 713                         }\r
 714                         if(head.team == COLOR_TEAM3)\r
 715                         {\r
 716                                 if(c3 >= 0)\r
 717                                 {\r
 718                                         c3 = c3 + value;\r
 719                                         cb3 = cb3 + bvalue;\r
 720                                 }\r
 721                         }\r
 722                         if(head.team == COLOR_TEAM4)\r
 723                         {\r
 724                                 if(c4 >= 0)\r
 725                                 {\r
 726                                         c4 = c4 + value;\r
 727                                         cb4 = cb4 + bvalue;\r
 728                                 }\r
 729                         }\r
 730                 }\r
 731         }\r
 732 }\r
 733 \r
 734 // returns # of smallest team (1, 2, 3, 4)\r
 735 // NOTE: Assumes CheckAllowedTeams has already been called!\r
 736 float FindSmallestTeam(entity pl, float ignore_pl)\r
 737 {\r
 738         float totalteams, balance_type, maxc;\r
 739         totalteams = 0;\r
 740 \r
 741         // find out what teams are available\r
 742         //CheckAllowedTeams();\r
 743 \r
 744         // make sure there are at least 2 teams to join\r
 745         if(c1 >= 0)\r
 746                 totalteams = totalteams + 1;\r
 747         if(c2 >= 0)\r
 748                 totalteams = totalteams + 1;\r
 749         if(c3 >= 0)\r
 750                 totalteams = totalteams + 1;\r
 751         if(c4 >= 0)\r
 752                 totalteams = totalteams + 1;\r
 753 \r
 754         if(cvar("bot_vs_human"))\r
 755                 totalteams += 1;\r
 756 \r
 757         if(totalteams <= 1)\r
 758         {\r
 759                 if(g_domination)\r
 760                         error("Too few teams available for domination\n");\r
 761                 else if(g_ctf)\r
 762                         error("Too few teams available for ctf\n");\r
 763                 else if(g_keyhunt)\r
 764                         error("Too few teams available for key hunt\n");\r
 765                 else\r
 766                         error("Too few teams available for team deathmatch\n");\r
 767         }\r
 768 \r
 769         // count how many players are in each team\r
 770         if(ignore_pl)\r
 771                 GetTeamCounts(pl);\r
 772         else\r
 773                 GetTeamCounts(world);\r
 774 \r
 775         // c1...c4 now have counts of each team\r
 776         // figure out which is smallest, giving priority to the team the player is already on as a tie-breaker\r
 777 \r
 778         // 2 gives priority to what team you're already on, 1 goes in order\r
 779         // 2 doesn't seem to work though...\r
 780         balance_type = 1;\r
 781 \r
 782         if(bots_would_leave)\r
 783         //if(pl.classname != "player")\r
 784         if(clienttype(pl) != CLIENTTYPE_BOT)\r
 785         {\r
 786                 c1 -= cb1 * 255.0/256.0;\r
 787                 c2 -= cb2 * 255.0/256.0;\r
 788                 c3 -= cb3 * 255.0/256.0;\r
 789                 c4 -= cb4 * 255.0/256.0;\r
 790         }\r
 791         maxc = max4(c1, c2, c3, c4);\r
 792 \r
 793         RandomSelection_Init();\r
 794         if(balance_type == 1)\r
 795         {\r
 796                 // 1: use team count, then score (note: can only use 8 significant bits of score)\r
 797                 if(c1 >= 0) RandomSelection_Add(world, 1, string_null, 1, (maxc - c1) + float2range01(-team1_score) / 256.0);\r
 798                 if(c2 >= 0) RandomSelection_Add(world, 2, string_null, 1, (maxc - c2) + float2range01(-team2_score) / 256.0);\r
 799                 if(c3 >= 0) RandomSelection_Add(world, 3, string_null, 1, (maxc - c3) + float2range01(-team3_score) / 256.0);\r
 800                 if(c4 >= 0) RandomSelection_Add(world, 4, string_null, 1, (maxc - c4) + float2range01(-team4_score) / 256.0);\r
 801         }\r
 802         else if(balance_type == 2)\r
 803         {\r
 804                 // 1: use team count, if equal prefer own team\r
 805                 if(c1 >= 0) RandomSelection_Add(world, 1, string_null, 1, (maxc - c1) + (self.team == COLOR_TEAM1) / 512.0);\r
 806                 if(c2 >= 0) RandomSelection_Add(world, 1, string_null, 1, (maxc - c1) + (self.team == COLOR_TEAM2) / 512.0);\r
 807                 if(c3 >= 0) RandomSelection_Add(world, 1, string_null, 1, (maxc - c1) + (self.team == COLOR_TEAM3) / 512.0);\r
 808                 if(c4 >= 0) RandomSelection_Add(world, 1, string_null, 1, (maxc - c1) + (self.team == COLOR_TEAM4) / 512.0);\r
 809         }\r
 810         else if(balance_type == 3)\r
 811         {\r
 812                 // 1: use team count, then score, if equal prefer own team (probably fails due to float accuracy problems)\r
 813                 if(c1 >= 0) RandomSelection_Add(world, 1, string_null, 1, (maxc - c1) + float2range01(-team1_score + 0.5 * (self.team == COLOR_TEAM1)) / 256.0);\r
 814                 if(c2 >= 0) RandomSelection_Add(world, 2, string_null, 1, (maxc - c2) + float2range01(-team2_score + 0.5 * (self.team == COLOR_TEAM2)) / 256.0);\r
 815                 if(c3 >= 0) RandomSelection_Add(world, 3, string_null, 1, (maxc - c3) + float2range01(-team3_score + 0.5 * (self.team == COLOR_TEAM3)) / 256.0);\r
 816                 if(c4 >= 0) RandomSelection_Add(world, 4, string_null, 1, (maxc - c4) + float2range01(-team4_score + 0.5 * (self.team == COLOR_TEAM4)) / 256.0);\r
 817         }\r
 818         return RandomSelection_chosen_float;\r
 819 }\r
 820 \r
 821 float JoinBestTeam(entity pl, float only_return_best, float forcebestteam)\r
 822 {\r
 823         float smallest, selectedteam;\r
 824 \r
 825         // don't join a team if we're not playing a team game\r
 826         if(!teams_matter)\r
 827                 return 0;\r
 828 \r
 829         // find out what teams are available\r
 830         CheckAllowedTeams(pl);\r
 831 \r
 832         // if we want the player in a certain team for campaign, force him there\r
 833         if(cvar("g_campaign"))\r
 834         if(clienttype(pl) == CLIENTTYPE_REAL) // only players, not bots\r
 835         {\r
 836                 switch(cvar("g_campaign_forceteam"))\r
 837                 {\r
 838                         case 1:\r
 839                                 SetPlayerColors(pl, COLOR_TEAM1 - 1);\r
 840                                 LogTeamchange(pl.playerid, pl.team, 2);\r
 841                                 return COLOR_TEAM1;\r
 842                         case 2:\r
 843                                 SetPlayerColors(pl, COLOR_TEAM2 - 1);\r
 844                                 LogTeamchange(pl.playerid, pl.team, 2);\r
 845                                 return COLOR_TEAM2;\r
 846                         case 3:\r
 847                                 SetPlayerColors(pl, COLOR_TEAM3 - 1);\r
 848                                 LogTeamchange(pl.playerid, pl.team, 2);\r
 849                                 return COLOR_TEAM3;\r
 850                         case 4:\r
 851                                 SetPlayerColors(pl, COLOR_TEAM4 - 1);\r
 852                                 LogTeamchange(pl.playerid, pl.team, 2);\r
 853                                 return COLOR_TEAM4;\r
 854                         default:\r
 855                                 break;\r
 856                 }\r
 857         }\r
 858 \r
 859         // if we don't care what team he ends up on, put him on whatever team he entered as.\r
 860         // if he's not on a valid team, then let other code put him on the smallest team\r
 861         if(!forcebestteam)\r
 862         {\r
 863                 if(     c1 >= 0 && pl.team == COLOR_TEAM1)\r
 864                         selectedteam = pl.team;\r
 865                 else if(c2 >= 0 && pl.team == COLOR_TEAM2)\r
 866                         selectedteam = pl.team;\r
 867                 else if(c3 >= 0 && pl.team == COLOR_TEAM3)\r
 868                         selectedteam = pl.team;\r
 869                 else if(c4 >= 0 && pl.team == COLOR_TEAM4)\r
 870                         selectedteam = pl.team;\r
 871                 else\r
 872                         selectedteam = -1;\r
 873 \r
 874                 if(selectedteam > 0)\r
 875                 {\r
 876                         if(!only_return_best)\r
 877                         {\r
 878                                 SetPlayerColors(pl, selectedteam - 1);\r
 879 \r
 880                                 // when JoinBestTeam is called by client.qc/ClientKill_Now_TeamChange the players team is -1 and thus skipped\r
 881                                 // when JoinBestTeam is called by cl_client.qc/ClientConnect the player_id is 0 the log attempt is rejected\r
 882                                 LogTeamchange(pl.playerid, pl.team, 99);\r
 883                         }\r
 884                         return selectedteam;\r
 885                 }\r
 886                 // otherwise end up on the smallest team (handled below)\r
 887         }\r
 888 \r
 889         smallest = FindSmallestTeam(pl, TRUE);\r
 890 \r
 891         if(!only_return_best && !pl.bot_forced_team)\r
 892         {\r
 893                 TeamchangeFrags(self);\r
 894                 if(smallest == 1)\r
 895                 {\r
 896                         SetPlayerColors(pl, COLOR_TEAM1 - 1);\r
 897                 }\r
 898                 else if(smallest == 2)\r
 899                 {\r
 900                         SetPlayerColors(pl, COLOR_TEAM2 - 1);\r
 901                 }\r
 902                 else if(smallest == 3)\r
 903                 {\r
 904                         SetPlayerColors(pl, COLOR_TEAM3 - 1);\r
 905                 }\r
 906                 else if(smallest == 4)\r
 907                 {\r
 908                         SetPlayerColors(pl, COLOR_TEAM4 - 1);\r
 909                 }\r
 910                 else\r
 911                 {\r
 912                         error("smallest team: invalid team\n");\r
 913                 }\r
 914 \r
 915                 LogTeamchange(pl.playerid, pl.team, 2); // log auto join\r
 916 \r
 917                 if(pl.deadflag == DEAD_NO)\r
 918                         Damage(pl, pl, pl, 100000, DEATH_TEAMCHANGE, pl.origin, '0 0 0');\r
 919         }\r
 920 \r
 921         return smallest;\r
 922 }\r
 923 \r
 924 //void() ctf_playerchanged;\r
 925 void SV_ChangeTeam(float _color)\r
 926 {\r
 927         float scolor, dcolor, steam, dteam, dbotcount, scount, dcount;\r
 928 \r
 929         // in normal deathmatch we can just apply the color and we're done\r
 930         if(!teams_matter) {\r
 931                 SetPlayerColors(self, _color);\r
 932                 return;\r
 933         }\r
 934 \r
 935         scolor = self.clientcolors & 0x0F;\r
 936         dcolor = _color & 0x0F;\r
 937 \r
 938         if(scolor == COLOR_TEAM1 - 1)\r
 939                 steam = 1;\r
 940         else if(scolor == COLOR_TEAM2 - 1)\r
 941                 steam = 2;\r
 942         else if(scolor == COLOR_TEAM3 - 1)\r
 943                 steam = 3;\r
 944         else // if(scolor == COLOR_TEAM4 - 1)\r
 945                 steam = 4;\r
 946         if(dcolor == COLOR_TEAM1 - 1)\r
 947                 dteam = 1;\r
 948         else if(dcolor == COLOR_TEAM2 - 1)\r
 949                 dteam = 2;\r
 950         else if(dcolor == COLOR_TEAM3 - 1)\r
 951                 dteam = 3;\r
 952         else // if(dcolor == COLOR_TEAM4 - 1)\r
 953                 dteam = 4;\r
 954 \r
 955         CheckAllowedTeams(self);\r
 956 \r
 957         if(dteam == 1 && c1 < 0) dteam = 4;\r
 958         if(dteam == 4 && c4 < 0) dteam = 3;\r
 959         if(dteam == 3 && c3 < 0) dteam = 2;\r
 960         if(dteam == 2 && c2 < 0) dteam = 1;\r
 961 \r
 962         // not changing teams\r
 963         if(scolor == dcolor)\r
 964         {\r
 965                 //bprint("same team change\n");\r
 966                 SetPlayerTeam(self, dteam, steam, TRUE);\r
 967                 return;\r
 968         }\r
 969 \r
 970         if((cvar("g_campaign")) || (cvar("g_changeteam_banned") && self.wasplayer)) {\r
 971                 sprint(self, "Team changes not allowed\n");\r
 972                 return; // changing teams is not allowed\r
 973         }\r
 974 \r
 975         if(cvar("g_balance_teams_prevent_imbalance"))\r
 976         {\r
 977                 // only allow changing to a smaller or equal size team\r
 978 \r
 979                 // find out what teams are available\r
 980                 //CheckAllowedTeams();\r
 981                 // count how many players on each team\r
 982                 GetTeamCounts(world);\r
 983 \r
 984                 // get desired team\r
 985                 if(dteam == 1 && c1 >= 0)//dcolor == COLOR_TEAM1 - 1)\r
 986                 {\r
 987                         dcount = c1;\r
 988                         dbotcount = cb1;\r
 989                 }\r
 990                 else if(dteam == 2 && c2 >= 0)//dcolor == COLOR_TEAM2 - 1)\r
 991                 {\r
 992                         dcount = c2;\r
 993                         dbotcount = cb2;\r
 994                 }\r
 995                 else if(dteam == 3 && c3 >= 0)//dcolor == COLOR_TEAM3 - 1)\r
 996                 {\r
 997                         dcount = c3;\r
 998                         dbotcount = cb3;\r
 999                 }\r
1000                 else if(dteam == 4 && c4 >= 0)//dcolor == COLOR_TEAM4 - 1)\r
1001                 {\r
1002                         dcount = c4;\r
1003                         dbotcount = cb4;\r
1004                 }\r
1005                 else\r
1006                 {\r
1007                         sprint(self, "Cannot change to an invalid team\n");\r
1008 \r
1009                         return;\r
1010                 }\r
1011 \r
1012                 // get starting team\r
1013                 if(steam == 1)//scolor == COLOR_TEAM1 - 1)\r
1014                         scount = c1;\r
1015                 else if(steam == 2)//scolor == COLOR_TEAM2 - 1)\r
1016                         scount = c2;\r
1017                 else if(steam == 3)//scolor == COLOR_TEAM3 - 1)\r
1018                         scount = c3;\r
1019                 else if(steam == 4)//scolor == COLOR_TEAM4 - 1)\r
1020                         scount = c4;\r
1021 \r
1022                 if(scount) // started at a valid, nonempty team\r
1023                 {\r
1024                         // check if we're trying to change to a larger team that doens't have bots to swap with\r
1025                         if(dcount >= scount && dbotcount <= 0)\r
1026                         {\r
1027                                 sprint(self, "Cannot change to a larger team\n");\r
1028                                 return; // can't change to a larger team\r
1029                         }\r
1030                 }\r
1031         }\r
1032 \r
1033 //      bprint("allow change teams from ", ftos(steam), " to ", ftos(dteam), "\n");\r
1034 \r
1035         if(self.classname == "player" && steam != dteam)\r
1036         {\r
1037                 // reduce frags during a team change\r
1038                 TeamchangeFrags(self);\r
1039         }\r
1040 \r
1041         SetPlayerTeam(self, dteam, steam, FALSE);\r
1042 \r
1043         if(self.classname == "player" && steam != dteam)\r
1044         {\r
1045                 // kill player when changing teams\r
1046                 if(self.deadflag == DEAD_NO)\r
1047                         Damage(self, self, self, 100000, DEATH_TEAMCHANGE, self.origin, '0 0 0');\r
1048         }\r
1049         //ctf_playerchanged();\r
1050 }\r
1051 \r
1052 void ShufflePlayerOutOfTeam (float source_team)\r
1053 {\r
1054         float smallestteam, smallestteam_count, steam;\r
1055         float lowest_bot_score, lowest_player_score;\r
1056         entity head, lowest_bot, lowest_player, selected;\r
1057 \r
1058         smallestteam = 0;\r
1059         smallestteam_count = 999999999;\r
1060 \r
1061         if(c1 >= 0 && c1 < smallestteam_count)\r
1062         {\r
1063                 smallestteam = 1;\r
1064                 smallestteam_count = c1;\r
1065         }\r
1066         if(c2 >= 0 && c2 < smallestteam_count)\r
1067         {\r
1068                 smallestteam = 2;\r
1069                 smallestteam_count = c2;\r
1070         }\r
1071         if(c3 >= 0 && c3 < smallestteam_count)\r
1072         {\r
1073                 smallestteam = 3;\r
1074                 smallestteam_count = c3;\r
1075         }\r
1076         if(c4 >= 0 && c4 < smallestteam_count)\r
1077         {\r
1078                 smallestteam = 4;\r
1079                 smallestteam_count = c4;\r
1080         }\r
1081 \r
1082         if(!smallestteam)\r
1083         {\r
1084                 bprint("warning: no smallest team\n");\r
1085                 return;\r
1086         }\r
1087 \r
1088         if(source_team == 1)\r
1089                 steam = COLOR_TEAM1;\r
1090         else if(source_team == 2)\r
1091                 steam = COLOR_TEAM2;\r
1092         else if(source_team == 3)\r
1093                 steam = COLOR_TEAM3;\r
1094         else if(source_team == 4)\r
1095                 steam = COLOR_TEAM4;\r
1096 \r
1097         lowest_bot = world;\r
1098         lowest_bot_score = 999999999;\r
1099         lowest_player = world;\r
1100         lowest_player_score = 999999999;\r
1101 \r
1102         // find the lowest-scoring player & bot of that team\r
1103         FOR_EACH_PLAYER(head)\r
1104         {\r
1105                 if(head.team == steam)\r
1106                 {\r
1107                         if(head.isbot)\r
1108                         {\r
1109                                 if(head.totalfrags < lowest_bot_score)\r
1110                                 {\r
1111                                         lowest_bot = head;\r
1112                                         lowest_bot_score = head.totalfrags;\r
1113                                 }\r
1114                         }\r
1115                         else\r
1116                         {\r
1117                                 if(head.totalfrags < lowest_player_score)\r
1118                                 {\r
1119                                         lowest_player = head;\r
1120                                         lowest_player_score = head.totalfrags;\r
1121                                 }\r
1122                         }\r
1123                 }\r
1124         }\r
1125 \r
1126         // prefers to move a bot...\r
1127         if(lowest_bot != world)\r
1128                 selected = lowest_bot;\r
1129         // but it will move a player if it has to\r
1130         else\r
1131                 selected = lowest_player;\r
1132         // don't do anything if it couldn't find anyone\r
1133         if(!selected)\r
1134         {\r
1135                 bprint("warning: couldn't find a player to move from team\n");\r
1136                 return;\r
1137         }\r
1138 \r
1139         // smallest team gains a member\r
1140         if(smallestteam == 1)\r
1141         {\r
1142                 c1 = c1 + 1;\r
1143         }\r
1144         else if(smallestteam == 2)\r
1145         {\r
1146                 c2 = c2 + 1;\r
1147         }\r
1148         else if(smallestteam == 3)\r
1149         {\r
1150                 c3 = c3 + 1;\r
1151         }\r
1152         else if(smallestteam == 4)\r
1153         {\r
1154                 c4 = c4 + 1;\r
1155         }\r
1156         else\r
1157         {\r
1158                 bprint("warning: destination team invalid\n");\r
1159                 return;\r
1160         }\r
1161         // source team loses a member\r
1162         if(source_team == 1)\r
1163         {\r
1164                 c1 = c1 + 1;\r
1165         }\r
1166         else if(source_team == 2)\r
1167         {\r
1168                 c2 = c2 + 2;\r
1169         }\r
1170         else if(source_team == 3)\r
1171         {\r
1172                 c3 = c3 + 3;\r
1173         }\r
1174         else if(source_team == 4)\r
1175         {\r
1176                 c4 = c4 + 4;\r
1177         }\r
1178         else\r
1179         {\r
1180                 bprint("warning: source team invalid\n");\r
1181                 return;\r
1182         }\r
1183 \r
1184         // move the player to the new team\r
1185         TeamchangeFrags(selected);\r
1186         SetPlayerTeam(selected, smallestteam, source_team, FALSE);\r
1187 \r
1188         if(selected.deadflag == DEAD_NO)\r
1189                 Damage(selected, selected, selected, 100000, DEATH_AUTOTEAMCHANGE, selected.origin, '0 0 0');\r
1190         centerprint(selected, strcat("You have been moved into a different team to improve team balance\nYou are now on: ", ColoredTeamName(selected.team)));\r
1191 }\r
1192 \r
1193 void CauseRebalance(float source_team, float howmany_toomany)\r
1194 {\r
1195         if(IsTeamBalanceForced() == 1)\r
1196         {\r
1197                 bprint("Rebalancing Teams\n");\r
1198                 ShufflePlayerOutOfTeam(source_team);\r
1199         }\r
1200 }\r
1201 \r
1202 // part of g_balance_teams_force\r
1203 // occasionally perform an audit of the teams to make\r
1204 // sure they're more or less balanced in player count.\r
1205 void AuditTeams()\r
1206 {\r
1207         float numplayers, numteams, smallest, toomany;\r
1208         float balance;\r
1209         balance = IsTeamBalanceForced();\r
1210         if(balance == 0)\r
1211                 return;\r
1212 \r
1213         if(audit_teams_time > time)\r
1214                 return;\r
1215 \r
1216         audit_teams_time = time + 4 + random();\r
1217 \r
1218 //      bprint("Auditing teams\n");\r
1219 \r
1220         CheckAllowedTeams(world);\r
1221         GetTeamCounts(world);\r
1222 \r
1223 \r
1224         numteams = numplayers = smallest = 0;\r
1225         if(c1 >= 0)\r
1226         {\r
1227                 numteams = numteams + 1;\r
1228                 numplayers = numplayers + c1;\r
1229                 smallest = c1;\r
1230         }\r
1231         if(c2 >= 0)\r
1232         {\r
1233                 numteams = numteams + 1;\r
1234                 numplayers = numplayers + c2;\r
1235                 if(c2 < smallest)\r
1236                         smallest = c2;\r
1237         }\r
1238         if(c3 >= 0)\r
1239         {\r
1240                 numteams = numteams + 1;\r
1241                 numplayers = numplayers + c3;\r
1242                 if(c3 < smallest)\r
1243                         smallest = c3;\r
1244         }\r
1245         if(c4 >= 0)\r
1246         {\r
1247                 numteams = numteams + 1;\r
1248                 numplayers = numplayers + c4;\r
1249                 if(c4 < smallest)\r
1250                         smallest = c4;\r
1251         }\r
1252 \r
1253         if(numplayers <= 0)\r
1254                 return; // no players to move around\r
1255         if(numteams < 2)\r
1256                 return; // don't bother shuffling if for some reason there aren't any teams\r
1257 \r
1258         toomany = smallest + 1;\r
1259 \r
1260         if(c1 && c1 > toomany)\r
1261                 CauseRebalance(1, c1 - toomany);\r
1262         if(c2 && c2 > toomany)\r
1263                 CauseRebalance(2, c2 - toomany);\r
1264         if(c3 && c3 > toomany)\r
1265                 CauseRebalance(3, c3 - toomany);\r
1266         if(c4 && c4 > toomany)\r
1267                 CauseRebalance(4, c4 - toomany);\r
1268 \r
1269         // if teams are still unbalanced, balance them further in the next audit,\r
1270         // which will happen sooner (keep doing rapid audits until things are in order)\r
1271         audit_teams_time = time + 0.7 + random()*0.3;\r
1272 }\r
1273 \r
1274 // code from here on is just to support maps that don't have team entities\r
1275 void tdm_spawnteam (string teamname, float teamcolor)\r
1276 {\r
1277         local entity e;\r
1278         e = spawn();\r
1279         e.classname = "tdm_team";\r
1280         e.netname = teamname;\r
1281         e.cnt = teamcolor;\r
1282         e.team = e.cnt + 1;\r
1283 };\r
1284 \r
1285 // spawn some default teams if the map is not set up for tdm\r
1286 void tdm_spawnteams()\r
1287 {\r
1288         float numteams;\r
1289 \r
1290         numteams = cvar("g_tdm_teams_override");\r
1291         if(numteams < 2)\r
1292                 numteams = cvar("g_tdm_teams");\r
1293         numteams = bound(2, numteams, 4);\r
1294 \r
1295         tdm_spawnteam("Red", COLOR_TEAM1-1);\r
1296         tdm_spawnteam("Blue", COLOR_TEAM2-1);\r
1297         if(numteams >= 3)\r
1298                 tdm_spawnteam("Yellow", COLOR_TEAM3-1);\r
1299         if(numteams >= 4)\r
1300                 tdm_spawnteam("Pink", COLOR_TEAM4-1);\r
1301 };\r
1302 \r
1303 void tdm_delayedinit()\r
1304 {\r
1305         // if no teams are found, spawn defaults\r
1306         if (find(world, classname, "tdm_team") == world)\r
1307                 tdm_spawnteams();\r
1308 };\r
1309 \r
1310 void tdm_init()\r
1311 {\r
1312         InitializeEntity(world, tdm_delayedinit, INITPRIO_GAMETYPE);\r
1313 };\r