diff --git a/OpenRA.Mods.Common/Traits/Render/WithAmmoPipsDecoration.cs b/OpenRA.Mods.Common/Traits/Render/WithAmmoPipsDecoration.cs index 2a7e3624c4..61853c163e 100644 --- a/OpenRA.Mods.Common/Traits/Render/WithAmmoPipsDecoration.cs +++ b/OpenRA.Mods.Common/Traits/Render/WithAmmoPipsDecoration.cs @@ -38,6 +38,9 @@ namespace OpenRA.Mods.Common.Traits.Render [PaletteReference] public readonly string Palette = "chrome"; + [Desc("Name(s) of AmmoPool(s) that use this decoration. Leave empty to include all pools.")] + public readonly string[] AmmoPools = { }; + public override object Create(ActorInitializer init) { return new WithAmmoPipsDecoration(init.Self, this); } } @@ -49,7 +52,13 @@ namespace OpenRA.Mods.Common.Traits.Render public WithAmmoPipsDecoration(Actor self, WithAmmoPipsDecorationInfo info) : base(self, info) { - ammo = self.TraitsImplementing().ToArray(); + if (info.AmmoPools.Any()) + ammo = self.TraitsImplementing() + .Where(ap => info.AmmoPools.Contains(ap.Info.Name)) + .ToArray(); + else + ammo = self.TraitsImplementing().ToArray(); + pips = new Animation(self.World, info.Image); }