BonusPack.MutSquadAI

class MutSquadAI extends DMSquad;

// This only runs on server
function bool SetEnemy( Bot B, Pawn NewEnemy )
{
    local xMutantGame mutantGame;
    local pawn mutantPawn;
    local int i;

    mutantGame = xMutantGame(Level.Game);
    
    // If no-one is a mutant (or we are), or we are the bottom feeder, do as normal DM. 
    if( mutantGame.CurrentMutant == None || mutantGame.CurrentMutant == B || mutantGame.CurrentBottomFeeder == B )
        return Super.SetEnemy(B, NewEnemy);

    mutantPawn = mutantGame.CurrentMutant.Pawn;

    // ??? If there is a mutant, remove all non-mutant enemies.
    for(i=0; i<8; i++)
    {
        if(Enemies[i] != mutantPawn)
            RemoveEnemy(Enemies[i]);
    }

    // Return false if NewEnemy != mutant.
    if( NewEnemy != mutantGame.CurrentMutant.Pawn )
        return false;

    if ( !AddEnemy(NewEnemy) )
        return false;

    return FindNewEnemyFor(B,(B.Enemy !=None) && B.LineOfSightTo(SquadMembers.Enemy));
}

function bool FindNewEnemyFor(Bot B, bool bSeeEnemy)
{
    local xMutantGame mutantGame;
    local Pawn OldEnemy, MutantPawn;

    mutantGame = xMutantGame(Level.Game);

    // If no-one is a mutant (or we are), or we are the bottom feeder, do as normal DM. 
    if( mutantGame.CurrentMutant == None || mutantGame.CurrentMutant == B || mutantGame.CurrentBottomFeeder == B )
        return Super.FindNewEnemyFor(B, bSeeEnemy);

    OldEnemy = B.Enemy;
    
    MutantPawn = mutantGame.CurrentMutant.Pawn;

    if(MutantPawn != None)
    {
        if ( VSize(MutantPawn.Location - B.Pawn.Location) < 1500 )
            bSeeEnemy = B.LineOfSightTo(MutantPawn);
        else
            bSeeEnemy = B.CanSee(MutantPawn);   // only if looking at him
    }
    B.Enemy = MutantPawn;

    if(OldEnemy != MutantPawn)
    {
        B.EnemyChanged(bSeeEnemy);
        return true;
    }

    return false;
}

function bool AssignSquadResponsibility(Bot B)
{
    local xMutantGame mutantGame;
    local pawn mutantPawn;

    mutantGame = xMutantGame(Level.Game);

    // If there is a mutant (and I'm not it)
    if(mutantGame.CurrentMutant != None && mutantGame.CurrentMutant != B)
    {
        mutantPawn = mutantGame.CurrentMutant.Pawn;

        // If I'm not the bottom feeder but i'm hunting someone, it can only be the mutant.
        if(B.Enemy != None && mutantGame.CurrentBottomFeeder != B && B.Enemy != mutantPawn)
        {
            FindNewEnemyFor(B, false);
        }
    }

    return Super.AssignSquadResponsibility(B);
}

defaultproperties
{
}