Core/SmartAI: Various fixes and extensions for smart scripts:

Port From (https://github.com/TrinityCore/TrinityCore/commit/a32d5cfa1762ae1158ee2f40b4c9b36e5b41913a)
2019-08-30 12:00:42 -04:00
parent b85860dd78
commit 1657f720dc
15 changed files with 863 additions and 257 deletions
@@ -571,4 +571,74 @@ namespace Game.AI
        Unit _source;
        bool _playerOnly;
    }
+
+    // Simple selector for units using mana
+    class PowerUsersSelector : ISelector
+    {
+        public PowerUsersSelector(Unit unit, PowerType power, float dist, bool playerOnly)
+        {
+            _me = unit;
+            _power = power;
+            _dist = dist;
+            _playerOnly = playerOnly;
+        }
+
+        public bool Check(Unit target)
+        {
+            if (_me == null || target == null)
+                return false;
+
+            if (target.GetPowerType() != _power)
+                return false;
+
+            if (_playerOnly && target.GetTypeId() != TypeId.Player)
+                return false;
+
+            if (_dist > 0.0f && !_me.IsWithinCombatRange(target, _dist))
+                return false;
+
+            if (_dist < 0.0f && _me.IsWithinCombatRange(target, -_dist))
+                return false;
+
+            return true;
+        }
+
+        Unit _me;
+        PowerType _power;
+        float _dist;
+        bool _playerOnly;
+    }
+
+    class FarthestTargetSelector : ISelector
+    {
+        public FarthestTargetSelector(Unit unit, float dist, bool playerOnly, bool inLos)
+        {
+            _me = unit;
+            _dist = dist;
+            _playerOnly = playerOnly;
+            _inLos = inLos;
+        }
+
+        public bool Check(Unit target)
+        {
+            if (_me == null || target == null)
+                return false;
+
+            if (_playerOnly && target.GetTypeId() != TypeId.Player)
+                return false;
+
+            if (_dist > 0.0f && !_me.IsWithinCombatRange(target, _dist))
+                return false;
+
+            if (_inLos && !_me.IsWithinLOSInMap(target))
+                return false;
+
+            return true;
+        }
+
+        Unit _me;
+        float _dist;
+        bool _playerOnly;
+        bool _inLos;
+    }
 }