Medical - Add AI Unconsciousness exception (when using zeus module)

AUTHORS.txt

@@ -175,6 +175,7 @@ VyMajoris(W-Cephei)<vycanismajoriscsa@gmail.com>
 Walthzer <walthzer.69@gmail.com>
 Winter <simon@agius-muscat.net>
 wizpig64
+YetheSamartaka
 xrufix
 Zakant <Zakant@gmx.de>
-zGuba
+zGuba

addons/medical_statemachine/Statemachine.hpp

@@ -46,7 +46,7 @@ class ACE_Medical_StateMachine {
         onStateEntered = QFUNC(enteredStateUnconscious);
         class DeathAI {
             targetState = "Dead";
-            condition = QUOTE(!GVAR(AIUnconsciousness) && {!isPlayer _this});
+            condition = QUOTE(!(_this call FUNC(AIUnconsciousException) || GVAR(AIUnconsciousness)) && {!isPlayer _this});
         };
         class WakeUp {
             targetState = "Injured";

addons/medical_statemachine/XEH_PREP.hpp

@@ -13,3 +13,4 @@ PREP(leftStateCardiacArrest);
 PREP(localityChangedEH);
 PREP(resetStateDefault);
 PREP(transitionSecondChance);
+PREP(AIUnconsciousException);

addons/medical_statemachine/functions/fnc_AIUnconsciousException.sqf

@@ -0,0 +1,25 @@
+#include "script_component.hpp"
+/*
+ * Author: YetheSamartaka
+ * Checks for AI Unconsciousness Exception variable
+ *
+ * Arguments:
+ * 0: The Unit <OBJECT>
+ *
+ * Return Value:
+ * Has Unconsciousness Exception variable <BOOL>
+ *
+ * Example:
+ * [_unit] call ace_medical_statemachine_fnc_AIUnconsciousException
+ *
+ * Public: No
+ */
+
+params ["_unit"];
+
+// If the unit died the loop is finished
+if (!alive _unit || {!local _unit}) exitWith {};
+
+if (_unit getVariable ["ACE_AI_Unconsciousness_ExceptionVar", false]) exitWith { true };
+
+false

addons/medical_statemachine/initSettings.sqf

@@ -33,6 +33,15 @@
     true
 ] call CBA_fnc_addSetting;
 
+[
+    QGVAR(AIUnconsciousness_Exception),
+    "CHECKBOX",
+    [LSTRING(AIUnconsciousness_Exception_DisplayName), LSTRING(AIUnconsciousness_Exception_Description)],
+    [ELSTRING(medical,Category), LSTRING(SubCategory)],
+    true,
+    true
+] call CBA_fnc_addSetting;
+
 [
     QGVAR(cardiacArrestTime),
     "TIME",

addons/medical_statemachine/stringtable.xml

@@ -184,5 +184,13 @@
             <Chinesesimp>控制单位是否会在心脏骤停时间耗完之前因失血过多而死亡。</Chinesesimp>
             <Korean>지정한 심정지 시간이 다 되기전에 출혈로 인해 사망할 수 있는지를 결정합니다.</Korean>
         </Key>
+        <Key ID="STR_ACE_Medical_Statemachine_AIUnconsciousness_Exception_DisplayName">
+            <English>AI Unconsciousness exception</English>
+            <Czech>Výjimka pro AI bezvědomí</Czech>
+        </Key>
+        <Key ID="STR_ACE_Medical_Statemachine_AIUnconsciousness_Exception_Description">
+            <English>If enabled, it allows you to put AI into unconsciousness via Zeus Module even though the AI unconsciousness is disabled.</English>
+            <Czech>Pokud povoleno, povolí vám uvrhnout AI do bezvědomí pomocí Zeus modulu i přes to, že bezvědomí pro AI je zakázáno.</Czech>
+        </Key>
     </Package>
 </Project>

addons/zeus/functions/fnc_moduleUnconscious.sqf

@@ -38,6 +38,12 @@ if (isNil QEFUNC(medical,setUnconscious)) then {
                 [LSTRING(OnlyAlive)] call FUNC(showMessage);
             } else {
                 private _unconscious = GETVAR(_unit,ACE_isUnconscious,false);
+                if (ace_medical_statemachine_AIUnconsciousness_Exception) then {
+                _unit setVariable ["ACE_AI_Unconsciousness_ExceptionVar", true, true];
+                    if (_unconscious) then {
+                    _unit setVariable ["ACE_AI_Unconsciousness_ExceptionVar", false, true];
+                    };
+                };
                 // Function handles locality for me
                 [_unit, !_unconscious, 10e10] call EFUNC(medical,setUnconscious);
             };