From 31e1e2964ba41eb3ddb9d1927594053c34e3fa0d Mon Sep 17 00:00:00 2001
From: Arcanum <lubos.len@gmail.com>
Date: Mon, 22 May 2017 20:53:00 +0200
Subject: [PATCH] Add function to check if AI and condition from fatal

---
 addons/medical/ACE_Medical_StateMachine.hpp   |  9 ++++++-
 .../fnc_conditionFromFatalToCardiac.sqf       | 17 ++++++++++++
 addons/medical/functions/fnc_isAI.sqf         | 27 +++++++++++++++++++
 3 files changed, 52 insertions(+), 1 deletion(-)
 create mode 100644 addons/medical/functions/fnc_conditionFromFatalToCardiac.sqf
 create mode 100644 addons/medical/functions/fnc_isAI.sqf

diff --git a/addons/medical/ACE_Medical_StateMachine.hpp b/addons/medical/ACE_Medical_StateMachine.hpp
index 6d551aad0f..4151ef8311 100644
--- a/addons/medical/ACE_Medical_StateMachine.hpp
+++ b/addons/medical/ACE_Medical_StateMachine.hpp
@@ -61,7 +61,14 @@ class ACE_Medical_StateMachine {
         class SecondChance {
             events[] = {QGVAR(FatalInjuryInstantTransition)};
             targetState = "CardiacArrest";
-            condition = QUOTE(GVAR(fatalInjuryCondition) > 0);
+            condition = QUOTE(DFUNC(conditionFromFatalToCardiac));
+            onTransition = QUOTE(DFUNC(transitionSecondChance));
+        };
+        class SecondChanceAI {
+            events[] = {QGVAR(FatalInjuryInstantTransition)};
+            targetState = "CardiacArrest";
+            //conditionFromFatalToCardiac
+            condition = QUOTE(DFUNC(conditionFromFatalToCardiac));
             onTransition = QUOTE(DFUNC(transitionSecondChance));
         };
         class Death {
diff --git a/addons/medical/functions/fnc_conditionFromFatalToCardiac.sqf b/addons/medical/functions/fnc_conditionFromFatalToCardiac.sqf
new file mode 100644
index 0000000000..2c19cc926f
--- /dev/null
+++ b/addons/medical/functions/fnc_conditionFromFatalToCardiac.sqf
@@ -0,0 +1,17 @@
+/*
+ * Author: Arcanum
+ * Condition for trasfer from FatalInjury to Second chance.
+ *
+ * Arguments:
+ * 0: The Unit <OBJECT>
+ * 1: 1 for AI 0 for player <BOOL>
+ * 
+ * Return Value:
+ * Condition fo state machine <BOOL>
+ *
+ * Public: No
+ */
+#include "script_component.hpp"
+params ["_unit","_wantAI"];
+
+(GVAR(fatalInjuryConditionAI) > 0) && (QUOTE(_this call FUNC(isAi)) == _wantAI)
diff --git a/addons/medical/functions/fnc_isAI.sqf b/addons/medical/functions/fnc_isAI.sqf
new file mode 100644
index 0000000000..e30563c8fc
--- /dev/null
+++ b/addons/medical/functions/fnc_isAI.sqf
@@ -0,0 +1,27 @@
+/*
+ * Author: Arcanum
+ * Check if a unit is AI
+ *
+ * Arguments:
+ * 0: The Unit <OBJECT>
+ *
+ * ReturnValue:
+ * Is AI <BOOL>
+ *
+ * Example:
+ * [player] call ace_medical_fnc_isAI
+ *
+ * Public: Yes
+ */
+
+#include "script_component.hpp"
+
+params ["_unit"];
+private _return = 0;
+
+if (isPlayer _unit) then {
+  _return=0;
+else {
+  _return=1;
+};
+_return