data/qcsrc/server/teamplay.qc

   1 float COLOR_TEAM1       = 5;  // red
   2 float COLOR_TEAM2       = 14; // blue
   3 float COLOR_TEAM3       = 10; // pink
   4 float COLOR_TEAM4       = 13; // yellow
   5
   6
   7 float GAME_DEATHMATCH           = 1;
   8 float GAME_TEAM_DEATHMATCH      = 2;
   9 float GAME_DOMINATION           = 3;
  10 float GAME_CTF                  = 4;
  11 float GAME_RUNEMATCH            = 5;
  12 float GAME_LMS                  = 6;
  13 float GAME_ARENA                = 7;
  14
  15
  16 // client counts for each team
  17 float c1, c2, c3, c4;
  18 // # of bots on those teams
  19 float cb1, cb2, cb3, cb4;
  20
  21 float g_domination, g_ctf, g_tdm;
  22
  23 float audit_teams_time;
  24
  25 void TeamchangeFrags(entity e)
  26 {
  27         if(e.classname == "player")
  28         {
  29                 // reduce frags during a team change
  30                 DistributeFragsAmongTeam(e, (e.clientcolors & 0x0F) + 1, 1 - cvar("g_changeteam_fragtransfer") / 100);
  31         }
  32 }
  33
  34 string TeamName(float t)
  35 {
  36         // fixme: Search for team entities and get their .netname's!
  37         if(t == COLOR_TEAM1)
  38                 return "Red Team";
  39         if(t == COLOR_TEAM2)
  40                 return "Blue Team";
  41         if(t == COLOR_TEAM3)
  42                 return "Pink Team";
  43         if(t == COLOR_TEAM4)
  44                 return "Yellow Team";
  45         return "Neutral Team";
  46 }
  47 string ColoredTeamName(float t)
  48 {
  49         // fixme: Search for team entities and get their .netname's!
  50         if(t == COLOR_TEAM1)
  51                 return "^1Red Team^7";
  52         if(t == COLOR_TEAM2)
  53                 return "^4Blue Team^7";
  54         if(t == COLOR_TEAM3)
  55                 return "^6Pink Team^7";
  56         if(t == COLOR_TEAM4)
  57                 return "^3Yellow Team^7";
  58         return "Neutral Team";
  59 }
  60 string TeamNoName(float t)
  61 {
  62         // fixme: Search for team entities and get their .netname's!
  63         if(t == 1)
  64                 return "Red Team";
  65         if(t == 2)
  66                 return "Blue Team";
  67         if(t == 3)
  68                 return "Pink Team";
  69         if(t == 4)
  70                 return "Yellow Team";
  71         return "Neutral Team";
  72 }
  73
  74 void dom_init();
  75 void ctf_init();
  76 void runematch_init();
  77 void tdm_init();
  78
  79 void(entity pl) LogTeamchange =
  80 {
  81         string str;
  82         if(!cvar("sv_eventlog"))
  83                 return;
  84         if(pl.playerid < 1)
  85                 return;
  86         str = strcat(":team:", ftos(pl.playerid), ":");
  87         str = strcat(str, ftos(pl.team));
  88         GameLogEcho(str, FALSE);
  89 }
  90
  91 void ResetGameCvars()
  92 {
  93         cvar_set("g_tdm", "0");
  94         cvar_set("g_domination", "0");
  95         cvar_set("g_ctf", "0");
  96         cvar_set("g_runematch", "0");
  97         cvar_set("g_lms", "0");
  98         cvar_set("g_arena", "0");
  99         cvar_set("teamplay", "0");
 100
 101
 102         cvar_set("exit_cfg", "");
 103 }
 104
 105 void ActivateTeamplay()
 106 {
 107         float teamplay_default;
 108         teamplay_default = cvar("teamplay_default");
 109
 110         if(teamplay_default)
 111                 cvar_set("teamplay", ftos(teamplay_default));
 112         else
 113                 cvar_set("teamplay", "3");
 114 }
 115
 116 void InitGameplayMode()
 117 {
 118         float fraglimit_override, timelimit_override;
 119
 120         VoteReset();
 121
 122         game = cvar ("gamecfg");        // load game options
 123
 124         // game cvars get reset before map changes
 125         // then map's cfg sets them as desired
 126
 127         // FIXME: also set a message or game mode name to print to players when the join
 128
 129         // set both here, gamemode can override it later
 130         timelimit_override = cvar("timelimit_override");
 131         fraglimit_override = cvar("fraglimit_override");
 132
 133         if(game == GAME_DOMINATION || cvar("g_domination"))
 134         {
 135                 ResetGameCvars();
 136                 game = GAME_DOMINATION;
 137                 cvar_set("g_domination", "1");
 138
 139                 ActivateTeamplay();
 140
 141                 fraglimit_override = cvar("g_domination_point_limit");
 142
 143                 gamemode_name = "Domination";
 144                 teams_matter = 1;
 145         }
 146         else if(game == GAME_CTF || cvar("g_ctf"))
 147         {
 148                 ResetGameCvars();
 149                 game = GAME_CTF;
 150                 cvar_set("g_ctf", "1");
 151
 152                 ActivateTeamplay();
 153
 154                 fraglimit_override = cvar("g_ctf_capture_limit");
 155
 156                 gamemode_name = "Capture the Flag";
 157                 teams_matter = 1;
 158         }
 159         else if((game == GAME_RUNEMATCH || cvar("g_runematch")) && !cvar("g_minstagib"))
 160         {
 161                 game = GAME_RUNEMATCH;
 162                 cvar_set("g_runematch", "1");
 163
 164                 if(cvar("deathmatch_force_teamplay"))
 165                         ActivateTeamplay();
 166
 167                 fraglimit_override = cvar("g_runematch_point_limit");
 168
 169                 gamemode_name = "Rune Match";
 170                 if(cvar("teamplay"))
 171                         teams_matter = 1;
 172                 else
 173                         teams_matter = 0;
 174         }
 175         else if(game == GAME_DEATHMATCH || game == GAME_TEAM_DEATHMATCH || cvar("g_tdm"))
 176         {
 177                 if(!cvar("deathmatch"))
 178                         cvar_set("deathmatch", "1");
 179
 180
 181                 if(game == GAME_TEAM_DEATHMATCH || cvar("g_tdm") || cvar("deathmatch_force_teamplay"))
 182                 {
 183                         ResetGameCvars();
 184                         game = GAME_TEAM_DEATHMATCH;
 185                         gamemode_name = "Team Deathmatch";
 186                         ActivateTeamplay();
 187                         teams_matter = 1;
 188                         cvar_set("g_tdm", "1");
 189                 }
 190                 else
 191                 {
 192                         ResetGameCvars();
 193                         game = GAME_DEATHMATCH;
 194                         gamemode_name = "Deathmatch";
 195                         teams_matter = 0;
 196                 }
 197
 198                 fraglimit_override = cvar("fraglimit_override");
 199         }
 200         else if(game == GAME_LMS || cvar("g_lms"))
 201         {
 202                 ResetGameCvars();
 203                 game = GAME_LMS;
 204                 cvar_set("g_lms", "1");
 205                 fraglimit_override = cvar("g_lms_lives_override");
 206                 if(fraglimit_override == 0)
 207                         fraglimit_override = -1;
 208                 gamemode_name = "Last Man Standing";
 209                 teams_matter = 0;
 210                 lms_lowest_lives = 999;
 211         }
 212         else if(game == GAME_ARENA || cvar("g_arena"))
 213         {
 214                 ResetGameCvars();
 215                 game = GAME_ARENA;
 216                 cvar_set("g_arena", "1");
 217                 fraglimit_override = cvar("g_arena_point_limit");
 218                 maxspawned = cvar("g_arena_maxspawned");
 219                 if(maxspawned < 2)
 220                         maxspawned = 2;
 221                 arena_roundbased = cvar("g_arena_roundbased");
 222                 gamemode_name = "Arena";
 223                 teams_matter = 0;
 224         }
 225         else
 226         {
 227                 // we can only assume...
 228                 ResetGameCvars();
 229                 gamemode_name = "Deathmatch";
 230                 teams_matter = 0;
 231         }
 232 /*      else if(game == GAME_TEAM_DEATHMATCH)
 233         {
 234                 if(!cvar("deathmatch"))
 235                         cvar_set("deathmatch", "1");
 236
 237                 //if(!cvar("teamplay"))
 238                 //      cvar_set("teamplay", "3");
 239                 ActivateTeamplay();
 240
 241                 fraglimit_override = cvar("fraglimit_override");
 242         }*/
 243
 244         // enforce the server's universal frag/time limits
 245         if(!cvar("g_campaign"))
 246         {
 247                 if(fraglimit_override >= 0)
 248                         cvar_set("fraglimit", ftos(fraglimit_override));
 249                 if(timelimit_override >= 0)
 250                         cvar_set("timelimit", ftos(timelimit_override));
 251         }
 252
 253         if (game == GAME_DOMINATION)//cvar("g_domination"))
 254                 dom_init();
 255         else if (game == GAME_CTF)//cvar("g_ctf"))
 256                 ctf_init();
 257         else if (game == GAME_RUNEMATCH)//cvar("g_runematch"))
 258                 runematch_init();
 259         else if (game == GAME_TEAM_DEATHMATCH)//cvar("g_runematch"))
 260                 tdm_init();
 261
 262         // those mutators rule each other out
 263         if(cvar("g_minstagib"))
 264         {
 265                 cvar_set("g_instagib", "0");
 266                 cvar_set("g_rocketarena", "0");
 267         }
 268         if(cvar("g_instagib"))
 269         {
 270                 cvar_set("g_minstagib", "0");
 271                 cvar_set("g_rocketarena", "0");
 272         }
 273         if(cvar("g_rocketarena"))
 274         {
 275                 cvar_set("g_instagib", "0");
 276                 cvar_set("g_minstagib", "0");
 277         }
 278
 279         registercvar("_motd", "");
 280         registercvar("_mutatormsg", "");
 281         cvar_set("_motd", wordwrap(cvar_string("sv_motd"), 50));
 282         cvar_set("_mutatormsg", wordwrap(cvar_string("g_mutatormsg"), 50));
 283
 284         g_domination = cvar("g_domination");
 285         g_ctf = cvar("g_ctf");
 286         g_tdm = cvar("g_tdm");
 287 }
 288
 289 string GetClientVersionMessage(float v) {
 290         local string versionmsg;
 291         if (v == 1) {
 292                 versionmsg = "^1client is too old to get versioninfo.\n\n\n### YOU WON'T BE ABLE TO PLAY ON THIS SERVER ###\n\n\nUPDATE!!! (http://www.nexuiz.com)^8";
 293                 // either that or someone wants to be funny
 294         } else if (v != cvar("gameversion")) {
 295                 if(v < cvar("gameversion")) {
 296                         versionmsg = "^3Your client version is outdated.\n\n\n### YOU WON'T BE ABLE TO PLAY ON THIS SERVER ###\n\n\nPlease update!!!^8";
 297                 } else {
 298                         versionmsg = "^3This server is using an outdated Nexuiz version.\n\n\n ### THIS SERVER IS INCOMPATIBLE AND THUS YOU CANNOT JOIN ###.^8";
 299                 }
 300         } else {
 301                 versionmsg = "^2client version and server version are compatible.^8";
 302         }
 303         return strzone(versionmsg);
 304
 305 }
 306
 307
 308 void PrintWelcomeMessage(entity pl)
 309 {
 310         string s, mutator, modifications;
 311
 312         /*if(self.welcomemessage_time < time)
 313                 return;
 314         if(self.welcomemessage_time2 > time)
 315                 return;
 316         self.welcomemessage_time2 = time + 0.8; */
 317
 318         if(self.classname == "observer")
 319         {
 320                 if(cvar("g_lms") && self.frags <= 0 && self.frags > -666)
 321                         return centermsg_setall(strcat(newlines, "^1You have no more lives left\nwait for next round\n\n\n^7press attack to spectate other players"));
 322                 else if(cvar("g_lms") && self.frags == -666)
 323                         return centermsg_setall(strcat(newlines, "^1Match has already begun\nwait for next round\n\n\n^7press attack to spectate other players"));
 324         }
 325         else if(self.classname == "spectator")
 326         {
 327                 if ((cvar("g_lms") && self.frags < 1) || cvar("g_arena"))
 328                         return centermsg_setall(strcat(newlines, "spectating ", self.enemy.netname, "\n\n\n^7press attack for next player\npress attack2 for free fly mode"));
 329                 else
 330                         return centermsg_setall(strcat(newlines, "spectating ", self.enemy.netname, "\n\n\n^7press jump to play\n^7press attack for next player\npress attack2 for free fly mode"));
 331         }
 332
 333
 334         if(self.welcomemessage_time2 > time) return;
 335         self.welcomemessage_time2 = time + 1.0;
 336
 337         if(cvar("g_campaign"))
 338         {
 339                 centermsg_setallfor(pl, campaign_message);
 340                 return;
 341         }
 342
 343         if(cvar("g_minstagib"))
 344                 mutator = "^2Minstagib ^1";
 345         else if(cvar("g_instagib"))
 346                 mutator = "^2Instagib ^1";
 347         else if(cvar("g_rocketarena"))
 348                 mutator = "^2Rocketarena ^1";
 349         else if(cvar("g_nixnex"))
 350                 mutator = "^2No Items Nexuiz ^1";
 351
 352         if(cvar("g_midair")) {
 353                 // to protect against unheedingly made changes
 354                 if (modifications) {
 355                         modifications = strcat(modifications, ", ");
 356                 }
 357                 modifications = "midair";
 358         }
 359         if(cvar("g_vampire")) {
 360                 if (modifications) {
 361                         modifications = strcat(modifications, ", ");
 362                 }
 363                 modifications = strcat(modifications, "vampire");
 364         }
 365         if(cvar("g_laserguided_missile")) {
 366                 if (modifications) {
 367                         modifications = strcat(modifications, ", ");
 368                 }
 369                 modifications = strcat(modifications, "laser-guided-missiles");
 370         }
 371
 372         local string versionmessage;
 373         versionmessage = GetClientVersionMessage(self.version);
 374
 375         s = strcat(s, newlines, "This is Nexuiz ", cvar_string("g_nexuizversion"), "\n", versionmessage);
 376         s = strcat(s, "^8\n\nmatch type is ^1", mutator, gamemode_name, "^8\n");
 377
 378         if(modifications != "")
 379                 s = strcat(s, "^8\nactive modifications: ^3", modifications, "^8\n");
 380
 381         if((self.classname == "observer" || self.classname == "spectator") && self.version == cvar("gameversion")) {
 382                 if(!cvar("g_arena"))
 383                         s = strcat(s,"^7\n\n\npress jump to play\npress attack to spectate other players\n\n");
 384                 else if(player_count < 2 && arena_roundbased)
 385                 {
 386                         s = strcat(s, "\n\n\n^1waiting for second player to start match^7\n\n");
 387                 }
 388                 else
 389                 {
 390                         s = strcat(s, "\n\n\n");
 391                         if(champion)
 392                                 s = strcat(s, "^7current champion is: ", champion.netname, "\n\n");
 393                         s = strcat(s,"^7press attack to spectate other players\n\n");
 394                 }
 395         }
 396
 397
 398         s = strzone(s);
 399
 400         if (cvar("g_grappling_hook"))
 401                 s = strcat(s, "\n\n^8grappling hook is enabled, press 'e' to use it\n");
 402
 403         if (cvar_string("_mutatormsg") != "") {
 404                 s = strcat(s, "\n\n^8special gameplay tips: ^7", cvar_string("_mutatormsg"));
 405         }
 406
 407         if (cvar_string("_motd") != "") {
 408                 s = strcat(s, "\n\n^8MOTD: ^7", cvar_string("_motd"));
 409         }
 410
 411         s = strcat(s, "\n");
 412         if(cvar("fraglimit"))
 413                 s = strcat(s, "\n^8frag limit: ^7", cvar_string("fraglimit"));
 414         if(cvar("timelimit"))
 415                 s = strcat(s, "\n^8time limit: ^7", cvar_string("timelimit"), " minutes");
 416
 417         s = strzone(s);
 418
 419         centermsg_setallfor(pl, s);
 420         //sprint(pl, s);
 421
 422         strunzone(s);
 423 }
 424
 425
 426 void SetPlayerColors(entity pl, float _color)
 427 {
 428         /*string s;
 429         s = ftos(cl);
 430         stuffcmd(pl, strcat("color ", s, " ", s, "\n")  );
 431         pl.team = cl + 1;
 432         //pl.clientcolors = pl.clientcolors - (pl.clientcolors & 15) + cl;
 433         pl.clientcolors = 16*cl + cl;*/
 434
 435         float pants, shirt;
 436         pants = _color & 0x0F;
 437         shirt = _color & 0xF0;
 438
 439
 440         if(teamplay) {
 441                 setcolor(pl, 16*pants + pants);
 442         } else {
 443                 setcolor(pl, shirt + pants);
 444         }
 445 }
 446
 447 void SetPlayerTeam(entity pl, float t, float s, float noprint)
 448 {
 449         float _color;
 450
 451         if(t == 4)
 452                 _color = COLOR_TEAM4 - 1;
 453         else if(t == 3)
 454                 _color = COLOR_TEAM3 - 1;
 455         else if(t == 2)
 456                 _color = COLOR_TEAM2 - 1;
 457         else
 458                 _color = COLOR_TEAM1 - 1;
 459
 460         SetPlayerColors(pl,_color);
 461
 462         if(!noprint && t != s)
 463         {
 464                 //bprint(strcat(pl.netname, " has changed to ", TeamNoName(t), "\n"));
 465                 bprint(strcat(pl.netname, "^7 has changed from ", TeamNoName(s), " to ", TeamNoName(t), "\n"));
 466         }
 467
 468         if(t != s)
 469                 LogTeamchange(pl);
 470 }
 471
 472
 473
 474
 475
 476
 477 // set c1...c4 to show what teams are allowed
 478 void CheckAllowedTeams ()
 479 {
 480         string teament_name;
 481         float dm;
 482         entity head;
 483
 484 //      if(!dom && !ctf)
 485 //              dm = 1;
 486
 487         c1 = c2 = c3 = c4 = -1;
 488         cb1 = cb2 = cb3 = cb4 = 0;
 489
 490         if(g_domination)
 491                 teament_name = "dom_team";
 492         else if(g_ctf)
 493                 teament_name = "ctf_team";
 494         else if(g_tdm)
 495                 teament_name = "tdm_team";
 496         else
 497         {
 498                 // cover anything else by treating it like tdm with no teams spawned
 499                 dm = cvar("g_tdm_teams");
 500                 if(dm < 2)
 501                         error("g_tdm_teams < 2, not enough teams to play team deathmatch\n");
 502
 503                 if(dm >= 4)
 504                 {
 505                         c1 = c2 = c3 = c4 = 0;
 506                 }
 507                 else if(dm >= 3)
 508                 {
 509                         c1 = c2 = c3 = 0;
 510                 }
 511                 else// if(dm >= 2)
 512                 {
 513                         c1 = c2 = 0;
 514                 }
 515                 return;
 516         }
 517
 518         // first find out what teams are allowed
 519         head = find(world, classname, teament_name);
 520         while(head)
 521         {
 522                 if(!(g_domination && head.netname == ""))
 523                 {
 524                         if(head.team == COLOR_TEAM1)
 525                         {
 526                                 c1 = 0;
 527                         }
 528                         if(head.team == COLOR_TEAM2)
 529                         {
 530                                 c2 = 0;
 531                         }
 532                         if(head.team == COLOR_TEAM3)
 533                         {
 534                                 c3 = 0;
 535                         }
 536                         if(head.team == COLOR_TEAM4)
 537                         {
 538                                 c4 = 0;
 539                         }
 540                 }
 541                 head = find(head, classname, teament_name);
 542         }
 543 }
 544
 545 // c1...c4 should be set to -1 (not allowed) or 0 (allowed).
 546 // teams that are allowed will now have their player counts stored in c1...c4
 547 void GetTeamCounts(entity ignore)
 548 {
 549         entity head;
 550         // now count how many players are on each team already
 551
 552         // FIXME: also find and memorize the lowest-scoring bot on each team (in case players must be shuffled around)
 553         // also remember the lowest-scoring player
 554
 555         head = find(world, classname, "player");
 556         while(head)
 557         {
 558                 if(head != ignore)// && head.netname != "")
 559                 {
 560                         if(head.team == COLOR_TEAM1)
 561                         {
 562                                 if(c1 >= 0)
 563                                 {
 564                                         c1 = c1 + 1;
 565                                         cb1 = cb1 + 1;
 566                                 }
 567                         }
 568                         if(head.team == COLOR_TEAM2)
 569                         {
 570                                 if(c2 >= 0)
 571                                 {
 572                                         c2 = c2 + 1;
 573                                         cb2 = cb2 + 1;
 574                                 }
 575                         }
 576                         if(head.team == COLOR_TEAM3)
 577                         {
 578                                 if(c3 >= 0)
 579                                 {
 580                                         c3 = c3 + 1;
 581                                         cb3 = cb3 + 1;
 582                                 }
 583                         }
 584                         if(head.team == COLOR_TEAM4)
 585                         {
 586                                 if(c4 >= 0)
 587                                 {
 588                                         c4 = c4 + 1;
 589                                         cb4 = cb4 + 1;
 590                                 }
 591                         }
 592                 }
 593                 head = find(head, classname, "player");
 594         }
 595 }
 596
 597 // returns # of smallest team (1, 2, 3, 4)
 598 // NOTE: Assumes CheckAllowedTeams has already been called!
 599 float FindSmallestTeam(entity pl, float ignore_pl)
 600 {
 601         float totalteams, smallestteam, smallestteam_count, balance_type;
 602         totalteams = 0;
 603
 604         // find out what teams are available
 605         //CheckAllowedTeams();
 606
 607         // make sure there are at least 2 teams to join
 608         if(c1 >= 0)
 609                 totalteams = totalteams + 1;
 610         if(c2 >= 0)
 611                 totalteams = totalteams + 1;
 612         if(c3 >= 0)
 613                 totalteams = totalteams + 1;
 614         if(c4 >= 0)
 615                 totalteams = totalteams + 1;
 616
 617         if(totalteams <= 1)
 618         {
 619                 if(g_domination)
 620                         error("Too few teams available for domination\n");
 621                 else if(g_ctf)
 622                         error("Too few teams available for ctf\n");
 623                 else
 624                         error("Too few teams available for team deathmatch\n");
 625         }
 626
 627
 628         // count how many players are in each team
 629         if(ignore_pl)
 630                 GetTeamCounts(pl);
 631         else
 632                 GetTeamCounts(world);
 633
 634         // c1...c4 now have counts of each team
 635         // figure out which is smallest, giving priority to the team the player is already on as a tie-breaker
 636
 637         smallestteam = 0;
 638         smallestteam_count = 999;
 639
 640         // 2 gives priority to what team you're already on, 1 goes in order
 641         // 2 doesn't seem to work though...
 642         balance_type = 1;
 643
 644         if(balance_type == 1)
 645         {
 646                 if(c1 >= 0 && c1 < smallestteam_count)
 647                 {
 648                         smallestteam = 1;
 649                         smallestteam_count = c1;
 650                 }
 651                 if(c2 >= 0 && c2 < smallestteam_count)
 652                 {
 653                         smallestteam = 2;
 654                         smallestteam_count = c2;
 655                 }
 656                 if(c3 >= 0 && c3 < smallestteam_count)
 657                 {
 658                         smallestteam = 3;
 659                         smallestteam_count = c3;
 660                 }
 661                 if(c4 >= 0 && c4 < smallestteam_count)
 662                 {
 663                         smallestteam = 4;
 664                         smallestteam_count = c4;
 665                 }
 666         }
 667         else
 668         {
 669                 if(c1 >= 0 && (c1 < smallestteam_count ||
 670                                         (c1 == smallestteam_count && self.team == COLOR_TEAM1) ) )
 671                 {
 672                         smallestteam = 1;
 673                         smallestteam_count = c1;
 674                 }
 675                 if(c2 >= 0 && c2 < (c2 < smallestteam_count ||
 676                                         (c2 == smallestteam_count && self.team == COLOR_TEAM2) ) )
 677                 {
 678                         smallestteam = 2;
 679                         smallestteam_count = c2;
 680                 }
 681                 if(c3 >= 0 && c3 < (c3 < smallestteam_count ||
 682                                         (c3 == smallestteam_count && self.team == COLOR_TEAM3) ) )
 683                 {
 684                         smallestteam = 3;
 685                         smallestteam_count = c3;
 686                 }
 687                 if(c4 >= 0 && c4 < (c4 < smallestteam_count ||
 688                                         (c4 == smallestteam_count && self.team == COLOR_TEAM4) ) )
 689                 {
 690                         smallestteam = 4;
 691                         smallestteam_count = c4;
 692                 }
 693         }
 694
 695         return smallestteam;
 696 }
 697
 698 float JoinBestTeam(entity pl, float only_return_best)
 699 {
 700         float smallest, selectedteam;
 701
 702         // don't join a team if we're not playing a team game
 703         if(!cvar("teamplay") && !g_domination && !g_ctf)
 704                 return 0;
 705
 706         // find out what teams are available
 707         CheckAllowedTeams();
 708
 709         if(cvar("g_domination"))
 710         {
 711                 if(cvar("g_domination_default_teams") < 3)
 712                         c3 = 9999;
 713                 if(cvar("g_domination_default_teams") < 4)
 714                         c4 = 9999;
 715         }
 716
 717         // if we don't care what team he ends up on, put him on whatever team he entered as.
 718         // if he's not on a valid team, then let other code put him on the smallest team
 719         if(!cvar("g_campaign") && !cvar("g_balance_teams") && !cvar("g_balance_teams_force"))
 720         {
 721                 if(     c1 >= 0 && pl.team == COLOR_TEAM1)
 722                         selectedteam = pl.team;
 723                 else if(c2 >= 0 && pl.team == COLOR_TEAM2)
 724                         selectedteam = pl.team;
 725                 else if(c3 >= 0 && pl.team == COLOR_TEAM3)
 726                         selectedteam = pl.team;
 727                 else if(c4 >= 0 && pl.team == COLOR_TEAM4)
 728                         selectedteam = pl.team;
 729                 else
 730                         selectedteam = -1;
 731                 if(selectedteam > 0)
 732                 {
 733                         if(!only_return_best)
 734                         {
 735                                 SetPlayerColors(pl, selectedteam - 1);
 736                                 LogTeamchange(pl);
 737                         }
 738                         return selectedteam;
 739                 }
 740                 // otherwise end up on the smallest team (handled below)
 741         }
 742
 743         smallest = FindSmallestTeam(pl, TRUE);
 744
 745
 746         if(!only_return_best)
 747         {
 748                 TeamchangeFrags(self);
 749                 if(smallest == 1)
 750                 {
 751                         SetPlayerColors(pl, COLOR_TEAM1 - 1);
 752                 }
 753                 else if(smallest == 2)
 754                 {
 755                         SetPlayerColors(pl, COLOR_TEAM2 - 1);
 756                 }
 757                 else if(smallest == 3)
 758                 {
 759                         SetPlayerColors(pl, COLOR_TEAM3 - 1);
 760                 }
 761                 else if(smallest == 4)
 762                 {
 763                         SetPlayerColors(pl, COLOR_TEAM4 - 1);
 764                 }
 765                 else
 766                 {
 767                         error("smallest team: invalid team\n");
 768                 }
 769                 LogTeamchange(pl);
 770                 if(pl.deadflag == DEAD_NO)
 771                         Damage(pl, pl, pl, 100000, DEATH_TEAMCHANGE, pl.origin, '0 0 0');
 772         }
 773
 774         return smallest;
 775 }
 776
 777
 778 void SV_ChangeTeam(float _color)
 779 {
 780         float scolor, dcolor, steam, dteam, dbotcount, scount, dcount;
 781
 782         // in normal deathmatch we can just apply the color and we're done
 783         if(!cvar("teamplay")) {
 784                 SetPlayerColors(self, _color);
 785                 return;
 786         }
 787
 788         scolor = self.clientcolors & 0x0F;
 789         dcolor = _color & 0x0F;
 790
 791         if(scolor == COLOR_TEAM1 - 1)
 792                 steam = 1;
 793         else if(scolor == COLOR_TEAM2 - 1)
 794                 steam = 2;
 795         else if(scolor == COLOR_TEAM3 - 1)
 796                 steam = 3;
 797         else if(scolor == COLOR_TEAM4 - 1)
 798                 steam = 4;
 799         if(dcolor == COLOR_TEAM1 - 1)
 800                 dteam = 1;
 801         else if(dcolor == COLOR_TEAM2 - 1)
 802                 dteam = 2;
 803         else if(dcolor == COLOR_TEAM3 - 1)
 804                 dteam = 3;
 805         else if(dcolor == COLOR_TEAM4 - 1)
 806                 dteam = 4;
 807
 808         // remap invalid teams in dom & ctf
 809         /*
 810         if(cvar("g_ctf") && dteam == 3)
 811                 dteam = 2;
 812         else if(cvar("g_ctf") && dteam == 4)
 813                 dteam = 1;
 814         else if((cvar("g_domination") && cvar("g_domination_default_teams") < 3) || (cvar("g_tdm") && cvar("g_tdm_teams") < 3))
 815         {
 816                 if(dteam == 3)
 817                         dteam = 2;
 818                 else if(dteam == 4)
 819                         dteam = 1;
 820         }
 821         else if((cvar("g_domination") && cvar("g_domination_default_teams") < 4) || (cvar("g_tdm") && cvar("g_tdm_teams") < 4))
 822         {
 823                 if(dteam == 4)
 824                         dteam = 1;
 825         }
 826         */
 827
 828         CheckAllowedTeams();
 829         if(dteam == 3 && c3 < 0)
 830                 dteam = 2;
 831         if(dteam == 4 && c4 < 0)
 832                 dteam = 1;
 833
 834         // not changing teams
 835         if(scolor == dcolor)
 836         {
 837                 //bprint("same team change\n");
 838                 SetPlayerTeam(self, dteam, steam, TRUE);
 839                 return;
 840         }
 841
 842         if(cvar("teamplay"))
 843         {
 844                 if(cvar("g_campaign") || cvar("g_changeteam_banned"))
 845                 {
 846                         sprint(self, "Team changes not allowed\n");
 847                         return; // changing teams is not allowed
 848                 }
 849
 850                 if(!cvar("g_campaign") && cvar("g_balance_teams_prevent_imbalance"))
 851                 {
 852                         // only allow changing to a smaller or equal size team
 853
 854                         // find out what teams are available
 855                         CheckAllowedTeams();
 856                         // count how many players on each team
 857                         GetTeamCounts(world);
 858
 859                         // get desired team
 860                         if(dteam == 1 && c1 >= 0)//dcolor == COLOR_TEAM1 - 1)
 861                         {
 862                                 dcount = c1;
 863                                 dbotcount = cb1;
 864                         }
 865                         else if(dteam == 2 && c2 >= 0)//dcolor == COLOR_TEAM2 - 1)
 866                         {
 867                                 dcount = c2;
 868                                 dbotcount = cb2;
 869                         }
 870                         else if(dteam == 3 && c3 >= 0)//dcolor == COLOR_TEAM3 - 1)
 871                         {
 872                                 dcount = c3;
 873                                 dbotcount = cb3;
 874                         }
 875                         else if(dteam == 4 && c4 >= 0)//dcolor == COLOR_TEAM4 - 1)
 876                         {
 877                                 dcount = c4;
 878                                 dbotcount = cb4;
 879                         }
 880                         else
 881                         {
 882                                 sprint(self, "Cannot change to an invalid team\n");
 883
 884                                 return;
 885                         }
 886
 887                         // get starting team
 888                         if(steam == 1)//scolor == COLOR_TEAM1 - 1)
 889                                 scount = c1;
 890                         else if(steam == 2)//scolor == COLOR_TEAM2 - 1)
 891                                 scount = c2;
 892                         else if(steam == 3)//scolor == COLOR_TEAM3 - 1)
 893                                 scount = c3;
 894                         else if(steam == 4)//scolor == COLOR_TEAM4 - 1)
 895                                 scount = c4;
 896
 897                         if(scount) // started at a valid, nonempty team
 898                         {
 899                                 // check if we're trying to change to a larger team that doens't have bots to swap with
 900                                 if(dcount >= scount && dbotcount <= 0)
 901                                 {
 902                                         sprint(self, "Cannot change to a larger team\n");
 903                                         return; // can't change to a larger team
 904                                 }
 905                         }
 906                 }
 907         }
 908
 909 //      bprint(strcat("allow change teams from ", ftos(steam), " to ", ftos(dteam), "\n"));
 910
 911         if(cvar("teamplay") && self.classname == "player" && steam != dteam)
 912         {
 913                 // reduce frags during a team change
 914                 TeamchangeFrags(self);
 915         }
 916
 917         SetPlayerTeam(self, dteam, steam, FALSE);
 918
 919         if(cvar("teamplay") && self.classname == "player" && steam != dteam)
 920         {
 921                 // kill player when changing teams
 922                 if(self.deadflag == DEAD_NO)
 923                         Damage(self, self, self, 100000, DEATH_TEAMCHANGE, self.origin, '0 0 0');
 924         }
 925 }
 926
 927 void ShufflePlayerOutOfTeam (float source_team)
 928 {
 929         float smallestteam, smallestteam_count, steam;
 930         float lowest_bot_score, lowest_player_score;
 931         entity head, lowest_bot, lowest_player, selected;
 932         string m;
 933
 934         smallestteam = 0;
 935         smallestteam_count = 999;
 936
 937         if(c1 >= 0 && c1 < smallestteam_count)
 938         {
 939                 smallestteam = 1;
 940                 smallestteam_count = c1;
 941         }
 942         if(c2 >= 0 && c2 < smallestteam_count)
 943         {
 944                 smallestteam = 2;
 945                 smallestteam_count = c2;
 946         }
 947         if(c3 >= 0 && c3 < smallestteam_count)
 948         {
 949                 smallestteam = 3;
 950                 smallestteam_count = c3;
 951         }
 952         if(c4 >= 0 && c4 < smallestteam_count)
 953         {
 954                 smallestteam = 4;
 955                 smallestteam_count = c4;
 956         }
 957
 958         if(!smallestteam)
 959         {
 960                 bprint("warning: no smallest team\n");
 961                 return;
 962         }
 963
 964         if(source_team == 1)
 965                 steam = COLOR_TEAM1;
 966         else if(source_team == 2)
 967                 steam = COLOR_TEAM2;
 968         else if(source_team == 3)
 969                 steam = COLOR_TEAM3;
 970         else if(source_team == 4)
 971                 steam = COLOR_TEAM4;
 972
 973         lowest_bot = world;
 974         lowest_bot_score = 9999;
 975         lowest_player = world;
 976         lowest_player_score = 9999;
 977
 978         // find the lowest-scoring player & bot of that team
 979         head = find(world, classname, "player");
 980         while(head)
 981         {
 982                 if(head.team == steam)
 983                 {
 984                         if(head.isbot)
 985                         {
 986                                 if(head.frags < lowest_bot_score)
 987                                 {
 988                                         lowest_bot = head;
 989                                         lowest_bot_score = head.frags;
 990                                 }
 991                         }
 992                         else
 993                         {
 994                                 if(head.frags < lowest_player_score)
 995                                 {
 996                                         lowest_player = head;
 997                                         lowest_player_score = head.frags;
 998                                 }
 999                         }
1000                 }
1001                 head = find(head, classname, "player");
1002         }
1003
1004         // prefers to move a bot...
1005         if(lowest_bot != world)
1006                 selected = lowest_bot;
1007         // but it will move a player if it has to
1008         else
1009                 selected = lowest_player;
1010         // don't do anything if it couldn't find anyone
1011         if(!selected)
1012         {
1013                 bprint("warning: couldn't find a player to move from team\n");
1014                 return;
1015         }
1016
1017         // smallest team gains a member
1018         if(smallestteam == 1)
1019         {
1020                 c1 = c1 + 1;
1021         }
1022         else if(smallestteam == 2)
1023         {
1024                 c2 = c2 + 1;
1025         }
1026         else if(smallestteam == 3)
1027         {
1028                 c3 = c3 + 1;
1029         }
1030         else if(smallestteam == 4)
1031         {
1032                 c4 = c4 + 1;
1033         }
1034         else
1035         {
1036                 bprint("warning: destination team invalid\n");
1037                 return;
1038         }
1039         // source team loses a member
1040         if(source_team == 1)
1041         {
1042                 c1 = c1 + 1;
1043         }
1044         else if(source_team == 2)
1045         {
1046                 c2 = c2 + 2;
1047         }
1048         else if(source_team == 3)
1049         {
1050                 c3 = c3 + 3;
1051         }
1052         else if(source_team == 4)
1053         {
1054                 c4 = c4 + 4;
1055         }
1056         else
1057         {
1058                 bprint("warning: source team invalid\n");
1059                 return;
1060         }
1061
1062         // move the player to the new team
1063         SetPlayerTeam(selected, smallestteam, source_team, FALSE);
1064
1065         if(selected.deadflag == DEAD_NO)
1066                         Damage(selected, selected, selected, 100000, DEATH_AUTOTEAMCHANGE, selected.origin, '0 0 0');
1067         /*
1068         m = "You have been moved into a different team to improve team balance\nYou are now on: ";
1069         if (selected.team == 5)
1070                 m = strcat(m, "^1Red Team");
1071         else if (selected.team == 14)
1072                 m = strcat(m, "^4Blue Team");
1073         else if (selected.team == 10)
1074                 m = strcat(m, "^6Pink Team");
1075         else if (selected.team == 13)
1076                 m = strcat(m, "^3Yellow Team");
1077         centerprint(selected, m);
1078         */
1079 }
1080
1081 // part of g_balance_teams_force
1082 // occasionally perform an audit of the teams to make
1083 // sure they're more or less balanced in player count.
1084 void AuditTeams()
1085 {
1086         float numplayers, numteams, average;
1087         if(cvar("g_campaign"))
1088                 return;
1089         if(!cvar("g_balance_teams_force"))
1090                 return;
1091         if(!cvar("teamplay"))
1092                 return;
1093
1094         if(audit_teams_time > time)
1095                 return;
1096
1097         audit_teams_time = time + 4 + random();
1098
1099 //      bprint("Auditing teams\n");
1100
1101         CheckAllowedTeams();
1102         GetTeamCounts(world);
1103
1104
1105         numteams = numplayers = 0;
1106         if(c1 >= 0)
1107         {
1108                 numteams = numteams + 1;
1109                 numplayers = numplayers + c1;
1110         }
1111         if(c2 >= 0)
1112         {
1113                 numteams = numteams + 1;
1114                 numplayers = numplayers + c2;
1115         }
1116         if(c3 >= 0)
1117         {
1118                 numteams = numteams + 1;
1119                 numplayers = numplayers + c3;
1120         }
1121         if(c4 >= 0)
1122         {
1123                 numteams = numteams + 1;
1124                 numplayers = numplayers + c4;
1125         }
1126
1127         if(numplayers <= 0)
1128                 return; // no players to move around
1129         if(numteams < 2)
1130                 return; // don't bother shuffling if for some reason there aren't any teams
1131
1132         average = ceil(numplayers / numteams);
1133
1134         if(average <= 0)
1135                 return; // that's weird...
1136
1137         if(c1 && c1 > average)
1138         {
1139                 bprint("Rebalancing Teams\n");
1140                 //bprint("Shuffle from team 1\n");
1141                 ShufflePlayerOutOfTeam(1);
1142         }
1143         if(c2 && c2 > average)
1144         {
1145                 bprint("Rebalancing Teams\n");
1146                 //bprint("Shuffle from team 2\n");
1147                 ShufflePlayerOutOfTeam(2);
1148         }
1149         if(c3 && c3 > average)
1150         {
1151                 bprint("Rebalancing Teams\n");
1152                 //bprint("Shuffle from team 3\n");
1153                 ShufflePlayerOutOfTeam(3);
1154         }
1155         if(c4 && c4 > average)
1156         {
1157                 bprint("Rebalancing Teams\n");
1158                 //bprint("Shuffle from team 4\n");
1159                 ShufflePlayerOutOfTeam(4);
1160         }
1161
1162         // if teams are still unbalanced, balance them further in the next audit,
1163         // which will happen sooner (keep doing rapid audits until things are in order)
1164         audit_teams_time = time + 0.7 + random()*0.3;
1165 }
1166
1167
1168
1169 /*void(entity e, float first) UpdateTeamScore =
1170 {
1171         clientno = e.FIXME;
1172         if(first)
1173         {
1174                 WriteByte (MSG_ALL, SVC_UPDATENAME);
1175                 WriteByte (MSG_ALL, clientno);
1176                 WriteString (MSG_ALL, e.netname);
1177
1178                 WriteByte (MSG_ALL, SVC_UPDATECOLORS);
1179                 WriteByte (MSG_ALL, clientno);
1180                 WriteByte (MSG_ALL, e.b_shirt * 16 + who.b_pants);
1181         }
1182
1183         WriteByte (MSG_ALL, SVC_UPDATEFRAGS);
1184         WriteByte (MSG_ALL, clientno);
1185         WriteShort (MSG_ALL, e.frags + 10000);
1186 };
1187
1188 */
1189
1190
1191 // code from here on is just to support maps that don't have team entities
1192 void tdm_spawnteam (string teamname, float teamcolor)
1193 {
1194         local entity e;
1195         e = spawn();
1196         e.classname = "tdm_team";
1197         e.netname = teamname;
1198         e.cnt = teamcolor;
1199         e.team = e.cnt + 1;
1200 };
1201
1202 // spawn some default teams if the map is not set up for tdm
1203 void() tdm_spawnteams =
1204 {
1205         float numteams;
1206
1207         numteams = cvar("g_tdm_teams");
1208
1209         tdm_spawnteam("Red", 4);
1210         tdm_spawnteam("Blue", 13);
1211 };
1212
1213 void() tdm_delayedinit =
1214 {
1215         self.think = SUB_Remove;
1216         self.nextthink = time;
1217         // if no teams are found, spawn defaults
1218         if (find(world, classname, "tdm_team") == world)
1219                 tdm_spawnteams();
1220 };
1221
1222 void() tdm_init =
1223 {
1224         local entity e;
1225         e = spawn();
1226         e.think = tdm_delayedinit;
1227         e.nextthink = time + 0.1;
1228 };
1229
1230