diff --git a/OpenRA.Mods.RA/Explodes.cs b/OpenRA.Mods.RA/Explodes.cs index 7d64a67aa2..8d23da8c3f 100644 --- a/OpenRA.Mods.RA/Explodes.cs +++ b/OpenRA.Mods.RA/Explodes.cs @@ -13,7 +13,7 @@ using OpenRA.Traits; namespace OpenRA.Mods.RA { - class ExplodesInfo : TraitInfo + class ExplodesInfo : ITraitInfo { [WeaponReference] public readonly string Weapon = "UnitExplode"; @@ -21,13 +21,23 @@ namespace OpenRA.Mods.RA public readonly string EmptyWeapon = "UnitExplode"; public readonly int Chance = 100; + public readonly int[] InfDeath = null; + + public object Create (ActorInitializer init) { return new Explodes(this); } } class Explodes : INotifyKilled { + readonly ExplodesInfo Info; + + public Explodes( ExplodesInfo info ) { Info = info; } + public void Killed(Actor self, AttackInfo e) { - if (self.World.SharedRandom.Next(100) > self.Info.Traits.Get().Chance) + if (self.World.SharedRandom.Next(100) > Info.Chance) + return; + + if (Info.InfDeath != null && e.Warhead != null && !Info.InfDeath.Contains(e.Warhead.InfDeath)) return; var weapon = ChooseWeaponForExplosion(self); @@ -42,9 +52,7 @@ namespace OpenRA.Mods.RA string ChooseWeaponForExplosion(Actor self) { var shouldExplode = self.TraitsImplementing().All(a => a.ShouldExplode(self)); - - var info = self.Info.Traits.Get(); - return shouldExplode ? info.Weapon : info.EmptyWeapon; + return shouldExplode ? Info.Weapon : Info.EmptyWeapon; } } }