cp_fightequip_callbacks.cpp

Go to the documentation of this file.

/*
Copyright (C) 2002-2020 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 "../../ui/ui_dataids.h"
#include "cp_campaign.h"
#include "cp_fightequip_callbacks.h"
#include "cp_mapfightequip.h"
#include "cp_ufo.h"

static aircraftItemType_t airequipID = MAX_ACITEMS;             
static int airequipSelectedZone = ZONE_NONE;        
static int airequipSelectedSlot = ZONE_NONE;            
static technology_t* aimSelectedTechnology = nullptr;       
static void AIM_CheckAirequipID (void)
{
    switch (airequipID) {
    case AC_ITEM_AMMO:
    case AC_ITEM_WEAPON:
    case AC_ITEM_SHIELD:
    case AC_ITEM_ELECTRONICS:
        return;
    default:
        airequipID = AC_ITEM_WEAPON;
        break;
    }
}

static void AIM_CheckAirequipSelectedSlot (const aircraft_t* aircraft)
{
    switch (airequipID) {
    case AC_ITEM_AMMO:
    case AC_ITEM_WEAPON:
        if (airequipSelectedSlot >= aircraft->maxWeapons) {
            airequipSelectedSlot = 0;
        }
        break;
    case AC_ITEM_ELECTRONICS:
        if (airequipSelectedSlot >= aircraft->maxElectronics) {
            airequipSelectedSlot = 0;
        }
        break;
    default:
        break;
    }
}

aircraftSlot_t* AII_SelectAircraftSlot (aircraft_t* aircraft, aircraftItemType_t type)
{
    aircraftSlot_t* slot;

    AIM_CheckAirequipSelectedSlot(aircraft);
    switch (type) {
    case AC_ITEM_SHIELD:
        slot = &aircraft->shield;
        break;
    case AC_ITEM_ELECTRONICS:
        slot = aircraft->electronics + airequipSelectedSlot;
        break;
    case AC_ITEM_AMMO:
    case AC_ITEM_WEAPON:
        slot = aircraft->weapons + airequipSelectedSlot;
        break;
    default:
        cgi->Com_Printf("AII_SelectAircraftSlot: Unknown airequipID: %i\n", type);
        return nullptr;
    }

    return slot;
}

static void AIM_CheckAirequipSelectedZone (aircraftSlot_t* slot)
{
    assert(slot);

    if (airequipSelectedZone == ZONE_AMMO && airequipID < AC_ITEM_AMMO && airequipID > AC_ITEM_WEAPON) {
        /* you can select ammo only for weapons and ammo */
        airequipSelectedZone = ZONE_MAIN;
    }

    /* You can choose an ammo only if a weapon has already been selected */
    if (airequipID >= AC_ITEM_AMMO && !slot->item) {
        airequipSelectedZone = ZONE_MAIN;
    }
}

static inline const char* AIM_AircraftItemtypeName (const int equiptype)
{
    switch (equiptype) {
    case AC_ITEM_WEAPON:
        return _("Weapons");
    case AC_ITEM_SHIELD:
        return _("Armour");
    case AC_ITEM_ELECTRONICS:
        return _("Items");
    case AC_ITEM_AMMO:
        /* ammo - only available from weapons */
        return _("Ammo");
    default:
        return _("Unknown");
    }
}

static bool AIM_CrafttypeFilter (const base_t* base, aircraftItemType_t filterType, const technology_t* tech)
{
    const objDef_t* item;
    if (!base)
        return false;

    if (!RS_IsResearched_ptr(tech))
        return false;

    item = INVSH_GetItemByID(tech->provides);
    if (!item)
        return false;
    if (item->isVirtual)
        return false;
    if (!B_BaseHasItem(base, item))
        return false;

    /* filter by type: special case for ammo because more than 1 type is an ammo type */
    if (filterType != AC_ITEM_AMMO) {
        if (item->craftitem.type != filterType)
            return false;
    } else {
        if (item->craftitem.type < AC_ITEM_AMMO)
            return false;
    }

    /* you can't install an item that does not have an installation time (alien item)
     * except for ammo which does not have installation time */
    if (item->craftitem.installationTime == -1 && filterType >= AC_ITEM_AMMO)
        return false;

    return true;
}

static void AIM_UpdateAircraftItemList (const aircraftSlot_t* slot)
{
    linkedList_t* amountList = nullptr;
    technology_t** techList;
    technology_t** currentTech;
    const base_t* base = slot->aircraft->homebase;
    int count = 0;
    uiNode_t* AIM_items = nullptr;

    /* Add all items corresponding to airequipID to list */
    techList = AII_GetCraftitemTechsByType(airequipID);

    /* Count only those which are researched to buffer */
    currentTech = techList;
    while (*currentTech) {
        if (AIM_CrafttypeFilter(base, airequipID, *currentTech))
            count++;
        currentTech++;
    }

    /* List only those which are researched to buffer */
    currentTech = techList;
    while (*currentTech) {
        if (AIM_CrafttypeFilter(base, airequipID, *currentTech)) {
            const objDef_t* item = INVSH_GetItemByID((*currentTech)->provides);
            const int amount = B_ItemInBase(item, base);

            cgi->LIST_AddString(&amountList, va("%d", amount));
            uiNode_t* option = cgi->UI_AddOption(&AIM_items, (*currentTech)->name, _((*currentTech)->name), va("%d", (*currentTech)->idx));
            if (!AIM_SelectableCraftItem(slot, *currentTech))
                option->disabled = true;
        }
        currentTech++;
    }

    cgi->UI_RegisterOption(TEXT_LIST, AIM_items);
    cgi->UI_RegisterLinkedListText(TEXT_LIST2, amountList);
}

static void AIM_DrawAircraftSlots (const aircraft_t* aircraft)
{
    /* initialise model cvars */
    for (int i = 0; i < AIR_POSITIONS_MAX; i++)
        cgi->Cvar_Set(va("mn_aircraft_item_model_slot%i", i), "");

    for (int i = 0; i < AIR_POSITIONS_MAX; i++) {
        const aircraftSlot_t* slot;
        int max;

        /* Default value */
        cgi->UI_ExecuteConfunc("airequip_display_slot %i 0", i);

        /* Draw available slots */
        switch (airequipID) {
        case AC_ITEM_AMMO:
        case AC_ITEM_WEAPON:
            max = aircraft->maxWeapons;
            slot = aircraft->weapons;
            break;
        case AC_ITEM_ELECTRONICS:
            max = aircraft->maxElectronics;
            slot = aircraft->electronics;
            break;
        /* do nothing for shield: there is only one slot */
        default:
            continue;
        }
        for (int j = 0; j < max; j++, slot++) {
            /* check if one of the aircraft slots is at this position */
            if (slot->pos == i) {
                /* draw in white if this is the selected slot */
                if (j == airequipSelectedSlot) {
                    cgi->UI_ExecuteConfunc("airequip_display_slot %i 2", i);
                } else {
                    cgi->UI_ExecuteConfunc("airequip_display_slot %i 1", i);
                }
                if (slot->item) {
                    cgi->Cvar_Set(va("mn_aircraft_item_model_slot%i", i), "%s", RS_GetTechForItem(slot->item)->mdl);
                } else {
                    cgi->Cvar_Set(va("mn_aircraft_item_model_slot%i", i), "");
                }
            }
        }
    }
}

static inline void AIM_EmphazeAmmoSlotText (void)
{
    cgi->UI_ExecuteConfunc("airequip_zone2_color \"1 0 0 1\"");
}

static inline void AIM_NoEmphazeAmmoSlotText (void)
{
    cgi->UI_ExecuteConfunc("airequip_zone2_color \"1 1 1 1\"");
}

static void AIM_AircraftEquipMenuUpdate (void)
{
    static char smallbuffer1[256];
    static char smallbuffer2[128];
    const char* typeName;
    aircraft_t* aircraft;
    aircraftSlot_t* slot;
    base_t* base = B_GetCurrentSelectedBase();

    if (!base)
        return;

    /* don't let old links appear on this menu */
    cgi->UI_ResetData(TEXT_AIREQUIP_1);
    cgi->UI_ResetData(TEXT_AIREQUIP_2);
    cgi->UI_ResetData(TEXT_ITEMDESCRIPTION);
    cgi->UI_ResetData(TEXT_LIST);

    aircraft = base->aircraftCurrent;

    assert(aircraft);

    /* Check that airequipSelectedSlot corresponds to an existing slot for this aircraft */
    AIM_CheckAirequipSelectedSlot(aircraft);

    /* Select slot */
    slot = AII_SelectAircraftSlot(aircraft, airequipID);

    /* Check that the selected zone is OK */
    AIM_CheckAirequipSelectedZone(slot);

    /* Fill the list of item you can equip your aircraft with */
    AIM_UpdateAircraftItemList(slot);

    cgi->Cvar_Set("mn_equip_itemtype_name", "%s", AIM_AircraftItemtypeName(airequipID));
    switch (airequipID) {
    case AC_ITEM_ELECTRONICS:
        typeName = "item";
        break;
    case AC_ITEM_SHIELD:
        typeName = "armour";
        break;
    case AC_ITEM_AMMO:
        typeName = "ammo";
        break;
    case AC_ITEM_WEAPON:
        typeName = "weapon";
        break;
    default:
        typeName = "unknown";
        break;
    }
    cgi->Cvar_Set("mn_equip_itemtype", "%s", typeName);

    /* First slot: item currently assigned */
    if (!slot->item) {
        Com_sprintf(smallbuffer1, sizeof(smallbuffer1), "%s", _("No item assigned.\n"));
        Q_strcat(smallbuffer1, sizeof(smallbuffer1), _("This slot is for %s or smaller items."),
            AII_WeightToName(slot->size));
    } else {
        technology_t* itemTech = RS_GetTechForItem(slot->item);
        technology_t* nextItemTech = slot->nextItem ? RS_GetTechForItem(slot->nextItem) : nullptr;
        /* Print next item if we are removing item currently installed and a new item has been added. */
        Com_sprintf(smallbuffer1, sizeof(smallbuffer1), "%s\n", slot->nextItem ? _(nextItemTech->name) : _(itemTech->name));
        if (!slot->installationTime) {
            Q_strcat(smallbuffer1, sizeof(smallbuffer1), _("This item is functional.\n"));
        } else if (slot->installationTime > 0) {
            Q_strcat(smallbuffer1, sizeof(smallbuffer1), ngettext("This item will be installed in %i hour.\n",
                    "This item will be installed in %i hours.\n", slot->installationTime), slot->installationTime);
        } else if (slot->nextItem) {
            Q_strcat(smallbuffer1, sizeof(smallbuffer1), ngettext("%s will be removed in %i hour.\n", "%s will be removed in %i hours.\n",
                    -slot->installationTime), _(itemTech->name), -slot->installationTime);
            Q_strcat(smallbuffer1, sizeof(smallbuffer1), ngettext("%s will be installed in %i hour.\n",
                    "%s will be installed in %i hours.\n", slot->nextItem->craftitem.installationTime - slot->installationTime),
                    _(nextItemTech->name), slot->nextItem->craftitem.installationTime - slot->installationTime);
        } else {
            Q_strcat(smallbuffer1, sizeof(smallbuffer1), ngettext("This item will be removed in %i hour.\n",
                    "This item will be removed in %i hours.\n", -slot->installationTime), -slot->installationTime);
        }
    }
    cgi->UI_RegisterText(TEXT_AIREQUIP_1, smallbuffer1);

    /* Second slot: ammo slot (only used for weapons) */
    if ((airequipID == AC_ITEM_WEAPON || airequipID == AC_ITEM_AMMO) && slot->item) {
        if (!slot->ammo) {
            AIM_EmphazeAmmoSlotText();
            Com_sprintf(smallbuffer2, sizeof(smallbuffer2), "%s", _("No ammo assigned to this weapon."));
        } else {
            const objDef_t* ammo = slot->nextAmmo ? slot->nextAmmo : slot->ammo;
            const technology_t* tech = RS_GetTechForItem(ammo);
            AIM_NoEmphazeAmmoSlotText();
            if (!ammo->isVirtual)
                Q_strncpyz(smallbuffer2, _(tech->name), sizeof(smallbuffer2));
            else
                Q_strncpyz(smallbuffer2, _("No ammo needed"), sizeof(smallbuffer2));
        }
    } else
        *smallbuffer2 = '\0';

    cgi->UI_RegisterText(TEXT_AIREQUIP_2, smallbuffer2);

    /* Draw existing slots for this aircraft */
    AIM_DrawAircraftSlots(aircraft);
}

#define AIM_LOADING_OK                          0
#define AIM_LOADING_NOSLOTSELECTED              1
#define AIM_LOADING_NOTECHNOLOGYSELECTED        2
#define AIM_LOADING_ALIENTECH                   3
#define AIM_LOADING_TECHNOLOGYNOTRESEARCHED     4
#define AIM_LOADING_TOOHEAVY                    5
#define AIM_LOADING_UNKNOWNPROBLEM              6
#define AIM_LOADING_NOWEAPON                    7
#define AIM_LOADING_NOTUSABLEWITHWEAPON         8

static int AIM_CheckTechnologyIntoSlot (const aircraftSlot_t* slot, const technology_t* tech)
{
    const objDef_t* item;

    if (!tech)
        return AIM_LOADING_NOTECHNOLOGYSELECTED;

    if (!slot)
        return AIM_LOADING_NOSLOTSELECTED;

    if (!RS_IsResearched_ptr(tech))
        return AIM_LOADING_TECHNOLOGYNOTRESEARCHED;

    item = INVSH_GetItemByID(tech->provides);
    if (!item)
        return AIM_LOADING_NOTECHNOLOGYSELECTED;

    if (item->craftitem.type >= AC_ITEM_AMMO) {
        const objDef_t* weapon = slot->item;
        int k;
        if (slot->nextItem != nullptr)
            weapon = slot->nextItem;

        if (weapon == nullptr)
            return AIM_LOADING_NOWEAPON;

        /* Is the ammo is usable with the slot */
        for (k = 0; k < weapon->numAmmos; k++) {
            const objDef_t* usable = weapon->ammos[k];
            if (usable && item->idx == usable->idx)
                break;
        }
        if (k >= weapon->numAmmos)
            return AIM_LOADING_NOTUSABLEWITHWEAPON;

#if 0

        if (!slot->nextItem && item->weapons[0] != slot->item)
            return AIM_LOADING_UNKNOWNPROBLEM;

        /* are we trying to change ammos for nextItem? */
        if (slot->nextItem && item->weapons[0] != slot->nextItem)
            return AIM_LOADING_UNKNOWNPROBLEM;
#endif
    }

    /* you can install an item only if its weight is small enough for the slot */
    if (AII_GetItemWeightBySize(item) > slot->size)
        return AIM_LOADING_TOOHEAVY;

    /* you can't install an item that you don't possess
     * virtual ammo don't need to be possessed
     * installations always have weapon and ammo */
    if (slot->aircraft) {
        if (!B_BaseHasItem(slot->aircraft->homebase, item))
            return AIM_LOADING_UNKNOWNPROBLEM;
    } else if (slot->base) {
        if (!B_BaseHasItem(slot->base, item))
            return AIM_LOADING_UNKNOWNPROBLEM;
    }

    /* you can't install an item that does not have an installation time (alien item)
     * except for ammo which does not have installation time */
    if (item->craftitem.installationTime == -1 && slot->type < AC_ITEM_AMMO)
        return AIM_LOADING_ALIENTECH;

    return AIM_LOADING_OK;
}

static void AIM_UpdateItemDescription (bool fromList, bool fromSlot)
{
    int status;
    aircraft_t* aircraft;
    aircraftSlot_t* slot;
    base_t* base = B_GetCurrentSelectedBase();
    assert(base);

    aircraft = base->aircraftCurrent;
    assert(aircraft);
    slot = AII_SelectAircraftSlot(aircraft, airequipID);

    /* update mini ufopedia */
    if (fromList)
        UP_AircraftItemDescription(INVSH_GetItemByIDSilent(aimSelectedTechnology ? aimSelectedTechnology->provides : nullptr));
    else if (fromSlot) {
        if (airequipID == AC_ITEM_AMMO)
            UP_AircraftItemDescription(slot->ammo);
        else
            UP_AircraftItemDescription(slot->item);
    }

    /* update status */
    status = AIM_CheckTechnologyIntoSlot(slot, aimSelectedTechnology);
    switch (status) {
    case AIM_LOADING_NOSLOTSELECTED:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("No slot selected."));
        break;
    case AIM_LOADING_NOTECHNOLOGYSELECTED:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("No item selected."));
        break;
    case AIM_LOADING_ALIENTECH:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("You can't equip an alien technology."));
        break;
    case AIM_LOADING_TECHNOLOGYNOTRESEARCHED:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("Technology requested is not yet completed."));
        break;
    case AIM_LOADING_TOOHEAVY:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("This item is too heavy for the selected slot."));
        break;
    case AIM_LOADING_NOWEAPON:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("Equip a weapon first."));
        break;
    case AIM_LOADING_NOTUSABLEWITHWEAPON:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("Ammo not usable with current weapon."));
        break;
    case AIM_LOADING_UNKNOWNPROBLEM:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("Unknown problem."));
        break;
    case AIM_LOADING_OK:
        cgi->Cvar_Set("mn_aircraft_item_warning", _("Ok"));
        break;
    }

    if (*cgi->Cvar_GetString("mn_item") == '\0') {
        cgi->UI_ExecuteConfunc("airequip_no_item");
    } else {
        if (fromSlot) {
            cgi->UI_ExecuteConfunc("airequip_installed_item");
        } else {
            if (status == AIM_LOADING_OK)
                cgi->UI_ExecuteConfunc("airequip_installable_item");
            else
                cgi->UI_ExecuteConfunc("airequip_noinstallable_item");
        }
    }
}

static void AIM_AircraftEquipMenuUpdate_f (void)
{
    if (cgi->Cmd_Argc() != 2) {
        if (airequipID == MAX_ACITEMS) {
            cgi->Com_Printf("Usage: %s <num>\n", cgi->Cmd_Argv(0));
            return;
        }
        AIM_CheckAirequipID();
    } else {
        const aircraftItemType_t type = (aircraftItemType_t)atoi(cgi->Cmd_Argv(1));
        switch (type) {
        case AC_ITEM_ELECTRONICS:
        case AC_ITEM_SHIELD:
            airequipID = type;
            cgi->UI_ExecuteConfunc("airequip_zone2_off");
            break;
        case AC_ITEM_AMMO:
        case AC_ITEM_WEAPON:
            airequipID = type;
            cgi->UI_ExecuteConfunc("airequip_zone2_on");
            break;
        default:
            airequipID = AC_ITEM_WEAPON;
            break;
        }
    }

    AIM_AircraftEquipMenuUpdate();
}

static void AIM_AircraftEquipSlotSelect_f (void)
{
    int i;
    itemPos_t pos;
    aircraft_t* aircraft;
    base_t* base = B_GetCurrentSelectedBase();
    int updateZone = 0;

    if (!base)
        return;

    if (cgi->Cmd_Argc() < 2) {
        cgi->Com_Printf("Usage: %s <arg> <zone1|zone2|item>\n", cgi->Cmd_Argv(0));
        return;
    }

    aircraft = base->aircraftCurrent;
    assert(aircraft);

    i = atoi(cgi->Cmd_Argv(1));
    pos = (itemPos_t)i;

    if (cgi->Cmd_Argc() == 3) {
        if (Q_streq(cgi->Cmd_Argv(2), "zone1")) {
            updateZone = 1;
        } else if (Q_streq(cgi->Cmd_Argv(2), "zone2")) {
            updateZone = 2;
        }
    }

    airequipSelectedSlot = ZONE_NONE;

    /* select the slot corresponding to pos, and set airequipSelectedSlot to this slot */
    switch (airequipID) {
    case AC_ITEM_ELECTRONICS:
        /* electronics selected */
        for (i = 0; i < aircraft->maxElectronics; i++) {
            if (aircraft->electronics[i].pos == pos) {
                airequipSelectedSlot = i;
                break;
            }
        }
        if (i == aircraft->maxElectronics)
            cgi->Com_Printf("this slot hasn't been found in aircraft electronics slots\n");
        break;
    case AC_ITEM_AMMO:
    case AC_ITEM_WEAPON:
        /* weapon selected */
        for (i = 0; i < aircraft->maxWeapons; i++) {
            if (aircraft->weapons[i].pos == pos) {
                airequipSelectedSlot = i;
                break;
            }
        }
        if (i == aircraft->maxWeapons)
            cgi->Com_Printf("this slot hasn't been found in aircraft weapon slots\n");
        break;
    default:
        cgi->Com_Printf("AIM_AircraftEquipSlotSelect_f : only weapons and electronics have several slots\n");
        break;
    }

    /* Update menu after changing slot */
    AIM_AircraftEquipMenuUpdate();

    /* update description with the selected slot */
    if (updateZone > 0)
        AIM_UpdateItemDescription(false, true);
    else
        AIM_UpdateItemDescription(true, false);
}

static void AIM_AircraftEquipZoneSelect_f (void)
{
    int zone;
    aircraft_t* aircraft;
    aircraftSlot_t* slot;
    base_t* base = B_GetCurrentSelectedBase();

    if (!base)
        return;

    if (cgi->Cmd_Argc() < 2) {
        cgi->Com_Printf("Usage: %s <arg>\n", cgi->Cmd_Argv(0));
        return;
    }

    zone = atoi(cgi->Cmd_Argv(1));

    aircraft = base->aircraftCurrent;
    assert(aircraft);
    /* Select slot */
    slot = AII_SelectAircraftSlot(aircraft, airequipID);

    /* ammos are only available for weapons */
    switch (airequipID) {
    /* a weapon was selected - select ammo type corresponding to this weapon */
    case AC_ITEM_WEAPON:
        if (zone == ZONE_AMMO) {
            if (slot->item)
                airequipID = AC_ITEM_AMMO;
        }
        break;
    /* an ammo was selected - select weapon type corresponding to this ammo */
    case AC_ITEM_AMMO:
        if (zone != ZONE_AMMO)
            airequipID = AC_ITEM_WEAPON;
        break;
    default :
        /* ZONE_AMMO is not available for electronics and shields */
        if (zone == ZONE_AMMO)
            return;
    }
    airequipSelectedZone = zone;

    /* update menu */
    AIM_AircraftEquipMenuUpdate();
    /* Check that the selected zone is OK */
    AIM_CheckAirequipSelectedZone(slot);

    AIM_UpdateItemDescription(false, true);
}

static void AIM_AircraftEquipAddItem_f (void)
{
    int zone;
    aircraftSlot_t* slot;
    aircraft_t* aircraft = nullptr;
    base_t* base = B_GetCurrentSelectedBase();

    zone = (airequipID == AC_ITEM_AMMO) ? 2 : 1;

    /* proceed only if an item has been selected */
    if (!aimSelectedTechnology)
        return;

    assert(base);
    aircraft = base->aircraftCurrent;
    assert(aircraft);
    base = aircraft->homebase;  /* we need to know where items will be removed */
    slot = AII_SelectAircraftSlot(aircraft, airequipID);
    if (slot == nullptr)
        return;

    /* the clicked button doesn't correspond to the selected zone */
    if (zone != airequipSelectedZone)
        return;

    /* check if the zone exists */
    if (zone >= ZONE_MAX)
        return;

    /* update the new item to slot */

    switch (zone) {
    case ZONE_MAIN:
        if (!slot->nextItem) {
            /* we add the weapon, shield, item if slot is free or the installation of current item just began */
            if (!slot->item || (slot->item && slot->installationTime == slot->item->craftitem.installationTime)) {
                AII_RemoveItemFromSlot(base, slot, false);
                AII_AddItemToSlot(base, aimSelectedTechnology, slot, false); /* Aircraft stats are updated below */
                AII_AutoAddAmmo(slot);
                break;
            } else if (slot->item == INVSH_GetItemByID(aimSelectedTechnology->provides)) {
                /* the added item is the same than the one in current slot */
                if (slot->installationTime == -slot->item->craftitem.installationTime) {
                    /* player changed his mind: he just want to re-add the item he just removed */
                    slot->installationTime = 0;
                    break;
                } else if (!slot->installationTime) {
                    /* player try to add a weapon he already have: just skip */
                    return;
                }
            } else {
                /* We start removing current item in slot, and the selected item will be installed afterwards */
                slot->installationTime = -slot->item->craftitem.installationTime;
                /* more below */
            }
        } else {
            /* remove weapon and ammo of next item */
            AII_RemoveNextItemFromSlot(base, slot, false);
            /* more below */
        }

        /* we change the weapon, shield, item, or base defence that will be installed AFTER the removal
         * of the one in the slot atm */
        AII_AddItemToSlot(base, aimSelectedTechnology, slot, true);
        AII_AutoAddAmmo(slot);
        break;
    case ZONE_AMMO:
        /* we can change ammo only if the selected item is an ammo (for weapon or base defence system) */
        if (airequipID >= AC_ITEM_AMMO) {
            AII_AddAmmoToSlot(base, aimSelectedTechnology, slot);
        }
        break;
    default:
        /* Zone higher than ZONE_AMMO shouldn't exist */
        return;
    }

    /* Update the values of aircraft stats (just in case an item has an installationTime of 0) */
    AII_UpdateAircraftStats(aircraft);

    AIM_AircraftEquipMenuUpdate();
}

static void AIM_AircraftEquipRemoveItem_f (void)
{
    int zone;
    aircraftSlot_t* slot;
    aircraft_t* aircraft = nullptr;
    base_t* base = B_GetCurrentSelectedBase();

    zone = (airequipID == AC_ITEM_AMMO) ? 2 : 1;

    assert(base);
    aircraft = base->aircraftCurrent;
    assert(aircraft);
    slot = AII_SelectAircraftSlot(aircraft, airequipID);

    /* no item in slot: nothing to remove */
    if (!slot->item)
        return;

    /* update the new item to slot */

    switch (zone) {
    case ZONE_MAIN:
        if (!slot->nextItem) {
            /* we change the weapon, shield, item, or base defence that is already in the slot */
            /* if the item has been installed since less than 1 hour, you don't need time to remove it */
            if (slot->installationTime < slot->item->craftitem.installationTime) {
                slot->installationTime = -slot->item->craftitem.installationTime;
                AII_RemoveItemFromSlot(base, slot, true); /* we remove only ammo, not item */
            } else {
                AII_RemoveItemFromSlot(base, slot, false); /* we remove weapon and ammo */
            }
            /* aircraft stats are updated below */
        } else {
            /* we change the weapon, shield, item, or base defence that will be installed AFTER the removal
             * of the one in the slot atm */
            AII_RemoveNextItemFromSlot(base, slot, false); /* we remove weapon and ammo */
            /* if you canceled next item for less than 1 hour, previous item is still functional */
            if (slot->installationTime == -slot->item->craftitem.installationTime) {
                slot->installationTime = 0;
            }
        }
        break;
    case ZONE_AMMO:
        /* we can change ammo only if the selected item is an ammo (for weapon or base defence system) */
        if (airequipID >= AC_ITEM_AMMO) {
            if (slot->nextAmmo)
                AII_RemoveNextItemFromSlot(base, slot, true);
            else
                AII_RemoveItemFromSlot(base, slot, true);
        }
        break;
    default:
        /* Zone higher than ZONE_AMMO shouldn't exist */
        return;
    }

    /* Update the values of aircraft stats */
    AII_UpdateAircraftStats(aircraft);

    AIM_AircraftEquipMenuUpdate();
}

static void AIM_AircraftEquipMenuClick_f (void)
{
    int techIdx;

    if (cgi->Cmd_Argc() < 2) {
        cgi->Com_Printf("Usage: %s <num>\n", cgi->Cmd_Argv(0));
        return;
    }

    /* Which tech? */
    techIdx = atoi(cgi->Cmd_Argv(1));
    aimSelectedTechnology = RS_GetTechByIDX(techIdx);
    AIM_UpdateItemDescription(true, false);
}

static void AIM_AircraftItemtypeByName_f (void)
{
    aircraftItemType_t i;
    const char* name;

    if (cgi->Cmd_Argc() != 2) {
        cgi->Com_Printf("Usage: %s <weapon|ammo|armour|item>\n", cgi->Cmd_Argv(0));
        return;
    }

    name = cgi->Cmd_Argv(1);

    if (Q_streq(name, "weapon"))
        i = AC_ITEM_WEAPON;
    else if (Q_streq(name, "ammo"))
        i = AC_ITEM_AMMO;
    else if (Q_streq(name, "armour"))
        i = AC_ITEM_SHIELD;
    else if (Q_streq(name, "item"))
        i = AC_ITEM_ELECTRONICS;
    else {
        cgi->Com_Printf("AIM_AircraftItemtypeByName_f: Invalid itemtype!\n");
        return;
    }

    airequipID = i;
    cgi->Cmd_ExecuteString("airequip_updatemenu %d", airequipID);
}

static const cmdList_t airequipCmds[] = {
    {"airequip_updatemenu", AIM_AircraftEquipMenuUpdate_f, "Init function for the aircraft equip menu"},
    {"airequip_selectcategory", AIM_AircraftItemtypeByName_f, "Select an item category and update the GUI"},
    {"airequip_list_click", AIM_AircraftEquipMenuClick_f, nullptr},
    {"airequip_slot_select", AIM_AircraftEquipSlotSelect_f, nullptr},
    {"airequip_add_item", AIM_AircraftEquipAddItem_f, "Add item to slot"},
    {"airequip_remove_item", AIM_AircraftEquipRemoveItem_f, "Remove item from slot"},
    {"airequip_zone_select", AIM_AircraftEquipZoneSelect_f, nullptr},
    {nullptr, nullptr, nullptr}
};
void AIM_InitCallbacks (void)
{
    cgi->Cmd_TableAddList(airequipCmds);
}

void AIM_ShutdownCallbacks (void)
{
    cgi->Cmd_TableRemoveList(airequipCmds);
}