Safemode - add public setWeaponSafety function

addons/safemode/XEH_PREP.hpp

@@ -3,3 +3,4 @@ PREP(lockSafety);
 PREP(playChangeFiremodeSound);
 PREP(setSafeModeVisual);
 PREP(unlockSafety);
+PREP(safeWeapon);

addons/safemode/functions/fn_safeWeapon.sqf

@@ -0,0 +1,33 @@
+/*
+ * Author: Brostrom.A
+ * Safe or unsafe the given weapon based on weapon state; locked or unlocked.
+ *
+ * Arguments:
+ * 0: Unit <OBJECT>
+ * 1: Weapon <STRING>
+ * 2: Muzzle <STRING>
+ * 3: State <BOOL>
+ *
+ * Return Value:
+ * None
+ *
+ * Example:
+ * [ACE_player, currentWeapon ACE_player, currentMuzzle Ace_player, true] call ace_safemode_fnc_safeWeapon
+ *
+ * Public: yes
+ */
+
+params [
+    ["_unit", objNull, [objNull]], 
+    ["_weapon", objNull, [objNull]],
+    ["_muzzle", objNull, [objNull]],
+    ["_state", true, [true]]
+];
+
+private _safedWeapons = _unit getVariable [QGVAR(safedWeapons), []];
+
+_weapon = configName (configFile >> "CfgWeapons" >> _weapon);
+
+if (((_state isEqualTo true) && ((_safedWeapons findIf {_weapon == _x}) == -1)) || ((_state isEqualTo false) && (!(_safedWeapons findIf {_weapon == _x}) == -1)))then {
+    [_unit, _weapon, _muzzle] call FUNC(lockSafety);
+};