ui_node_container.cpp

Go to the documentation of this file.

/*
Copyright (C) 2002-2023 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 "../ui_main.h"
#include "../ui_parse.h"
#include "../ui_behaviour.h"
#include "../ui_actions.h"
#include "../ui_dragndrop.h"
#include "../ui_tooltip.h"
#include "../ui_nodes.h"
#include "../ui_input.h"
#include "../ui_render.h"
#include "../ui_lua.h"

#include "ui_node_model.h"
#include "ui_node_container.h"
#include "ui_node_abstractnode.h"

#include "../../client.h"
#include "../../renderer/r_draw.h"
#include "../../renderer/r_mesh.h"
#include "../../cgame/cl_game.h"
#include "../../battlescape/cl_actor.h"
#include "../../cl_inventory.h"

#include "../../../common/scripts_lua.h"

Inventory* ui_inventory = nullptr;

#define EXTRADATA_TYPE containerExtraData_t
#define EXTRADATA(node) UI_EXTRADATA(node, EXTRADATA_TYPE)
#define EXTRADATACONST(node) UI_EXTRADATACONST(node, EXTRADATA_TYPE)

static int dragInfoFromX = -1;
static int dragInfoFromY = -1;

static int dragInfoToX = -1;
static int dragInfoToY = -1;

static const Item* dragInfoIC;

static inline bool UI_IsScrollContainerNode (const uiNode_t* const node)
{
    return EXTRADATACONST(node).container && EXTRADATACONST(node).container->scroll;
}

void UI_ContainerNodeUpdateEquipment (Inventory* inv, const equipDef_t* ed)
{
    /* make a copy, because we are not really using these items,
     * but only putting them into a temp container */
    int* const numItems = Mem_Dup(int, ed->numItems, lengthof(ed->numItems));

    /* a 'tiny hack' to add the remaining equipment (not carried)
     * correctly into buy categories, reloading at the same time;
     * it is valid only due to the following property: */
    assert(MAX_CONTAINERS >= FILTER_AIRCRAFT);

    for (int i = 0; i < csi.numODs; i++) {
        const objDef_t* od = INVSH_GetItemByIDX(i);
        /* Don't allow to show unuseable items. */
        if (!GAME_ItemIsUseable(od))
            continue;

        while (numItems[i]) {
            const Item item(od);
            if (!cls.i.addToInventory(inv, &item, INVDEF(CID_EQUIP), NONE, NONE, 1)) {
                /* no space left in the inventory */
                break;
            }
            numItems[item.def()->idx]--;
        }
    }

    Mem_Free(numItems);

    /* First-time linking of ui_inventory. */
    if (ui_inventory && !ui_inventory->getEquipContainer()) {
        ui_inventory->setContainer(CID_EQUIP, inv->getEquipContainer());
    }
}

void UI_DrawItem (uiNode_t* node, const vec3_t org, const Item* item, int x, int y, const vec3_t scale, const vec4_t color)
{
    const objDef_t* od = item->def();
    vec4_t col;
    vec3_t origin;

    assert(od);
    assert(org[2] > -1000 && org[2] < 1000);    /*< prevent use of vec2_t for org */

    Vector4Copy(color, col);
    /* no ammo in this weapon - highlight this item */
    if (od->weapon && item->mustReload()) {
        col[1] *= 0.5;
        col[2] *= 0.5;
    }

    VectorCopy(org, origin);

    /* Calculate correct location of the image or the model (depends on rotation) */
    if (x >= 0 || y >= 0) {
        /* Add offset of location in container. */
        origin[0] += x * C_UNIT;
        origin[1] += y * C_UNIT;

        /* Add offset for item-center (depends on rotation). */
        if (item->rotated) {
            origin[0] += od->sy * C_UNIT / 2.0;
            origin[1] += od->sx * C_UNIT / 2.0;
        } else {
            origin[0] += od->sx * C_UNIT / 2.0;
            origin[1] += od->sy * C_UNIT / 2.0;
        }
    }

    /* don't handle the od->tech->image here - it's very ufopedia specific in most cases */
    if (od->image[0] != '\0') {
        const int imgWidth = od->sx * C_UNIT;
        const int imgHeight = od->sy * C_UNIT;
        origin[0] -= od->sx * C_UNIT / 2.0;
        origin[1] -= od->sy * C_UNIT / 2.0;

        /* Draw the image. */
        R_Color(color);
        UI_DrawNormImageByName(false, origin[0], origin[1], imgWidth, imgHeight, 0, 0, 0, 0, od->image);
        R_Color(nullptr);
    } else {
        uiModel_t* model = nullptr;
        const char* modelName = GAME_GetModelForItem(od, &model);

        /* no model definition in the tech struct, not in the fallback object definition */
        if (Q_strnull(modelName)) {
            Com_Printf("UI_DrawItem: No model given for item: '%s'\n", od->id);
            return;
        }

        if (model && node) {
            UI_DrawModelNode(node, modelName);
        } else {
            modelInfo_t mi;
            vec3_t angles = {-10, 160, 70};
            vec3_t size = {scale[0], scale[1], scale[2]};
            vec3_t center;

            if (item->rotated)
                angles[0] -= 90;

            if (od->scale)
                VectorScale(size, od->scale, size);

            VectorNegate(od->center, center);

            OBJZERO(mi);
            mi.origin = origin;
            mi.angles = angles;
            mi.center = center;
            mi.scale = size;
            mi.color = col;
            mi.name = modelName;

            /* draw the model */
            R_DrawModelDirect(&mi, nullptr, nullptr);
        }
    }
}

void UI_GetItemTooltip (const Item& item, char* tooltipText, size_t stringMaxLength)
{
    assert(item.def());

    if (item.getAmount() > 1)
        Com_sprintf(tooltipText, stringMaxLength, "%i x %s\n", item.getAmount(), _(item.def()->name));
    else
        Com_sprintf(tooltipText, stringMaxLength, "%s\n", _(item.def()->name));

    /* Only display further info if item.t is researched */
    if (GAME_ItemIsUseable(item.def())) {
        if (item.isWeapon()) {
            /* Get info about used ammo (if there is any) */
            if (item.def() == item.ammoDef()) {
                /* Item has no ammo but might have shot-count */
                if (item.getAmmoLeft()) {
                    Q_strcat(tooltipText, stringMaxLength, _("Ammo: %i\n"), item.getAmmoLeft());
                }
            } else if (item.ammoDef()) {
                /* Search for used ammo and display name + ammo count */
                Q_strcat(tooltipText, stringMaxLength, _("%s loaded\n"), _(item.ammoDef()->name));
                Q_strcat(tooltipText, stringMaxLength, _("Ammo: %i\n"), item.getAmmoLeft());
            }
        } else if (item.def()->numWeapons) {
            /* Check if this is a non-weapon and non-ammo item */
            if (!(item.def()->numWeapons == 1 && item.def()->weapons[0] == item.def())) {
                /* If it's ammo get the weapon names it can be used in */
                Q_strcat(tooltipText, stringMaxLength, _("Usable in:\n"));
                for (int i = 0; i < item.def()->numWeapons; i++) {
                    const objDef_t* weapon = item.def()->weapons[i];
                    if (GAME_ItemIsUseable(weapon)) {
                        Q_strcat(tooltipText, stringMaxLength, "* %s\n", _(weapon->name));
                    }
                }
            }
        }
    }
}

uiNode_t* UI_GetContainerNodeByContainerIDX (const uiNode_t* const parent, const int index)
{
    const invDef_t* const container = INVDEF(index);
    uiNode_t* containerNode = UI_GetNode(parent, container->name);
    return containerNode;
}

static void UI_DrawDisabled (const uiNode_t* node)
{
    const vec4_t color = { 0.3f, 0.3f, 0.3f, 0.7f };
    vec2_t nodepos;

    UI_GetNodeAbsPos(node, nodepos);
    UI_DrawFill(nodepos[0], nodepos[1], node->box.size[0], node->box.size[1], color);
}

static void UI_DrawFree (containerIndex_t container, const uiNode_t* node, int posx, int posy, int sizex, int sizey, bool showTUs)
{
    const vec4_t color = { 0.0f, 1.0f, 0.0f, 0.7f };
    invDef_t* inv = INVDEF(container);
    vec2_t nodepos;

    UI_GetNodeAbsPos(node, nodepos);
    UI_DrawFill(posx, posy, sizex, sizey, color);

    /* if showTUs is true (only the first time in none single containers)
     * and we are connected to a game */
    if (showTUs && CL_BattlescapeRunning()) {
        UI_DrawString("f_verysmall", ALIGN_UL, nodepos[0] + 3, nodepos[1] + 3,
            nodepos[0] + 3, node->box.size[0] - 6, 0,
            va(_("In: %i Out: %i"), inv->in, inv->out));
    }
}

static void UI_ContainerNodeDrawFreeSpace (uiNode_t* node, Inventory* inv)
{
    const objDef_t* od = UI_DNDGetItem()->def();    
    vec2_t nodepos;

    /* Draw only in dragging-mode and not for the equip-floor */
    assert(UI_DNDIsDragging());
    assert(inv);

    UI_GetNodeAbsPos(node, nodepos);
    /* if single container (hands, headgear) */
    if (EXTRADATA(node).container->single) {
        /* if container is free or the dragged-item is in it */
        if (UI_DNDIsSourceNode(node) || inv->canHoldItem(EXTRADATA(node).container, od, 0, 0, dragInfoIC))
            UI_DrawFree(EXTRADATA(node).container->id, node, nodepos[0], nodepos[1], node->box.size[0], node->box.size[1], true);
    } else {
        /* The shape of the free positions. */
        uint32_t free[SHAPE_BIG_MAX_HEIGHT];
        bool showTUs = true;

        OBJZERO(free);

        for (int y = 0; y < SHAPE_BIG_MAX_HEIGHT; y++) {
            for (int x = 0; x < SHAPE_BIG_MAX_WIDTH; x++) {
                /* Check if the current position is usable (topleft of the item). */

                /* Add '1's to each position the item is 'blocking'. */
                const int checkedTo = inv->canHoldItem(EXTRADATA(node).container, od, x, y, dragInfoIC);
                if (checkedTo & INV_FITS)               /* Item can be placed normally. */
                    INVSH_MergeShapes(free, od->shape, x, y);
                if (checkedTo & INV_FITS_ONLY_ROTATED)  /* Item can be placed rotated. */
                    INVSH_MergeShapes(free, od->getShapeRotated(), x, y);

                /* Only draw on existing positions. */
                if (INVSH_CheckShape(EXTRADATA(node).container->shape, x, y)) {
                    if (INVSH_CheckShape(free, x, y)) {
                        UI_DrawFree(EXTRADATA(node).container->id, node, nodepos[0] + x * C_UNIT, nodepos[1] + y * C_UNIT, C_UNIT, C_UNIT, showTUs);
                        showTUs = false;
                    }
                }
            }
        }
    }
}

void uiContainerNode::onLoaded (uiNode_t* const node)
{
    const char* name;
    const invDef_t* container;

    name = node->name;
    if (Q_strstart(node->name, "equip_"))
        name = "equip";

    container = INVSH_GetInventoryDefinitionByID(name);
    if (container == nullptr)
        return;

    EXTRADATA(node).container = container;

    if (UI_IsScrollContainerNode(node)) {
        /* No need to compute the size, the script provide it */
    } else {
        int i, j;
        /* Start on the last bit of the shape mask. */
        for (i = SHAPE_BIG_MAX_WIDTH - 1; i >= 0; i--) {
            for (j = 0; j < SHAPE_BIG_MAX_HEIGHT; j++)
                if (container->shape[j] & (1 << i))
                    break;
            if (j < SHAPE_BIG_MAX_HEIGHT)
                break;
        }
        node->box.size[0] = C_UNIT * (i + 1) + 0.01;

        /* start on the lower row of the shape mask */
        for (i = SHAPE_BIG_MAX_HEIGHT - 1; i >= 0; i--)
            if (container->shape[i] & ~0x0)
                break;
        node->box.size[1] = C_UNIT * (i + 1) + 0.01;
    }
}

static const vec3_t scale = {3.5, 3.5, 3.5};
static const vec4_t colorDefault = {1, 1, 1, 1};
static const vec4_t colorLoadable = {0.5, 1, 0.5, 1};
static const vec4_t colorDisabled = {0.5, 0.5, 0.5, 1};
static const vec4_t colorDisabledLoadable = {0.5, 0.25, 0.25, 1.0};
static const vec4_t colorPreview = { 0.5, 0.5, 1, 1 };  
static void UI_ContainerNodeDrawSingle (uiNode_t* node, const objDef_t* highlightType)
{
    vec4_t color;
    vec3_t pos;

    UI_GetNodeAbsPos(node, pos);
    pos[0] += node->box.size[0] / 2.0;
    pos[1] += node->box.size[1] / 2.0;
    pos[2] = 0;

    /* Single item container (special case for left hand). */
    const invDef_t* contType = EXTRADATA(node).container;
    if (contType->isLeftDef() && !ui_inventory->getLeftHandContainer()) {
        if (ui_inventory->getRightHandContainer()) {
            const Item* item = ui_inventory->getRightHandContainer();
            assert(item);
            assert(item->def());

            if (item->isHeldTwoHanded()) {
                if (highlightType && highlightType->isLoadableInWeapon(item->def()))
                    memcpy(color, colorLoadable, sizeof(vec4_t));
                else
                    memcpy(color, colorDefault, sizeof(vec4_t));
                color[3] = 0.5;
                UI_DrawItem(node, pos, item, -1, -1, scale, color);
            }
        }
    } else if (ui_inventory->getContainer2(contType->id)) {
        bool disabled = false;
        const Item* item;

        if (ui_inventory->getRightHandContainer()) {
            item = ui_inventory->getRightHandContainer();
            /* If there is a weapon in the right hand that needs two hands to shoot it
             * and there is a weapon in the left, then draw a disabled marker for the
             * fireTwoHanded weapon. */
            assert(item);
            assert(item->def());
            if (contType->isRightDef() && item->def()->fireTwoHanded && ui_inventory->getLeftHandContainer()) {
                disabled = true;
                UI_DrawDisabled(node);
            }
        }

        item = ui_inventory->getContainer2(contType->id);
        assert(item);
        assert(item->def());
        if (highlightType && highlightType->isLoadableInWeapon(item->def())) {
            if (disabled)
                Vector4Copy(colorDisabledLoadable, color);
            else
                Vector4Copy(colorLoadable, color);
        } else {
            if (disabled)
                Vector4Copy(colorDisabled, color);
            else
                Vector4Copy(colorDefault, color);
        }
        if (disabled)
            color[3] = 0.5;
        UI_DrawItem(node, pos, item, -1, -1, scale, color);
    }
}

static void UI_ContainerNodeDrawGrid (uiNode_t* node, const objDef_t* highlightType)
{
    vec3_t pos;

    UI_GetNodeAbsPos(node, pos);
    pos[2] = 0;

    const Container& cont = ui_inventory->getContainer(EXTRADATA(node).container->id);
    Item* item = nullptr;
    while ((item = cont.getNextItem(item))) {
        assert(item->def());
        if (highlightType && highlightType->isLoadableInWeapon(item->def()))
            UI_DrawItem(node, pos, item, item->getX(), item->getY(), scale, colorLoadable);
        else
            UI_DrawItem(node, pos, item, item->getX(), item->getY(), scale, colorDefault);
    }
}

static void UI_ContainerNodeDrawDropPreview (uiNode_t* target)
{
    Item previewItem;
    int checkedTo;
    vec3_t origine;

    /* no preview into scrollable list */
    if (UI_IsScrollContainerNode(target))
        return;

    /* copy the DND item to not change the original one */
    previewItem = *UI_DNDGetItem();
    previewItem.rotated = false;
    checkedTo = ui_inventory->canHoldItem(EXTRADATA(target).container, previewItem.def(), dragInfoToX, dragInfoToY, dragInfoIC);
    switch (checkedTo) {
    case INV_DOES_NOT_FIT:
        return;
    case INV_FITS:
        break;
    case INV_FITS_BOTH:
        if (!Key_IsDown(K_SHIFT) || CL_BattlescapeRunning())
            break;
        /* else fits rotated, fall through */
    case INV_FITS_ONLY_ROTATED:
        previewItem.rotated = true;
    }

    /* Hack, no preview for armour, we don't want it out of the armour container (and armour container is not visible) */
    if (previewItem.isArmour())
        return;

    UI_GetNodeAbsPos(target, origine);
    origine[2] = -40;

    /* Get center of single container for placement of preview item */
    if (EXTRADATA(target).container->single) {
        origine[0] += target->box.size[0] / 2.0;
        origine[1] += target->box.size[1] / 2.0;
    /* This is a "grid" container - we need to calculate the item-position
     * (on the screen) from stored placement in the container and the
     * calculated rotation info. */
    } else {
        if (previewItem.rotated) {
            origine[0] += (dragInfoToX + previewItem.def()->sy / 2.0) * C_UNIT;
            origine[1] += (dragInfoToY + previewItem.def()->sx / 2.0) * C_UNIT;
        } else {
            origine[0] += (dragInfoToX + previewItem.def()->sx / 2.0) * C_UNIT;
            origine[1] += (dragInfoToY + previewItem.def()->sy / 2.0) * C_UNIT;
        }
    }

    UI_DrawItem(nullptr, origine, &previewItem, -1, -1, scale, colorPreview);
}

void uiContainerNode::draw (uiNode_t* node)
{
    const objDef_t* highlightType = nullptr;

    if (!EXTRADATA(node).container)
        return;
    if (!ui_inventory)
        return;
    /* is container invisible */
    if (node->color[3] < 0.001)
        return;

    /* Highlight weapons that the dragged ammo (if it is one) can be loaded into. */
    if (UI_DNDIsDragging() && UI_DNDGetType() == DND_ITEM) {
        highlightType = UI_DNDGetItem()->def();
    }

    if (EXTRADATA(node).container->single) {
        UI_ContainerNodeDrawSingle(node, highlightType);
    } else {
        if (UI_IsScrollContainerNode(node)) {
            assert(false);
        } else {
            UI_ContainerNodeDrawGrid(node, highlightType);
        }
    }

    /* Draw free space if dragging - but not for CID_EQUIP */
    if (UI_DNDIsDragging() && EXTRADATA(node).container->id != CID_EQUIP)
        UI_ContainerNodeDrawFreeSpace(node, ui_inventory);

    if (UI_DNDIsTargetNode(node))
        UI_ContainerNodeDrawDropPreview(node);
}

static Item* UI_ContainerNodeGetItemAtPosition (const uiNode_t* const node, int mouseX, int mouseY, int* contX = nullptr, int* contY = nullptr)
{
    if (!ui_inventory)
        return nullptr;

    /* Get coordinates inside a scrollable container (if it is one). */
    if (UI_IsScrollContainerNode(node)) {
        Sys_Error("UI_ContainerNodeGetItemAtPosition is not usable for scrollable containers!");
    }

    vec2_t nodepos;
    UI_GetNodeAbsPos(node, nodepos);
    /* Normalize screen coordinates to container coordinates. */
    const int fromX = (int) (mouseX - nodepos[0]) / C_UNIT;
    const int fromY = (int) (mouseY - nodepos[1]) / C_UNIT;
    if (contX)
        *contX = fromX;
    if (contY)
        *contY = fromY;

    Item* result = ui_inventory->getItemAtPos(EXTRADATACONST(node).container, fromX, fromY);
    return result;
}

void uiContainerNode::drawTooltip (const uiNode_t* node, int x, int y) const
{
    /* Find out where the mouse is. */
    const Item* itemHover = UI_ContainerNodeGetItemAtPosition(node, x, y);

    if (itemHover) {
        static char tooltiptext[MAX_VAR * 2];
        const int itemToolTipWidth = 250;

        /* Get name and info about item */
        UI_GetItemTooltip(*itemHover, tooltiptext, sizeof(tooltiptext));
#ifdef DEBUG
        /* Display stored container-coordinates of the item. */
        Q_strcat(tooltiptext, sizeof(tooltiptext), "\n%i/%i", itemHover->getX(), itemHover->getY());
#endif
        UI_DrawTooltip(tooltiptext, x, y, itemToolTipWidth);
    }
}

static bool UI_ContainerNodeAddItem (const invDef_t* container, Item* ic, containerIndex_t containerID, Item** icp)
{
    int px, py;
    const invDef_t* target = INVDEF(containerID);
    ui_inventory->findSpace(target, ic, &px, &py, nullptr);
    return INV_MoveItem(ui_inventory, target, px, py, container, ic, icp);
}

void UI_ContainerNodeAutoPlaceItem (uiNode_t* node, Item* ic)
{
    containerIndex_t target;
    bool ammoChanged = false;
    const invDef_t* container = EXTRADATA(node).container;

    /* Right click: automatic item assignment/removal. */
    if (container->id != CID_EQUIP) {
        target = CID_EQUIP;
        if (ic->ammoDef()) {
            /* Remove ammo on removing weapon from a soldier */
            ammoChanged = INV_UnloadWeapon(ic, ui_inventory, INVDEF(target));
        }
        if (!ammoChanged && !ic->ammoDef()) {
            /* Move back to CID_EQUIP (ground, floor) container. */
            INV_MoveItem(ui_inventory, INVDEF(target), NONE, NONE, container, ic, nullptr);
        }
    } else {
        bool packed = false;
        assert(ic->def());
        /* armour can only have one target */
        if (ic->isArmour()) {
            target = CID_ARMOUR;
            packed = INV_MoveItem(ui_inventory, INVDEF(target), 0, 0, container, ic, nullptr);
        /* ammo or item */
        } else if (ic->def()->isAmmo()) {
            /* Finally try left and right hand. There is no other place to put it now. */
            const containerIndex_t idxArray[] = { CID_BELT, CID_HOLSTER, CID_BACKPACK, CID_LEFT, CID_RIGHT };
            const size_t size = lengthof(idxArray);
            for (int i = 0; i < size; i++) {
                target = idxArray[i];
                packed = UI_ContainerNodeAddItem(container, ic, target, nullptr);
                if (packed)
                    break;
            }
        } else if (ic->def()->headgear) {
            target = CID_HEADGEAR;
            packed = UI_ContainerNodeAddItem(container, ic, target, nullptr);
        } else if (ic->def()->implant) {
            target = CID_IMPLANT;
            packed = UI_ContainerNodeAddItem(container, ic, target, nullptr);
        } else {
            /* left and right are single containers, but this might change - it's cleaner to check
             * for available space here, too */
            const containerIndex_t idxArray[] = { CID_RIGHT, CID_BELT, CID_HOLSTER, CID_BACKPACK, CID_LEFT };
            const size_t size = lengthof(idxArray);
            Item* tItem = nullptr;
            for (int i = 0; i < size; i++) {
                target = idxArray[i];
                packed = UI_ContainerNodeAddItem(container, ic, target, &tItem);
                if (packed) {
                    if ((ic->isWeapon() && !ic->getAmmoLeft()) || ic->def()->weapons[0])
                        ammoChanged = INV_LoadWeapon(tItem, ui_inventory, container, INVDEF(target));
                    break;
                }
            }
        }
        /* no need to continue here - placement wasn't successful at all */
        if (!packed)
            return;
    }

    EXTRADATA(node).lastSelectedId = ic->def()->idx;
    if (EXTRADATA(node).onSelect) {
        UI_ExecuteEventActions(node, EXTRADATA(node).onSelect);
    }
    if (EXTRADATA(node).lua_onSelect != LUA_NOREF) {
        UI_ExecuteLuaEventScript(node, EXTRADATA(node).lua_onSelect);
    }
    /* Run onChange events */
    uiNode_t* targetNode = UI_GetContainerNodeByContainerIDX(node->parent, target);
    if (node->onChange) {
        UI_ExecuteEventActions(node, node->onChange);
    }
    if (node->lua_onChange != LUA_NOREF) {
        UI_ExecuteLuaEventScript(node, node->lua_onChange);
    }
    if (targetNode != nullptr && node != targetNode) {
        if (targetNode->onChange) {
            UI_ExecuteEventActions(targetNode, targetNode->onChange);
        }
        if (targetNode->lua_onChange != LUA_NOREF) {
            UI_ExecuteLuaEventScript(targetNode, targetNode->lua_onChange);
        }
    }
    /* Also call onChange for equip_ammo if ammo moved
     * Maybe there's a better way to do this? */
    if (ic->def()->isAmmo() || ammoChanged) {
        uiNode_t* ammoNode = UI_GetNode(node->root, "equip_ammo");
        if (ammoNode != nullptr && node != ammoNode) {
            if (ammoNode->onChange) {
                UI_ExecuteEventActions(ammoNode, ammoNode->onChange);
            }
            if (ammoNode->lua_onChange != LUA_NOREF) {
                UI_ExecuteLuaEventScript(ammoNode, ammoNode->lua_onChange);
            }
        }
    }
}

static void UI_ContainerNodeAutoPlace (uiNode_t* node, int mouseX, int mouseY)
{
    if (!ui_inventory)
        return;

    /* don't allow this in tactical missions */
    if (CL_BattlescapeRunning())
        return;

    const int sel = cl_selected->integer;
    if (sel < 0)
        return;

    assert(EXTRADATA(node).container);

    Item* ic = UI_ContainerNodeGetItemAtPosition(node, mouseX, mouseY);
    if (!ic)
        return;
    UI_ContainerNodeAutoPlaceItem(node, ic);
}

static int oldMouseX = 0;
static int oldMouseY = 0;

void uiContainerNode::onCapturedMouseMove (uiNode_t* node, int x, int y)
{
    const int delta = abs(oldMouseX - x) + abs(oldMouseY - y);
    if (delta > 15) {
        UI_DNDDragItem(node, dragInfoIC);
        UI_MouseRelease();
    }
}

void uiContainerNode::onMouseDown (uiNode_t* node, int x, int y, int button)
{
    switch (button) {
    case K_MOUSE1:
    {
        /* start drag and drop */
        int fromX, fromY;
        dragInfoIC = UI_ContainerNodeGetItemAtPosition(node, x, y, &fromX, &fromY);
        if (dragInfoIC) {
            dragInfoFromX = fromX;
            dragInfoFromY = fromY;
            oldMouseX = x;
            oldMouseY = y;
            UI_SetMouseCapture(node);
            EXTRADATA(node).lastSelectedId = dragInfoIC->def()->idx;
            if (EXTRADATA(node).onSelect) {
                UI_ExecuteEventActions(node, EXTRADATA(node).onSelect);
            }
            if (EXTRADATA(node).lua_onSelect != LUA_NOREF) {
                UI_ExecuteLuaEventScript(node, EXTRADATA(node).lua_onSelect);
            }
        }
        break;
    }
    case K_MOUSE2:
        if (UI_DNDIsDragging()) {
            UI_DNDAbort();
        } else {
            /* auto place */
            UI_ContainerNodeAutoPlace(node, x, y);
        }
        break;
    default:
        break;
    }
}

void uiContainerNode::onMouseUp (uiNode_t* node, int x, int y, int button)
{
    if (button != K_MOUSE1)
        return;
    if (UI_GetMouseCapture() == node) {
        UI_MouseRelease();
    }
    if (UI_DNDIsDragging()) {
        UI_DNDDrop();
    }
}

void uiContainerNode::initNode (uiNode_t* node) {
    /* initialize lua events */
    uiLocatedNode::initNode(node);
    EXTRADATA(node).lua_onSelect = LUA_NOREF;
}

void uiContainerNode::onLoading (uiNode_t* node)
{
    EXTRADATA(node).container = nullptr;
    node->color[3] = 1.0;
}

bool uiContainerNode::onDndEnter (uiNode_t* target)
{
    /* accept items only, if we have a container */
    return UI_DNDGetType() == DND_ITEM && EXTRADATA(target).container && (!UI_IsScrollContainerNode(target) || UI_DNDGetSourceNode() != target);
}

bool uiContainerNode::onDndMove (uiNode_t* target, int x, int y)
{
    vec2_t nodepos;
    bool exists;
    int itemX = 0;
    int itemY = 0;
    Item* dragItem = UI_DNDGetItem();

    /* we already check it when the node accept the DND */
    assert(EXTRADATA(target).container);
    const invDef_t* containerType = EXTRADATA(target).container;

    UI_GetNodeAbsPos(target, nodepos);

    if (dragItem->def()) {
        itemX = C_UNIT * dragItem->def()->sx / 2;   /* Half item-width. */
        itemY = C_UNIT * dragItem->def()->sy / 2;   /* Half item-height. */

        /* Place relative center in the middle of the square. */
        itemX -= C_UNIT / 2;
        itemY -= C_UNIT / 2;
    }

    dragInfoToX = (mousePosX - nodepos[0] - itemX) / C_UNIT;
    dragInfoToY = (mousePosY - nodepos[1] - itemY) / C_UNIT;

    /* Check if the items already exists in the container. i.e. there is already at least one item. */
    exists = false;
    if ((containerType->isFloorDef() || containerType->isEquipDef())
        && (dragInfoToX < 0 || dragInfoToY < 0 || dragInfoToX >= SHAPE_BIG_MAX_WIDTH || dragInfoToY >= SHAPE_BIG_MAX_HEIGHT)
        && ui_inventory->containsItem(containerType->id, dragItem)) {
        exists = true;
    }

    /* Search for a suitable position to render the item at if
     * the container is "single", the cursor is out of bound of the container. */
    if (!exists && dragItem->def() && (containerType->single
        || dragInfoToX < 0 || dragInfoToY < 0
        || dragInfoToX >= SHAPE_BIG_MAX_WIDTH || dragInfoToY >= SHAPE_BIG_MAX_HEIGHT)) {
#if 0
/* ... or there is something in the way. */
/* We would need to check for weapon/ammo as well here, otherwise a preview would be drawn as well when hovering over the correct weapon to reload. */
        || (ui_inventory->canHoldItem(containerType, dragItem->item, dragInfoToX, dragInfoToY) == INV_DOES_NOT_FIT)) {
#endif
        ui_inventory->findSpace(containerType, dragItem, &dragInfoToX, &dragInfoToY, dragInfoIC);
    }

    /* we can drag every thing */
    if (UI_IsScrollContainerNode(target)) {
        return true;
    }

    {
        Item* fItem;

        /* is there empty slot? */
        const int checkedTo = ui_inventory->canHoldItem(containerType, dragItem->def(), dragInfoToX, dragInfoToY, dragInfoIC);
        if (checkedTo != INV_DOES_NOT_FIT)
            return true;

        /* can we equip dragging item into the target item? */
        fItem = ui_inventory->getItemAtPos(containerType, dragInfoToX, dragInfoToY);
        if (!fItem)
            return false;
        if (containerType->single)
            return true;
        return dragItem->def()->isLoadableInWeapon(fItem->def());
    }
}

void uiContainerNode::onDndLeave (uiNode_t* node)
{
    dragInfoToX = -1;
    dragInfoToY = -1;
}

bool uiContainerNode::onDndFinished (uiNode_t* source, bool isDropped)
{
    Item* dragItem = UI_DNDGetItem();
    const invDef_t* sourceContainer = EXTRADATACONST(source).container;

    /* if the target can't finalize the DND we stop */
    if (!isDropped) {
        return false;
    }

    assert(sourceContainer);

    /* on tactical mission */
    if (CL_BattlescapeRunning()) {
        const uiNode_t* target = UI_DNDGetTargetNode();
        const invDef_t* targetContainer = EXTRADATACONST(target).container;
        assert(targetContainer);
        CL_ActorInvMove(selActor, sourceContainer->id, dragInfoFromX, dragInfoFromY,
                targetContainer->id, dragInfoToX, dragInfoToY);
    } else {
        uiNode_t* target = UI_DNDGetTargetNode();
        if (target) {
            Item* fItem;
            Item* tItem;
            bool moved = false;
            const invDef_t* targetContainer = EXTRADATACONST(target).container;
            assert(targetContainer);
            if (UI_IsScrollContainerNode(source)) {
                fItem = INV_SearchInInventoryWithFilter(ui_inventory, sourceContainer, dragItem->def(), MAX_FILTERTYPES);
            } else
                fItem = ui_inventory->getItemAtPos(sourceContainer, dragInfoFromX, dragInfoFromY);
            assert(fItem);

            /* Remove ammo on removing weapon from a soldier */
            if (UI_IsScrollContainerNode(target) && fItem->ammoDef())
                INV_UnloadWeapon(fItem, ui_inventory, targetContainer);

            /* Save rotation: in case the move fails we don't want the item overlapping other items or
             * ending partially out of the container due to changed rotation */
            const bool rotated = fItem->rotated;
            /* rotate on Shift */
            fItem->rotated = Key_IsDown(K_SHIFT) && !CL_BattlescapeRunning();

            /* move the item */
            moved = INV_MoveItem(ui_inventory, targetContainer, dragInfoToX, dragInfoToY, sourceContainer, fItem, &tItem);

            /* No need to continue move wasn't successful */
            if (!moved) {
                fItem->rotated = rotated;
                return false;
            }

            /* Add ammo on adding weapon to a soldier */
            if (UI_IsScrollContainerNode(source) && ((fItem->isWeapon() && !fItem->getAmmoLeft())
                    || (fItem->def()->weapons[0] && !fItem->def()->isAmmo())))
                INV_LoadWeapon(tItem, ui_inventory, sourceContainer, targetContainer);

            /* Run onChange events */
            if (source->onChange) {
                UI_ExecuteEventActions(source, source->onChange);
            }
            if (source->lua_onChange != LUA_NOREF) {
                UI_ExecuteLuaEventScript(source, source->lua_onChange);
            }

            if (source != target) {
                if (target->onChange) {
                    UI_ExecuteEventActions(target, target->onChange);
                }
                if (target->lua_onChange != LUA_NOREF) {
                    UI_ExecuteLuaEventScript(target, target->lua_onChange);
                }
            }
        }
    }

    dragInfoFromX = -1;
    dragInfoFromY = -1;
    return true;
}

void UI_RegisterContainerNode (uiBehaviour_t* behaviour)
{
    behaviour->name = "container";
    behaviour->manager = UINodePtr(new uiContainerNode());
    behaviour->extraDataSize = sizeof(EXTRADATA_TYPE);
    behaviour->lua_SWIG_typeinfo = UI_SWIG_TypeQuery("uiContainerNode_t *");

    /* Callback value set before calling onSelect. It is used to know the item selected */
    UI_RegisterExtradataNodeProperty(behaviour, "lastselectedid", V_INT, containerExtraData_t, lastSelectedId);
    /* Callback event called when the user select an item */
    UI_RegisterExtradataNodeProperty(behaviour, "onselect", V_UI_ACTION, containerExtraData_t, onSelect);
}