// Team Autobalance Script
//
// Determines which team needs players and proceeds to find candidates
// Candidates are based on low scores then switches the candidate
//
// Enable or Disable Autobalance
// $Host::EnableAutobalance = 1;
//


// Run from TeamBalanceNotify.cs via UnbalancedSound( %game )
function Autobalance( %game, %AutobalanceSafetynetTrys )
{	
	//Debug: Uncomment to enable
	//%AutobalanceDebug = true;
	
	//Reset
	%lastclient1 = "";
	%lastclient2 = "";
		
	//Team Count code by Keen
	$PlayerCount[0] = 0;
	$PlayerCount[1] = 0;
	$PlayerCount[2] = 0;
			

	for(%i = 0; %i < ClientGroup.getCount(); %i++)
	{
		%client = ClientGroup.getObject(%i);
			
		//Pick a client for autobalance
		//Try to pick a low scoring player
		%team = %client.team;
		if(%client.score < %lastclient[%team].score || %lastclient[%team] $= "") 
			%teamcanidate[%team] = %client; 

		%lastclient[%team] = %client;
			
		//if(!%client.isAIControlled())
			$PlayerCount[%client.team]++;
	}
	
	//Difference Variables
	%team1difference = $PlayerCount[1] - $PlayerCount[2];
	%team2difference = $PlayerCount[2] - $PlayerCount[1];
	
	//If even, stop.
	if( %team1difference == 1 || %team2difference == 1 || $PlayerCount[1] == $PlayerCount[2] )
	{
		$StatsMsgPlayed = 0;
		return;
	}
	//Determine bigTeam
	else if( %team1difference >= 2 )
		%bigTeam = 1;
	else if( %team2difference >= 2 )
		%bigTeam = 2;

	//Debug
	if( %AutobalanceDebug )
		AutobalanceDebug(%teamcanidate1, %teamcanidate2, %team1difference, %team2difference, %bigTeam, %AutobalanceSafetynetTrys);
				
	%client = %teamcanidate[%bigTeam];
	%team = %teamcanidate[%bigTeam].team;
	%otherTeam = ( %team == 1 ) ? 2 : 1;
			
	if( %teamcanidate[%bigTeam].team $= %bigTeam )
	{
		// Fire Autobalance
		Game.clientChangeTeam( %client, %otherTeam, 0 );
		messageAll('MsgTeamBalanceNotify', '~wfx/powered/vehicle_screen_on.wav');
	}
	else
		messageAll('MsgTeamBalanceNotify', '\c0Autobalance error: Team %1 mismatch.', %bigTeam );
			
	//Trigger GetCounts
	ResetClientChangedTeams();
	//Reset Unbalanced
	$UnbalancedMsgPlayed = 0;
	return;
}

function AutobalanceDebug(%teamcanidate1, %teamcanidate2, %team1difference, %team2difference, %bigTeam, %AutobalanceSafetynetTrys)
{
	if( %teamcanidate[%bigTeam] $= "" )
	{
		%AutobalanceSafetynetTrys++; 
		if(%AutobalanceSafetynetTrys $= 3) 
			return;
			
		if( %teamcanidate1 $= "" && %teamcanidate2 $= "" )
			%error = "Both Teams";
		else if( %teamcanidate[%bigTeam] $= "" )
			%error = "Team " @ %bigTeam;
				
		if( %error !$= "" )
			messageAll('MsgTeamBalanceNotify', '\c0Autobalance error: %1', %error );
				
		//Trigger GetCounts
		ResetClientChangedTeams();
		//Rerun in 10 secs
		schedule(10000, 0, "Autobalance", %game, %AutobalanceSafetynetTrys );
	}
		
	if( %teamcanidate1 $= "" ) 
		%teamcanidate1 = "NULL"; 
	if( %teamcanidate2 $= "" ) 
		%teamcanidate2 = "NULL";
		
	messageAll('MsgTeamBalanceNotify', '\c0Autobalance stat: %1, %2, %3, %4', %teamcanidate1, %team1difference, %teamcanidate2, %team2difference );
	
	return;
}