Merge pull request CBATeam#1330 from Dystopian/fnc_unitHasSpace

Add CBA_fnc_canAddItem
Origoadam · Oct 10, 2020 · 65e3698 · 65e3698
2 parents 578b409 + 13ca845
commit 65e3698
Show file tree

Hide file tree

Showing 2 changed files with 137 additions and 0 deletions.
diff --git a/addons/common/CfgFunctions.hpp b/addons/common/CfgFunctions.hpp
@@ -78,6 +78,7 @@ class CfgFunctions {
             PATHTO_FNC(addBinocularMagazine);
             PATHTO_FNC(removeBinocularMagazine);
             PATHTO_FNC(randomizeFacewear);
+            PATHTO_FNC(canAddItem);
         };
 
         class Cargo {

diff --git a/addons/common/fnc_canAddItem.sqf b/addons/common/fnc_canAddItem.sqf
@@ -0,0 +1,136 @@
+#include "script_component.hpp"
+/* ----------------------------------------------------------------------------
+Function: CBA_fnc_canAddItem
+
+Description:
+    Checks if unit has enough free space in inventory to store item.
+
+    Doesn't take current unit load into account unlike canAdd command.
+
+Parameters:
+    _unit          - Unit <OBJECT>
+    _item          - Item to store <STRING>
+    _count         - Item count <NUMBER> (Default: 1)
+    _checkUniform  - Check space in uniform <BOOLEAN> (Default: true)
+    _checkVest     - Check space in vest <BOOLEAN> (Default: true)
+    _checkBackpack - Check space in backpack <BOOLEAN> (Default: true)
+
+Returns:
+    True if unit has free space, false otherwise <BOOLEAN>
+
+Examples:
+    (begin example)
+        [player, "acc_flashlight"] call CBA_fnc_canAddItem
+        [player, "30Rnd_556x45_Stanag", 7, false, true, false] call CBA_fnc_canAddItem
+    (end)
+
+Author:
+    Dystopian
+---------------------------------------------------------------------------- */
+SCRIPT(canAddItem);
+
+params [
+    ["_unit", objNull, [objNull]],
+    ["_item", "", [""]],
+    ["_count", 1, [0]],
+    ["_checkUniform", true, [false]],
+    ["_checkVest", true, [false]],
+    ["_checkBackpack", true, [false]]
+];
+
+if (isNull _unit || {_item isEqualTo ""}) exitWith {false};
+
+#define TYPE_VEST 701
+#define TYPE_UNIFORM 801
+#define TYPE_BACKPACK 901
+
+if (isNil QGVAR(itemMassAllowedSlots)) then {
+    GVAR(itemMassAllowedSlots) = [] call CBA_fnc_createNamespace;
+};
+
+(GVAR(itemMassAllowedSlots) getVariable [_item, []]) params ["_mass", "_allowedSlots"];
+
+if (isNil "_mass") then {
+    _allowedSlots = [TYPE_UNIFORM, TYPE_VEST, TYPE_BACKPACK];
+    private _cfgWeaponsItem = configFile >> "CfgWeapons" >> _item;
+    private _cfgMagazinesItem = configFile >> "CfgMagazines" >> _item;
+    private _cfgGlassesItem = configFile >> "CfgGlasses" >> _item;
+    switch true do {
+        case (isClass _cfgWeaponsItem): {
+            private _cfgItemInfo = _cfgWeaponsItem >> "ItemInfo";
+            private _cfgWeaponSlotsInfo = _cfgWeaponsItem >> "WeaponSlotsInfo";
+            if (isNumber (_cfgItemInfo >> "mass")) then {
+                _mass = getNumber (_cfgItemInfo >> "mass");
+            } else {
+                _mass = getNumber (_cfgWeaponSlotsInfo >> "mass");
+            };
+
+            {
+                if (isArray _x) exitWith {
+                    _allowedSlots = getArray _x;
+                };
+            } forEach [
+                _cfgWeaponsItem >> "allowedSlots",
+                _cfgItemInfo >> "allowedSlots",
+                _cfgWeaponSlotsInfo >> "allowedSlots"
+            ];
+        };
+        case (isClass _cfgMagazinesItem): {
+            _mass = getNumber (_cfgMagazinesItem >> "mass");
+            private _cfgAllowedSlots = _cfgMagazinesItem >> "allowedSlots";
+            if (isArray _cfgAllowedSlots) then {
+                _allowedSlots = getArray _cfgAllowedSlots;
+            };
+        };
+        case (isClass _cfgGlassesItem): {
+            _mass = getNumber (_cfgGlassesItem >> "mass");
+        };
+        default {
+            _mass = -1;
+            _allowedSlots = [];
+        };
+    };
+    TRACE_3("caching",_item,_mass,_allowedSlots);
+    GVAR(itemMassAllowedSlots) setVariable [_item, [_mass, _allowedSlots]];
+};
+
+if (_mass == -1) exitWith {false}; // item doesn't exist
+
+if (
+    _checkUniform
+    && {TYPE_UNIFORM in _allowedSlots}
+    && {
+        _mass == 0
+        || {
+            // each time subtract whole number of items which can be put in container
+            _count = _count - floor (getContainerMaxLoad uniform _unit * (1 - loadUniform _unit) / _mass);
+            _count <= 0
+        }
+    }
+) exitWith {true};
+
+if (
+    _checkVest
+    && {TYPE_VEST in _allowedSlots}
+    && {
+        _mass == 0
+        || {
+            _count = _count - floor (getContainerMaxLoad vest _unit * (1 - loadVest _unit) / _mass);
+            _count <= 0
+        }
+    }
+) exitWith {true};
+
+if (
+    _checkBackpack
+    && {TYPE_BACKPACK in _allowedSlots}
+    && {
+        _mass == 0
+        || {
+            _count = _count - floor (getContainerMaxLoad backpack _unit * (1 - loadBackpack _unit) / _mass);
+            _count <= 0
+        }
+    }
+) exitWith {true};
+
+false