Medical Treatement - Add Grave Digging To Body Bags

addons/medical_treatment/ACE_Medical_Treatment_Actions.hpp

@@ -256,7 +256,7 @@ class GVAR(actions) {
     class Grave: BodyBag {
         displayName = CSTRING(DigGrave);
         displayNameProgress = CSTRING(DiggingGrave);
-        icon = QPATHTOEF(medical_gui,ui\cross_grave.paa);
+        icon = QPATHTOEF(medical_gui,ui\grave.paa);
         treatmentTime = QGVAR(treatmentTimeGrave);
         condition = QFUNC(canDigGrave);
         callbackSuccess = QFUNC(placeInGrave);

addons/medical_treatment/CfgVehicles.hpp

@@ -29,6 +29,13 @@ class CfgVehicles {
                 statement = "";
                 icon = "\a3\ui_f\data\IGUI\Cfg\Actions\eject_ca.paa";
                 selection = "";
+                class GVAR(buryBodyBag) {
+                    displayName = CSTRING(DigGrave);
+                    distance = HANDCUFFS_DISTANCE;
+                    condition = QUOTE([_this#1] call FUNC(canDigGrave));
+                    statement = QUOTE(_this call FUNC(placeBodyBagInGrave));
+                    icon = QPATHTOEF(medical_gui,ui\grave.paa);
+                };
             };
         };
     };

addons/medical_treatment/XEH_postInit.sqf

@@ -67,28 +67,31 @@ if (["ace_trenches"] call EFUNC(common,isModLoaded)) then {
         private _checkHeadstoneAction = [
             QGVAR(checkHeadstone),
             LLSTRING(checkHeadstoneName),
-            QPATHTOEF(medical_gui,ui\cross_grave.paa),
+            QPATHTOEF(medical_gui,ui\grave.paa),
             {
                 [
                     [_target getVariable QGVAR(headstoneData)],
                     true
                 ] call CBA_fnc_notify;
             },
-            {!isNil {_target getVariable QGVAR(headstoneData)}},
-            {},
-            [],
-            [1.05, 0.02, 0.3] //position in centre of cross
+            {!isNil {_target getVariable QGVAR(headstoneData)}}
         ] call EFUNC(interact_menu,createAction);
 
-        ["Land_Grave_dirt_F", 0, [], _checkHeadstoneAction] call EFUNC(interact_menu,addActionToClass);
+        [missionNameSpace getVariable [QGVAR(graveClassname), "ACE_Grave"], 0, [], _checkHeadstoneAction] call EFUNC(interact_menu,addActionToClass);
     };
 
     if (isServer) then {
         ["ace_placedInBodyBag", {
             params ["_target", "_restingPlace"];
             TRACE_2("ace_placedInBodyBag eh",_target,_restingPlace);
 
-            private _targetName = [_target, false, true] call EFUNC(common,getName);
+            private _targetName = "";
+            if (_patient isKindOf "ACE_bodyBagObject") then {
+                _targetName = _target getVariable [QGVAR(headstoneData), ""];
+            } else {
+                _targetName = [_target, false, true] call EFUNC(common,getName);
+            };
+
             _restingPlace setVariable [QGVAR(headstoneData), _targetName, true];
         }] call CBA_fnc_addEventHandler;
     };

addons/medical_treatment/functions/fnc_placeBodyBagInGrave.sqf

@@ -0,0 +1,30 @@
+#include "..\script_component.hpp"
+/*
+ * Author: drofseh
+ * Places a body bag inside a grave.
+ *
+ * Arguments:
+ * 0: Medic <OBJECT>
+ * 1: Patient <OBJECT>
+ *
+ * Return Value:
+ * None
+ *
+ * Example:
+ * [cursorObject, player] call ace_medical_treatment_fnc_placeBodyBagInGrave
+ *
+ * Public: No
+ */
+
+params ["_bodybag"];
+TRACE_1("placeBodyBagInGrave",_bodybag);
+
+[
+    QGVAR(treatmentTimeGrave),
+    _this,
+    {
+        [[_this#1, _this#0], missionNameSpace getVariable [QGVAR(graveClassname), "ACE_Grave"], [0,0,0], missionNameSpace getVariable [QGVAR(graveRotation), 0]] call FUNC(placeInBodyBagOrGrave);
+    },
+    {},
+    LLSTRING(DiggingGrave)
+] call EFUNC(common,progressBar);

addons/medical_treatment/functions/fnc_placeInBodyBagOrGrave.sqf

@@ -34,10 +34,17 @@ if (alive _patient) then {
     [_patient, "buried_alive", _medic] call EFUNC(medical_status,setDead);
 };
 
-private _headPos = _patient modelToWorldVisual (_patient selectionPosition "head");
-private _spinePos = _patient modelToWorldVisual (_patient selectionPosition "Spine3");
-private _direction = (_headPos vectorFromTo _spinePos) call CBA_fnc_vectDir;
 private _position = getPosASL _patient;
+private _direction = 0;
+
+if (_patient isKindOf "ACE_bodyBagObject") then {
+    _direction getDir _patient;
+} else {
+    private _headPos = _patient modelToWorldVisual (_patient selectionPosition "head");
+    private _spinePos = _patient modelToWorldVisual (_patient selectionPosition "Spine3");
+    _direction = (_headPos vectorFromTo _spinePos) call CBA_fnc_vectDir;
+};
+
 // apply adjustments
 _position = _position vectorAdd _offset;
 _direction = _direction + _rotation;

addons/medical_treatment/functions/fnc_placeInGrave.sqf

@@ -28,4 +28,3 @@ private _graveClassname = missionNameSpace getVariable [QGVAR(graveClassname), "
 private _graveRotation = missionNameSpace getVariable [QGVAR(graveRotation), 90]; 
 
 [_this, _graveClassname, [0,0,0], _graveRotation] call FUNC(placeInBodyBagOrGrave);
-