Improve dragging getWeight function (#7261)

addons/dragging/functions/fnc_getWeight.sqf

@@ -1,45 +1,95 @@
 #include "script_component.hpp"
 /*
- * Author: L-H, edited by commy2, rewritten by joko // Jonas
- * Returns the weight of a crate.
+ * Author: L-H, edited by commy2, rewritten by joko // Jonas, re-rewritten by mharis001
+ * Returns the weight of the given object.
+ * Weight is calculated from the object's mass and its current inventory.
  *
  * Arguments:
- * 0: Crate to get weight of <OBJECT>
+ * 0: Object <OBJECT>
  *
  * Return Value:
- * Total Weight <NUMBER>
+ * Weight <NUMBER>
  *
  * Example:
- * [Crate1] call ace_dragging_fnc_getweight;
+ * [_object] call ace_dragging_fnc_getWeight
  *
  * Public: No
  */
 
 params ["_object"];
 
-// Initialize the total weight.
-private _totalWeight = 0;
+private _weight = 0;
+private _cfgWeapons   = configFile >> "CfgWeapons";
+private _cfgGlasses   = configFile >> "CfgGlasses";
+private _cfgVehicles  = configFile >> "CfgVehicles";
+private _cfgMagazines = configFile >> "CfgMagazines";
+
+// Add the masses of "regular" items in the object's inventory
+// Handle separating CfgGlasses items from this cargo array
+getItemCargo _object params ["_itemTypes", "_itemCounts"];
 
-// Cycle through all item types with their assigned config paths.
 {
-    _x params ["_items", "_getConfigCode"];
-    _items params ["_item", "_count"];
-    // Cycle through all items and read their mass out of the config.
+    private _itemConfig = if (isClass (_cfgGlasses >> _x)) then {
+        _cfgGlasses >> _x
+    } else {
+        _cfgWeapons >> _x >> "ItemInfo"
+    };
+
+    _weight = _weight + getNumber (_itemConfig >> "mass") * (_itemCounts select _forEachIndex);
+} forEach _itemTypes;
+
+// Add the masses of magazine items in the object's inventory
+getMagazineCargo _object params ["_magazineTypes", "_magazineCounts"];
+
 {
-        // Multiply mass with amount of items and add the mass to the total weight.
-        _totalWeight = _totalWeight + (getNumber ((call _getConfigCode) >> "mass") * (_count select _forEachIndex));
-    } forEach _item;
-    true
-} count [
-    //IGNORE_PRIVATE_WARNING ["_x"];
-    [getMagazineCargo _object, {configFile >> "CfgMagazines" >> _x}],
-    [getBackpackCargo _object, {configFile >> "CfgVehicles" >> _x}],
-    [getItemCargo _object, {configFile >> "CfgWeapons" >> _x >> "ItemInfo"}],
-    [getWeaponCargo _object, {configFile >> "CfgWeapons" >> _x >> "WeaponSlotsInfo"}]
-];
+    _weight = _weight + getNumber (_cfgMagazines >> _x >> "mass") * (_magazineCounts select _forEachIndex);
+} forEach _magazineTypes;
 
-// add Weight of create to totalWeight
-_totalWeight = _totalWeight + (getNumber (configFile >> "CfgVehicles" >> typeOf _object >> "mass"));
+// Add the masses of backpack items in the object's inventory
+getBackpackCargo _object params ["_backpackTypes", "_backpackCounts"];
 
-// Mass in Arma isn't an exact amount but rather a volume/weight value. This attempts to work around that by making it a usable value. (sort of).
-_totalWeight * 0.5
+{
+    _weight = _weight + getNumber (_cfgVehicles >> _x >> "mass") * (_backpackCounts select _forEachIndex);
+} forEach _backpackTypes;
+
+{
+    _x params ["_weapon", "_muzzle", "_pointer", "_optic", "_primaryMagazine", "_secondaryMagazine", "_bipod"];
+
+    // Add the weapon's mass
+    _weight = _weight + getNumber (_cfgWeapons >> _weapon >> "WeaponSlotsInfo" >> "mass");
+
+    // Add the masses of the weapon's attachments if they exist
+    {
+        if (_x != "") then {
+            _weight = _weight + getNumber (_cfgWeapons >> _x >> "ItemInfo" >> "mass");
+        };
+    } forEach [_muzzle, _pointer, _optic, _bipod];
+
+    // Add the masses of the weapon's magazines if they exist
+    {
+        _x params ["_magazine"];
+
+        if (!isNil "_magazine") then {
+            _weight = _weight + getNumber (_cfgMagazines >> _magazine >> "mass");
+        };
+    } forEach [_primaryMagazine, _secondaryMagazine];
+} forEach weaponsItemsCargo _object;
+
+// Add the mass of the object itself
+// The mass of sub-containers such as vests was added through the items cargo
+// The container object is generally of type SupplyX and has mass of zero
+_weight = _weight + getNumber (_cfgVehicles >> typeOf _object >> "mass");
+
+// Mass in Arma isn't an exact amount but rather a volume/weight value
+// This attempts to work around that by making it a usable value (sort of)
+// Note: this is done before the recursive calls to avoid reducing the weight multiple times
+_weight = _weight * 0.5;
+
+// Handle sub-containers within the object's inventory
+{
+    _x params ["", "_container"];
+
+    _weight = _weight + (_container call FUNC(getWeight));
+} forEach everyContainer _object;
+
+_weight

addons/dragging/script_component.hpp

@@ -2,7 +2,6 @@
 #define COMPONENT_BEAUTIFIED Dragging
 #include "\z\ace\addons\main\script_mod.hpp"
 
-// #define DEBUG_ENABLED_DRAGGING
 // #define DEBUG_MODE_FULL
 // #define DISABLE_COMPILE_CACHE
 // #define ENABLE_PERFORMANCE_COUNTERS