cp_auto_mission.cpp

Go to the documentation of this file.

/*
Copyright (C) 2002-2022 UFO: Alien Invasion.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
*/

#include "../../cl_shared.h"
#include "../../cl_inventory.h"
#include "cp_auto_mission.h"
#include "cp_campaign.h"
#include "cp_character.h"
#include "cp_geoscape.h"
#include "cp_missions.h"
#include "cp_mission_triggers.h"
#include "../../../shared/mathlib_extra.h"
#include "math.h"
#include "cp_mission_callbacks.h"

typedef enum autoMission_teamType_s {
    AUTOMISSION_TEAM_TYPE_PLAYER,               
    AUTOMISSION_TEAM_TYPE_ALIEN,                
    AUTOMISSION_TEAM_TYPE_CIVILIAN,             
    AUTOMISSION_TEAM_TYPE_MAX
} autoMissionTeamType_t;

#define MAX_SOLDIERS_AUTOMISSION MAX_TEAMS * AUTOMISSION_TEAM_TYPE_MAX

typedef struct autoUnit_s {
    int idx;
    autoMissionTeamType_t team;                 
    character_t* chr;                           
    double attackStrength;                      
    double defendStrength;                      
} autoUnit_t;

typedef struct autoMissionBattle_s {
    bool isHostile[AUTOMISSION_TEAM_TYPE_MAX][AUTOMISSION_TEAM_TYPE_MAX];       
    short nUnits[AUTOMISSION_TEAM_TYPE_MAX];                            
    short actUnits[AUTOMISSION_TEAM_TYPE_MAX];                          
    double scoreTeamEquipment[AUTOMISSION_TEAM_TYPE_MAX];           
    double scoreTeamSkill[AUTOMISSION_TEAM_TYPE_MAX];               
    double scoreTeamDifficulty[AUTOMISSION_TEAM_TYPE_MAX];      
    autoUnit_t units[AUTOMISSION_TEAM_TYPE_MAX][MAX_SOLDIERS_AUTOMISSION];      
    int teamAccomplishment[AUTOMISSION_TEAM_TYPE_MAX];                          
    int winningTeam;                                
    missionResults_t* results;                      
} autoMissionBattle_t;

#define SKILL_AWARD_SCALE 0.3f
#define ABILITY_AWARD_SCALE 0.06f

#define AM_IsPlayer(type) ((type) == AUTOMISSION_TEAM_TYPE_PLAYER)
#define AM_IsAlien(type)  ((type) == AUTOMISSION_TEAM_TYPE_ALIEN)
#define AM_IsCivilian(type) ((type) == AUTOMISSION_TEAM_TYPE_CIVILIAN)
#define AM_SetHostile(battle, team, otherTeam, value) (battle)->isHostile[(team)][(otherTeam)] = (value)
#define AM_IsHostile(battle, team, otherTeam) ((battle)->isHostile[(team)][(otherTeam)])

#define AM_GetUnit(battle, teamIdx, unitIdx) (&battle->units[teamIdx][unitIdx])
#define AM_IsUnitActive(unit) (((unit)->chr->HP > 0) && ((unit)->chr->HP > (unit)->chr->STUN))

static void AM_ClearBattle (autoMissionBattle_t* battle)
{
    assert(battle != nullptr);

    OBJZERO(*battle);

    for (int team = 0; team < AUTOMISSION_TEAM_TYPE_MAX; team++) {
        battle->scoreTeamDifficulty[team] = 0.5;
        battle->scoreTeamEquipment[team] = 0.5;
        battle->scoreTeamSkill[team] = 0.5;

        for (int otherTeam = 0; otherTeam < AUTOMISSION_TEAM_TYPE_MAX; otherTeam++) {
            /* If you forget to set this and run a battle, everyone will just kill each other by default */
            battle->isHostile[team][otherTeam] = true;
        }
    }

    battle->winningTeam = -1;
    battle->results = nullptr;
}

static void AM_FillTeamFromAircraft (autoMissionBattle_t* battle, const autoMissionTeamType_t teamNum, const aircraft_t* aircraft, const campaign_t* campaign)
{
    int teamSize;
    int unitsAlive;

    assert(teamNum < AUTOMISSION_TEAM_TYPE_MAX);
    assert(battle != nullptr);
    assert(aircraft != nullptr);

    teamSize = 0;
    unitsAlive = 0;
    LIST_Foreach(aircraft->acTeam, Employee, employee) {
        autoUnit_t* unit = AM_GetUnit(battle, teamNum, teamSize);

        unit->chr = &employee->chr;
        unit->team = teamNum;
        unit->idx = teamSize;

        teamSize++;
        if (employee->chr.HP > 0)
            unitsAlive++;

        if (teamSize >= MAX_SOLDIERS_AUTOMISSION)
            break;
    }
    battle->nUnits[teamNum] = teamSize;
    battle->actUnits[teamNum] = unitsAlive;

    if (teamSize == 0) {
        cgi->Com_DPrintf(DEBUG_CLIENT, "Warning: Attempt to add soldiers to an auto-mission from an aircraft with no soldiers onboard.\n");
        cgi->Com_DPrintf(DEBUG_CLIENT, "--- Note: Aliens might win this mission by default because they are un-challenged, with no resistance!\n");
    }
    if (unitsAlive == 0) {
        cgi->Com_DPrintf(DEBUG_CLIENT, "Warning: Attempt to add team to auto battle where all the units on the team are DEAD!\n");
        cgi->Com_DPrintf(DEBUG_CLIENT, "--- Note: This team will LOSE the battle by default.\n");
    }

    /* NOTE:  For now these are hard-coded to values based upon general campaign difficulty.
     * --- In the future, it might be easier to set this according to a scripted value in a .ufo
     * --- file, with other campaign info.  Reminder:  Higher floating point values mean better
     * --- soldiers, and therefore make an easier fight for the player. */
    switch (campaign->difficulty) {
    case 4:
        battle->scoreTeamDifficulty[teamNum] = 0.30;
        break;
    case 3:
        battle->scoreTeamDifficulty[teamNum] = 0.35;
        break;
    case 2:
        battle->scoreTeamDifficulty[teamNum] = 0.40;
        break;
    case 1:
        battle->scoreTeamDifficulty[teamNum] = 0.45;
        break;
    case 0:
        battle->scoreTeamDifficulty[teamNum] = 0.50;
        break;
    case -1:
        battle->scoreTeamDifficulty[teamNum] = 0.55;
        break;
    case -2:
        battle->scoreTeamDifficulty[teamNum] = 0.60;
        break;
    case -3:
        battle->scoreTeamDifficulty[teamNum] = 0.65;
        break;
    case -4:
        battle->scoreTeamDifficulty[teamNum] = 0.70;
        break;
    default:
        battle->scoreTeamDifficulty[teamNum] = 0.50;
    }
}

static void AM_CreateUnitChr (autoUnit_t* unit, const teamDef_t* teamDef, const equipDef_t* ed)
{
    unit->chr = Mem_PoolAllocType(character_t, cp_campaignPool);
    cgi->CL_GenerateCharacter(unit->chr, teamDef->id);

    cgi->INV_EquipActor(unit->chr, ed, unit->chr->teamDef->onlyWeapon, cgi->GAME_GetChrMaxLoad(unit->chr));
}

static void AM_DestroyUnitChr (autoUnit_t* unit)
{
    cgi->INV_DestroyInventory(&unit->chr->inv);
    cgi->Free(unit->chr);
}

static void AM_FillTeamFromBattleParams (autoMissionBattle_t* battle, const battleParam_t* missionParams)
{
    assert(battle);
    assert(missionParams);

    /* Aliens */
    battle->nUnits[AUTOMISSION_TEAM_TYPE_ALIEN] = missionParams->aliens;
    battle->actUnits[AUTOMISSION_TEAM_TYPE_ALIEN] = missionParams->aliens;
    if (missionParams->aliens > 0) {
        const equipDef_t* ed = cgi->INV_GetEquipmentDefinitionByID(missionParams->alienEquipment);
        const alienTeamGroup_t* alienTeamGroup = missionParams->alienTeamGroup;
        for (int unitIDX = 0; unitIDX < missionParams->aliens; unitIDX++) {
            const teamDef_t* teamDef = alienTeamGroup->alienTeams[rand() % alienTeamGroup->numAlienTeams];
            autoUnit_t* unit = AM_GetUnit(battle, AUTOMISSION_TEAM_TYPE_ALIEN, unitIDX);

            AM_CreateUnitChr(unit, teamDef, ed);
            unit->team = AUTOMISSION_TEAM_TYPE_ALIEN;
            unit->idx = unitIDX;
        }
        battle->scoreTeamSkill[AUTOMISSION_TEAM_TYPE_ALIEN] = (frand() * 0.6f) + 0.2f;
    }

    /* Civilians (if any) */
    battle->nUnits[AUTOMISSION_TEAM_TYPE_CIVILIAN] = missionParams->civilians;
    battle->actUnits[AUTOMISSION_TEAM_TYPE_CIVILIAN] = missionParams->civilians;
    if (missionParams->civilians > 0) {
        const teamDef_t* teamDef = cgi->Com_GetTeamDefinitionByID(missionParams->civTeam);
        for (int unitIDX = 0; unitIDX < missionParams->civilians; unitIDX++) {
            autoUnit_t* unit = AM_GetUnit(battle, AUTOMISSION_TEAM_TYPE_CIVILIAN, unitIDX);

            AM_CreateUnitChr(unit, teamDef, nullptr);
            unit->team = AUTOMISSION_TEAM_TYPE_CIVILIAN;
            unit->idx = unitIDX;
        }
        battle->scoreTeamSkill[AUTOMISSION_TEAM_TYPE_CIVILIAN] = (frand() * 0.5f) + 0.05f;
    }
}

static void AM_SetDefaultHostilities (autoMissionBattle_t* battle, const bool civsInfected)
{
    bool civsInverted = !civsInfected;

    for (int i = AUTOMISSION_TEAM_TYPE_PLAYER; i < AUTOMISSION_TEAM_TYPE_MAX; i++) {
        int j;
        const autoMissionTeamType_t team = (autoMissionTeamType_t)i;
        if (battle->actUnits[team] <= 0)
            continue;

        for (j = AUTOMISSION_TEAM_TYPE_PLAYER; j < AUTOMISSION_TEAM_TYPE_MAX; j++) {
            const autoMissionTeamType_t otherTeam = (autoMissionTeamType_t)j;
            if (battle->actUnits[otherTeam] <= 0)
                continue;

            if (AM_IsPlayer(team)) {
                if (AM_IsAlien(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, true);
                else if (AM_IsPlayer(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, false);
                else if (AM_IsCivilian(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, civsInfected);
            } else if (AM_IsAlien(team)) {
                if (AM_IsAlien(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, false);
                else if (AM_IsPlayer(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, true);
                else if (AM_IsCivilian(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, civsInverted);
            } else if (AM_IsCivilian(team)) {
                if (AM_IsAlien(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, civsInverted);
                else if (AM_IsPlayer(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, civsInfected);
                else if (AM_IsCivilian(otherTeam))
                    AM_SetHostile(battle, team, otherTeam, false);
            }
        }
    }
}

static void AM_CalculateTeamScores (autoMissionBattle_t* battle)
{
    int unitTotal = 0;
    int isHostileTotal = 0;
    int totalActiveTeams = 0;
    int lastActiveTeam = -1;
    int isHostileCount;
    int team;
    /* Sums of various values */
    double teamPooledHealth[AUTOMISSION_TEAM_TYPE_MAX];
    double teamPooledHealthMax[AUTOMISSION_TEAM_TYPE_MAX];
    double teamPooledUnitsHealthy[AUTOMISSION_TEAM_TYPE_MAX];
    double teamPooledUnitsTotal[AUTOMISSION_TEAM_TYPE_MAX];
    /* Ratios */
    double teamRatioHealthyUnits[AUTOMISSION_TEAM_TYPE_MAX];
    double teamRatioHealthTotal[AUTOMISSION_TEAM_TYPE_MAX];

    for (team = 0; team < AUTOMISSION_TEAM_TYPE_MAX; team++) {
        unitTotal += battle->nUnits[team];

        if (battle->actUnits[team] > 0) {
            lastActiveTeam = team;
            totalActiveTeams++;
        }
        for (isHostileCount = 0; isHostileCount < AUTOMISSION_TEAM_TYPE_MAX; isHostileCount++) {
            if (battle->nUnits[isHostileCount] <= 0)
                continue;

            if (battle->isHostile[team][isHostileCount] && battle->actUnits[team] > 0)
                isHostileTotal++;
        }
    }

    /* sanity checks */
    if (unitTotal == 0)
        cgi->Com_Error(ERR_DROP, "Grand total of ZERO units are fighting in auto battle, something is wrong.");

    if (unitTotal < 0)
        cgi->Com_Error(ERR_DROP, "Negative number of total units are fighting in auto battle, something is VERY wrong!");

    if (isHostileTotal <= 0)
        cgi->Com_Error(ERR_DROP, "No team has any other team hostile toward it, no battle is possible!");

    if (totalActiveTeams <= 0)
        cgi->Com_Error(ERR_DROP, "No Active teams detected in Auto Battle!");

    if (totalActiveTeams == 1) {
        cgi->Com_DPrintf(DEBUG_CLIENT, "Note: Only one active team detected, this team will win the auto mission battle by default.\n");
        battle->winningTeam = lastActiveTeam;
        return;
    }

    /* Set up teams */
    for (team = 0; team < AUTOMISSION_TEAM_TYPE_MAX; team++) {
        teamPooledHealth[team] = 0.0;
        teamPooledHealthMax[team] = 0.0;
        teamPooledUnitsHealthy[team] = 0.0;
        teamPooledUnitsTotal[team] = 0.0;

        if (battle->actUnits[team] > 0) {
            double skillAdjCalc;
            double skillAdjCalcAbs;

            for (int currentUnit = 0; currentUnit < battle->nUnits[team]; currentUnit++) {
                autoUnit_t* unit = AM_GetUnit(battle, team, currentUnit);
                const character_t* chr = unit->chr;

                if (chr->HP <= 0)
                    continue;

                teamPooledHealth[team] += chr->HP;
                teamPooledHealthMax[team] += chr->maxHP;
                teamPooledUnitsTotal[team] += 1.0;
                if (chr->HP == chr->maxHP)
                    teamPooledUnitsHealthy[team] += 1.0;
            }
            /* We shouldn't be dividing by zero here. */
            assert(teamPooledHealthMax[team] > 0.0);
            assert(teamPooledUnitsTotal[team] > 0.0);

            teamRatioHealthTotal[team] = teamPooledHealth[team] / teamPooledHealthMax[team];
            teamRatioHealthyUnits[team] = teamPooledUnitsHealthy[team] / teamPooledUnitsTotal[team];

            /* In DEBUG mode, these should help with telling where things are at what time, for bug-hunting purposes. */
            /* Note (Destructavator):  Is there a better way to implement this?  Is there a set protocol for this type of thing? */
            cgi->Com_DPrintf(DEBUG_CLIENT, "Team %i has calculated ratio of healthy units of %f.\n",
                team, teamRatioHealthyUnits[team]);
            cgi->Com_DPrintf(DEBUG_CLIENT, "Team %i has calculated ratio of health values of %f.\n",
                team, teamRatioHealthTotal[team]);

            skillAdjCalc = teamRatioHealthyUnits[team] + teamRatioHealthTotal[team];
            skillAdjCalc *= 0.50;
            skillAdjCalc = FpCurve1D_u_in(skillAdjCalc, 0.50, 0.50);
            skillAdjCalc -= 0.50;
            skillAdjCalcAbs = fabs(skillAdjCalc);
            if (skillAdjCalc > 0.0)
                battle->scoreTeamSkill[team] = ChkDNorm_Inv (FpCurveUp (battle->scoreTeamSkill[team], skillAdjCalcAbs) );
            else if (skillAdjCalc < 0.0)
                battle->scoreTeamSkill[team] = ChkDNorm (FpCurveDn (battle->scoreTeamSkill[team], skillAdjCalcAbs) );
            /* if (skillAdjCalc == exact 0.0), no change to team's skill. */

            cgi->Com_DPrintf(DEBUG_CLIENT, "Team %i has adjusted skill rating of %f.\n",
                team, battle->scoreTeamSkill[team]);
        }
    }
}

static int AM_GetRandomTeam (autoMissionBattle_t* battle, int currTeam, bool enemy)
{
    int eTeam;

    assert(battle);
    assert(currTeam >= 0 && currTeam < AUTOMISSION_TEAM_TYPE_MAX);

    /* select a team randomly */
    eTeam = rand () % AUTOMISSION_TEAM_TYPE_MAX;
    /* if selected team is active and it's hostility match, we're ready */
    if (battle->actUnits[eTeam] > 0 && AM_IsHostile(battle, currTeam, eTeam) == enemy) {
        return eTeam;
    } else {
        int nextTeam;

        /* if not, check next */
        for (nextTeam = (eTeam + 1) % AUTOMISSION_TEAM_TYPE_MAX; nextTeam != eTeam; nextTeam = (nextTeam + 1) % AUTOMISSION_TEAM_TYPE_MAX) {
            if (battle->actUnits[nextTeam] > 0 && AM_IsHostile(battle, currTeam, nextTeam) == enemy)
                return nextTeam;
        }
        /* none found */
        return AUTOMISSION_TEAM_TYPE_MAX;
    }
}

static autoUnit_t* AM_GetRandomActiveUnitOfTeam (autoMissionBattle_t* battle, int team)
{
    int idx;
    autoUnit_t* unit;

    assert(battle);
    if (team < 0 || team >= AUTOMISSION_TEAM_TYPE_MAX)
        return nullptr;
    if (battle->actUnits[team] <= 0)
        return nullptr;
    if (battle->nUnits[team] <= 0)
        return nullptr;

    /* select a unit randomly */
    idx = rand() % battle->nUnits[team];
    unit = AM_GetUnit(battle, team, idx);

    /* if (s)he is active (alive, not stunned), we're ready */
    if (AM_IsUnitActive(unit)) {
        return unit;
    } else {
        int nextIdx;

        /* if not active, check next */
        for (nextIdx = (idx + 1) % battle->nUnits[team]; nextIdx != idx; nextIdx = (nextIdx + 1) % battle->nUnits[team]) {
            unit = AM_GetUnit(battle, team, nextIdx);
            if (AM_IsUnitActive(unit))
                return unit;
        }
        /* none found */
        return nullptr;
    }
}

static autoUnit_t* AM_GetRandomActiveUnit (autoMissionBattle_t* battle, int currTeam, bool enemy)
{
    int eTeam;
    int nextTeam;
    autoUnit_t* unit;

    assert(battle);
    assert(currTeam >= 0 && currTeam < AUTOMISSION_TEAM_TYPE_MAX);

    eTeam = AM_GetRandomTeam(battle, currTeam, enemy);
    if (eTeam >= AUTOMISSION_TEAM_TYPE_MAX)
        return nullptr;

    unit = AM_GetRandomActiveUnitOfTeam(battle, eTeam);
    if (unit)
        return unit;

    /* if not, check next */
    for (nextTeam = (eTeam + 1) % AUTOMISSION_TEAM_TYPE_MAX; nextTeam != eTeam; nextTeam = (nextTeam + 1) % AUTOMISSION_TEAM_TYPE_MAX) {
        if (battle->actUnits[nextTeam] > 0 && AM_IsHostile(battle, currTeam, nextTeam) == enemy) {
            unit = AM_GetRandomActiveUnitOfTeam(battle, nextTeam);
            if (unit)
                return unit;
        }
    }
    /* none found */
    return nullptr;
}

static bool AM_CheckFire (autoMissionBattle_t* battle, autoUnit_t* currUnit, autoUnit_t* eUnit, const double effective)
{
    character_t* currChr = currUnit->chr;
    chrScoreGlobal_t* score = &currChr->score;
    character_t* eChr = eUnit->chr;
    double calcRand = frand();
    int strikeDamage;

    if (AM_IsHostile(battle, currUnit->team, eUnit->team)) {
        if (calcRand > effective)
            return false;
        strikeDamage = (int) (100.0 * battle->scoreTeamDifficulty[currUnit->team] * (effective - calcRand) / effective);
        battle->teamAccomplishment[currUnit->team] += strikeDamage;
    } else {
        if (calcRand >= (0.050 - (effective * 0.050)))
            return false;
        strikeDamage = (int) (100.0 * (1.0 - battle->scoreTeamDifficulty[currUnit->team]) * calcRand);
        battle->teamAccomplishment[currUnit->team] -= strikeDamage;
    }

    eChr->HP = std::max(0, eChr->HP - strikeDamage);
    /* Wound the target */
    if (eChr->HP > 0)
        eChr->wounds.treatmentLevel[eChr->teamDef->bodyTemplate->getRandomBodyPart()] += strikeDamage;

    /* If target is still active, continue */
    if (AM_IsUnitActive(eUnit))
        return true;

#if DEBUG
    cgi->Com_Printf("AutoBattle: Team: %d Unit: %d killed Team: %d Unit: %d\n", currUnit->team, currUnit->idx, eUnit->team, eUnit->idx);
#endif
    battle->actUnits[eUnit->team]--;

    switch (currUnit->team) {
    case AUTOMISSION_TEAM_TYPE_PLAYER:
        switch (eUnit->team) {
        case AUTOMISSION_TEAM_TYPE_PLAYER:
            battle->results->ownSurvived--;
            battle->results->ownKilledFriendlyFire++;
            score->kills[KILLED_TEAM] += 1;
            break;
        case AUTOMISSION_TEAM_TYPE_ALIEN:
            battle->results->aliensSurvived--;
            battle->results->aliensKilled++;
            score->kills[KILLED_ENEMIES] += 1;
            break;
        case AUTOMISSION_TEAM_TYPE_CIVILIAN:
            battle->results->civiliansSurvived--;
            battle->results->civiliansKilledFriendlyFire++;
            score->kills[KILLED_CIVILIANS] += 1;
            break;
        default:
            break;
        }
        break;
    case AUTOMISSION_TEAM_TYPE_ALIEN:
        switch (eUnit->team) {
        case AUTOMISSION_TEAM_TYPE_PLAYER:
            battle->results->ownSurvived--;
            battle->results->ownKilled++;
            break;
        case AUTOMISSION_TEAM_TYPE_ALIEN:
            battle->results->aliensSurvived--;
            battle->results->aliensKilled++;
            break;
        case AUTOMISSION_TEAM_TYPE_CIVILIAN:
            battle->results->civiliansSurvived--;
            battle->results->civiliansKilled++;
            break;
        default:
            break;
        }
        break;
    case AUTOMISSION_TEAM_TYPE_CIVILIAN:
        switch (eUnit->team) {
        case AUTOMISSION_TEAM_TYPE_PLAYER:
            battle->results->ownSurvived--;
            battle->results->ownKilledFriendlyFire++;
            break;
        case AUTOMISSION_TEAM_TYPE_ALIEN:
            battle->results->aliensSurvived--;
            battle->results->aliensKilled++;
            break;
        case AUTOMISSION_TEAM_TYPE_CIVILIAN:
            battle->results->civiliansSurvived--;
            battle->results->civiliansKilledFriendlyFire++;
            break;
        default:
            break;
        }
        break;
    default:
        break;
    }
    return true;
}

static bool AM_UnitAttackEnemy (autoMissionBattle_t* battle, autoUnit_t* currUnit, const double effective)
{
    autoUnit_t* eUnit;

    eUnit = AM_GetRandomActiveUnit(battle, currUnit->team, true);
    /* no more enemies */
    if (eUnit == nullptr)
        return false;

    /* shot an enemy */
    if (!AM_CheckFire(battle, currUnit, eUnit, effective)) {
        /* if failed, attack a friendly */
        eUnit = AM_GetRandomActiveUnit(battle, currUnit->team, false);
        if (eUnit != nullptr)
            AM_CheckFire(battle, currUnit, eUnit, effective);
    }

    return true;
}

static void AM_DoFight (autoMissionBattle_t* battle)
{
    bool combatActive = true;

#ifdef DEBUG
    cgi->Com_Printf("Auto battle started\n");
    for (int teamID = 0; teamID < AUTOMISSION_TEAM_TYPE_MAX; teamID++) {
        cgi->Com_Printf("Team %d Units: %d\n", teamID, battle->nUnits[teamID]);
    }
#endif

    while (combatActive) {
        for (int team = 0; team < AUTOMISSION_TEAM_TYPE_MAX; team++) {
            int aliveUnits;

            if (team == AUTOMISSION_TEAM_TYPE_CIVILIAN)
                continue;

            if (battle->actUnits[team] <= 0)
                continue;

            aliveUnits = 0;
            /* Is this unit still alive (has any health left?) */
            for (int currentUnit = 0; currentUnit < battle->nUnits[team]; currentUnit++) {
                autoUnit_t* unit = AM_GetUnit(battle, team, currentUnit);
                character_t* chr = unit->chr;
                /* Wounded units don't fight quite as well */
                const double hpLeftRatio = chr->HP / chr->maxHP;
                const double effective = FpCurveDn(battle->scoreTeamSkill[team], hpLeftRatio * 0.50);

                if (!AM_IsUnitActive(unit))
                    continue;

                cgi->Com_DPrintf(DEBUG_CLIENT, "Unit %i on team %i has adjusted attack rating of %f.\n",
                    currentUnit, team, battle->scoreTeamSkill[team]);

                aliveUnits++;
                combatActive = AM_UnitAttackEnemy(battle, unit, effective);
            }
        }
    }

    /* Set results */
    if (battle->actUnits[AUTOMISSION_TEAM_TYPE_PLAYER] <= 0) {
        battle->results->state = LOST;
        battle->winningTeam = AUTOMISSION_TEAM_TYPE_ALIEN;
    } else {
        battle->winningTeam = AUTOMISSION_TEAM_TYPE_PLAYER;
        battle->results->state = WON;
    }
}

static void AM_DisplayResults (const autoMissionBattle_t* battle)
{
    assert(battle);

    cgi->Cvar_SetValue("cp_mission_tryagain", 0);
    if (battle->results->state == WON) {
        cgi->UI_PushWindow("won");
        if (battle->teamAccomplishment[AUTOMISSION_TEAM_TYPE_PLAYER] > battle->teamAccomplishment[AUTOMISSION_TEAM_TYPE_ALIEN])
            MS_AddNewMessage(_("Notice"), _("You've won the battle"));
        else
            MS_AddNewMessage(_("Notice"), _("You've defeated the enemy, but did poorly, and many civilians were killed"));
    } else {
        cgi->UI_PushWindow("lost");
        MS_AddNewMessage(_("Notice"), _("You've lost the battle"));
    }
}

static void AM_MoveCharacterInventoryIntoItemCargo (aircraft_t* aircraft, character_t* chr)
{
    assert(aircraft != nullptr);
    assert(chr != nullptr);

    /* add items to itemcargo */
    const Container* cont = nullptr;
    while ((cont = chr->inv.getNextCont(cont))) {
        Item* item = nullptr;
        while ((item = cont->getNextItem(item))) {
            if (item->def()) {
                AII_CollectItem(aircraft, item->def(), 1);

                if (item->getAmmoLeft() && item->ammoDef())
                    AII_CollectItem(aircraft, item->ammoDef(), 1);
            }
        }
    }
}

static void AM_AlienCollect (aircraft_t* aircraft, const autoMissionBattle_t* battle)
{
    assert(aircraft);
    assert(battle);

    /* Aliens */
    int collected = 0;
    for (int unitIDX = 0; unitIDX < battle->nUnits[AUTOMISSION_TEAM_TYPE_ALIEN]; unitIDX++) {
        const autoUnit_t* unit = AM_GetUnit(battle, AUTOMISSION_TEAM_TYPE_ALIEN, unitIDX);

        if (AM_IsUnitActive(unit))
            continue;

        AM_MoveCharacterInventoryIntoItemCargo(aircraft, unit->chr);
        AL_AddAlienTypeToAircraftCargo(aircraft, unit->chr->teamDef, 1, unit->chr->HP <= 0);
        collected++;
    }

    if (collected > 0)
        MS_AddNewMessage(_("Notice"), _("Collected alien bodies"));
}

static void AM_UpdateSurivorsAfterBattle (const autoMissionBattle_t* battle, struct aircraft_s* aircraft)
{
    assert(battle);
    assert(battle->results);

    const int battleExperience = std::max(0, battle->teamAccomplishment[AUTOMISSION_TEAM_TYPE_PLAYER]);
    int unit = 0;

    LIST_Foreach(aircraft->acTeam, Employee, soldier) {
        if (unit >= MAX_SOLDIERS_AUTOMISSION)
            break;

        unit++;

        character_t* chr = &soldier->chr;
        /* dead soldiers are removed in CP_MissionEnd, just move their inventory to itemCargo */
        if (chr->HP <= 0) {
            if (battle->results->state == WON)
                AM_MoveCharacterInventoryIntoItemCargo(aircraft, &soldier->chr);
            E_RemoveInventoryFromStorage(soldier);
            continue;
        }

        chrScoreGlobal_t* score = &chr->score;
        for (int expCount = 0; expCount < ABILITY_NUM_TYPES; expCount++) {
            const int maxXP = CHAR_GetMaxExperiencePerMission(static_cast<abilityskills_t>(expCount));
            const int gainedXP = std::min(maxXP, static_cast<int>(battleExperience * ABILITY_AWARD_SCALE * frand()));
            score->experience[expCount] += gainedXP;
            cgi->Com_DPrintf(DEBUG_CLIENT, "AM_UpdateSurivorsAfterBattle: Soldier %s earned %d experience points in skill #%d (total experience: %d).\n",
                        chr->name, gainedXP, expCount, chr->score.experience[expCount]);
        }

        for (int expCount = ABILITY_NUM_TYPES; expCount < SKILL_NUM_TYPES; expCount++) {
            const int maxXP = CHAR_GetMaxExperiencePerMission(static_cast<abilityskills_t>(expCount));
            const int gainedXP = std::min(maxXP, static_cast<int>(battleExperience * SKILL_AWARD_SCALE * frand()));
            score->experience[expCount] += gainedXP;
            cgi->Com_DPrintf(DEBUG_CLIENT, "AM_UpdateSurivorsAfterBattle: Soldier %s earned %d experience points in skill #%d (total experience: %d).\n",
                        chr->name, gainedXP, expCount, chr->score.experience[expCount]);
        }
        /* Health isn't part of abilityskills_t, so it needs to be handled separately. */
        const int maxXP = CHAR_GetMaxExperiencePerMission(SKILL_NUM_TYPES);
        const int gainedXP = std::min(maxXP, static_cast<int>(battleExperience * ABILITY_AWARD_SCALE * frand()));
        score->experience[SKILL_NUM_TYPES] += gainedXP;
        cgi->Com_DPrintf(DEBUG_CLIENT, "AM_UpdateSurivorsAfterBattle: Soldier %s earned %d experience points in skill #%d (total experience: %d).\n",
                    chr->name, gainedXP, SKILL_NUM_TYPES, chr->score.experience[SKILL_NUM_TYPES]);

        CHAR_UpdateSkills(chr);
    }
}

static void AM_CleanBattleParameters (autoMissionBattle_t* battle)
{
    int unitIDX;

    assert(battle);

    /* Aliens */
    for (unitIDX = 0; unitIDX < battle->nUnits[AUTOMISSION_TEAM_TYPE_ALIEN]; unitIDX++) {
        autoUnit_t* unit = AM_GetUnit(battle, AUTOMISSION_TEAM_TYPE_ALIEN, unitIDX);

        AM_DestroyUnitChr(unit);
    }

    /* Civilians */
    for (unitIDX = 0; unitIDX < battle->nUnits[AUTOMISSION_TEAM_TYPE_CIVILIAN]; unitIDX++) {
        autoUnit_t* unit = AM_GetUnit(battle, AUTOMISSION_TEAM_TYPE_CIVILIAN, unitIDX);

        AM_DestroyUnitChr(unit);
    }
}

void AM_Go (mission_t* mission, aircraft_t* aircraft, const campaign_t* campaign, const battleParam_t* battleParameters, missionResults_t* results)
{
    autoMissionBattle_t autoBattle;

    assert(mission);
    assert(aircraft);
    assert(aircraft->homebase);

    if (mission && mission->mapDef && mission->mapDef->storyRelated) {
        cgi->Com_Printf("Story-related mission cannot be done via automission\n");
        return;
    }

    AM_ClearBattle(&autoBattle);
    autoBattle.results = results;
    AM_FillTeamFromAircraft(&autoBattle, AUTOMISSION_TEAM_TYPE_PLAYER, aircraft, campaign);
    AM_FillTeamFromBattleParams(&autoBattle, battleParameters);
    AM_SetDefaultHostilities(&autoBattle, false);
    AM_CalculateTeamScores(&autoBattle);

    OBJZERO(*results);
    results->ownSurvived = autoBattle.nUnits[AUTOMISSION_TEAM_TYPE_PLAYER];
    results->aliensSurvived = autoBattle.nUnits[AUTOMISSION_TEAM_TYPE_ALIEN];
    results->civiliansSurvived = autoBattle.nUnits[AUTOMISSION_TEAM_TYPE_CIVILIAN];
    results->mission = mission;

    ccs.eMission = aircraft->homebase->storage; /* copied, including arrays inside! */

    AM_DoFight(&autoBattle);

    AM_UpdateSurivorsAfterBattle(&autoBattle, aircraft);
    if (results->state == WON)
        AM_AlienCollect(aircraft, &autoBattle);

    MIS_InitResultScreen(results);
    if (ccs.missionResultCallback) {
        ccs.missionResultCallback(results);
    }

    AM_DisplayResults(&autoBattle);
    AM_CleanBattleParameters(&autoBattle);
}

void AM_InitStartup (void)
{
}

void AM_Shutdown (void)
{
}