diff --git a/MiraAPI/Hud/MultiTargetButton.cs b/MiraAPI/Hud/MultiTargetButton.cs index a700233..8c1ae34 100644 --- a/MiraAPI/Hud/MultiTargetButton.cs +++ b/MiraAPI/Hud/MultiTargetButton.cs @@ -19,6 +19,11 @@ public abstract class MultiTargetButton : CustomActionButton where T : MonoBe /// public virtual float Distance => PlayerControl.LocalPlayer.Data.Role.GetAbilityDistance(); + /// + /// Gets the maximum amount of targets the button can have. + /// + public virtual int MaxTargets { get; } = int.MaxValue; + /// public virtual bool IsTargetValid(T target) { @@ -47,7 +52,10 @@ public abstract class MultiTargetButton : CustomActionButton where T : MonoBe } } - Targets = newTargets.Where(IsTargetValid).ToArray(); + Targets = newTargets + .Where(IsTargetValid) + .Take(MaxTargets) + .ToArray(); Targets.Do(t => SetOutline(t, true)); return base.CanUse() && Targets.Length > 0;